git.sesse.net Git - casparcg/blob - core/frame/frame.cpp

   1 /*
   2 * Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>
   3 *
   4 * This file is part of CasparCG (www.casparcg.com).
   5 *
   6 * CasparCG is free software: you can redistribute it and/or modify
   7 * it under the terms of the GNU General Public License as published by
   8 * the Free Software Foundation, either version 3 of the License, or
   9 * (at your option) any later version.
  10 *
  11 * CasparCG is distributed in the hope that it will be useful,
  12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 * GNU General Public License for more details.
  15 *
  16 * You should have received a copy of the GNU General Public License
  17 * along with CasparCG. If not, see <http://www.gnu.org/licenses/>.
  18 *
  19 * Author: Robert Nagy, ronag89@gmail.com
  20 */
  21
  22 #include "../stdafx.h"
  23
  24 #include "frame.h"
  25
  26 #include <common/except.h>
  27 #include <common/array.h>
  28
  29 #include <core/frame/frame_visitor.h>
  30 #include <core/frame/pixel_format.h>
  31
  32 #include <boost/lexical_cast.hpp>
  33 #include <boost/thread/future.hpp>
  34
  35 namespace caspar { namespace core {
  36
  37 struct mutable_frame::impl : boost::noncopyable
  38 {
  39         std::vector<array<std::uint8_t>>                        buffers_;
  40         core::audio_buffer                                                      audio_data_;
  41         const core::pixel_format_desc                           desc_;
  42         const void*                                                                     tag_;
  43
  44         impl(std::vector<array<std::uint8_t>> buffers, audio_buffer audio_buffer, const void* tag, const core::pixel_format_desc& desc)
  45                 : buffers_(std::move(buffers))
  46                 , audio_data_(std::move(audio_buffer))
  47                 , desc_(desc)
  48                 , tag_(tag)
  49         {
  50                 BOOST_FOREACH(auto& buffer, buffers_)
  51                         if(!buffer.data())
  52                                 CASPAR_THROW_EXCEPTION(invalid_argument() << msg_info("mutable_frame: null argument"));
  53         }
  54 };
  55
  56 mutable_frame::mutable_frame(std::vector<array<std::uint8_t>> image_buffers, audio_buffer audio_buffer, const void* tag, const core::pixel_format_desc& desc)
  57         : impl_(new impl(std::move(image_buffers), std::move(audio_buffer), tag, desc)){}
  58 mutable_frame::~mutable_frame(){}
  59 mutable_frame::mutable_frame(mutable_frame&& other) : impl_(std::move(other.impl_)){}
  60 mutable_frame& mutable_frame::operator=(mutable_frame&& other)
  61 {
  62         impl_ = std::move(other.impl_);
  63         return *this;
  64 }
  65 void mutable_frame::swap(mutable_frame& other){impl_.swap(other.impl_);}
  66 const core::pixel_format_desc& mutable_frame::pixel_format_desc() const{return impl_->desc_;}
  67 const array<std::uint8_t>& mutable_frame::image_data(std::size_t index) const{return impl_->buffers_.at(index);}
  68 const core::audio_buffer& mutable_frame::audio_data() const{return impl_->audio_data_;}
  69 array<std::uint8_t>& mutable_frame::image_data(std::size_t index){return impl_->buffers_.at(index);}
  70 core::audio_buffer& mutable_frame::audio_data(){return impl_->audio_data_;}
  71 std::size_t mutable_frame::width() const{return impl_->desc_.planes.at(0).width;}
  72 std::size_t mutable_frame::height() const{return impl_->desc_.planes.at(0).height;}
  73 const void* mutable_frame::stream_tag()const{return impl_->tag_;}
  74 const void* mutable_frame::data_tag()const{return impl_.get();}
  75
  76 const const_frame& const_frame::empty()
  77 {
  78         static int dummy;
  79         static const_frame empty(&dummy);
  80         return empty;
  81 }
  82
  83 struct const_frame::impl : boost::noncopyable
  84 {
  85         mutable std::vector<boost::shared_future<array<const std::uint8_t>>>    future_buffers_;
  86         int                                                                                     id_;
  87         core::audio_buffer                                                      audio_data_;
  88         const core::pixel_format_desc                           desc_;
  89         const void*                                                                     tag_;
  90
  91         impl(const void* tag)
  92                 : desc_(core::pixel_format::invalid)
  93                 , tag_(tag)
  94                 , id_(0)
  95         {
  96         }
  97
  98         impl(boost::shared_future<array<const std::uint8_t>> image, audio_buffer audio_buffer, const void* tag, const core::pixel_format_desc& desc)
  99                 : audio_data_(std::move(audio_buffer))
 100                 , desc_(desc)
 101                 , tag_(tag)
 102                 , id_(reinterpret_cast<int>(this))
 103         {
 104                 if(desc.format != core::pixel_format::bgra)
 105                         CASPAR_THROW_EXCEPTION(not_implemented());
 106
 107                 future_buffers_.push_back(std::move(image));
 108         }
 109
 110         impl(mutable_frame&& other)
 111                 : audio_data_(other.audio_data())
 112                 , desc_(other.pixel_format_desc())
 113                 , tag_(other.stream_tag())
 114                 , id_(reinterpret_cast<int>(this))
 115         {
 116                 for(std::size_t n = 0; n < desc_.planes.size(); ++n)
 117                 {
 118                         boost::promise<array<const std::uint8_t>> p;
 119                         p.set_value(std::move(other.image_data(n)));
 120                         future_buffers_.push_back(p.get_future());
 121                 }
 122         }
 123
 124         array<const std::uint8_t> image_data(int index) const
 125         {
 126                 return tag_ != empty().stream_tag() ? future_buffers_.at(index).get() : array<const std::uint8_t>(nullptr, 0, true, 0);
 127         }
 128
 129         std::size_t width() const
 130         {
 131                 return tag_ != empty().stream_tag() ? desc_.planes.at(0).width : 0;
 132         }
 133
 134         std::size_t height() const
 135         {
 136                 return tag_ != empty().stream_tag() ? desc_.planes.at(0).height : 0;
 137         }
 138
 139         std::size_t size() const
 140         {
 141                 return tag_ != empty().stream_tag() ? desc_.planes.at(0).size : 0;
 142         }
 143 };
 144
 145 const_frame::const_frame(const void* tag) : impl_(new impl(tag)){}
 146 const_frame::const_frame(boost::shared_future<array<const std::uint8_t>> image, audio_buffer audio_buffer, const void* tag, const core::pixel_format_desc& desc)
 147         : impl_(new impl(std::move(image), std::move(audio_buffer), tag, desc)){}
 148 const_frame::const_frame(mutable_frame&& other) : impl_(new impl(std::move(other))){}
 149 const_frame::~const_frame(){}
 150 const_frame::const_frame(const_frame&& other) : impl_(std::move(other.impl_)){}
 151 const_frame& const_frame::operator=(const_frame&& other)
 152 {
 153         impl_ = std::move(other.impl_);
 154         return *this;
 155 }
 156 const_frame::const_frame(const const_frame& other) : impl_(other.impl_){}
 157 const_frame& const_frame::operator=(const const_frame& other)
 158 {
 159         impl_ = other.impl_;
 160         return *this;
 161 }
 162 bool const_frame::operator==(const const_frame& other){return impl_ == other.impl_;}
 163 bool const_frame::operator!=(const const_frame& other){return !(*this == other);}
 164 bool const_frame::operator<(const const_frame& other){return impl_< other.impl_;}
 165 bool const_frame::operator>(const const_frame& other){return impl_> other.impl_;}
 166 const core::pixel_format_desc& const_frame::pixel_format_desc()const{return impl_->desc_;}
 167 array<const std::uint8_t> const_frame::image_data(int index)const{return impl_->image_data(index);}
 168 const core::audio_buffer& const_frame::audio_data()const{return impl_->audio_data_;}
 169 std::size_t const_frame::width()const{return impl_->width();}
 170 std::size_t const_frame::height()const{return impl_->height();}
 171 std::size_t const_frame::size()const{return impl_->size();}
 172 const void* const_frame::stream_tag()const{return impl_->tag_;}
 173 const void* const_frame::data_tag()const{return impl_.get();}
 174
 175 }}