Handle multiple audio streams in a single piece of content
[dcpomatic.git] / src / wx / timeline.cc
index 52707e45c725de3bbd3ffbc661eb445cef4137aa..c1713bb61262e2bcc27047a60eadb2b9186baf47 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2013-2015 Carl Hetherington <cth@carlh.net>
 
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
 
 #include <list>
 #include <wx/graphics.h>
+#include <boost/weak_ptr.hpp>
+#include "lib/film.h"
+#include "lib/playlist.h"
+#include "lib/image_content.h"
+#include "film_editor.h"
 #include "timeline.h"
+#include "timeline_time_axis_view.h"
+#include "timeline_video_content_view.h"
+#include "timeline_audio_content_view.h"
+#include "timeline_subtitle_content_view.h"
+#include "content_panel.h"
 #include "wx_util.h"
-#include "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;
+using boost::optional;
 
-int const Timeline::_track_height = 64;
-
-Timeline::Timeline (wxWindow* parent, shared_ptr<Playlist> pl)
-       : wxPanel (parent)
-       , _playlist (pl)
+Timeline::Timeline (wxWindow* parent, ContentPanel* cp, shared_ptr<Film> film)
+       : wxPanel (parent, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxFULL_REPAINT_ON_RESIZE)
+       , _content_panel (cp)
+       , _film (film)
+       , _time_axis_view (new TimelineTimeAxisView (*this, 32))
+       , _tracks (0)
+       , _left_down (false)
+       , _down_view_position (0)
+       , _first_move (false)
+       , _menu (this)
+       , _snap (true)
 {
+#ifndef __WXOSX__
        SetDoubleBuffered (true);
-       
-       Connect (wxID_ANY, wxEVT_PAINT, wxPaintEventHandler (Timeline::paint), 0, this);
+#endif 
+
+       Bind (wxEVT_PAINT,      boost::bind (&Timeline::paint,       this));
+       Bind (wxEVT_LEFT_DOWN,  boost::bind (&Timeline::left_down,   this, _1));
+       Bind (wxEVT_LEFT_UP,    boost::bind (&Timeline::left_up,     this, _1));
+       Bind (wxEVT_RIGHT_DOWN, boost::bind (&Timeline::right_down,  this, _1));
+       Bind (wxEVT_MOTION,     boost::bind (&Timeline::mouse_moved, this, _1));
+       Bind (wxEVT_SIZE,       boost::bind (&Timeline::resized,     this));
+
+       playlist_changed ();
+
+       SetMinSize (wxSize (640, tracks() * track_height() + 96));
+
+       _playlist_changed_connection = film->playlist()->Changed.connect (bind (&Timeline::playlist_changed, this));
+       _playlist_content_changed_connection = film->playlist()->ContentChanged.connect (bind (&Timeline::playlist_content_changed, this, _2));
+}
+
+void
+Timeline::paint ()
+{
+       wxPaintDC dc (this);
 
-       if (pl->audio_from() == Playlist::AUDIO_FFMPEG) {
-               SetMinSize (wxSize (640, _track_height * 2 + 96));
-       } else {
-               SetMinSize (wxSize (640, _track_height * (max (1UL, pl->audio().size()) + 1) + 96));
+       wxGraphicsContext* gc = wxGraphicsContext::Create (dc);
+       if (!gc) {
+               return;
+       }
+
+       for (TimelineViewList::iterator i = _views.begin(); i != _views.end(); ++i) {
+               (*i)->paint (gc);
        }
 
-       pl->Changed.connect (bind (&Timeline::playlist_changed, this));
-       pl->ContentChanged.connect (bind (&Timeline::playlist_changed, this));
+       delete gc;
+}
+
+void
+Timeline::playlist_changed ()
+{
+       ensure_ui_thread ();
+       recreate_views ();
+}
+
+void
+Timeline::recreate_views ()
+{
+       shared_ptr<const Film> fl = _film.lock ();
+       if (!fl) {
+               return;
+       }
+
+       _views.clear ();
+       _views.push_back (_time_axis_view);
+
+       ContentList content = fl->playlist()->content ();
+
+       for (ContentList::iterator i = content.begin(); i != content.end(); ++i) {
+               if (dynamic_pointer_cast<VideoContent> (*i)) {
+                       _views.push_back (shared_ptr<TimelineView> (new TimelineVideoContentView (*this, *i)));
+               }
+
+               shared_ptr<AudioContent> ac = dynamic_pointer_cast<AudioContent> (*i);
+               if (ac && !ac->audio_mapping().mapped_dcp_channels().empty ()) {
+                       _views.push_back (shared_ptr<TimelineView> (new TimelineAudioContentView (*this, *i)));
+               }
+
+               shared_ptr<SubtitleContent> sc = dynamic_pointer_cast<SubtitleContent> (*i);
+               if (sc && sc->has_subtitles ()) {
+                       _views.push_back (shared_ptr<TimelineView> (new TimelineSubtitleContentView (*this, sc)));
+               }
+       }
+
+       assign_tracks ();
+       setup_pixels_per_second ();
+       Refresh ();
+}
+
+void
+Timeline::playlist_content_changed (int property)
+{
+       ensure_ui_thread ();
+
+       if (property == ContentProperty::POSITION) {
+               assign_tracks ();
+               if (!_left_down) {
+                       /* Only do this if we are not dragging, as it's confusing otherwise */
+                       setup_pixels_per_second ();
+               }
+               Refresh ();
+       } else if (property == AudioContentProperty::AUDIO_STREAMS) {
+               recreate_views ();
+       }
+}
+
+void
+Timeline::assign_tracks ()
+{
+       for (TimelineViewList::iterator i = _views.begin(); i != _views.end(); ++i) {
+               shared_ptr<TimelineContentView> c = dynamic_pointer_cast<TimelineContentView> (*i);
+               if (c) {
+                       c->unset_track ();
+               }
+       }
+
+       for (TimelineViewList::iterator i = _views.begin(); i != _views.end(); ++i) {
+               shared_ptr<TimelineContentView> cv = dynamic_pointer_cast<TimelineContentView> (*i);
+               if (!cv) {
+                       continue;
+               }
+
+               shared_ptr<Content> content = cv->content();
+
+               int t = 0;
+               while (true) {
+                       TimelineViewList::iterator j = _views.begin();
+                       while (j != _views.end()) {
+                               shared_ptr<TimelineContentView> test = dynamic_pointer_cast<TimelineContentView> (*j);
+                               if (!test) {
+                                       ++j;
+                                       continue;
+                               }
+                               
+                               shared_ptr<Content> test_content = test->content();
+                                       
+                               if (test && test->track() && test->track().get() == t) {
+                                       bool const no_overlap =
+                                               (content->position() < test_content->position() && content->end() < test_content->position()) ||
+                                               (content->position() > test_content->end()      && content->end() > test_content->end());
+                                       
+                                       if (!no_overlap) {
+                                               /* we have an overlap on track `t' */
+                                               ++t;
+                                               break;
+                                       }
+                               }
+                               
+                               ++j;
+                       }
+
+                       if (j == _views.end ()) {
+                               /* no overlap on `t' */
+                               break;
+                       }
+               }
+
+               cv->set_track (t);
+               _tracks = max (_tracks, t + 1);
+       }
+
+       _time_axis_view->set_y (tracks() * track_height() + 32);
 }
 
-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
-       )
-{
-       Time t = 0;
-       for (typename list<shared_ptr<const T> >::iterator i = content.begin(); i != content.end(); ++i) {
-               Time const len = (*i)->temporal_length ();
-               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);
-               gc->StrokePath (path);
-               gc->FillPath (path);
-
-               wxString name = wxString::Format ("%s [%s]", std_to_wx ((*i)->file().filename().string()), type);
-               wxDouble name_width;
-               wxDouble name_height;
-               wxDouble name_descent;
-               wxDouble name_leading;
-               gc->GetTextExtent (name, &name_width, &name_height, &name_descent, &name_leading);
+Timeline::tracks () const
+{
+       return _tracks;
+}
+
+void
+Timeline::setup_pixels_per_second ()
+{
+       shared_ptr<const Film> film = _film.lock ();
+       if (!film || film->length() == DCPTime ()) {
+               return;
+       }
+
+       _pixels_per_second = static_cast<double>(width() - x_offset() * 2) / film->length().seconds ();
+}
+
+shared_ptr<TimelineView>
+Timeline::event_to_view (wxMouseEvent& ev)
+{
+       TimelineViewList::iterator i = _views.begin();
+       Position<int> const p (ev.GetX(), ev.GetY());
+       while (i != _views.end() && !(*i)->bbox().contains (p)) {
+               ++i;
+       }
+
+       if (i == _views.end ()) {
+               return shared_ptr<TimelineView> ();
+       }
+
+       return *i;
+}
+
+void
+Timeline::left_down (wxMouseEvent& ev)
+{
+       shared_ptr<TimelineView> view = event_to_view (ev);
+       shared_ptr<TimelineContentView> content_view = dynamic_pointer_cast<TimelineContentView> (view);
+
+       _down_view.reset ();
+
+       if (content_view) {
+               _down_view = content_view;
+               _down_view_position = content_view->content()->position ();
+       }
+
+       for (TimelineViewList::iterator i = _views.begin(); i != _views.end(); ++i) {
+               shared_ptr<TimelineContentView> cv = dynamic_pointer_cast<TimelineContentView> (*i);
+               if (!cv) {
+                       continue;
+               }
+               
+               if (!ev.ShiftDown ()) {
+                       cv->set_selected (view == *i);
+               }
                
-               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->ResetClip ();
-
-               if (consecutive) {
-                       t += len;
-               } else {
-                       y += track_height;
+               if (view == *i) {
+                       _content_panel->set_selection (cv->content ());
                }
        }
 
-       if (consecutive) {
-               y += track_height;
+       if (content_view && ev.ShiftDown ()) {
+               content_view->set_selected (!content_view->selected ());
        }
 
-       return y;
+       _left_down = true;
+       _down_point = ev.GetPosition ();
+       _first_move = false;
+
+       if (_down_view) {
+               _down_view->content()->set_change_signals_frequent (true);
+       }
 }
 
 void
-Timeline::paint (wxPaintEvent &)
+Timeline::left_up (wxMouseEvent& ev)
 {
-       wxPaintDC dc (this);
+       _left_down = false;
+
+       if (_down_view) {
+               _down_view->content()->set_change_signals_frequent (false);
+       }
+
+       set_position_from_event (ev);
+
+       /* We don't do this during drag, and set_position_from_event above
+          might not have changed the position, so do it now.
+       */
+       setup_pixels_per_second ();
+       Refresh ();
+}
 
-       shared_ptr<Playlist> pl = _playlist.lock ();
-       if (!pl) {
+void
+Timeline::mouse_moved (wxMouseEvent& ev)
+{
+       if (!_left_down) {
                return;
        }
 
-       wxGraphicsContext* gc = wxGraphicsContext::Create (dc);
-       if (!gc) {
+       set_position_from_event (ev);
+}
+
+void
+Timeline::right_down (wxMouseEvent& ev)
+{
+       shared_ptr<TimelineView> view = event_to_view (ev);
+       shared_ptr<TimelineContentView> cv = dynamic_pointer_cast<TimelineContentView> (view);
+       if (!cv) {
+               return;
+       }
+
+       if (!cv->selected ()) {
+               clear_selection ();
+               cv->set_selected (true);
+       }
+
+       _menu.popup (_film, selected_content (), selected_views (), ev.GetPosition ());
+}
+
+void
+Timeline::maybe_snap (DCPTime a, DCPTime b, optional<DCPTime>& nearest_distance) const
+{
+       DCPTime const d = a - b;
+       if (!nearest_distance || d.abs() < nearest_distance.get().abs()) {
+               nearest_distance = d;
+       }
+}
+
+void
+Timeline::set_position_from_event (wxMouseEvent& ev)
+{
+       if (!_pixels_per_second) {
                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());
+       double const pps = _pixels_per_second.get ();
+
+       wxPoint const p = ev.GetPosition();
 
-       gc->SetFont (gc->CreateFont (*wxNORMAL_FONT));
+       if (!_first_move) {
+               /* We haven't moved yet; in that case, we must move the mouse some reasonable distance
+                  before the drag is considered to have started.
+               */
+               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;
+       }
 
-       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);
+       if (!_down_view) {
+               return;
+       }
+       
+       DCPTime new_position = _down_view_position + DCPTime::from_seconds ((p.x - _down_point.x) / pps);
        
-#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);
-
-       /* Time axis */
-
-#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 (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 (x_offset, y + 40);
-       path.AddLineToPoint (width, y + 40);
-       gc->StrokePath (path);
-
-       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);
-
-               int tc = t;
-               int const h = tc / 3600;
-               tc -= h * 3600;
-               int const m = tc / 60;
-               tc -= m * 60;
-               int const s = tc;
-
-               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);
-
-               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 (_snap) {
+
+               DCPTime const new_end = new_position + _down_view->content()->length_after_trim () - DCPTime (1);
+               /* Signed `distance' to nearest thing (i.e. negative is left on the timeline,
+                  positive is right).
+               */
+               optional<DCPTime> nearest_distance;
+               
+               /* Find the nearest content edge; this is inefficient */
+               for (TimelineViewList::iterator i = _views.begin(); i != _views.end(); ++i) {
+                       shared_ptr<TimelineContentView> cv = dynamic_pointer_cast<TimelineContentView> (*i);
+                       if (!cv || cv == _down_view || cv->content() == _down_view->content()) {
+                               continue;
+                       }
+
+                       maybe_snap (cv->content()->position(), new_position, nearest_distance);
+                       maybe_snap (cv->content()->position(), new_end + DCPTime (1), nearest_distance);
+                       maybe_snap (cv->content()->end() + DCPTime (1), new_position, nearest_distance);
+                       maybe_snap (cv->content()->end() + DCPTime (1), new_end, nearest_distance);
+               }
+               
+               if (nearest_distance) {
+                       /* Snap if it's close; `close' means within a proportion of the time on the timeline */
+                       if (nearest_distance.get().abs() < DCPTime::from_seconds ((width() / pps) / 64)) {
+                               new_position += nearest_distance.get ();
+                       }
                }
-               t += mark_interval;
+       }
+       
+       if (new_position < DCPTime ()) {
+               new_position = DCPTime ();
        }
 
-       delete gc;
+       _down_view->content()->set_position (new_position);
+       
+       shared_ptr<Film> film = _film.lock ();
+       DCPOMATIC_ASSERT (film);
+       film->set_sequence_video (false);
 }
 
 void
-Timeline::playlist_changed ()
+Timeline::force_redraw (dcpomatic::Rect<int> 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 ()
+{
+       setup_pixels_per_second ();
+}
+
+void
+Timeline::clear_selection ()
+{
+       for (TimelineViewList::iterator i = _views.begin(); i != _views.end(); ++i) {
+               shared_ptr<TimelineContentView> cv = dynamic_pointer_cast<TimelineContentView> (*i);
+               if (cv) {
+                       cv->set_selected (false);
+               }
+       }
+}
+
+TimelineContentViewList
+Timeline::selected_views () const
+{
+       TimelineContentViewList sel;
+       
+       for (TimelineViewList::const_iterator i = _views.begin(); i != _views.end(); ++i) {
+               shared_ptr<TimelineContentView> cv = dynamic_pointer_cast<TimelineContentView> (*i);
+               if (cv && cv->selected()) {
+                       sel.push_back (cv);
+               }
+       }
+
+       return sel;
+}
+
+ContentList
+Timeline::selected_content () const
+{
+       ContentList sel;
+       TimelineContentViewList views = selected_views ();
+       
+       for (TimelineContentViewList::const_iterator i = views.begin(); i != views.end(); ++i) {
+               sel.push_back ((*i)->content ());
+       }
+
+       return sel;
 }