More various tweaks.
[dcpomatic.git] / src / wx / timeline.cc
index 2539038881ff903cd5dbb97f161e34a07bf630ca..96bd2ccd4866a9585875b7cf8b232d7e5eefacff 100644 (file)
@@ -1,3 +1,5 @@
+/* -*- c-basic-offset: 8; default-tab-width: 8; -*- */
+
 /*
     Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
 
@@ -19,6 +21,8 @@
 
 #include <list>
 #include <wx/graphics.h>
+#include <boost/weak_ptr.hpp>
+#include "film.h"
 #include "timeline.h"
 #include "wx_util.h"
 #include "playlist.h"
@@ -27,69 +31,302 @@ 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)
+       {
+               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;
+       }
+
+       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));
+               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));
                gc->StrokePath (path);
                gc->FillPath (path);
 
-               wxString name = wxString::Format (wxT ("%s [%s]"), std_to_wx ((*i)->file().filename().string()).data(), type.data());
+               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 == VideoContentProperty::VIDEO_LENGTH) {
+                       force_redraw ();
+               }
+       }
+
+       boost::weak_ptr<Content> _content;
+       int _track;
+       bool _selected;
+};
+
+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, shared_ptr<const Film> film)
+       : wxPanel (parent, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxFULL_REPAINT_ON_RESIZE)
+       , _film (film)
+       , _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);
+
+       SetMinSize (wxSize (640, tracks() * track_height() + 96));
+
+       playlist_changed ();
+
+       film->playlist()->Changed.connect (bind (&Timeline::playlist_changed, this));
 }
 
 void
@@ -97,104 +334,143 @@ 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, wxSOLID));
-       gc->SetBrush (*wxTheBrushList->FindOrCreateBrush (wxColour (149, 121, 232, 255), wxSOLID));
-#endif
-       y = plot_content_list (pl->video (), gc, x_offset, y, pixels_per_second, _track_height, _("video"), true);
-       
-#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), wxSOLID));
-#endif
-       y = plot_content_list (pl->audio (), gc, x_offset, y, pixels_per_second, _track_height, _("audio"), pl->audio_from() == Playlist::AUDIO_FFMPEG);
+       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, 1)));
+               }
+       }
 
-       /* Time axis */
+       _views.push_back (shared_ptr<View> (new TimeAxisView (*this, tracks() * track_height() + 32)));
+               
+       Refresh ();
+}
 
-#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 / pixels_per_second);
-        if (mark_interval > 5) {
-               mark_interval -= mark_interval % 5;
+int
+Timeline::tracks () const
+{
+       /* XXX */
+       return 2;
+}
+
+void
+Timeline::setup_pixels_per_time_unit ()
+{
+       shared_ptr<const Film> film = _film.lock ();
+       if (!film) {
+               return;
        }
-        if (mark_interval > 10) {
-               mark_interval -= mark_interval % 10;
+
+       _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 > 60) {
-               mark_interval -= mark_interval % 60;
+
+       _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 > 3600) {
-               mark_interval -= mark_interval % 3600;
+
+       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 (mark_interval < 1) {
-               mark_interval = 1;
-        }
+       _left_down = true;
+       _down_point = ev.GetPosition ();
+       _first_move = false;
+}
 
-       wxGraphicsPath path = gc->CreatePath ();
-       path.MoveToPoint (x_offset, y + 40);
-       path.AddLineToPoint (width, y + 40);
-       gc->StrokePath (path);
+void
+Timeline::left_up (wxMouseEvent &)
+{
+       _left_down = 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::mouse_moved (wxMouseEvent& ev)
+{
+       if (!_left_down) {
+               return;
+       }
 
-               int tc = t;
-               int const h = tc / 3600;
-               tc -= h * 3600;
-               int const m = tc / 60;
-               tc -= m * 60;
-               int const s = tc;
+       wxPoint const p = ev.GetPosition();
 
-               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);
+       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;
+       }
 
-               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);
+       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 (_down_view_start + time_diff);
                }
-               t += mark_interval;
        }
+}
 
-       delete gc;
+void
+Timeline::force_redraw (Rect const & r)
+{
+       RefreshRect (wxRect (r.x, r.y, r.width, r.height), false);
+}
+
+shared_ptr<const Film>
+Timeline::film () const
+{
+       return _film.lock ();
 }
 
 void
-Timeline::playlist_changed ()
+Timeline::resized (wxSizeEvent &)
 {
-       Refresh ();
+       setup_pixels_per_time_unit ();
 }