2 * Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>
\r
4 * This file is part of CasparCG (www.casparcg.com).
\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
19 * Author: Robert Nagy, ronag89@gmail.com
\r
22 #include "../../stdafx.h"
\r
24 #include "color_producer.h"
\r
26 #include <core/producer/frame_producer.h>
\r
27 #include <core/frame/draw_frame.h>
\r
28 #include <core/frame/frame_factory.h>
\r
29 #include <core/frame/pixel_format.h>
\r
30 #include <core/frame/write_frame.h>
\r
32 #include <common/except.h>
\r
34 #include <boost/algorithm/string.hpp>
\r
38 namespace caspar { namespace core {
\r
40 class color_producer : public frame_producer
\r
42 spl::shared_ptr<draw_frame> frame_;
\r
43 const std::wstring color_str_;
\r
46 explicit color_producer(const spl::shared_ptr<core::frame_factory>& frame_factory, const std::wstring& color)
\r
48 , frame_(create_color_frame(this, frame_factory, color))
\r
53 virtual spl::shared_ptr<draw_frame> receive(int) override
\r
58 virtual spl::shared_ptr<draw_frame> last_frame() const override
\r
63 virtual std::wstring print() const override
\r
65 return L"color[" + color_str_ + L"]";
\r
68 virtual std::wstring name() const override
\r
73 boost::property_tree::wptree info() const override
\r
75 boost::property_tree::wptree info;
\r
76 info.add(L"type", L"color");
\r
77 info.add(L"color", color_str_);
\r
82 std::wstring get_hex_color(const std::wstring& str)
\r
84 if(str.at(0) == '#')
\r
85 return str.length() == 7 ? L"#FF" + str.substr(1) : str;
\r
87 if(boost::iequals(str, L"EMPTY"))
\r
88 return L"#00000000";
\r
90 if(boost::iequals(str, L"BLACK"))
\r
91 return L"#FF000000";
\r
93 if(boost::iequals(str, L"WHITE"))
\r
94 return L"#FFFFFFFF";
\r
96 if(boost::iequals(str, L"RED"))
\r
97 return L"#FFFF0000";
\r
99 if(boost::iequals(str, L"GREEN"))
\r
100 return L"#FF00FF00";
\r
102 if(boost::iequals(str, L"BLUE"))
\r
103 return L"#FF0000FF";
\r
105 if(boost::iequals(str, L"ORANGE"))
\r
106 return L"#FFFFA500";
\r
108 if(boost::iequals(str, L"YELLOW"))
\r
109 return L"#FFFFFF00";
\r
111 if(boost::iequals(str, L"BROWN"))
\r
112 return L"#FFA52A2A";
\r
114 if(boost::iequals(str, L"GRAY"))
\r
115 return L"#FF808080";
\r
117 if(boost::iequals(str, L"TEAL"))
\r
118 return L"#FF008080";
\r
123 spl::shared_ptr<frame_producer> create_color_producer(const spl::shared_ptr<frame_factory>& frame_factory, const std::vector<std::wstring>& params)
\r
125 if(params.size() < 0)
\r
126 return core::frame_producer::empty();
\r
128 auto color2 = get_hex_color(params[0]);
\r
129 if(color2.length() != 9 || color2[0] != '#')
\r
130 return core::frame_producer::empty();
\r
132 return core::wrap_producer(spl::make_shared<color_producer>(frame_factory, color2));
\r
134 spl::shared_ptr<write_frame> create_color_frame(void* tag, const spl::shared_ptr<frame_factory>& frame_factory, const std::wstring& color)
\r
136 auto color2 = get_hex_color(color);
\r
137 if(color2.length() != 9 || color2[0] != '#')
\r
138 BOOST_THROW_EXCEPTION(invalid_argument() << arg_name_info("color") << arg_value_info(color2) << msg_info("Invalid color."));
\r
140 core::pixel_format_desc desc(pixel_format::bgra);
\r
141 desc.planes.push_back(core::pixel_format_desc::plane(1, 1, 4));
\r
142 auto frame = frame_factory->create_frame(tag, desc);
\r
144 // Read color from hex-string and write to frame pixel.
\r
146 auto& value = *reinterpret_cast<uint32_t*>(frame->image_data(0).begin());
\r
147 std::wstringstream str(color2.substr(1));
\r
148 if(!(str >> std::hex >> value) || !str.eof())
\r
149 BOOST_THROW_EXCEPTION(invalid_argument() << arg_name_info("color") << arg_value_info(color2) << msg_info("Invalid color."));
\r