2 * copyright (c) 2010 Sveriges Television AB <info@casparcg.com>
\r
4 * This file is part of CasparCG.
\r
6 * CasparCG is free software: you can redistribute it and/or modify
\r
7 * it under the terms of the GNU General Public License as published by
\r
8 * the Free Software Foundation, either version 3 of the License, or
\r
9 * (at your option) any later version.
\r
11 * CasparCG is distributed in the hope that it will be useful,
\r
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
\r
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
\r
14 * GNU General Public License for more details.
\r
16 * You should have received a copy of the GNU General Public License
\r
17 * along with CasparCG. If not, see <http://www.gnu.org/licenses/>.
\r
20 #include "../stdafx.h"
\r
22 #include "write_frame.h"
\r
24 #include "gpu/ogl_device.h"
\r
25 #include "gpu/host_buffer.h"
\r
26 #include "gpu/device_buffer.h"
\r
28 #include <core/producer/frame/frame_visitor.h>
\r
29 #include <core/producer/frame/pixel_format.h>
\r
31 namespace caspar { namespace core {
\r
33 struct write_frame::implementation
\r
36 std::vector<std::shared_ptr<host_buffer>> buffers_;
\r
37 std::vector<safe_ptr<device_buffer>> textures_;
\r
38 std::vector<int16_t> audio_data_;
\r
39 const core::pixel_format_desc desc_;
\r
41 core::video_mode::type mode_;
\r
43 implementation(const void* tag)
\r
48 implementation(ogl_device& ogl, const void* tag, const core::pixel_format_desc& desc)
\r
52 , mode_(core::video_mode::progressive)
\r
56 std::transform(desc.planes.begin(), desc.planes.end(), std::back_inserter(buffers_), [&](const core::pixel_format_desc::plane& plane)
\r
58 return ogl_->create_host_buffer(plane.size, host_buffer::write_only);
\r
60 std::transform(desc.planes.begin(), desc.planes.end(), std::back_inserter(textures_), [&](const core::pixel_format_desc::plane& plane)
\r
62 return ogl_->create_device_buffer(plane.width, plane.height, plane.channels);
\r
67 void accept(write_frame& self, core::frame_visitor& visitor)
\r
69 visitor.begin(self);
\r
70 visitor.visit(self);
\r
74 boost::iterator_range<uint8_t*> image_data(size_t index)
\r
76 if(index >= buffers_.size() || !buffers_[index]->data())
\r
77 return boost::iterator_range<const uint8_t*>();
\r
78 auto ptr = static_cast<uint8_t*>(buffers_[index]->data());
\r
79 return boost::iterator_range<uint8_t*>(ptr, ptr+buffers_[index]->size());
\r
82 const boost::iterator_range<const uint8_t*> image_data(size_t index) const
\r
84 if(index >= buffers_.size() || !buffers_[index]->data())
\r
85 return boost::iterator_range<const uint8_t*>();
\r
86 auto ptr = static_cast<const uint8_t*>(buffers_[index]->data());
\r
87 return boost::iterator_range<const uint8_t*>(ptr, ptr+buffers_[index]->size());
\r
92 for(size_t n = 0; n < buffers_.size(); ++n)
\r
96 void commit(size_t plane_index)
\r
98 if(plane_index >= buffers_.size())
\r
101 auto buffer = std::move(buffers_[plane_index]); // Release buffer once done.
\r
106 auto texture = textures_.at(plane_index);
\r
108 ogl_->begin_invoke([=]
\r
110 texture->read(*buffer);
\r
115 write_frame::write_frame(const void* tag) : impl_(new implementation(tag)){}
\r
116 write_frame::write_frame(ogl_device& ogl, const void* tag, const core::pixel_format_desc& desc)
\r
117 : impl_(new implementation(ogl, tag, desc)){}
\r
118 write_frame::write_frame(const write_frame& other) : impl_(new implementation(*other.impl_)){}
\r
119 void write_frame::accept(core::frame_visitor& visitor){impl_->accept(*this, visitor);}
\r
121 boost::iterator_range<uint8_t*> write_frame::image_data(size_t index){return impl_->image_data(index);}
\r
122 std::vector<int16_t>& write_frame::audio_data() { return impl_->audio_data_; }
\r
123 const boost::iterator_range<const uint8_t*> write_frame::image_data(size_t index) const
\r
125 return boost::iterator_range<const uint8_t*>(impl_->image_data(index).begin(), impl_->image_data(index).end());
\r
127 const boost::iterator_range<const int16_t*> write_frame::audio_data() const
\r
129 return boost::iterator_range<const int16_t*>(impl_->audio_data_.data(), impl_->audio_data_.data() + impl_->audio_data_.size());
\r
131 const void* write_frame::tag() const {return impl_->tag_;}
\r
132 const core::pixel_format_desc& write_frame::get_pixel_format_desc() const{return impl_->desc_;}
\r
133 const std::vector<safe_ptr<device_buffer>>& write_frame::get_textures() const{return impl_->textures_;}
\r
134 void write_frame::commit(size_t plane_index){impl_->commit(plane_index);}
\r
135 void write_frame::commit(){impl_->commit();}
\r
136 void write_frame::set_type(const video_mode::type& mode){impl_->mode_ = mode;}
\r
137 core::video_mode::type write_frame::get_type() const{return impl_->mode_;}
\r