Support for keeping video in sequence when changing lengths; tie selection in timelin...
[dcpomatic.git] / src / wx / timeline.cc
index 7e0ecf27e5e93bfa752bdc01ccb5ef2041c84f56..42f489c2f5d438c4cdb96944ab821511b9a84793 100644 (file)
 
 #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;
 
-int const Timeline::_track_height = 64;
-
-Timeline::Timeline (wxWindow* parent, shared_ptr<Playlist> pl)
-       : wxPanel (parent)
-       , _playlist (pl)
+class View
 {
-       SetDoubleBuffered (true);
+public:
+       View (Timeline& t)
+               : _timeline (t)
+       {
+
+       }
+               
+       void paint (wxGraphicsContext* g)
+       {
+               _last_paint_bbox = bbox ();
+               do_paint (g);
+       }
        
-       Connect (wxID_ANY, wxEVT_PAINT, wxPaintEventHandler (Timeline::paint), 0, this);
+       void force_redraw ()
+       {
+               _timeline.force_redraw (_last_paint_bbox);
+               _timeline.force_redraw (bbox ());
+       }
+
+       virtual Rect bbox () const = 0;
 
-       if (pl->audio_from() == Playlist::AUDIO_FFMPEG) {
-               SetMinSize (wxSize (640, _track_height * 2 + 96));
-       } else {
-               SetMinSize (wxSize (640, _track_height * (max (size_t (1), pl->audio().size()) + 1) + 96));
+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;
 
-       pl->Changed.connect (bind (&Timeline::playlist_changed, this));
-       pl->ContentChanged.connect (bind (&Timeline::playlist_changed, this));
-}
+private:
+       Rect _last_paint_bbox;
+};
 
-template <class T>
-int
-plot_content_list (
-       list<shared_ptr<const T> > content, wxGraphicsContext* gc, int x, int y, double pixels_per_second, int track_height, wxString type, bool consecutive
-       )
+class ContentView : public View
 {
-       Time t = 0;
-       for (typename list<shared_ptr<const T> >::iterator i = content.begin(); i != content.end(); ++i) {
-               Time const len = (*i)->temporal_length ();
+public:
+       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;
+       }
+
+       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 ();
+               if (!film || !content) {
+                       return;
+               }
+
+               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 (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 (selected, wxSOLID));
+               } else {
+                       gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (colour(), wxSOLID));
+               }
+#endif
+               
                wxGraphicsPath path = gc->CreatePath ();
-               path.MoveToPoint (x + t * pixels_per_second, y);
-               path.AddLineToPoint (x + (t + len) * pixels_per_second, y);
-               path.AddLineToPoint (x + (t + len) * pixels_per_second, y + track_height);
-               path.AddLineToPoint (x + t * pixels_per_second, y + track_height);
-               path.AddLineToPoint (x + t * pixels_per_second, y);
+               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);
 
-               wxString name = wxString::Format ("%s [%s]", std_to_wx ((*i)->file().filename().string()), type);
+               wxString name = wxString::Format (wxT ("%s [%s]"), std_to_wx (content->file().filename().string()).data(), type().data());
                wxDouble name_width;
                wxDouble name_height;
                wxDouble name_descent;
                wxDouble name_leading;
                gc->GetTextExtent (name, &name_width, &name_height, &name_descent, &name_leading);
                
-               gc->Clip (wxRegion (x + t * pixels_per_second, y, len * pixels_per_second, track_height));
-               gc->DrawText (name, t * pixels_per_second + 12, y + track_height - name_height - 4);
+               gc->Clip (wxRegion (time_x (start), y_pos (_track), len * _timeline.pixels_per_time_unit(), _timeline.track_height()));
+               gc->DrawText (name, time_x (start) + 12, y_pos (_track + 1) - name_height - 4);
                gc->ResetClip ();
+       }
 
-               if (consecutive) {
-                       t += len;
-               } else {
-                       y += track_height;
+       int y_pos (int t) const
+       {
+               return _timeline.tracks_position().y + t * _timeline.track_height();
+       }
+
+       void content_changed (int p)
+       {
+               if (p == ContentProperty::START || p == ContentProperty::LENGTH) {
+                       force_redraw ();
+               }
+       }
+
+       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<Content> c, int t)
+               : ContentView (tl, c, t)
+       {}
+       
+private:
+       wxString type () const
+       {
+               return _("audio");
+       }
+
+       wxColour colour () const
+       {
+               return wxColour (149, 121, 232, 255);
+       }
+};
+
+class VideoContentView : public ContentView
+{
+public:
+       VideoContentView (Timeline& tl, shared_ptr<Content> c, int t)
+               : ContentView (tl, c, t)
+       {}
+
+private:       
+
+       wxString type () const
+       {
+               return _("video");
+       }
+
+       wxColour colour () const
+       {
+               return wxColour (242, 92, 120, 255);
+       }
+};
+
+class TimeAxisView : public View
+{
+public:
+       TimeAxisView (Timeline& tl, int y)
+               : View (tl)
+               , _y (y)
+       {}
+
+       void do_paint (wxGraphicsContext* gc)
+       {
+#if wxMAJOR_VERSION == 2 && wxMINOR_VERSION >= 9
+               gc->SetPen (*wxThePenList->FindOrCreatePen (wxColour (0, 0, 0), 1, wxPENSTYLE_SOLID));
+#else              
+               gc->SetPen (*wxThePenList->FindOrCreatePen (wxColour (0, 0, 0), 1, wxSOLID));
+#endif             
+               
+               int mark_interval = rint (128 / (TIME_HZ * _timeline.pixels_per_time_unit ()));
+               if (mark_interval > 5) {
+                       mark_interval -= mark_interval % 5;
+               }
+               if (mark_interval > 10) {
+                       mark_interval -= mark_interval % 10;
+               }
+               if (mark_interval > 60) {
+                       mark_interval -= mark_interval % 60;
+               }
+               if (mark_interval > 3600) {
+                       mark_interval -= mark_interval % 3600;
+               }
+               
+               if (mark_interval < 1) {
+                       mark_interval = 1;
+               }
+
+               wxGraphicsPath path = gc->CreatePath ();
+               path.MoveToPoint (_timeline.x_offset(), _y);
+               path.AddLineToPoint (_timeline.width(), _y);
+               gc->StrokePath (path);
+
+               Time t = 0;
+               while ((t * _timeline.pixels_per_time_unit()) < _timeline.width()) {
+                       wxGraphicsPath path = gc->CreatePath ();
+                       path.MoveToPoint (time_x (t), _y - 4);
+                       path.AddLineToPoint (time_x (t), _y + 4);
+                       gc->StrokePath (path);
+
+                       int tc = t / TIME_HZ;
+                       int const h = tc / 3600;
+                       tc -= h * 3600;
+                       int const m = tc / 60;
+                       tc -= m * 60;
+                       int const s = tc;
+                       
+                       wxString str = wxString::Format (wxT ("%02d:%02d:%02d"), h, m, s);
+                       wxDouble str_width;
+                       wxDouble str_height;
+                       wxDouble str_descent;
+                       wxDouble str_leading;
+                       gc->GetTextExtent (str, &str_width, &str_height, &str_descent, &str_leading);
+                       
+                       int const tx = _timeline.x_offset() + t * _timeline.pixels_per_time_unit();
+                       if ((tx + str_width) < _timeline.width()) {
+                               gc->DrawText (str, time_x (t), _y + 16);
+                       }
+                       
+                       t += mark_interval * TIME_HZ;
                }
        }
 
-       if (consecutive) {
-               y += track_height;
+       Rect bbox () const
+       {
+               return Rect (0, _y - 4, _timeline.width(), 24);
        }
 
-       return y;
+private:
+       int _y;
+};
+
+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);
+
+       setup_pixels_per_time_unit ();
+       
+       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);
+
+       playlist_changed ();
+
+       SetMinSize (wxSize (640, tracks() * track_height() + 96));
+
+       _playlist_connection = film->playlist()->Changed.connect (bind (&Timeline::playlist_changed, this));
 }
 
 void
@@ -97,104 +345,206 @@ Timeline::paint (wxPaintEvent &)
 {
        wxPaintDC dc (this);
 
-       shared_ptr<Playlist> pl = _playlist.lock ();
-       if (!pl) {
+       wxGraphicsContext* gc = wxGraphicsContext::Create (dc);
+       if (!gc) {
                return;
        }
 
-       wxGraphicsContext* gc = wxGraphicsContext::Create (dc);
-       if (!gc) {
+       gc->SetFont (gc->CreateFont (*wxNORMAL_FONT));
+
+       for (list<shared_ptr<View> >::iterator i = _views.begin(); i != _views.end(); ++i) {
+               (*i)->paint (gc);
+       }
+
+       delete gc;
+}
+
+void
+Timeline::playlist_changed ()
+{
+       shared_ptr<const Film> fl = _film.lock ();
+       if (!fl) {
                return;
        }
 
-       int const x_offset = 8;
-       int y = 8;
-       int const width = GetSize().GetWidth();
-       double const pixels_per_second = (width - x_offset * 2) / (pl->content_length() / pl->video_frame_rate());
+       _views.clear ();
 
-       gc->SetFont (gc->CreateFont (*wxNORMAL_FONT));
+       Playlist::ContentList content = fl->playlist()->content ();
 
-       gc->SetPen (*wxBLACK_PEN);
-#if wxMAJOR_VERSION == 2 && wxMINOR_VERSION >= 9
-       gc->SetPen (*wxThePenList->FindOrCreatePen (wxColour (0, 0, 0), 4, wxPENSTYLE_SOLID));
-       gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (wxColour (149, 121, 232, 255), wxBRUSHSTYLE_SOLID));
-#else                  
-       gc->SetPen (*wxThePenList->FindOrCreatePen (wxColour (0, 0, 0), 4, SOLID));
-       gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (wxColour (149, 121, 232, 255), SOLID));
-#endif
-       y = plot_content_list (pl->video (), gc, x_offset, y, pixels_per_second, _track_height, _("video"), true);
+       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)));
+               }
+               if (dynamic_pointer_cast<AudioContent> (*i)) {
+                       _views.push_back (shared_ptr<View> (new AudioContentView (*this, *i, 0)));
+               }
+       }
+
+       assign_tracks ();
        
-#if wxMAJOR_VERSION == 2 && wxMINOR_VERSION >= 9
-       gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (wxColour (242, 92, 120, 255), wxBRUSHSTYLE_SOLID));
-#else                  
-       gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (wxColour (242, 92, 120, 255), SOLID));
-#endif
-       y = plot_content_list (pl->audio (), gc, x_offset, y, pixels_per_second, _track_height, _("audio"), pl->audio_from() == Playlist::AUDIO_FFMPEG);
+       _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);
+               }
+       }
 
-       /* Time axis */
+       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 wxMAJOR_VERSION == 2 && wxMINOR_VERSION >= 9
-        gc->SetPen (*wxThePenList->FindOrCreatePen (wxColour (0, 0, 0), 1, wxPENSTYLE_SOLID));
-#else              
-       gc->SetPen (*wxThePenList->FindOrCreatePen (wxColour (0, 0, 0), 1, SOLID));
-#endif             
-                   
-       int mark_interval = rint (128 / pixels_per_second);
-        if (mark_interval > 5) {
-               mark_interval -= mark_interval % 5;
+                       if (j == _views.end ()) {
+                               /* no overlap on `t' */
+                               break;
+                       }
+               }
+
+               acv->set_track (t);
+               _tracks = max (_tracks, t + 1);
        }
-        if (mark_interval > 10) {
-               mark_interval -= mark_interval % 10;
+}
+
+int
+Timeline::tracks () const
+{
+       return _tracks;
+}
+
+void
+Timeline::setup_pixels_per_time_unit ()
+{
+       shared_ptr<const Film> film = _film.lock ();
+       if (!film) {
+               return;
        }
-       if (mark_interval > 60) {
-               mark_interval -= mark_interval % 60;
+
+       _pixels_per_time_unit = static_cast<double>(width() - x_offset() * 2) / film->length();
+}
+
+void
+Timeline::left_down (wxMouseEvent& ev)
+{
+       list<shared_ptr<View> >::iterator i = _views.begin();
+       Position const p (ev.GetX(), ev.GetY());
+       while (i != _views.end() && !(*i)->bbox().contains (p)) {
+               ++i;
        }
-       if (mark_interval > 3600) {
-               mark_interval -= mark_interval % 3600;
+
+       _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 ();
+               }
        }
 
-        if (mark_interval < 1) {
-               mark_interval = 1;
-        }
+       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 ());
+                       }
+               }
+       }
 
-       wxGraphicsPath path = gc->CreatePath ();
-       path.MoveToPoint (x_offset, y + 40);
-       path.AddLineToPoint (width, y + 40);
-       gc->StrokePath (path);
+       _left_down = true;
+       _down_point = ev.GetPosition ();
+       _first_move = false;
+}
 
-       double t = 0;
-       while ((t * pixels_per_second) < width) {
-               wxGraphicsPath path = gc->CreatePath ();
-               path.MoveToPoint (x_offset + t * pixels_per_second, y + 36);
-               path.AddLineToPoint (x_offset + t * pixels_per_second, y + 44);
-               gc->StrokePath (path);
+void
+Timeline::left_up (wxMouseEvent &)
+{
+       _left_down = false;
+}
 
-               int tc = t;
-               int const h = tc / 3600;
-               tc -= h * 3600;
-               int const m = tc / 60;
-               tc -= m * 60;
-               int const s = tc;
+void
+Timeline::mouse_moved (wxMouseEvent& ev)
+{
+       if (!_left_down) {
+               return;
+       }
 
-               wxString str = wxString::Format ("%02d:%02d:%02d", h, m, s);
-               wxDouble str_width;
-               wxDouble str_height;
-               wxDouble str_descent;
-               wxDouble str_leading;
-               gc->GetTextExtent (str, &str_width, &str_height, &str_descent, &str_leading);
+       wxPoint const p = ev.GetPosition();
 
-               int const tx = x_offset + t * pixels_per_second;
-               if ((tx + str_width) < width) {
-                       gc->DrawText (str, x_offset + t * pixels_per_second, y + 60);
+       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;
                }
-               t += mark_interval;
+               _first_move = true;
        }
 
-       delete gc;
+       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);
+               }
+       }
 }
 
 void
-Timeline::playlist_changed ()
+Timeline::force_redraw (Rect const & r)
 {
-       Refresh ();
+       RefreshRect (wxRect (r.x, r.y, r.width, r.height), false);
+}
+
+shared_ptr<const Film>
+Timeline::film () const
+{
+       return _film.lock ();
+}
+
+void
+Timeline::resized (wxSizeEvent &)
+{
+       setup_pixels_per_time_unit ();
 }