Home | History | Annotate | Download | only in base 
      1 /*
      2  * libjingle
      3  * Copyright 2011 Google Inc.
      4  *
      5  * Redistribution and use in source and binary forms, with or without
      6  * modification, are permitted provided that the following conditions are met:
      7  *
      8  *  1. Redistributions of source code must retain the above copyright notice,
      9  *     this list of conditions and the following disclaimer.
     10  *  2. Redistributions in binary form must reproduce the above copyright notice,
     11  *     this list of conditions and the following disclaimer in the documentation
     12  *     and/or other materials provided with the distribution.
     13  *  3. The name of the author may not be used to endorse or promote products
     14  *     derived from this software without specific prior written permission.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
     17  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
     18  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
     19  * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     20  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
     21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
     22  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
     23  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
     24  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
     25  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     26  */
     27 
     28 #ifndef TALK_MEDIA_BASE_FAKEVIDEORENDERER_H_
     29 #define TALK_MEDIA_BASE_FAKEVIDEORENDERER_H_
     30 
     31 #include "talk/media/base/videoframe.h"
     32 #include "talk/media/base/videorenderer.h"
     33 #include "webrtc/base/logging.h"
     34 #include "webrtc/base/sigslot.h"
     35 
     36 namespace cricket {
     37 
     38 // Faked video renderer that has a callback for actions on rendering.
     39 class FakeVideoRenderer : public VideoRenderer {
     40  public:
     41   FakeVideoRenderer()
     42       : errors_(0),
     43         width_(0),
     44         height_(0),
     45         num_set_sizes_(0),
     46         num_rendered_frames_(0),
     47         black_frame_(false) {
     48   }
     49 
     50   virtual bool SetSize(int width, int height, int reserved) {
     51     rtc::CritScope cs(&crit_);
     52     width_ = width;
     53     height_ = height;
     54     ++num_set_sizes_;
     55     SignalSetSize(width, height, reserved);
     56     return true;
     57   }
     58 
     59   virtual bool RenderFrame(const VideoFrame* frame) {
     60     rtc::CritScope cs(&crit_);
     61     // TODO(zhurunz) Check with VP8 team to see if we can remove this
     62     // tolerance on Y values.
     63     black_frame_ = CheckFrameColorYuv(6, 48, 128, 128, 128, 128, frame);
     64     // Treat unexpected frame size as error.
     65     if (!frame ||
     66         frame->GetWidth() != static_cast<size_t>(width_) ||
     67         frame->GetHeight() != static_cast<size_t>(height_)) {
     68       if (!frame) {
     69         LOG(LS_WARNING) << "RenderFrame expected non-null frame.";
     70       } else {
     71         LOG(LS_WARNING) << "RenderFrame expected frame of size " << width_
     72                         << "x" << height_ << " but received frame of size "
     73                         << frame->GetWidth() << "x" << frame->GetHeight();
     74       }
     75       ++errors_;
     76       return false;
     77     }
     78     ++num_rendered_frames_;
     79     SignalRenderFrame(frame);
     80     return true;
     81   }
     82 
     83   int errors() const { return errors_; }
     84   int width() const {
     85     rtc::CritScope cs(&crit_);
     86     return width_;
     87   }
     88   int height() const {
     89     rtc::CritScope cs(&crit_);
     90     return height_;
     91   }
     92   int num_set_sizes() const {
     93     rtc::CritScope cs(&crit_);
     94     return num_set_sizes_;
     95   }
     96   int num_rendered_frames() const {
     97     rtc::CritScope cs(&crit_);
     98     return num_rendered_frames_;
     99   }
    100   bool black_frame() const {
    101     rtc::CritScope cs(&crit_);
    102     return black_frame_;
    103   }
    104 
    105   sigslot::signal3<int, int, int> SignalSetSize;
    106   sigslot::signal1<const VideoFrame*> SignalRenderFrame;
    107 
    108  private:
    109   static bool CheckFrameColorYuv(uint8 y_min, uint8 y_max,
    110                                  uint8 u_min, uint8 u_max,
    111                                  uint8 v_min, uint8 v_max,
    112                                  const cricket::VideoFrame* frame) {
    113     if (!frame) {
    114       return false;
    115     }
    116     // Y
    117     size_t y_width = frame->GetWidth();
    118     size_t y_height = frame->GetHeight();
    119     const uint8* y_plane = frame->GetYPlane();
    120     const uint8* y_pos = y_plane;
    121     int32 y_pitch = frame->GetYPitch();
    122     for (size_t i = 0; i < y_height; ++i) {
    123       for (size_t j = 0; j < y_width; ++j) {
    124         uint8 y_value = *(y_pos + j);
    125         if (y_value < y_min || y_value > y_max) {
    126           return false;
    127         }
    128       }
    129       y_pos += y_pitch;
    130     }
    131     // U and V
    132     size_t chroma_width = frame->GetChromaWidth();
    133     size_t chroma_height = frame->GetChromaHeight();
    134     const uint8* u_plane = frame->GetUPlane();
    135     const uint8* v_plane = frame->GetVPlane();
    136     const uint8* u_pos = u_plane;
    137     const uint8* v_pos = v_plane;
    138     int32 u_pitch = frame->GetUPitch();
    139     int32 v_pitch = frame->GetVPitch();
    140     for (size_t i = 0; i < chroma_height; ++i) {
    141       for (size_t j = 0; j < chroma_width; ++j) {
    142         uint8 u_value = *(u_pos + j);
    143         if (u_value < u_min || u_value > u_max) {
    144           return false;
    145         }
    146         uint8 v_value = *(v_pos + j);
    147         if (v_value < v_min || v_value > v_max) {
    148           return false;
    149         }
    150       }
    151       u_pos += u_pitch;
    152       v_pos += v_pitch;
    153     }
    154     return true;
    155   }
    156 
    157   int errors_;
    158   int width_;
    159   int height_;
    160   int num_set_sizes_;
    161   int num_rendered_frames_;
    162   bool black_frame_;
    163   mutable rtc::CriticalSection crit_;
    164 };
    165 
    166 }  // namespace cricket
    167 
    168 #endif  // TALK_MEDIA_BASE_FAKEVIDEORENDERER_H_
    169