]> git.sesse.net Git - casparcg/blob - core/producer/scene/scene_producer.cpp
Refactored to use range based for instead of BOOST_FOREACH
[casparcg] / core / producer / scene / scene_producer.cpp
1 /*
2 * Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>
3 *
4 * This file is part of CasparCG (www.casparcg.com).
5 *
6 * CasparCG is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation, either version 3 of the License, or
9 * (at your option) any later version.
10 *
11 * CasparCG is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with CasparCG. If not, see <http://www.gnu.org/licenses/>.
18 *
19 * Author: Helge Norberg, helge.norberg@svt.se
20 */
21
22 #include "../../stdafx.h"
23
24 #include <common/future.h>
25 #include <boost/algorithm/string/split.hpp>
26 #include <boost/algorithm/string.hpp>
27
28 #include "scene_producer.h"
29
30 #include "../../frame/draw_frame.h"
31 #include "../../interaction/interaction_aggregator.h"
32 #include "../text/text_producer.h"
33
34 namespace caspar { namespace core { namespace scene {
35
36 layer::layer(const std::wstring& name, const spl::shared_ptr<frame_producer>& producer)
37         : name(name), producer(producer)
38 {
39         clipping.width.bind(producer.get()->pixel_constraints().width);
40         clipping.height.bind(producer.get()->pixel_constraints().height);
41 }
42
43 adjustments::adjustments()
44         : opacity(1.0)
45 {
46 }
47
48 struct timeline
49 {
50         std::map<int64_t, keyframe> keyframes;
51
52         void on_frame(int64_t frame)
53         {
54                 auto before = --keyframes.upper_bound(frame);
55                 bool found_before = before != keyframes.end() && before->first < frame;
56                 auto after = keyframes.upper_bound(frame);
57                 bool found_after = after != keyframes.end() && after->first > frame;
58                 auto exact_frame = keyframes.find(frame);
59                 bool found_exact_frame = exact_frame != keyframes.end();
60
61                 if (found_exact_frame)
62                 {
63                         exact_frame->second.on_destination_frame();
64
65                         auto next_frame = ++exact_frame;
66
67                         if (next_frame != keyframes.end() && next_frame->second.on_start_animate)
68                                 next_frame->second.on_start_animate();
69                 }
70                 else if (found_after)
71                 {
72                         int64_t start_frame = 0;
73
74                         if (found_before)
75                         {
76                                 start_frame = before->first;
77                         }
78
79                         if (after->second.on_start_animate && frame == 0)
80                                 after->second.on_start_animate();
81                         else if (after->second.on_animate_to)
82                                 after->second.on_animate_to(start_frame, frame);
83                 }
84         }
85 };
86
87 struct scene_producer::impl
88 {
89         constraints pixel_constraints_;
90         std::list<layer> layers_;
91         interaction_aggregator aggregator_;
92         binding<int64_t> frame_number_;
93         binding<double> speed_;
94         double frame_fraction_;
95         std::map<void*, timeline> timelines_;
96         std::map<std::wstring, std::shared_ptr<core::variable>> variables_;
97         std::vector<std::wstring> variable_names_;
98         monitor::subject monitor_subject_;
99
100         impl(int width, int height)
101                 : pixel_constraints_(width, height)
102                 , aggregator_([=] (double x, double y) { return collission_detect(x, y); })
103                 , frame_fraction_(0)
104         {
105                 auto speed_variable = std::make_shared<core::variable_impl<double>>(L"1.0", true, 1.0);
106                 store_variable(L"scene_speed", speed_variable);
107                 speed_ = speed_variable->value();
108                 auto frame_variable = std::make_shared<core::variable_impl<int64_t>>(L"0", true, 0);
109                 store_variable(L"frame", frame_variable);
110                 frame_number_ = frame_variable->value();
111         }
112
113         layer& create_layer(
114                         const spl::shared_ptr<frame_producer>& producer, int x, int y, const std::wstring& name)
115         {
116                 layer layer(name, producer);
117
118                 layer.position.x.set(x);
119                 layer.position.y.set(y);
120
121                 layers_.push_back(layer);
122
123                 return layers_.back();
124         }
125
126         void store_keyframe(void* timeline_identity, const keyframe& k)
127         {
128                 timelines_[timeline_identity].keyframes.insert(std::make_pair(k.destination_frame, k));
129         }
130
131         void store_variable(
132                         const std::wstring& name, const std::shared_ptr<core::variable>& var)
133         {
134                 variables_.insert(std::make_pair(name, var));
135                 variable_names_.push_back(name);
136         }
137
138         core::variable& get_variable(const std::wstring& name)
139         {
140                 auto found = variables_.find(name);
141
142                 if (found == variables_.end())
143                         CASPAR_THROW_EXCEPTION(caspar_exception() << msg_info(name + L" not found in scene"));
144
145                 return *found->second;
146         }
147
148         const std::vector<std::wstring>& get_variables() const
149         {
150                 return variable_names_;
151         }
152
153         binding<int64_t> frame()
154         {
155                 return frame_number_;
156         }
157
158         frame_transform get_transform(const layer& layer) const
159         {
160                 frame_transform transform;
161
162                 auto& pos = transform.image_transform.fill_translation;
163                 auto& scale = transform.image_transform.fill_scale;
164                 //auto& clip_pos = transform.image_transform.clip_translation;
165                 //auto& clip_scale = transform.image_transform.clip_scale;
166
167                 pos[0] = static_cast<double>(layer.position.x.get()) / static_cast<double>(pixel_constraints_.width.get());
168                 pos[1] = static_cast<double>(layer.position.y.get()) / static_cast<double>(pixel_constraints_.height.get());
169                 scale[0] = static_cast<double>(layer.producer.get()->pixel_constraints().width.get())
170                                 / static_cast<double>(pixel_constraints_.width.get());
171                 scale[1] = static_cast<double>(layer.producer.get()->pixel_constraints().height.get())
172                                 / static_cast<double>(pixel_constraints_.height.get());
173
174                 /*clip_pos[0] = static_cast<double>(layer.clipping.upper_left.x.get()) / static_cast<double>(pixel_constraints_.width.get());
175                 clip_pos[1] = static_cast<double>(layer.clipping.upper_left.y.get()) / static_cast<double>(pixel_constraints_.height.get());
176                 clip_scale[0] = static_cast<double>(layer.clipping.width.get()) / static_cast<double>(pixel_constraints_.width.get());
177                 clip_scale[1] = static_cast<double>(layer.clipping.height.get()) / static_cast<double>(pixel_constraints_.height.get());*/
178
179                 transform.image_transform.opacity = layer.adjustments.opacity.get();
180                 transform.image_transform.is_key = layer.is_key.get();
181
182                 return transform;
183         }
184
185         draw_frame render_frame()
186         {
187                 for (auto& timeline : timelines_)
188                         timeline.second.on_frame(frame_number_.get());
189
190                 std::vector<draw_frame> frames;
191
192                 for (auto& layer : layers_)
193                 {
194                         if (layer.hidden.get())
195                                 continue;
196
197                         draw_frame frame(layer.producer.get()->receive());
198                         frame.transform() = get_transform(layer);;
199                         frames.push_back(frame);
200                 }
201
202                 frame_fraction_ += speed_.get();
203
204                 if (std::abs(frame_fraction_) >= 1.0)
205                 {
206                         int64_t delta = static_cast<int64_t>(frame_fraction_);
207                         frame_number_.set(frame_number_.get() + delta);
208                         frame_fraction_ -= delta;
209                 }
210
211                 return draw_frame(frames);
212         }
213
214         void on_interaction(const interaction_event::ptr& event)
215         {
216                 aggregator_.translate_and_send(event);
217         }
218
219         bool collides(double x, double y) const
220         {
221                 return static_cast<bool>((collission_detect(x, y)));
222         }
223
224         boost::optional<interaction_target> collission_detect(double x, double y) const
225         {
226                 for (auto& layer : layers_ | boost::adaptors::reversed)
227                 {
228                         if (layer.hidden.get())
229                                 continue;
230
231                         auto transform = get_transform(layer);
232                         auto translated = translate(x, y, transform);
233
234                         if (translated.first >= 0.0
235                                 && translated.first <= 1.0
236                                 && translated.second >= 0.0
237                                 && translated.second <= 1.0
238                                 && layer.producer.get()->collides(translated.first, translated.second))
239                         {
240                                 return std::make_pair(transform, layer.producer.get().get());
241                         }
242                 }
243
244                 return boost::optional<interaction_target>();
245         }
246
247         std::future<std::wstring> call(const std::vector<std::wstring>& params) 
248         {
249                 for (int i = 0; i + 1 < params.size(); i += 2)
250                 {
251                         auto found = variables_.find(boost::to_lower_copy(params[i]));
252
253                         if (found != variables_.end() && found->second->is_public())
254                                 found->second->from_string(params[i + 1]);
255                 }
256
257                 return make_ready_future(std::wstring(L""));
258         }
259
260         std::wstring print() const
261         {
262                 return L"scene[]";
263         }
264
265         std::wstring name() const
266         {
267                 return L"scene";
268         }
269         
270         boost::property_tree::wptree info() const
271         {
272                 boost::property_tree::wptree info;
273                 info.add(L"type", L"scene");
274                 return info;
275         }
276
277         monitor::subject& monitor_output()
278         {
279                 return monitor_subject_;
280         }
281 };
282
283 scene_producer::scene_producer(int width, int height)
284         : impl_(new impl(width, height))
285 {
286 }
287
288 scene_producer::~scene_producer()
289 {
290 }
291
292 layer& scene_producer::create_layer(
293                 const spl::shared_ptr<frame_producer>& producer, int x, int y, const std::wstring& name)
294 {
295         return impl_->create_layer(producer, x, y, name);
296 }
297
298 layer& scene_producer::create_layer(
299                 const spl::shared_ptr<frame_producer>& producer, const std::wstring& name)
300 {
301         return impl_->create_layer(producer, 0, 0, name);
302 }
303
304 binding<int64_t> scene_producer::frame()
305 {
306         return impl_->frame();
307 }
308
309 draw_frame scene_producer::receive_impl()
310 {
311         return impl_->render_frame();
312 }
313
314 constraints& scene_producer::pixel_constraints() { return impl_->pixel_constraints_; }
315
316 void scene_producer::on_interaction(const interaction_event::ptr& event)
317 {
318         impl_->on_interaction(event);
319 }
320
321 bool scene_producer::collides(double x, double y) const
322 {
323         return impl_->collides(x, y);
324 }
325
326 std::wstring scene_producer::print() const
327 {
328         return impl_->print();
329 }
330
331 std::wstring scene_producer::name() const
332 {
333         return impl_->name();
334 }
335
336 boost::property_tree::wptree scene_producer::info() const
337 {
338         return impl_->info();
339 }
340
341 std::future<std::wstring> scene_producer::call(const std::vector<std::wstring>& params) 
342 {
343         return impl_->call(params);
344 }
345
346 monitor::subject& scene_producer::monitor_output()
347 {
348         return impl_->monitor_output();
349 }
350
351 void scene_producer::store_keyframe(void* timeline_identity, const keyframe& k)
352 {
353         impl_->store_keyframe(timeline_identity, k);
354 }
355
356 void scene_producer::store_variable(
357                 const std::wstring& name, const std::shared_ptr<core::variable>& var)
358 {
359         impl_->store_variable(name, var);
360 }
361
362 core::variable& scene_producer::get_variable(const std::wstring& name)
363 {
364         return impl_->get_variable(name);
365 }
366
367 const std::vector<std::wstring>& scene_producer::get_variables() const
368 {
369         return impl_->get_variables();
370 }
371
372 spl::shared_ptr<core::frame_producer> create_dummy_scene_producer(const spl::shared_ptr<core::frame_factory>& frame_factory, const video_format_desc& format_desc, const std::vector<std::wstring>& params)
373 {
374         if (params.size() < 1 || !boost::iequals(params.at(0), L"[SCENE]"))
375                 return core::frame_producer::empty();
376
377         auto scene = spl::make_shared<scene_producer>(format_desc.width, format_desc.height);
378
379         text::text_info text_info;
380         text_info.font = L"Arial";
381         text_info.size = 62;
382         text_info.color.r = 1;
383         text_info.color.g = 1;
384         text_info.color.b = 1;
385         text_info.color.a = 0.5;
386         auto text_area = text_producer::create(frame_factory, 0, 0, L"a", text_info, 1280, 720, false);
387
388         auto text_width = text_area->pixel_constraints().width;
389         binding<double> padding(1.0);
390         binding<double> panel_width = padding + text_width + padding;
391         binding<double> panel_height = padding + text_area->pixel_constraints().height + padding;
392
393         auto subscription = panel_width.on_change([&]
394         {
395                 CASPAR_LOG(info) << "Panel width: " << panel_width.get();
396         });
397
398         padding.set(2);
399
400         auto create_param = [](std::wstring elem) -> std::vector<std::wstring>
401         {
402                 std::vector<std::wstring> result;
403                 result.push_back(elem);
404                 return result;
405         };
406
407         auto& car_layer = scene->create_layer(create_producer(frame_factory, format_desc, create_param(L"car")), L"car");
408         car_layer.clipping.upper_left.x.set(80);
409         car_layer.clipping.upper_left.y.set(45);
410         car_layer.clipping.width.unbind();
411         car_layer.clipping.width.set(640);
412         car_layer.clipping.height.unbind();
413         car_layer.clipping.height.set(360);
414         car_layer.adjustments.opacity.set(0.5);
415         //car_layer.hidden = scene->frame() % 50 > 25 || !(scene->frame() < 1000);
416         std::vector<std::wstring> sub_params;
417         sub_params.push_back(L"[FREEHAND]");
418         sub_params.push_back(L"640");
419         sub_params.push_back(L"360");
420         scene->create_layer(create_producer(frame_factory, format_desc, sub_params), 10, 10, L"freehand");
421         sub_params.clear();
422
423         auto& color_layer = scene->create_layer(create_producer(frame_factory, format_desc, create_param(L"RED")), 110, 10, L"color");
424         color_layer.producer.get()->pixel_constraints().width.set(1000);
425         color_layer.producer.get()->pixel_constraints().height.set(550);
426
427         //scene->create_layer(create_producer(frame_factory, format_desc, create_param(L"SP")), 50, 50);
428
429         auto& upper_left = scene->create_layer(create_producer(frame_factory, format_desc, create_param(L"scene/upper_left")), L"upper_left");
430         auto& upper_right = scene->create_layer(create_producer(frame_factory, format_desc, create_param(L"scene/upper_right")), L"upper_right");
431         auto& lower_left = scene->create_layer(create_producer(frame_factory, format_desc, create_param(L"scene/lower_left")), L"lower_left");
432         auto& lower_right = scene->create_layer(create_producer(frame_factory, format_desc, create_param(L"scene/lower_right")), L"lower_right");
433         auto& text_layer = scene->create_layer(text_area, L"text_area");
434         upper_left.adjustments.opacity.bind(text_layer.adjustments.opacity);
435         upper_right.adjustments.opacity.bind(text_layer.adjustments.opacity);
436         lower_left.adjustments.opacity.bind(text_layer.adjustments.opacity);
437         lower_right.adjustments.opacity.bind(text_layer.adjustments.opacity);
438
439         /*
440         binding<double> panel_x = (scene->frame()
441                         .as<double>()
442                         .transformed([](double v) { return std::sin(v / 20.0); })
443                         * 20.0
444                         + 40.0)
445                         .transformed([](double v) { return std::floor(v); }); // snap to pixels instead of subpixels
446                         */
447         tweener tween(L"easeoutbounce");
448         binding<double> panel_x(0.0);
449
450         scene->add_keyframe(panel_x, -panel_width, 0);
451         scene->add_keyframe(panel_x, 300.0, 50, L"easeinoutsine");
452         scene->add_keyframe(panel_x, 300.0, 50 * 4);
453         scene->add_keyframe(panel_x, 1000.0, 50 * 5, L"easeinoutsine");
454         //panel_x = delay(panel_x, add_tween(panel_x, scene->frame(), 200.0, int64_t(50), L"linear"), scene->frame(), int64_t(100));
455         /*binding<double> panel_x = when(scene->frame() < 50)
456                 .then(scene->frame().as<double>().transformed([tween](double t) { return tween(t, 0.0, 200, 50); }))
457                 .otherwise(200.0);*/
458         //binding<double> panel_y = when(car_layer.hidden).then(500.0).otherwise(-panel_x + 300);
459         binding<double> panel_y(500.0);
460         scene->add_keyframe(panel_y, panel_y.get(), 50 * 4);
461         scene->add_keyframe(panel_y, 720.0, 50 * 5, L"easeinexpo");
462
463         scene->add_keyframe(text_layer.adjustments.opacity, 1.0, 100);
464         scene->add_keyframe(text_layer.adjustments.opacity, 0.0, 125, L"linear");
465         scene->add_keyframe(text_layer.adjustments.opacity, 1.0, 150, L"linear");
466
467         upper_left.position.x = panel_x;
468         upper_left.position.y = panel_y;
469         upper_right.position.x = upper_left.position.x + upper_left.producer.get()->pixel_constraints().width + panel_width;
470         upper_right.position.y = upper_left.position.y;
471         lower_left.position.x = upper_left.position.x;
472         lower_left.position.y = upper_left.position.y + upper_left.producer.get()->pixel_constraints().height + panel_height;
473         lower_right.position.x = upper_right.position.x;
474         lower_right.position.y = lower_left.position.y;
475         text_layer.position.x = upper_left.position.x + upper_left.producer.get()->pixel_constraints().width + padding;
476         text_layer.position.y = upper_left.position.y + upper_left.producer.get()->pixel_constraints().height + padding + text_area->current_bearing_y().as<double>();
477
478         text_area->text().bind(scene->create_variable<std::wstring>(L"text", true));
479
480         auto params2 = params;
481         params2.erase(params2.begin());
482
483         scene->call(params2);
484
485         return scene;
486 }
487
488 }}}