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, int32_t 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
61 void accept(write_frame& self, core::frame_visitor& visitor)
\r
63 visitor.begin(self);
\r
64 visitor.visit(self);
\r
68 boost::iterator_range<uint8_t*> image_data(size_t index)
\r
70 if(index >= buffers_.size() || !buffers_[index]->data())
\r
71 return boost::iterator_range<const uint8_t*>();
\r
72 auto ptr = static_cast<uint8_t*>(buffers_[index]->data());
\r
73 return boost::iterator_range<uint8_t*>(ptr, ptr+buffers_[index]->size());
\r
76 const boost::iterator_range<const uint8_t*> image_data(size_t index) const
\r
78 if(index >= buffers_.size() || !buffers_[index]->data())
\r
79 return boost::iterator_range<const uint8_t*>();
\r
80 auto ptr = static_cast<const uint8_t*>(buffers_[index]->data());
\r
81 return boost::iterator_range<const uint8_t*>(ptr, ptr+buffers_[index]->size());
\r
86 for(size_t n = 0; n < buffers_.size(); ++n)
\r
90 void commit(size_t plane_index)
\r
92 if(plane_index >= buffers_.size())
\r
95 auto texture = textures_[plane_index];
\r
96 auto buffer = std::move(buffers_[plane_index]); // Release buffer once done.
\r
98 ogl_.begin_invoke([=]
\r
100 texture->read(*buffer);
\r
105 write_frame::write_frame(ogl_device& ogl, int32_t tag, const core::pixel_format_desc& desc)
\r
106 : impl_(new implementation(ogl, tag, desc)){}
\r
107 void write_frame::accept(core::frame_visitor& visitor){impl_->accept(*this, visitor);}
\r
109 boost::iterator_range<uint8_t*> write_frame::image_data(size_t index){return impl_->image_data(index);}
\r
110 std::vector<int16_t>& write_frame::audio_data() { return impl_->audio_data_; }
\r
111 const boost::iterator_range<const uint8_t*> write_frame::image_data(size_t index) const
\r
113 return boost::iterator_range<const uint8_t*>(impl_->image_data(index).begin(), impl_->image_data(index).end());
\r
115 const boost::iterator_range<const int16_t*> write_frame::audio_data() const
\r
117 return boost::iterator_range<const int16_t*>(impl_->audio_data_.data(), impl_->audio_data_.data() + impl_->audio_data_.size());
\r
119 int32_t write_frame::tag() const {return impl_->tag_;}
\r
120 const core::pixel_format_desc& write_frame::get_pixel_format_desc() const{return impl_->desc_;}
\r
121 const std::vector<safe_ptr<device_buffer>>& write_frame::get_textures() const{return impl_->textures_;}
\r
122 const std::vector<safe_ptr<host_buffer>>& write_frame::get_buffers() const{return impl_->buffers_;}
\r
123 void write_frame::commit(size_t plane_index){impl_->commit(plane_index);}
\r
124 void write_frame::commit(){impl_->commit();}
\r