]> git.sesse.net Git - casparcg/blobdiff - core/producer/color/color_producer.cpp
set svn:eol-style native on .h and .cpp files
[casparcg] / core / producer / color / color_producer.cpp
index 0b448bf842456263cb5dc7f06685393fed44873d..ed932b446ee94c086e9f51c90dacbcbd1fa8faf5 100644 (file)
-/*\r
-* copyright (c) 2010 Sveriges Television AB <info@casparcg.com>\r
-*\r
-*  This file is part of CasparCG.\r
-*\r
-*    CasparCG is free software: you can redistribute it and/or modify\r
-*    it under the terms of the GNU General Public License as published by\r
-*    the Free Software Foundation, either version 3 of the License, or\r
-*    (at your option) any later version.\r
-*\r
-*    CasparCG is distributed in the hope that it will be useful,\r
-*    but WITHOUT ANY WARRANTY; without even the implied warranty of\r
-*    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
-*    GNU General Public License for more details.\r
-\r
-*    You should have received a copy of the GNU General Public License\r
-*    along with CasparCG.  If not, see <http://www.gnu.org/licenses/>.\r
-*\r
-*/\r
\r
-#include "../../stdafx.h"\r
-\r
-#include "color_producer.h"\r
-\r
-#include "../../format/video_format.h"\r
-\r
-#include <intrin.h>\r
-#pragma intrinsic(__movsd, __stosd)\r
-\r
-namespace caspar { namespace core {\r
-       \r
-unsigned int get_pixel_color_value(const std::wstring& parameter)\r
-{\r
-       union Color \r
-       {\r
-               struct Components \r
-               {\r
-                       unsigned char a;\r
-                       unsigned char r;\r
-                       unsigned char g;\r
-                       unsigned char b;\r
-               } comp;\r
-\r
-               unsigned int value;\r
-       };\r
-\r
-       std::wstring color_code;\r
-       if(parameter.length() != 9 || parameter[0] != '#')\r
-               BOOST_THROW_EXCEPTION(invalid_argument() << arg_name_info("parameter") << arg_value_info(common::narrow(parameter)) << msg_info("Invalid color code"));\r
-       \r
-       color_code = parameter.substr(1);\r
-\r
-       Color color;\r
-       color.value = _tcstoul(color_code.c_str(), 0, 16);\r
-       unsigned char temp = color.comp.a;\r
-       color.comp.a = color.comp.b;\r
-       color.comp.b = temp;\r
-       temp = color.comp.r;\r
-       color.comp.r = color.comp.g;\r
-       color.comp.g = temp;\r
-\r
-       return color.value;\r
-}\r
-\r
-class color_producer : public frame_producer\r
-{\r
-public:\r
-       explicit color_producer(const std::wstring& color) : color_str_(color), color_value_(get_pixel_color_value(color)){}\r
-       \r
-       gpu_frame_ptr render_frame()\r
-       { \r
-               return frame_;\r
-       }\r
-\r
-       void initialize(const frame_processor_device_ptr& frame_processor)\r
-       {\r
-               auto frame = frame_processor->create_frame();\r
-               __stosd(reinterpret_cast<unsigned long*>(frame->data().begin()), color_value_, frame->data().size() / sizeof(unsigned long));\r
-               frame_ = frame;\r
-       }\r
-       \r
-       std::wstring print() const\r
-       {\r
-               return + L"color_producer. color: " + color_str_;\r
-       }\r
-\r
-       gpu_frame_ptr frame_;\r
-       unsigned int color_value_;\r
-       std::wstring color_str_;\r
-};\r
-\r
-frame_producer_ptr create_color_producer(const std::vector<std::wstring>& params)\r
-{\r
-       if(params.empty() || params[0].at(0) != '#')\r
-               return nullptr;\r
-       return std::make_shared<color_producer>(params[0]);\r
-}\r
-\r
+/*
+* Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>
+*
+* This file is part of CasparCG (www.casparcg.com).
+*
+* CasparCG is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* CasparCG is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+* GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with CasparCG. If not, see <http://www.gnu.org/licenses/>.
+*
+* Author: Robert Nagy, ronag89@gmail.com
+*/
+
+#include "../../stdafx.h"
+
+#include "color_producer.h"
+
+#include <core/producer/frame_producer.h>
+#include <core/frame/frame.h>
+#include <core/frame/draw_frame.h>
+#include <core/frame/frame_factory.h>
+#include <core/frame/pixel_format.h>
+#include <core/monitor/monitor.h>
+
+#include <common/except.h>
+#include <common/array.h>
+
+#include <boost/algorithm/string.hpp>
+
+#include <sstream>
+
+namespace caspar { namespace core {
+       
+class color_producer : public frame_producer_base
+{
+       monitor::basic_subject  event_subject_;
+
+       draw_frame                              frame_;
+       const std::wstring              color_str_;
+
+public:
+       explicit color_producer(const spl::shared_ptr<core::frame_factory>& frame_factory, const std::wstring& color) 
+               : color_str_(color)
+               , frame_(create_color_frame(this, frame_factory, color))
+       {
+               CASPAR_LOG(info) << print() << L" Initialized";
+       }
+
+       // frame_producer
+                       
+       draw_frame receive_impl() override
+       {
+               event_subject_ << monitor::event("color") % color_str_;
+
+               return frame_;
+       }       
+       
+       std::wstring print() const override
+       {
+               return L"color[" + color_str_ + L"]";
+       }
+
+       std::wstring name() const override
+       {
+               return L"color";
+       }
+       
+       boost::property_tree::wptree info() const override
+       {
+               boost::property_tree::wptree info;
+               info.add(L"type", L"color");
+               info.add(L"color", color_str_);
+               return info;
+       }
+
+       void subscribe(const monitor::observable::observer_ptr& o) override                                                                                                                     
+       {
+               return event_subject_.subscribe(o);
+       }
+
+       void unsubscribe(const monitor::observable::observer_ptr& o) override           
+       {
+               return event_subject_.unsubscribe(o);
+       }
+};
+
+std::wstring get_hex_color(const std::wstring& str)
+{
+       if(str.at(0) == '#')
+               return str.length() == 7 ? L"#FF" + str.substr(1) : str;
+       
+       if(boost::iequals(str, L"EMPTY"))
+               return L"#00000000";
+
+       if(boost::iequals(str, L"BLACK"))
+               return L"#FF000000";
+       
+       if(boost::iequals(str, L"WHITE"))
+               return L"#FFFFFFFF";
+       
+       if(boost::iequals(str, L"RED"))
+               return L"#FFFF0000";
+       
+       if(boost::iequals(str, L"GREEN"))
+               return L"#FF00FF00";
+       
+       if(boost::iequals(str, L"BLUE"))
+               return L"#FF0000FF";    
+       
+       if(boost::iequals(str, L"ORANGE"))
+               return L"#FFFFA500";    
+       
+       if(boost::iequals(str, L"YELLOW"))
+               return L"#FFFFFF00";
+       
+       if(boost::iequals(str, L"BROWN"))
+               return L"#FFA52A2A";
+       
+       if(boost::iequals(str, L"GRAY"))
+               return L"#FF808080";
+       
+       if(boost::iequals(str, L"TEAL"))
+               return L"#FF008080";
+       
+       return str;
+}
+
+spl::shared_ptr<frame_producer> create_color_producer(const spl::shared_ptr<frame_factory>& frame_factory, const std::vector<std::wstring>& params)
+{
+       if(params.size() < 0)
+               return core::frame_producer::empty();
+
+       auto color2 = get_hex_color(params[0]);
+       if(color2.length() != 9 || color2[0] != '#')
+               return core::frame_producer::empty();
+
+       return spl::make_shared<color_producer>(frame_factory, color2);
+}
+
+draw_frame create_color_frame(void* tag, const spl::shared_ptr<frame_factory>& frame_factory, const std::wstring& color)
+{
+       auto color2 = get_hex_color(color);
+       if(color2.length() != 9 || color2[0] != '#')
+               CASPAR_THROW_EXCEPTION(invalid_argument() << arg_name_info("color") << arg_value_info(color2) << msg_info("Invalid color."));
+       
+       core::pixel_format_desc desc(pixel_format::bgra);
+       desc.planes.push_back(core::pixel_format_desc::plane(1, 1, 4));
+       auto frame = frame_factory->create_frame(tag, desc);
+               
+       // Read color from hex-string and write to frame pixel.
+
+       auto& value = *reinterpret_cast<uint32_t*>(frame.image_data(0).begin());
+       std::wstringstream str(color2.substr(1));
+       if(!(str >> std::hex >> value) || !str.eof())
+               CASPAR_THROW_EXCEPTION(invalid_argument() << arg_name_info("color") << arg_value_info(color2) << msg_info("Invalid color."));
+                       
+       return core::draw_frame(std::move(frame));
+}
+
 }}
\ No newline at end of file