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 : boost::noncopyable
\r
36 std::vector<safe_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
43 implementation(ogl_device& ogl, int tag, const core::pixel_format_desc& desc)
\r
50 std::transform(desc.planes.begin(), desc.planes.end(), std::back_inserter(buffers_), [&](const core::pixel_format_desc::plane& plane)
\r
52 return ogl_.create_host_buffer(plane.size, host_buffer::write_only);
\r
54 std::transform(desc.planes.begin(), desc.planes.end(), std::back_inserter(textures_), [&](const core::pixel_format_desc::plane& plane)
\r
56 return ogl_.create_device_buffer(plane.width, plane.height, plane.channels);
\r
63 auto buffers = buffers_;
\r
64 auto textures = textures_;
\r
65 ogl_.begin_invoke([=]() mutable
\r
72 void accept(write_frame& self, core::frame_visitor& visitor)
\r
74 visitor.begin(self);
\r
75 visitor.visit(self);
\r
79 boost::iterator_range<uint8_t*> image_data(size_t index)
\r
81 if(index >= buffers_.size() || !buffers_[index]->data())
\r
82 return boost::iterator_range<const uint8_t*>();
\r
83 auto ptr = static_cast<uint8_t*>(buffers_[index]->data());
\r
84 return boost::iterator_range<uint8_t*>(ptr, ptr+buffers_[index]->size());
\r
87 const boost::iterator_range<const uint8_t*> image_data(size_t index) const
\r
89 if(index >= buffers_.size() || !buffers_[index]->data())
\r
90 return boost::iterator_range<const uint8_t*>();
\r
91 auto ptr = static_cast<const uint8_t*>(buffers_[index]->data());
\r
92 return boost::iterator_range<const uint8_t*>(ptr, ptr+buffers_[index]->size());
\r
97 for(size_t n = 0; n < buffers_.size(); ++n)
\r
101 void commit(size_t plane_index)
\r
103 if(plane_index >= buffers_.size())
\r
106 auto texture = textures_[plane_index];
\r
107 auto buffer = std::move(buffers_[plane_index]); // Release buffer once done.
\r
109 ogl_.begin_invoke([=]
\r
111 texture->read(*buffer);
\r
116 write_frame::write_frame(ogl_device& ogl, int32_t tag, const core::pixel_format_desc& desc)
\r
117 : impl_(new implementation(ogl, tag, desc)){}
\r
118 void write_frame::accept(core::frame_visitor& visitor){impl_->accept(*this, visitor);}
\r
120 boost::iterator_range<uint8_t*> write_frame::image_data(size_t index){return impl_->image_data(index);}
\r
121 std::vector<int16_t>& write_frame::audio_data() { return impl_->audio_data_; }
\r
122 const boost::iterator_range<const uint8_t*> write_frame::image_data(size_t index) const
\r
124 return boost::iterator_range<const uint8_t*>(impl_->image_data(index).begin(), impl_->image_data(index).end());
\r
126 const boost::iterator_range<const int16_t*> write_frame::audio_data() const
\r
128 return boost::iterator_range<const int16_t*>(impl_->audio_data_.data(), impl_->audio_data_.data() + impl_->audio_data_.size());
\r
130 int write_frame::tag() const {return impl_->tag_;}
\r
131 const core::pixel_format_desc& write_frame::get_pixel_format_desc() const{return impl_->desc_;}
\r
132 const std::vector<safe_ptr<device_buffer>>& write_frame::get_textures() const{return impl_->textures_;}
\r
133 void write_frame::commit(size_t plane_index){impl_->commit(plane_index);}
\r
134 void write_frame::commit(){impl_->commit();}
\r