git.sesse.net Git - casparcg/blob - core/producer/ffmpeg/video/video_transformer.cpp

   1 #include "../../../stdafx.h"\r
   2 \r
   3 #include "video_transformer.h"\r
   4 \r
   5 #include "../../../frame/frame_format.h"\r
   6 #include "../../../../common/utility/memory.h"\r
   7 #include "../../../frame/gpu_frame.h"\r
   8 #include "../../../frame/gpu_frame.h"\r
   9 #include "../../../frame/frame_factory.h"\r
  10 \r
  11 #include <tbb/parallel_for.h>\r
  12 #include <tbb/atomic.h>\r
  13 #include <tbb/mutex.h>\r
  14 #include <tbb/concurrent_queue.h>\r
  15 #include <tbb/scalable_allocator.h>\r
  16 \r
  17 #if defined(_MSC_VER)\r
  18 #pragma warning (push)\r
  19 #pragma warning (disable : 4244)\r
  20 #endif\r
  21 extern "C" \r
  22 {\r
  23         #define __STDC_CONSTANT_MACROS\r
  24         #define __STDC_LIMIT_MACROS\r
  25         #include <libswscale/swscale.h>\r
  26 }\r
  27 #if defined(_MSC_VER)\r
  28 #pragma warning (pop)\r
  29 #endif\r
  30 \r
  31 namespace caspar { namespace core { namespace ffmpeg{\r
  32         \r
  33 struct video_transformer::implementation : boost::noncopyable\r
  34 {\r
  35         video_packet_ptr execute(const video_packet_ptr video_packet)\r
  36         {                               \r
  37                 assert(video_packet);\r
  38                 size_t width = video_packet->codec_context->width;\r
  39                 size_t height = video_packet->codec_context->height;\r
  40                 auto pix_fmt = video_packet->codec_context->pix_fmt;\r
  41 \r
  42                 if(!sws_context_)\r
  43                 {\r
  44                         double param;\r
  45                         sws_context_.reset(sws_getContext(width, height, pix_fmt, width, height, \r
  46                                                                                                 PIX_FMT_BGRA, SWS_BILINEAR, nullptr, nullptr, &param), sws_freeContext);\r
  47                 }\r
  48                 \r
  49                 video_packet->frame = factory_->create_frame(width, height);\r
  50                 AVFrame av_frame;       \r
  51                 avcodec_get_frame_defaults(&av_frame);\r
  52                 avpicture_fill(reinterpret_cast<AVPicture*>(&av_frame), video_packet->frame->data(), PIX_FMT_BGRA, width, height);\r
  53                 \r
  54                 sws_scale(sws_context_.get(), video_packet->decoded_frame->data, video_packet->decoded_frame->linesize, 0, height, av_frame.data, av_frame.linesize);\r
  55                 \r
  56                 if(video_packet->codec->id == CODEC_ID_DVVIDEO) // Move up one field\r
  57                         video_packet->frame->translate(0.0f, 1.0/static_cast<double>(video_packet->format_desc.height));\r
  58                 \r
  59                 return video_packet;    \r
  60         }\r
  61 \r
  62         frame_factory_ptr factory_;\r
  63         std::shared_ptr<SwsContext> sws_context_;\r
  64 };\r
  65 \r
  66 video_transformer::video_transformer() : impl_(new implementation()){}\r
  67 video_packet_ptr video_transformer::execute(const video_packet_ptr& video_packet){return impl_->execute(video_packet);}\r
  68 void video_transformer::initialize(const frame_factory_ptr& factory){impl_->factory_ = factory; }\r
  69 }}}