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 #include <core/frame/geometry.h>
  32
  33 #include <boost/lexical_cast.hpp>
  34 #include <boost/thread/future.hpp>
  35
  36 namespace caspar { namespace core {
  37
  38 struct mutable_frame::impl : boost::noncopyable
  39 {
  40         std::vector<array<std::uint8_t>>                        buffers_;
  41         core::audio_buffer                                                      audio_data_;
  42         const core::pixel_format_desc                           desc_;
  43         const void*                                                                     tag_;
  44         core::frame_geometry                                            geometry_;
  45
  46         impl(std::vector<array<std::uint8_t>> buffers, audio_buffer audio_buffer, const void* tag, const core::pixel_format_desc& desc)
  47                 : buffers_(std::move(buffers))
  48                 , audio_data_(std::move(audio_buffer))
  49                 , desc_(desc)
  50                 , tag_(tag)
  51                 , geometry_(frame_geometry::get_default())
  52         {
  53                 BOOST_FOREACH(auto& buffer, buffers_)
  54                         if(!buffer.data())
  55                                 CASPAR_THROW_EXCEPTION(invalid_argument() << msg_info("mutable_frame: null argument"));
  56         }
  57 };
  58
  59 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)
  60         : impl_(new impl(std::move(image_buffers), std::move(audio_buffer), tag, desc)){}
  61 mutable_frame::~mutable_frame(){}
  62 mutable_frame::mutable_frame(mutable_frame&& other) : impl_(std::move(other.impl_)){}
  63 mutable_frame& mutable_frame::operator=(mutable_frame&& other)
  64 {
  65         impl_ = std::move(other.impl_);
  66         return *this;
  67 }
  68 void mutable_frame::swap(mutable_frame& other){impl_.swap(other.impl_);}
  69 const core::pixel_format_desc& mutable_frame::pixel_format_desc() const{return impl_->desc_;}
  70 const array<std::uint8_t>& mutable_frame::image_data(std::size_t index) const{return impl_->buffers_.at(index);}
  71 const core::audio_buffer& mutable_frame::audio_data() const{return impl_->audio_data_;}
  72 array<std::uint8_t>& mutable_frame::image_data(std::size_t index){return impl_->buffers_.at(index);}
  73 core::audio_buffer& mutable_frame::audio_data(){return impl_->audio_data_;}
  74 std::size_t mutable_frame::width() const{return impl_->desc_.planes.at(0).width;}
  75 std::size_t mutable_frame::height() const{return impl_->desc_.planes.at(0).height;}
  76 const void* mutable_frame::stream_tag()const{return impl_->tag_;}
  77 const void* mutable_frame::data_tag()const{return impl_.get();}
  78 const frame_geometry& mutable_frame::geometry() const { return impl_->geometry_; }
  79 void mutable_frame::set_geometry(const frame_geometry& g) { impl_->geometry_ = g; }
  80
  81 const const_frame& const_frame::empty()
  82 {
  83         static int dummy;
  84         static const_frame empty(&dummy);
  85         return empty;
  86 }
  87
  88 struct const_frame::impl : boost::noncopyable
  89 {
  90         mutable std::vector<boost::shared_future<array<const std::uint8_t>>>    future_buffers_;
  91         int                                                                                     id_;
  92         core::audio_buffer                                                      audio_data_;
  93         const core::pixel_format_desc                           desc_;
  94         const void*                                                                     tag_;
  95         core::frame_geometry                                            geometry_;
  96
  97         impl(const void* tag)
  98                 : desc_(core::pixel_format::invalid)
  99                 , tag_(tag)
 100                 , id_(0)
 101                 , geometry_(frame_geometry::get_default())
 102         {
 103         }
 104
 105         impl(boost::shared_future<array<const std::uint8_t>> image, audio_buffer audio_buffer, const void* tag, const core::pixel_format_desc& desc)
 106                 : audio_data_(std::move(audio_buffer))
 107                 , desc_(desc)
 108                 , tag_(tag)
 109                 , id_(reinterpret_cast<int>(this))
 110                 , geometry_(frame_geometry::get_default())
 111         {
 112                 if(desc.format != core::pixel_format::bgra)
 113                         CASPAR_THROW_EXCEPTION(not_implemented());
 114
 115                 future_buffers_.push_back(std::move(image));
 116         }
 117
 118         impl(mutable_frame&& other)
 119                 : audio_data_(other.audio_data())
 120                 , desc_(other.pixel_format_desc())
 121                 , tag_(other.stream_tag())
 122                 , id_(reinterpret_cast<int>(this))
 123                 , geometry_(other.geometry())
 124         {
 125                 for(std::size_t n = 0; n < desc_.planes.size(); ++n)
 126                 {
 127                         boost::promise<array<const std::uint8_t>> p;
 128                         p.set_value(std::move(other.image_data(n)));
 129                         future_buffers_.push_back(p.get_future());
 130                 }
 131         }
 132
 133         array<const std::uint8_t> image_data(int index) const
 134         {
 135                 return tag_ != empty().stream_tag() ? future_buffers_.at(index).get() : array<const std::uint8_t>(nullptr, 0, true, 0);
 136         }
 137
 138         std::size_t width() const
 139         {
 140                 return tag_ != empty().stream_tag() ? desc_.planes.at(0).width : 0;
 141         }
 142
 143         std::size_t height() const
 144         {
 145                 return tag_ != empty().stream_tag() ? desc_.planes.at(0).height : 0;
 146         }
 147
 148         std::size_t size() const
 149         {
 150                 return tag_ != empty().stream_tag() ? desc_.planes.at(0).size : 0;
 151         }
 152 };
 153
 154 const_frame::const_frame(const void* tag) : impl_(new impl(tag)){}
 155 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)
 156         : impl_(new impl(std::move(image), std::move(audio_buffer), tag, desc)){}
 157 const_frame::const_frame(mutable_frame&& other) : impl_(new impl(std::move(other))){}
 158 const_frame::~const_frame(){}
 159 const_frame::const_frame(const_frame&& other) : impl_(std::move(other.impl_)){}
 160 const_frame& const_frame::operator=(const_frame&& other)
 161 {
 162         impl_ = std::move(other.impl_);
 163         return *this;
 164 }
 165 const_frame::const_frame(const const_frame& other) : impl_(other.impl_){}
 166 const_frame& const_frame::operator=(const const_frame& other)
 167 {
 168         impl_ = other.impl_;
 169         return *this;
 170 }
 171 bool const_frame::operator==(const const_frame& other){return impl_ == other.impl_;}
 172 bool const_frame::operator!=(const const_frame& other){return !(*this == other);}
 173 bool const_frame::operator<(const const_frame& other){return impl_< other.impl_;}
 174 bool const_frame::operator>(const const_frame& other){return impl_> other.impl_;}
 175 const core::pixel_format_desc& const_frame::pixel_format_desc()const{return impl_->desc_;}
 176 array<const std::uint8_t> const_frame::image_data(int index)const{return impl_->image_data(index);}
 177 const core::audio_buffer& const_frame::audio_data()const{return impl_->audio_data_;}
 178 std::size_t const_frame::width()const{return impl_->width();}
 179 std::size_t const_frame::height()const{return impl_->height();}
 180 std::size_t const_frame::size()const{return impl_->size();}
 181 const void* const_frame::stream_tag()const{return impl_->tag_;}
 182 const void* const_frame::data_tag()const{return impl_.get();}
 183 const frame_geometry& const_frame::geometry() const { return impl_->geometry_; }
 184 void const_frame::set_geometry(const frame_geometry& g) { impl_->geometry_ = g; }
 185
 186 }}