Support for keeping video in sequence when changing lengths; tie selection in timelin...
[dcpomatic.git] / src / wx / timeline.cc
index 3a833e701de09a6e4efc44214976f2ac7283ef2d..42f489c2f5d438c4cdb96944ab821511b9a84793 100644 (file)
@@ -1,5 +1,3 @@
-/* -*- c-basic-offset: 8; default-tab-width: 8; -*- */
-
 /*
     Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
 
 
 #include <list>
 #include <wx/graphics.h>
+#include <boost/weak_ptr.hpp>
 #include "film.h"
+#include "film_editor.h"
 #include "timeline.h"
 #include "wx_util.h"
-#include "playlist.h"
+#include "lib/playlist.h"
 
 using std::list;
 using std::cout;
 using std::max;
 using boost::shared_ptr;
+using boost::weak_ptr;
 using boost::dynamic_pointer_cast;
 using boost::bind;
 
@@ -42,31 +43,89 @@ public:
 
        }
                
-       virtual void paint (wxGraphicsContext *) = 0;
+       void paint (wxGraphicsContext* g)
+       {
+               _last_paint_bbox = bbox ();
+               do_paint (g);
+       }
+       
+       void force_redraw ()
+       {
+               _timeline.force_redraw (_last_paint_bbox);
+               _timeline.force_redraw (bbox ());
+       }
+
        virtual Rect bbox () const = 0;
 
 protected:
+       virtual void do_paint (wxGraphicsContext *) = 0;
+       
        int time_x (Time t) const
        {
                return _timeline.tracks_position().x + t * _timeline.pixels_per_time_unit();
        }
        
        Timeline& _timeline;
+
+private:
+       Rect _last_paint_bbox;
 };
 
 class ContentView : public View
 {
 public:
-       ContentView (Timeline& tl, shared_ptr<const Content> c, int t)
+       ContentView (Timeline& tl, shared_ptr<Content> c, int t)
                : View (tl)
                , _content (c)
                , _track (t)
                , _selected (false)
        {
+               _content_connection = c->Changed.connect (bind (&ContentView::content_changed, this, _2));
+       }
+
+       Rect bbox () const
+       {
+               shared_ptr<const Film> film = _timeline.film ();
+               shared_ptr<const Content> content = _content.lock ();
+               if (!film || !content) {
+                       return Rect ();
+               }
+               
+               return Rect (
+                       time_x (content->start ()) - 8,
+                       y_pos (_track) - 8,
+                       content->length () * _timeline.pixels_per_time_unit() + 16,
+                       _timeline.track_height() + 16
+                       );
+       }
+
+       void set_selected (bool s) {
+               _selected = s;
+               force_redraw ();
+       }
+       
+       bool selected () const {
+               return _selected;
+       }
+
+       weak_ptr<Content> content () const {
+               return _content;
+       }
+
+       void set_track (int t) {
+               _track = t;
+       }
 
+       int track () const {
+               return _track;
        }
 
-       void paint (wxGraphicsContext* gc)
+       virtual wxString type () const = 0;
+       virtual wxColour colour () const = 0;
+       
+private:
+
+       void do_paint (wxGraphicsContext* gc)
        {
                shared_ptr<const Film> film = _timeline.film ();
                shared_ptr<const Content> content = _content.lock ();
@@ -74,33 +133,35 @@ public:
                        return;
                }
 
-               Time const start = content->time ();
-               Time const len = content->length (film);
+               Time const start = content->start ();
+               Time const len = content->length ();
+
+               wxColour selected (colour().Red() / 2, colour().Green() / 2, colour().Blue() / 2);
 
                gc->SetPen (*wxBLACK_PEN);
                
 #if wxMAJOR_VERSION == 2 && wxMINOR_VERSION >= 9
                gc->SetPen (*wxThePenList->FindOrCreatePen (wxColour (0, 0, 0), 4, wxPENSTYLE_SOLID));
                if (_selected) {
-                       gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (wxColour (200, 200, 200), wxBRUSHSTYLE_SOLID));
+                       gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (selected, wxBRUSHSTYLE_SOLID));
                } else {
                        gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (colour(), wxBRUSHSTYLE_SOLID));
                }
 #else                  
                gc->SetPen (*wxThePenList->FindOrCreatePen (wxColour (0, 0, 0), 4, wxSOLID));
                if (_selected) {
-                       gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (wxColour (200, 200, 200), wxSOLID));
+                       gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (selected, wxSOLID));
                } else {
                        gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (colour(), wxSOLID));
                }
 #endif
                
                wxGraphicsPath path = gc->CreatePath ();
-               path.MoveToPoint    (time_x (start),       y_pos (_track));
-               path.AddLineToPoint (time_x (start + len), y_pos (_track));
-               path.AddLineToPoint (time_x (start + len), y_pos (_track + 1));
-               path.AddLineToPoint (time_x (start),       y_pos (_track + 1));
-               path.AddLineToPoint (time_x (start),       y_pos (_track));
+               path.MoveToPoint    (time_x (start),       y_pos (_track) + 4);
+               path.AddLineToPoint (time_x (start + len), y_pos (_track) + 4);
+               path.AddLineToPoint (time_x (start + len), y_pos (_track + 1) - 4);
+               path.AddLineToPoint (time_x (start),       y_pos (_track + 1) - 4);
+               path.AddLineToPoint (time_x (start),       y_pos (_track) + 4);
                gc->StrokePath (path);
                gc->FillPath (path);
 
@@ -116,50 +177,29 @@ public:
                gc->ResetClip ();
        }
 
-       Rect bbox () const
+       int y_pos (int t) const
        {
-               shared_ptr<const Film> film = _timeline.film ();
-               shared_ptr<const Content> content = _content.lock ();
-               if (!film || !content) {
-                       return Rect ();
-               }
-               
-               return Rect (
-                       time_x (content->time ()),
-                       y_pos (_track),
-                       content->length (film) * _timeline.pixels_per_time_unit(),
-                       _timeline.track_height()
-                       );
-       }
-
-       void set_selected (bool s) {
-               _selected = s;
-               _timeline.force_redraw (bbox ());
-       }
-       
-       bool selected () const {
-               return _selected;
+               return _timeline.tracks_position().y + t * _timeline.track_height();
        }
 
-       virtual wxString type () const = 0;
-       virtual wxColour colour () const = 0;
-       
-private:
-       
-       int y_pos (int t) const
+       void content_changed (int p)
        {
-               return _timeline.tracks_position().y + t * _timeline.track_height();
+               if (p == ContentProperty::START || p == ContentProperty::LENGTH) {
+                       force_redraw ();
+               }
        }
 
-       boost::weak_ptr<const Content> _content;
+       boost::weak_ptr<Content> _content;
        int _track;
        bool _selected;
+
+       boost::signals2::scoped_connection _content_connection;
 };
 
 class AudioContentView : public ContentView
 {
 public:
-       AudioContentView (Timeline& tl, shared_ptr<const Content> c, int t)
+       AudioContentView (Timeline& tl, shared_ptr<Content> c, int t)
                : ContentView (tl, c, t)
        {}
        
@@ -178,7 +218,7 @@ private:
 class VideoContentView : public ContentView
 {
 public:
-       VideoContentView (Timeline& tl, shared_ptr<const Content> c, int t)
+       VideoContentView (Timeline& tl, shared_ptr<Content> c, int t)
                : ContentView (tl, c, t)
        {}
 
@@ -203,7 +243,7 @@ public:
                , _y (y)
        {}
 
-       void paint (wxGraphicsContext* gc)
+       void do_paint (wxGraphicsContext* gc)
        {
 #if wxMAJOR_VERSION == 2 && wxMINOR_VERSION >= 9
                gc->SetPen (*wxThePenList->FindOrCreatePen (wxColour (0, 0, 0), 1, wxPENSTYLE_SOLID));
@@ -273,10 +313,15 @@ private:
        int _y;
 };
 
-Timeline::Timeline (wxWindow* parent, shared_ptr<const Film> film)
+Timeline::Timeline (wxWindow* parent, FilmEditor* ed, shared_ptr<Film> film)
        : wxPanel (parent, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxFULL_REPAINT_ON_RESIZE)
+       , _film_editor (ed)
        , _film (film)
+       , _tracks (0)
        , _pixels_per_time_unit (0)
+       , _left_down (false)
+       , _down_view_start (0)
+       , _first_move (false)
 {
        SetDoubleBuffered (true);
 
@@ -284,14 +329,15 @@ Timeline::Timeline (wxWindow* parent, shared_ptr<const Film> film)
        
        Connect (wxID_ANY, wxEVT_PAINT, wxPaintEventHandler (Timeline::paint), 0, this);
        Connect (wxID_ANY, wxEVT_LEFT_DOWN, wxMouseEventHandler (Timeline::left_down), 0, this);
+       Connect (wxID_ANY, wxEVT_LEFT_UP, wxMouseEventHandler (Timeline::left_up), 0, this);
+       Connect (wxID_ANY, wxEVT_MOTION, wxMouseEventHandler (Timeline::mouse_moved), 0, this);
        Connect (wxID_ANY, wxEVT_SIZE, wxSizeEventHandler (Timeline::resized), 0, this);
 
-       SetMinSize (wxSize (640, tracks() * track_height() + 96));
-
        playlist_changed ();
 
-       film->playlist()->Changed.connect (bind (&Timeline::playlist_changed, this));
-       film->playlist()->ContentChanged.connect (bind (&Timeline::playlist_changed, this));
+       SetMinSize (wxSize (640, tracks() * track_height() + 96));
+
+       _playlist_connection = film->playlist()->Changed.connect (bind (&Timeline::playlist_changed, this));
 }
 
 void
@@ -328,21 +374,78 @@ Timeline::playlist_changed ()
        for (Playlist::ContentList::iterator i = content.begin(); i != content.end(); ++i) {
                if (dynamic_pointer_cast<VideoContent> (*i)) {
                        _views.push_back (shared_ptr<View> (new VideoContentView (*this, *i, 0)));
-               } else {
-                       _views.push_back (shared_ptr<View> (new AudioContentView (*this, *i, 1)));
+               }
+               if (dynamic_pointer_cast<AudioContent> (*i)) {
+                       _views.push_back (shared_ptr<View> (new AudioContentView (*this, *i, 0)));
                }
        }
 
+       assign_tracks ();
+       
        _views.push_back (shared_ptr<View> (new TimeAxisView (*this, tracks() * track_height() + 32)));
                
        Refresh ();
 }
 
+void
+Timeline::assign_tracks ()
+{
+       for (list<shared_ptr<View> >::iterator i = _views.begin(); i != _views.end(); ++i) {
+               shared_ptr<ContentView> cv = dynamic_pointer_cast<ContentView> (*i);
+               if (cv) {
+                       cv->set_track (0);
+               }
+       }
+
+       for (list<shared_ptr<View> >::iterator i = _views.begin(); i != _views.end(); ++i) {
+               shared_ptr<AudioContentView> acv = dynamic_pointer_cast<AudioContentView> (*i);
+               if (!acv) {
+                       continue;
+               }
+       
+               shared_ptr<Content> acv_content = acv->content().lock ();
+               assert (acv_content);
+               
+               int t = 1;
+               while (1) {
+                       list<shared_ptr<View> >::iterator j = _views.begin();
+                       while (j != _views.end()) {
+                               shared_ptr<AudioContentView> test = dynamic_pointer_cast<AudioContentView> (*j);
+                               if (!test) {
+                                       ++j;
+                                       continue;
+                               }
+                               
+                               shared_ptr<Content> test_content = test->content().lock ();
+                               assert (test_content);
+                                       
+                               if (test && test->track() == t) {
+                                       if ((acv_content->start() <= test_content->start() && test_content->start() <= acv_content->end()) ||
+                                           (acv_content->start() <= test_content->end()   && test_content->end()   <= acv_content->end())) {
+                                               /* we have an overlap on track `t' */
+                                               ++t;
+                                               break;
+                                       }
+                               }
+                               
+                               ++j;
+                       }
+
+                       if (j == _views.end ()) {
+                               /* no overlap on `t' */
+                               break;
+                       }
+               }
+
+               acv->set_track (t);
+               _tracks = max (_tracks, t + 1);
+       }
+}
+
 int
 Timeline::tracks () const
 {
-       /* XXX */
-       return 2;
+       return _tracks;
 }
 
 void
@@ -365,10 +468,65 @@ Timeline::left_down (wxMouseEvent& ev)
                ++i;
        }
 
+       _down_view.reset ();
+
+       if (i != _views.end ()) {
+               shared_ptr<ContentView> cv = dynamic_pointer_cast<ContentView> (*i);
+               if (cv) {
+                       _down_view = cv;
+                       shared_ptr<Content> c = cv->content().lock();
+                       assert (c);
+                       _down_view_start = c->start ();
+               }
+       }
+
        for (list<shared_ptr<View> >::iterator j = _views.begin(); j != _views.end(); ++j) {
                shared_ptr<ContentView> cv = dynamic_pointer_cast<ContentView> (*j);
                if (cv) {
                        cv->set_selected (i == j);
+                       if (i == j) {
+                               _film_editor->set_selection (cv->content ());
+                       }
+               }
+       }
+
+       _left_down = true;
+       _down_point = ev.GetPosition ();
+       _first_move = false;
+}
+
+void
+Timeline::left_up (wxMouseEvent &)
+{
+       _left_down = false;
+}
+
+void
+Timeline::mouse_moved (wxMouseEvent& ev)
+{
+       if (!_left_down) {
+               return;
+       }
+
+       wxPoint const p = ev.GetPosition();
+
+       if (!_first_move) {
+               int const dist = sqrt (pow (p.x - _down_point.x, 2) + pow (p.y - _down_point.y, 2));
+               if (dist < 8) {
+                       return;
+               }
+               _first_move = true;
+       }
+
+       Time const time_diff = (p.x - _down_point.x) / _pixels_per_time_unit;
+       if (_down_view) {
+               shared_ptr<Content> c = _down_view->content().lock();
+               if (c) {
+                       c->set_start (max (static_cast<Time> (0), _down_view_start + time_diff));
+
+                       shared_ptr<Film> film = _film.lock ();
+                       assert (film);
+                       film->set_sequence_video (false);
                }
        }
 }