4 * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
6 * This program 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 2 of the License, or
9 * (at your option) any later version.
11 * This program 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.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 #include "bcdisplayinfo.h"
26 #include "edlsession.h"
28 #include "motion-cv.h"
29 #include "motionwindow-cv.h"
31 #include "pluginserver.h"
33 MotionCVWindow::MotionCVWindow(MotionCVMain *plugin)
34 : PluginClientWindow(plugin, 815, 650, 815, 650, 0)
36 this->plugin = plugin;
39 MotionCVWindow::~MotionCVWindow()
43 void MotionCVWindow::create_objects()
45 int x1 = 10, x = 10, y = 10;
49 add_subwindow(global = new MotionCVGlobal(plugin, this, x1, y));
50 add_subwindow(rotate = new MotionCVRotate(plugin, this, x2, y));
53 add_subwindow(title = new BC_Title(x1, y,
54 _("Translation search radius:\n(W/H Percent of image)")));
55 add_subwindow(global_range_w = new GlobalRange(plugin,
56 x1 + title->get_w() + 10, y,
57 &plugin->config.global_range_w));
58 add_subwindow(global_range_h = new GlobalRange(plugin,
59 x1 + title->get_w() + 30 + global_range_w->get_w(), y,
60 &plugin->config.global_range_h));
62 add_subwindow(title = new BC_Title(x2, y,
63 _("Rotation search radius:\n(Degrees)")));
64 add_subwindow(rotation_range =
65 new RotationRange(plugin, x2 + title->get_w() + 10, y));
68 add_subwindow(title = new BC_Title(x1, y,
69 _("Translation block size:\n(W/H Percent of image)")));
70 add_subwindow(global_block_w = new BlockSize(plugin,
71 x1 + title->get_w() + 10, y,
72 &plugin->config.global_block_w));
73 add_subwindow(global_block_h = new BlockSize(plugin,
74 x1 + title->get_w() + 30 + global_block_w->get_w(), y,
75 &plugin->config.global_block_h));
77 add_subwindow(title = new BC_Title(x2, y,
78 _("Rotation block size:\n(W/H Percent of image)")));
79 add_subwindow(rotation_block_w = new BlockSize(plugin,
80 x2 + title->get_w() + 10, y,
81 &plugin->config.rotation_block_w));
82 add_subwindow(rotation_block_h = new BlockSize(plugin,
83 x2 + title->get_w() + 30 + rotation_block_w->get_w(), y,
84 &plugin->config.rotation_block_h));
87 add_subwindow(title = new BC_Title(x1, y, _("Translation search steps:")));
88 add_subwindow(global_search_positions = new GlobalSearchPositions(plugin,
89 x1 + title->get_w() + 10, y, 80));
90 global_search_positions->create_objects();
92 add_subwindow(title = new BC_Title(x2, y, _("Rotation search steps:")));
93 add_subwindow(rotation_search_positions = new RotationSearchPositions(plugin,
94 x2 + title->get_w() + 10, y, 80));
95 rotation_search_positions->create_objects();
98 add_subwindow(title = new BC_Title(x, y, _("Translation direction:")));
99 add_subwindow(mode3 = new Mode3(plugin,
100 this, x + title->get_w() + 10, y));
101 mode3->create_objects();
104 add_subwindow(title = new BC_Title(x, y + 10, _("Block X:")));
105 add_subwindow(block_x = new MotionCVBlockX(plugin, this,
106 x + title->get_w() + 10, y));
107 add_subwindow(block_x_text = new MotionCVBlockXText(plugin,
108 this, x + title->get_w() + 10 + block_x->get_w() + 10, y + 10));
111 add_subwindow(title = new BC_Title(x, y + 10, _("Block Y:")));
112 add_subwindow(block_y = new MotionCVBlockY(plugin, this,
113 x + title->get_w() + 10, y));
114 add_subwindow(block_y_text = new MotionCVBlockYText(plugin, this,
115 x + title->get_w() + 10 + block_y->get_w() + 10, y + 10));
118 add_subwindow(title = new BC_Title(x, y + 10, _("Maximum absolute offset:")));
119 add_subwindow(magnitude = new MotionCVMagnitude(plugin,
120 x + title->get_w() + 10, y));
123 add_subwindow(title = new BC_Title(x, y + 10, _("Settling speed:")));
124 add_subwindow(return_speed = new MotionCVReturnSpeed(plugin,
125 x + title->get_w() + 10, y));
128 add_subwindow(vectors = new MotionCVDrawVectors(plugin, this, x, y));
130 add_subwindow(title = new BC_Title(x2, y, _("Tracking file:")));
131 add_subwindow(tracking_file = new MotionCVTrackingFile(plugin,
132 plugin->config. tracking_file, this,
133 x2 + title->get_w() + 20, y));
137 add_subwindow(track_single =
138 new TrackSingleFrame(plugin, this, x1, y1));
139 add_subwindow(title =
140 new BC_Title(x1 += track_single->get_w() + 20, y1, _("Frame number:")));
141 add_subwindow(track_frame_number =
142 new TrackFrameNumber(plugin, this, x1 += title->get_w(), y1));
143 add_subwindow(addtrackedframeoffset =
144 new AddTrackedFrameOffset(plugin, this, x1, y1+=track_frame_number->get_h()));
145 int pef = client->server->mwindow->edl->session->video_every_frame;
146 add_subwindow(pef_title = new BC_Title(x1, y1+=addtrackedframeoffset->get_h() + 5,
147 !pef ? _("For best results\n"
148 " Set: Play every frame\n"
149 " Preferences-> Playback-> Video Out") :
150 _("Currently using: Play every frame"), MEDIUMFONT,
151 !pef ? RED : GREEN));
154 add_subwindow(track_previous = new TrackPreviousFrame(plugin, this, x, y));
156 add_subwindow(previous_same = new PreviousFrameSameBlock(plugin, this, x, y));
160 add_subwindow(title = new BC_Title(x, y, _("Master layer:")));
161 add_subwindow(master_layer = new MasterLayer(plugin, this,
162 x + title->get_w() + 10, y));
163 master_layer->create_objects();
166 add_subwindow(title = new BC_Title(x, y, _("Action:")));
167 add_subwindow(mode1 = new Mode1(plugin, this,
168 x + title->get_w() + 10, y));
169 mode1->create_objects();
172 add_subwindow(title = new BC_Title(x, y, _("Calculation:")));
173 add_subwindow(mode2 = new Mode2(plugin, this,
174 x + title->get_w() + 10, y));
175 mode2->create_objects();
180 void MotionCVWindow::update_mode()
182 global_range_w->update(plugin->config.global_range_w, MIN_RADIUS, MAX_RADIUS);
183 global_range_h->update(plugin->config.global_range_h, MIN_RADIUS, MAX_RADIUS);
184 rotation_range->update(plugin->config.rotation_range, MIN_ROTATION, MAX_ROTATION);
185 vectors->update(plugin->config.draw_vectors);
186 tracking_file->update(plugin->config.tracking_file);
187 global->update(plugin->config.global);
188 rotate->update(plugin->config.rotate);
189 addtrackedframeoffset->update(plugin->config.addtrackedframeoffset);
192 GlobalRange::GlobalRange(MotionCVMain *plugin, int x, int y, int *value)
193 :BC_IPot(x, y, (int64_t) * value, (int64_t) MIN_RADIUS, (int64_t) MAX_RADIUS)
195 this->plugin = plugin;
199 int GlobalRange::handle_event()
201 *value = (int)get_value();
202 plugin->send_configure_change();
206 RotationRange::RotationRange(MotionCVMain *plugin, int x, int y)
209 y, (int64_t) plugin->config.rotation_range,
210 (int64_t) MIN_ROTATION, (int64_t) MAX_ROTATION)
212 this->plugin = plugin;
215 int RotationRange::handle_event()
217 plugin->config.rotation_range = (int)get_value();
218 plugin->send_configure_change();
222 BlockSize::BlockSize(MotionCVMain *plugin, int x, int y, int *value)
223 : BC_IPot(x, y, (int64_t) * value, (int64_t) MIN_BLOCK, (int64_t) MAX_BLOCK)
225 this->plugin = plugin;
229 int BlockSize::handle_event()
231 *value = (int)get_value();
232 plugin->send_configure_change();
236 GlobalSearchPositions::GlobalSearchPositions(MotionCVMain *plugin,
238 : BC_PopupMenu(x, y, w, "", 1)
240 this->plugin = plugin;
243 void GlobalSearchPositions::create_objects()
245 add_item(new BC_MenuItem("64"));
246 add_item(new BC_MenuItem("128"));
247 add_item(new BC_MenuItem("256"));
248 add_item(new BC_MenuItem("512"));
249 add_item(new BC_MenuItem("1024"));
250 add_item(new BC_MenuItem("2048"));
251 add_item(new BC_MenuItem("4096"));
252 add_item(new BC_MenuItem("8192"));
253 add_item(new BC_MenuItem("16384"));
254 add_item(new BC_MenuItem("32768"));
255 add_item(new BC_MenuItem("65536"));
256 add_item(new BC_MenuItem("131072"));
257 char string[BCTEXTLEN];
258 sprintf(string, "%d", plugin->config.global_positions);
262 int GlobalSearchPositions::handle_event()
264 plugin->config.global_positions = atoi(get_text());
265 plugin->send_configure_change();
269 RotationSearchPositions::RotationSearchPositions(MotionCVMain *plugin,
271 : BC_PopupMenu(x, y, w, "", 1)
273 this->plugin = plugin;
276 void RotationSearchPositions::create_objects()
278 add_item(new BC_MenuItem("4"));
279 add_item(new BC_MenuItem("8"));
280 add_item(new BC_MenuItem("16"));
281 add_item(new BC_MenuItem("32"));
282 char string[BCTEXTLEN];
283 sprintf(string, "%d", plugin->config.rotate_positions);
287 int RotationSearchPositions::handle_event()
289 plugin->config.rotate_positions = atoi(get_text());
290 plugin->send_configure_change();
294 MotionCVMagnitude::MotionCVMagnitude(MotionCVMain *plugin, int x, int y)
295 : BC_IPot(x, y, (int64_t) plugin->config.magnitude, (int64_t) 0, (int64_t) 100)
297 this->plugin = plugin;
300 int MotionCVMagnitude::handle_event()
302 plugin->config.magnitude = (int)get_value();
303 plugin->send_configure_change();
307 MotionCVReturnSpeed::MotionCVReturnSpeed(MotionCVMain *plugin, int x, int y)
308 : BC_IPot(x, y, (int64_t) plugin->config.return_speed, (int64_t) 0, (int64_t) 100)
310 this->plugin = plugin;
313 int MotionCVReturnSpeed::handle_event()
315 plugin->config.return_speed = (int)get_value();
316 plugin->send_configure_change();
320 AddTrackedFrameOffset::AddTrackedFrameOffset(MotionCVMain *plugin,
321 MotionCVWindow *gui, int x, int y)
322 : BC_CheckBox(x, y, plugin->config.addtrackedframeoffset,
323 _("Add (loaded) offset from tracked frame"))
325 this->plugin = plugin;
329 int AddTrackedFrameOffset::handle_event()
331 plugin->config.addtrackedframeoffset = get_value();
332 plugin->send_configure_change();
336 MotionCVTrackingFile::MotionCVTrackingFile(MotionCVMain *plugin,
337 const char *filename, MotionCVWindow *gui, int x, int y)
338 : BC_TextBox(x, y, 250, 1, filename)
340 this->plugin = plugin;
344 int MotionCVTrackingFile::handle_event()
346 strcpy(plugin->config.tracking_file, get_text());
347 plugin->send_configure_change();
351 MotionCVGlobal::MotionCVGlobal(MotionCVMain *plugin,
352 MotionCVWindow *gui, int x, int y)
353 :BC_CheckBox(x, y, plugin->config.global, _("Track translation"))
355 this->plugin = plugin;
359 int MotionCVGlobal::handle_event()
361 plugin->config.global = get_value();
362 plugin->send_configure_change();
366 MotionCVRotate::MotionCVRotate(MotionCVMain *plugin,
367 MotionCVWindow *gui, int x, int y)
368 : BC_CheckBox(x, y, plugin->config.rotate, _("Track rotation"))
370 this->plugin = plugin;
374 int MotionCVRotate::handle_event()
376 plugin->config.rotate = get_value();
377 plugin->send_configure_change();
381 MotionCVBlockX::MotionCVBlockX(MotionCVMain *plugin,
382 MotionCVWindow *gui, int x, int y)
383 : BC_FPot(x, y, plugin->config.block_x, (float)0, (float)100)
385 this->plugin = plugin;
389 int MotionCVBlockX::handle_event()
391 plugin->config.block_x = get_value();
392 gui->block_x_text->update((float)plugin->config.block_x);
393 plugin->send_configure_change();
397 MotionCVBlockY::MotionCVBlockY(MotionCVMain *plugin,
398 MotionCVWindow *gui, int x, int y)
399 : BC_FPot(x, y, (float)plugin->config.block_y, (float)0, (float)100)
401 this->plugin = plugin;
405 int MotionCVBlockY::handle_event()
407 plugin->config.block_y = get_value();
408 gui->block_y_text->update((float)plugin->config.block_y);
409 plugin->send_configure_change();
413 MotionCVBlockXText::MotionCVBlockXText(MotionCVMain *plugin,
414 MotionCVWindow *gui, int x, int y)
415 : BC_TextBox(x, y, 75, 1, (float)plugin->config.block_x)
417 this->plugin = plugin;
422 int MotionCVBlockXText::handle_event()
424 plugin->config.block_x = atof(get_text());
425 gui->block_x->update(plugin->config.block_x);
426 plugin->send_configure_change();
430 MotionCVBlockYText::MotionCVBlockYText(MotionCVMain *plugin,
431 MotionCVWindow *gui, int x, int y)
432 : BC_TextBox(x, y, 75, 1, (float)plugin->config.block_y)
434 this->plugin = plugin;
439 int MotionCVBlockYText::handle_event()
441 plugin->config.block_y = atof(get_text());
442 gui->block_y->update(plugin->config.block_y);
443 plugin->send_configure_change();
447 MotionCVDrawVectors::MotionCVDrawVectors(MotionCVMain *plugin,
448 MotionCVWindow *gui, int x, int y)
449 : BC_CheckBox(x, y, plugin->config.draw_vectors, _("Draw vectors"))
452 this->plugin = plugin;
455 int MotionCVDrawVectors::handle_event()
457 plugin->config.draw_vectors = get_value();
458 plugin->send_configure_change();
462 TrackSingleFrame::TrackSingleFrame(MotionCVMain *plugin,
463 MotionCVWindow *gui, int x, int y)
466 y, plugin->config.mode3 == MotionCVConfig::TRACK_SINGLE,
467 _("Track single frame"))
469 this->plugin = plugin;
473 int TrackSingleFrame::handle_event()
475 plugin->config.mode3 = MotionCVConfig::TRACK_SINGLE;
476 gui->track_previous->update(0);
477 gui->previous_same->update(0);
478 gui->track_frame_number->enable();
479 plugin->send_configure_change();
483 TrackFrameNumber::TrackFrameNumber(MotionCVMain *plugin,
484 MotionCVWindow *gui, int x, int y)
485 : BC_TextBox(x, y, 100, 1, plugin->config.track_frame)
487 this->plugin = plugin;
489 if( plugin->config.mode3 != MotionCVConfig::TRACK_SINGLE )
493 int TrackFrameNumber::handle_event()
495 plugin->config.track_frame = atol(get_text());
496 plugin->send_configure_change();
500 TrackPreviousFrame::TrackPreviousFrame(MotionCVMain *plugin,
501 MotionCVWindow *gui, int x, int y)
504 y, plugin->config.mode3 == MotionCVConfig::TRACK_PREVIOUS,
505 _("Track previous frame"))
507 this->plugin = plugin;
511 int TrackPreviousFrame::handle_event()
513 plugin->config.mode3 = MotionCVConfig::TRACK_PREVIOUS;
514 gui->track_single->update(0);
515 gui->previous_same->update(0);
516 gui->track_frame_number->disable();
517 plugin->send_configure_change();
521 PreviousFrameSameBlock::PreviousFrameSameBlock(MotionCVMain *plugin,
522 MotionCVWindow *gui, int x, int y)
525 y, plugin->config.mode3 == MotionCVConfig::PREVIOUS_SAME_BLOCK,
526 _("Previous frame same block"))
528 this->plugin = plugin;
532 int PreviousFrameSameBlock::handle_event()
534 plugin->config.mode3 = MotionCVConfig::PREVIOUS_SAME_BLOCK;
535 gui->track_single->update(0);
536 gui->track_previous->update(0);
537 gui->track_frame_number->disable();
538 plugin->send_configure_change();
542 MasterLayer::MasterLayer(MotionCVMain *plugin, MotionCVWindow *gui, int x,
544 : BC_PopupMenu(x, y, calculate_w(gui),
545 to_text(plugin->config.bottom_is_master))
547 this->plugin = plugin;
551 int MasterLayer::handle_event()
553 plugin->config.bottom_is_master = from_text(get_text());
554 plugin->send_configure_change();
558 void MasterLayer::create_objects()
560 add_item(new BC_MenuItem(to_text(0)));
561 add_item(new BC_MenuItem(to_text(1)));
564 int MasterLayer::from_text(char *text)
566 if( !strcmp(text, _("Top")) ) return 0;
570 const char *MasterLayer::to_text(int mode)
572 return mode ? _("Bottom") : _("Top");
575 int MasterLayer::calculate_w(MotionCVWindow *gui)
578 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(0)));
579 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(1)));
583 Mode1::Mode1(MotionCVMain *plugin, MotionCVWindow *gui, int x, int y)
584 : BC_PopupMenu(x, y, calculate_w(gui), to_text(plugin->config.mode1))
586 this->plugin = plugin;
590 int Mode1::handle_event()
592 plugin->config.mode1 = from_text(get_text());
593 plugin->send_configure_change();
597 void Mode1::create_objects()
599 add_item(new BC_MenuItem(to_text(MotionCVConfig::TRACK)));
600 add_item(new BC_MenuItem(to_text(MotionCVConfig::TRACK_PIXEL)));
601 add_item(new BC_MenuItem(to_text(MotionCVConfig::STABILIZE)));
602 add_item(new BC_MenuItem(to_text(MotionCVConfig::STABILIZE_PIXEL)));
603 add_item(new BC_MenuItem(to_text(MotionCVConfig::NOTHING)));
606 int Mode1::from_text(char *text)
608 if( !strcmp(text, _("Track Subpixel")) ) return MotionCVConfig::TRACK;
609 if( !strcmp(text, _("Track Pixel")) ) return MotionCVConfig::TRACK_PIXEL;
610 if( !strcmp(text, _("Stabilize Subpixel")) ) return MotionCVConfig::STABILIZE;
611 if( !strcmp(text, _("Stabilize Pixel")) ) return MotionCVConfig::STABILIZE_PIXEL;
612 //if( !strcmp(text, _("Do Nothing")) ) return MotionCVConfig::NOTHING;
613 return MotionCVConfig::NOTHING;
616 const char *Mode1::to_text(int mode)
619 case MotionCVConfig::TRACK: return _("Track Subpixel");
620 case MotionCVConfig::TRACK_PIXEL: return _("Track Pixel");
621 case MotionCVConfig::STABILIZE: return _("Stabilize Subpixel");
622 case MotionCVConfig::STABILIZE_PIXEL: return _("Stabilize Pixel");
623 case MotionCVConfig::NOTHING: return _("Do Nothing");
628 int Mode1::calculate_w(MotionCVWindow *gui)
631 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(MotionCVConfig::TRACK)));
632 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(MotionCVConfig::TRACK_PIXEL)));
633 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(MotionCVConfig::STABILIZE)));
634 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(MotionCVConfig::STABILIZE_PIXEL)));
635 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(MotionCVConfig::NOTHING)));
639 Mode2::Mode2(MotionCVMain *plugin, MotionCVWindow *gui, int x, int y)
640 : BC_PopupMenu(x, y, calculate_w(gui), to_text(plugin->config.mode2))
642 this->plugin = plugin;
646 int Mode2::handle_event()
648 plugin->config.mode2 = from_text(get_text());
649 plugin->send_configure_change();
653 void Mode2::create_objects()
655 add_item(new BC_MenuItem(to_text(MotionCVConfig::SAVE)));
656 add_item(new BC_MenuItem(to_text(MotionCVConfig::LOAD)));
657 add_item(new BC_MenuItem(to_text(MotionCVConfig::RECALCULATE)));
658 add_item(new BC_MenuItem(to_text(MotionCVConfig::NO_CALCULATE)));
661 int Mode2::from_text(char *text)
663 if( !strcmp(text, _("Recalculate")) ) return MotionCVConfig::RECALCULATE;
664 if( !strcmp(text, _("Save coords to tracking file")) ) return MotionCVConfig::SAVE;
665 if( !strcmp(text, _("Load coords from tracking file")) ) return MotionCVConfig::LOAD;
666 //if( !strcmp(text, _("Don't Calculate")) ) return MotionCVConfig::NO_CALCULATE;
667 return MotionCVConfig::NO_CALCULATE;
670 const char *Mode2::to_text(int mode)
673 case MotionCVConfig::NO_CALCULATE: return _("Don't Calculate");
674 case MotionCVConfig::RECALCULATE: return _("Recalculate");
675 case MotionCVConfig::SAVE: return _("Save coords to tracking file");
676 case MotionCVConfig::LOAD: return _("Load coords from tracking file");
681 int Mode2::calculate_w(MotionCVWindow *gui)
684 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(MotionCVConfig::NO_CALCULATE)));
685 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(MotionCVConfig::RECALCULATE)));
686 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(MotionCVConfig::SAVE)));
687 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(MotionCVConfig::LOAD)));
691 Mode3::Mode3(MotionCVMain *plugin, MotionCVWindow *gui, int x, int y)
693 BC_PopupMenu(x, y, calculate_w(gui),
694 to_text(plugin->config.horizontal_only, plugin->config.vertical_only))
696 this->plugin = plugin;
700 int Mode3::handle_event()
702 from_text(&plugin->config.horizontal_only,
703 &plugin->config.vertical_only, get_text());
704 plugin->send_configure_change();
708 void Mode3::create_objects()
710 add_item(new BC_MenuItem(to_text(1, 0)));
711 add_item(new BC_MenuItem(to_text(0, 1)));
712 add_item(new BC_MenuItem(to_text(0, 0)));
715 void Mode3::from_text(int *horizontal_only, int *vertical_only, char *text)
717 *horizontal_only = 0;
719 if( !strcmp(text, to_text(1, 0)) ) *horizontal_only = 1;
720 if( !strcmp(text, to_text(0, 1)) ) *vertical_only = 1;
723 const char *Mode3::to_text(int horizontal_only, int vertical_only)
725 if( horizontal_only ) return _("Horizontal only");
726 if( vertical_only ) return _("Vertical only");
730 int Mode3::calculate_w(MotionCVWindow *gui)
733 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(1, 0)));
734 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(0, 1)));
735 result = MAX(result, gui->get_text_width(MEDIUMFONT, to_text(0, 0)));