Merge branch 'windows+cc' into cairocanvas
[ardour.git] / gtk2_ardour / editor_timefx.cc
index 015a265a9bd608b13afeaf85f4e48666a92e3f64..2528c823b7996ac9f75346ece542476c47378f87 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2000 Paul Davis 
+    Copyright (C) 2000 Paul Davis
 
     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 <iostream>
 #include <cstdlib>
 #include <cmath>
-
+#include <ctime>
 #include <string>
+#include <set>
 
-#include <pbd/error.h>
-#include <pbd/pthread_utils.h>
-#include <pbd/memento_command.h>
+#include "pbd/error.h"
+#include "pbd/pthread_utils.h"
+#include "pbd/memento_command.h"
+#include "pbd/stateful_diff_command.h"
 
-#include <gtkmm2ext/window_title.h>
+#include <gtkmm2ext/utils.h>
 
-#include "editor.h"
-#include "audio_time_axis.h"
 #include "audio_region_view.h"
+#include "audio_time_axis.h"
+#include "editor.h"
 #include "region_selection.h"
+#include "time_fx_dialog.h"
+
+#include "ardour/audioregion.h"
+#include "ardour/midi_stretch.h"
+#include "ardour/pitch.h"
+#include "ardour/region.h"
+#include "ardour/session.h"
+#include "ardour/stretch.h"
 
-#include <ardour/session.h>
-#include <ardour/region.h>
-#include <ardour/audioplaylist.h>
-#include <ardour/audio_track.h>
-#include <ardour/audioregion.h>
-#include <ardour/audio_diskstream.h>
-#include <ardour/stretch.h>
-#include <ardour/pitch.h>
+#ifdef USE_RUBBERBAND
+#include <rubberband/RubberBandStretcher.h>
+using namespace RubberBand;
+#endif
 
 #include "i18n.h"
 
+using namespace std;
 using namespace ARDOUR;
 using namespace PBD;
-using namespace sigc;
 using namespace Gtk;
 using namespace Gtkmm2ext;
 
-Editor::TimeFXDialog::TimeFXDialog (Editor& e, bool pitch)
-       : ArdourDialog (X_("time fx dialog")),
-         editor (e),
-         pitching (pitch),
-         pitch_octave_adjustment (0.0, 0.0, 4.0, 1, 2.0),
-         pitch_semitone_adjustment (0.0, 0.0, 12.0, 1.0, 4.0),
-         pitch_cent_adjustment (0.0, 0.0, 150.0, 5.0, 15.0),
-         pitch_octave_spinner (pitch_octave_adjustment),
-         pitch_semitone_spinner (pitch_semitone_adjustment),
-         pitch_cent_spinner (pitch_cent_adjustment),
-         quick_button (_("Quick but Ugly")),
-         antialias_button (_("Skip Anti-aliasing"))
+/** @return -1 in case of error, 1 if operation was cancelled by the user, 0 if everything went ok */
+int
+Editor::time_stretch (RegionSelection& regions, float fraction)
 {
-       set_modal (true);
-       set_position (Gtk::WIN_POS_MOUSE);
-       set_name (N_("TimeFXDialog"));
+       RegionList audio;
+       RegionList midi;
+       int aret;
 
-       WindowTitle title(Glib::get_application_name());
-       if (pitching) {
-               title += _("Pitch Shift");
-       } else {
-               title += _("Time Stretch");
+       begin_reversible_command (_("stretch/shrink"));
+
+       for (RegionSelection::iterator i = regions.begin(); i != regions.end(); ++i) {
+               if  ((*i)->region()->data_type() == DataType::AUDIO) {
+                       audio.push_back ((*i)->region());
+               } else if  ((*i)->region()->data_type() == DataType::MIDI) {
+                       midi.push_back ((*i)->region());
+               }
        }
-       set_title(title.get_string());
 
-       cancel_button = add_button (_("Cancel"), Gtk::RESPONSE_CANCEL);
+       if ((aret = time_fx (audio, fraction, false)) != 0) {
+               return aret;
+       }
 
-       get_vbox()->set_spacing (5);
-       get_vbox()->set_border_width (12);
-       get_vbox()->pack_start (upper_button_box, false, false);
-       get_vbox()->pack_start (progress_bar);
+       set<boost::shared_ptr<Playlist> > midi_playlists_affected;
 
-       if (pitching) {
+       for (RegionList::iterator i = midi.begin(); i != midi.end(); ++i) {
+               boost::shared_ptr<Playlist> playlist = (*i)->playlist();
 
-               upper_button_box.set_spacing (5);
-               upper_button_box.set_border_width (5);
-               
-               Gtk::Label* l;
+               if (playlist) {
+                       playlist->clear_changes ();
+               }
 
-               l = manage (new Label (_("Octaves")));
-               upper_button_box.pack_start (*l, false, false);
-               upper_button_box.pack_start (pitch_octave_spinner, false, false);
+       }
 
-               l = manage (new Label (_("Semitones (12TET)")));
-               upper_button_box.pack_start (*l, false, false);
-               upper_button_box.pack_start (pitch_semitone_spinner, false, false);
+       ARDOUR::TimeFXRequest request;
+       request.time_fraction = fraction;
 
-               l = manage (new Label (_("Cents")));
-               upper_button_box.pack_start (*l, false, false);
-               upper_button_box.pack_start (pitch_cent_spinner, false, false);
+       for (RegionList::iterator i = midi.begin(); i != midi.end(); ++i) {
+               boost::shared_ptr<Playlist> playlist = (*i)->playlist();
 
-               pitch_cent_spinner.set_digits (1);
+               if (!playlist) {
+                       continue;
+               }
 
-               add_button (_("Shift"), Gtk::RESPONSE_ACCEPT);
+               MidiStretch stretch (*_session, request);
+               stretch.run (*i);
 
-       } else {
+               playlist->replace_region (regions.front()->region(), stretch.results[0],
+                                         regions.front()->region()->position());
+               midi_playlists_affected.insert (playlist);
+       }
 
-               upper_button_box.set_homogeneous (true);
-               upper_button_box.set_spacing (5);
-               upper_button_box.set_border_width (5);
-               upper_button_box.pack_start (quick_button, true, true);
-               upper_button_box.pack_start (antialias_button, true, true);
-       
-               add_button (_("Stretch/Shrink"), Gtk::RESPONSE_ACCEPT);
+       for (set<boost::shared_ptr<Playlist> >::iterator p = midi_playlists_affected.begin(); p != midi_playlists_affected.end(); ++p) {
+               _session->add_command (new StatefulDiffCommand (*p));
        }
 
-       quick_button.set_name (N_("TimeFXButton"));
-       antialias_button.set_name (N_("TimeFXButton"));
-       progress_bar.set_name (N_("TimeFXProgress"));
+       commit_reversible_command ();
 
-       show_all_children ();
+       return 0;
 }
 
-gint
-Editor::TimeFXDialog::update_progress ()
+int
+Editor::pitch_shift (RegionSelection& regions, float fraction)
 {
-       progress_bar.set_fraction (request.progress);
-       return !request.done;
-}
+       RegionList rl;
 
-void
-Editor::TimeFXDialog::cancel_in_progress ()
-{
-       status = -2;
-       request.cancel = true;
-       first_cancel.disconnect();
-}
+       for (RegionSelection::iterator i = regions.begin(); i != regions.end(); ++i) {
+               rl.push_back ((*i)->region());
+       }
 
-gint
-Editor::TimeFXDialog::delete_in_progress (GdkEventAny* ev)
-{
-       status = -2;
-       request.cancel = true;
-       first_delete.disconnect();
-       return TRUE;
-}
+       begin_reversible_command (_("pitch shift"));
 
-int
-Editor::time_stretch (RegionSelection& regions, float fraction)
-{
-       return time_fx (regions, fraction, false);
-}
+       int ret = time_fx (rl, fraction, true);
 
-int
-Editor::pitch_shift (RegionSelection& regions, float fraction)
-{
-       return time_fx (regions, fraction, true);
+       if (ret == 0) {
+               commit_reversible_command ();
+       }
+
+       return ret;
 }
 
+/** @param val Percentage to time stretch by; ignored if pitch-shifting.
+ *  @param pitching true to pitch shift, false to time stretch.
+ *  @return -1 in case of error, 1 if operation was cancelled by the user, 0 if everything went ok */
 int
-Editor::time_fx (RegionSelection& regions, float val, bool pitching)
+Editor::time_fx (RegionList& regions, float val, bool pitching)
 {
-       if (current_timefx != 0) {
-               delete current_timefx;
-       }
-
+       delete current_timefx;
        current_timefx = new TimeFXDialog (*this, pitching);
+       current_timefx->regions = regions;
 
-       current_timefx->progress_bar.set_fraction (0.0f);
+       /* See if we have any audio regions on our list */
+       RegionList::iterator i = regions.begin ();
+       while (i != regions.end() && boost::dynamic_pointer_cast<AudioRegion> (*i) == 0) {
+               ++i;
+       }
 
+       if (i == regions.end ()) {
+               /* No audio regions; we can just do the timefx without a dialogue */
+               do_timefx ();
+               return 0;
+       }
+       
        switch (current_timefx->run ()) {
        case RESPONSE_ACCEPT:
                break;
@@ -179,11 +168,11 @@ Editor::time_fx (RegionSelection& regions, float val, bool pitching)
        }
 
        current_timefx->status = 0;
-       current_timefx->regions = regions;
 
        if (pitching) {
 
                float cents = current_timefx->pitch_octave_adjustment.get_value() * 1200.0;
+               float pitch_fraction;
                cents += current_timefx->pitch_semitone_adjustment.get_value() * 100.0;
                cents += current_timefx->pitch_cent_adjustment.get_value();
 
@@ -193,18 +182,14 @@ Editor::time_fx (RegionSelection& regions, float val, bool pitching)
                        return 0;
                }
 
-               // we now have the pitch shift in cents. divide by 1200 to get octaves
-               // then multiply by 2.0 because 1 octave == doubling the frequency
-               
-               cents /= 1200.0;
-               cents /= 2.0;
-
-               // add 1.0 to convert to RB scale
+               // one octave == 1200 cents
+               // adding one octave doubles the frequency
+               // ratio is 2^^octaves
 
-               cents += 1.0;
+               pitch_fraction = pow(2, cents/1200);
 
                current_timefx->request.time_fraction = 1.0;
-               current_timefx->request.pitch_fraction = cents;
+               current_timefx->request.pitch_fraction = pitch_fraction;
 
        } else {
 
@@ -213,23 +198,105 @@ Editor::time_fx (RegionSelection& regions, float val, bool pitching)
 
        }
 
+#ifdef USE_RUBBERBAND
+       /* parse options */
+
+       RubberBandStretcher::Options options = 0;
+
+       bool realtime = false;
+       bool precise = false;
+       bool peaklock = true;
+       bool longwin = false;
+       bool shortwin = false;
+       bool preserve_formants = false;
+       string txt;
+
+       enum {
+               NoTransients,
+               BandLimitedTransients,
+               Transients
+       } transients = Transients;
+
+       precise = current_timefx->precise_button.get_active();
+       preserve_formants = current_timefx->preserve_formants_button.get_active();
+
+       txt = current_timefx->stretch_opts_selector.get_active_text ();
+
+       for (int i = 0; i <= 6; i++) {
+               if (txt == rb_opt_strings[i]) {
+                       rb_current_opt = i;
+                       break;
+               }
+       }
+
+       switch (rb_current_opt) {
+               case 0:
+                       transients = NoTransients; peaklock = false; longwin = true; shortwin = false;
+                       break;
+               case 1:
+                       transients = NoTransients; peaklock = false; longwin = false; shortwin = false;
+                       break;
+               case 2:
+                       transients = NoTransients; peaklock = true; longwin = false; shortwin = false;
+                       break;
+               case 3:
+                       transients = BandLimitedTransients; peaklock = true; longwin = false; shortwin = false;
+                       break;
+               case 5:
+                       transients = Transients; peaklock = false; longwin = false; shortwin = true;
+                       break;
+               case 6:
+                       transients = NoTransients;
+                       precise = true;
+                       preserve_formants = false;
+                       current_timefx->request.pitch_fraction = 1/val;
+                       shortwin = true;
+                       // peaklock = false;
+                       break;
+               default:
+                       /* default/4 */
+                       transients = Transients; peaklock = true; longwin = false; shortwin = false;
+                       break;
+       };
+
+       if (realtime)          options |= RubberBandStretcher::OptionProcessRealTime;
+       if (precise)           options |= RubberBandStretcher::OptionStretchPrecise;
+       if (preserve_formants) options |= RubberBandStretcher::OptionFormantPreserved;
+       if (!peaklock)         options |= RubberBandStretcher::OptionPhaseIndependent;
+       if (longwin)           options |= RubberBandStretcher::OptionWindowLong;
+       if (shortwin)          options |= RubberBandStretcher::OptionWindowShort;
+
+       switch (transients) {
+       case NoTransients:
+               options |= RubberBandStretcher::OptionTransientsSmooth;
+               break;
+       case BandLimitedTransients:
+               options |= RubberBandStretcher::OptionTransientsMixed;
+               break;
+       case Transients:
+               options |= RubberBandStretcher::OptionTransientsCrisp;
+               break;
+       }
+
+       current_timefx->request.opts = (int) options;
+#else
        current_timefx->request.quick_seek = current_timefx->quick_button.get_active();
        current_timefx->request.antialias = !current_timefx->antialias_button.get_active();
-       current_timefx->request.progress = 0.0f;
+#endif
        current_timefx->request.done = false;
        current_timefx->request.cancel = false;
-       
+
        /* re-connect the cancel button and delete events */
-       
+
        current_timefx->first_cancel.disconnect();
        current_timefx->first_delete.disconnect();
-       
-       current_timefx->first_cancel = current_timefx->cancel_button->signal_clicked().connect 
-               (mem_fun (current_timefx, &TimeFXDialog::cancel_in_progress));
-       current_timefx->first_delete = current_timefx->signal_delete_event().connect 
-               (mem_fun (current_timefx, &TimeFXDialog::delete_in_progress));
 
-       if (pthread_create_and_store ("timefx", &current_timefx->request.thread, 0, timefx_thread, current_timefx)) {
+       current_timefx->first_cancel = current_timefx->cancel_button->signal_clicked().connect
+               (sigc::mem_fun (current_timefx, &TimeFXDialog::cancel_in_progress));
+       current_timefx->first_delete = current_timefx->signal_delete_event().connect
+               (sigc::mem_fun (current_timefx, &TimeFXDialog::delete_in_progress));
+
+       if (pthread_create_and_store ("timefx", &current_timefx->request.thread, timefx_thread, current_timefx)) {
                current_timefx->hide ();
                error << _("timefx cannot be started - thread creation error") << endmsg;
                return -1;
@@ -237,73 +304,65 @@ Editor::time_fx (RegionSelection& regions, float val, bool pitching)
 
        pthread_detach (current_timefx->request.thread);
 
-       sigc::connection c = Glib::signal_timeout().connect (mem_fun (current_timefx, &TimeFXDialog::update_progress), 100);
-
-       while (!current_timefx->request.done) {
+       while (!current_timefx->request.done && !current_timefx->request.cancel) {
                gtk_main_iteration ();
        }
 
-       c.disconnect ();
-       
+       pthread_join (current_timefx->request.thread, 0);
+
        current_timefx->hide ();
        return current_timefx->status;
 }
 
 void
-Editor::do_timefx (TimeFXDialog& dialog)
+Editor::do_timefx ()
 {
-       Track*    t;
        boost::shared_ptr<Playlist> playlist;
        boost::shared_ptr<Region>   new_region;
-       bool in_command = false;
-       
-       for (RegionSelection::iterator i = dialog.regions.begin(); i != dialog.regions.end(); ) {
-               AudioRegionView* arv = dynamic_cast<AudioRegionView*>(*i);
+       set<boost::shared_ptr<Playlist> > playlists_affected;
 
-               if (!arv) {
-                       continue;
-               }
+       uint32_t const N = current_timefx->regions.size ();
 
-               boost::shared_ptr<AudioRegion> region (arv->audio_region());
-               TimeAxisView* tv = &(arv->get_time_axis_view());
-               RouteTimeAxisView* rtv;
-               RegionSelection::iterator tmp;
-               
-               tmp = i;
-               ++tmp;
+       for (RegionList::iterator i = current_timefx->regions.begin(); i != current_timefx->regions.end(); ++i) {
+               boost::shared_ptr<Playlist> playlist = (*i)->playlist();
 
-               if ((rtv = dynamic_cast<RouteTimeAxisView*> (tv)) == 0) {
-                       i = tmp;
-                       continue;
+               if (playlist) {
+                       playlist->clear_changes ();
                }
+       }
 
-               if ((t = dynamic_cast<Track*> (rtv->route().get())) == 0) {
-                       i = tmp;
-                       continue;
-               }
-       
-               if ((playlist = t->diskstream()->playlist()) == 0) {
-                       i = tmp;
+       for (RegionList::iterator i = current_timefx->regions.begin(); i != current_timefx->regions.end(); ++i) {
+
+               boost::shared_ptr<AudioRegion> region = boost::dynamic_pointer_cast<AudioRegion> (*i);
+
+               if (!region || (playlist = region->playlist()) == 0) {
                        continue;
                }
 
-               if (dialog.request.cancel) {
+               if (current_timefx->request.cancel) {
                        /* we were cancelled */
-                       dialog.status = 1;
+                       /* XXX what to do about playlists already affected ? */
+                       current_timefx->status = 1;
                        return;
                }
 
                Filter* fx;
 
-               if (dialog.pitching) {
-                       fx = new Pitch (*session, dialog.request);
+               if (current_timefx->pitching) {
+                       fx = new Pitch (*_session, current_timefx->request);
                } else {
-                       fx = new Stretch (*session, dialog.request);
+#ifdef USE_RUBBERBAND
+                       fx = new RBStretch (*_session, current_timefx->request);
+#else
+                       fx = new STStretch (*_session, current_timefx->request);
+#endif
                }
 
-               if (fx->run (region)) {
-                       dialog.status = -1;
-                       dialog.request.done = true;
+               current_timefx->descend (1.0 / N);
+
+               if (fx->run (region, current_timefx)) {
+                       current_timefx->status = -1;
+                       current_timefx->request.done = true;
                        delete fx;
                        return;
                }
@@ -311,40 +370,44 @@ Editor::do_timefx (TimeFXDialog& dialog)
                if (!fx->results.empty()) {
                        new_region = fx->results.front();
 
-                       if (!in_command) {
-                               begin_reversible_command (dialog.pitching ? _("pitch shift") : _("time stretch"));
-                               in_command = true;
-                       }
-
-                       XMLNode &before = playlist->get_state();
                        playlist->replace_region (region, new_region, region->position());
-                       XMLNode &after = playlist->get_state();
-                       session->add_command (new MementoCommand<Playlist>(*playlist, &before, &after));
+                       playlists_affected.insert (playlist);
                }
 
-               i = tmp;
+               current_timefx->ascend ();
                delete fx;
        }
 
-       if (in_command) {
-               commit_reversible_command ();
+       for (set<boost::shared_ptr<Playlist> >::iterator p = playlists_affected.begin(); p != playlists_affected.end(); ++p) {
+               _session->add_command (new StatefulDiffCommand (*p));
        }
 
-       dialog.status = 0;
-       dialog.request.done = true;
+       current_timefx->status = 0;
+       current_timefx->request.done = true;
 }
 
 void*
 Editor::timefx_thread (void *arg)
 {
-       PBD::ThreadCreated (pthread_self(), X_("TimeFX"));
+       SessionEvent::create_per_thread_pool ("timefx events", 64);
 
        TimeFXDialog* tsd = static_cast<TimeFXDialog*>(arg);
 
        pthread_setcanceltype (PTHREAD_CANCEL_ASYNCHRONOUS, 0);
 
-       tsd->editor.do_timefx (*tsd);
+       tsd->editor.do_timefx ();
+
+        /* GACK! HACK! sleep for a bit so that our request buffer for the GUI
+           event loop doesn't die before any changes we made are processed
+           by the GUI ...
+        */
 
+#ifdef PLATFORM_WINDOWS
+       Sleep(2000);
+#else
+        struct timespec t = { 2, 0 };
+        nanosleep (&t, 0);
+#endif
        return 0;
 }