Remove confusing track / group members in TimeSelection struct; time selection
[ardour.git] / gtk2_ardour / editor_timefx.cc
index 79772090f6b61b86de2d540327c43c750b7dd169..8538c7b8b3c79548a68e7188399357aefca6847e 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
     along with this program; if not, write to the Free Software
     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 
-    $Id$
 */
 
+#include <iostream>
 #include <cstdlib>
 #include <cmath>
 
 #include <string>
 
-#include <pbd/error.h>
-#include <pbd/pthread_utils.h>
+#include "pbd/error.h"
+#include "pbd/pthread_utils.h"
+#include "pbd/memento_command.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 <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 "time_fx_dialog.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/midi_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::TimeStretchDialog::TimeStretchDialog (Editor& e)
-       : ArdourDialog ("time stretch dialog"),
-         editor (e),
-         quick_button (_("Quick but Ugly")),
-         antialias_button (_("Skip Anti-aliasing"))
-{
-       set_modal (true);
-       set_position (Gtk::WIN_POS_MOUSE);
-       set_title (_("ardour: timestretch"));
-       set_name (N_("TimeStretchDialog"));
-
-       get_vbox()->set_spacing (5);
-       get_vbox()->set_border_width (5);
-       get_vbox()->pack_start (upper_button_box);
-       get_vbox()->pack_start (progress_bar);
-
-       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);
-
-       action_button = add_button (_("Stretch/Shrink it"), Gtk::RESPONSE_ACCEPT);
-       cancel_button = add_button (_("Cancel"), Gtk::RESPONSE_CANCEL);
-
-       quick_button.set_name (N_("TimeStretchButton"));
-       antialias_button.set_name (N_("TimeStretchButton"));
-       progress_bar.set_name (N_("TimeStretchProgress"));
-
-       show_all_children ();
-}
-
-gint
-Editor::TimeStretchDialog::update_progress ()
+int
+Editor::time_stretch (RegionSelection& regions, float fraction)
 {
-       progress_bar.set_fraction (request.progress);
-       return request.running;
-}
+       // FIXME: kludge, implement stretching of selection of both types
+
+       if (regions.front()->region()->data_type() == DataType::AUDIO) {
+               // Audio, pop up timefx dialog
+               return time_fx (regions, fraction, false);
+       } else {
+               // MIDI, just stretch
+               RouteTimeAxisView* rtv = dynamic_cast<RouteTimeAxisView*> (&regions.front()->get_time_axis_view());
+               if (!rtv)
+                       return -1;
+
+               boost::shared_ptr<Playlist> playlist
+                       = rtv->track()->diskstream()->playlist();
+
+           ARDOUR::TimeFXRequest request;
+               request.time_fraction = fraction;
+               MidiStretch stretch(*session, request);
+               begin_reversible_command ("midi stretch");
+               stretch.run(regions.front()->region());
+               XMLNode &before = playlist->get_state();
+               playlist->replace_region (regions.front()->region(), stretch.results[0],
+                               regions.front()->region()->position());
+               XMLNode &after = playlist->get_state();
+               session->add_command (new MementoCommand<Playlist>(*playlist, &before, &after));
+               commit_reversible_command ();
+       }
 
-void
-Editor::TimeStretchDialog::cancel_timestretch_in_progress ()
-{
-       status = -2;
-       request.running = false;
+       return 0;
 }
 
-gint
-Editor::TimeStretchDialog::delete_timestretch_in_progress (GdkEventAny* ev)
+int
+Editor::pitch_shift (RegionSelection& regions, float fraction)
 {
-       status = -2;
-       request.running = false;
-       return TRUE;
+       return time_fx (regions, fraction, true);
 }
 
 int
-Editor::run_timestretch (RegionSelection& regions, float fraction)
+Editor::time_fx (RegionSelection& regions, float val, bool pitching)
 {
-       pthread_t thread;
+       delete current_timefx;
 
-       if (current_timestretch == 0) {
-               current_timestretch = new TimeStretchDialog (*this);
-       }
+       current_timefx = new TimeFXDialog (*this, pitching);
 
-       current_timestretch->progress_bar.set_fraction (0.0f);
+       current_timefx->progress_bar.set_fraction (0.0f);
 
-       switch (current_timestretch->run ()) {
+       switch (current_timefx->run ()) {
        case RESPONSE_ACCEPT:
                break;
        default:
-               current_timestretch->hide ();
+               current_timefx->hide ();
                return 1;
        }
 
-       current_timestretch->status = 0;
-       current_timestretch->regions = regions;
-       current_timestretch->request.fraction = fraction;
-       current_timestretch->request.quick_seek = current_timestretch->quick_button.get_active();
-       current_timestretch->request.antialias = !current_timestretch->antialias_button.get_active();
-       current_timestretch->request.progress = 0.0f;
-       current_timestretch->request.running = true;
-       
+       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();
+
+               if (cents == 0.0) {
+                       // user didn't change anything
+                       current_timefx->hide ();
+                       return 0;
+               }
+
+               // one octave == 1200 cents
+               // adding one octave doubles the frequency
+               // ratio is 2^^octaves
+
+               pitch_fraction = pow(2, cents/1200);
+
+               current_timefx->request.time_fraction = 1.0;
+               current_timefx->request.pitch_fraction = pitch_fraction;
+
+       } else {
+
+               current_timefx->request.time_fraction = val;
+               current_timefx->request.pitch_fraction = 1.0;
+
+       }
+
+#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 ();
+
+       if (txt == rb_opt_strings[0]) {
+               transients = NoTransients; peaklock = false; longwin = true; shortwin = false;
+       } else if (txt == rb_opt_strings[1]) {
+               transients = NoTransients; peaklock = false; longwin = false; shortwin = false;
+       } else if (txt == rb_opt_strings[2]) {
+               transients = NoTransients; peaklock = true; longwin = false; shortwin = false;
+       } else if (txt == rb_opt_strings[3]) {
+               transients = BandLimitedTransients; peaklock = true; longwin = false; shortwin = false;
+       } else if (txt == rb_opt_strings[5]) {
+               transients = Transients; peaklock = false; longwin = false; shortwin = true;
+       } else {
+               /* default/4 */
+
+               transients = Transients; peaklock = true; longwin = false; shortwin = false;
+       }
+
+       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();
+#endif
+       current_timefx->request.progress = 0.0f;
+       current_timefx->request.done = false;
+       current_timefx->request.cancel = false;
+
        /* re-connect the cancel button and delete events */
-       
-       current_timestretch->first_cancel.disconnect();
-       current_timestretch->first_delete.disconnect();
-       
-       current_timestretch->cancel_button->signal_clicked().connect (mem_fun (current_timestretch, &TimeStretchDialog::cancel_timestretch_in_progress));
-       current_timestretch->signal_delete_event().connect (mem_fun (current_timestretch, &TimeStretchDialog::delete_timestretch_in_progress));
-
-       if (pthread_create_and_store ("timestretch", &thread, 0, timestretch_thread, current_timestretch)) {
-               current_timestretch->hide ();
-               error << _("timestretch cannot be started - thread creation error") << endmsg;
+
+       current_timefx->first_cancel.disconnect();
+       current_timefx->first_delete.disconnect();
+
+       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;
        }
 
-       pthread_detach (thread);
+       pthread_detach (current_timefx->request.thread);
 
-       sigc::connection c = Glib::signal_timeout().connect (mem_fun (current_timestretch, &TimeStretchDialog::update_progress), 100);
+       sigc::connection c = Glib::signal_timeout().connect (sigc::mem_fun (current_timefx, &TimeFXDialog::update_progress), 100);
 
-       while (current_timestretch->request.running) {
+       while (!current_timefx->request.done && !current_timefx->request.cancel) {
                gtk_main_iteration ();
        }
 
        c.disconnect ();
-       
-       current_timestretch->hide ();
-       return current_timestretch->status;
+
+       current_timefx->hide ();
+       return current_timefx->status;
 }
 
 void
-Editor::do_timestretch (TimeStretchDialog& dialog)
+Editor::do_timefx (TimeFXDialog& dialog)
 {
        Track*    t;
-       Playlist* playlist;
-       Region*   new_region;
-
+       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);
-               if (!arv)
+
+               if (!arv) {
                        continue;
+               }
 
-               AudioRegion& region (arv->audio_region());
+               boost::shared_ptr<AudioRegion> region (arv->audio_region());
                TimeAxisView* tv = &(arv->get_time_axis_view());
                RouteTimeAxisView* rtv;
                RegionSelection::iterator tmp;
-               
-               cerr << "stretch " << region.name() << endl;
 
                tmp = i;
                ++tmp;
@@ -186,47 +276,73 @@ Editor::do_timestretch (TimeStretchDialog& dialog)
                        i = tmp;
                        continue;
                }
-       
-               if ((playlist = t->diskstream().playlist()) == 0) {
+
+               if ((playlist = t->diskstream()->playlist()) == 0) {
                        i = tmp;
                        continue;
                }
 
-               dialog.request.region = &region;
-
-               if (!dialog.request.running) {
+               if (dialog.request.cancel) {
                        /* we were cancelled */
                        dialog.status = 1;
                        return;
                }
 
-               if ((new_region = session->tempoize_region (dialog.request)) == 0) {
+               Filter* fx;
+
+               if (dialog.pitching) {
+                       fx = new Pitch (*session, dialog.request);
+               } else {
+#ifdef USE_RUBBERBAND
+                       fx = new RBStretch (*session, dialog.request);
+#else
+                       fx = new STStretch (*session, dialog.request);
+#endif
+               }
+
+               if (fx->run (region)) {
                        dialog.status = -1;
-                       dialog.request.running = false;
+                       dialog.request.done = true;
+                       delete fx;
                        return;
                }
 
-               session->add_undo (playlist->get_memento());
-               playlist->replace_region (region, *new_region, region.position());
-               session->add_redo_no_execute (playlist->get_memento());
+               if (!fx->results.empty()) {
+                       new_region = fx->results.front();
+
+                       if (!in_command) {
+                               session->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));
+               }
 
                i = tmp;
+               delete fx;
+       }
+
+       if (in_command) {
+               session->commit_reversible_command ();
        }
 
        dialog.status = 0;
-       dialog.request.running = false;
+       dialog.request.done = true;
 }
 
 void*
-Editor::timestretch_thread (void *arg)
+Editor::timefx_thread (void *arg)
 {
-       PBD::ThreadCreated (pthread_self(), X_("TimeFX"));
+       SessionEvent::create_per_thread_pool ("timefx events", 64);
 
-       TimeStretchDialog* tsd = static_cast<TimeStretchDialog*>(arg);
+       TimeFXDialog* tsd = static_cast<TimeFXDialog*>(arg);
 
        pthread_setcanceltype (PTHREAD_CANCEL_ASYNCHRONOUS, 0);
 
-       tsd->editor.do_timestretch (*tsd);
+       tsd->editor.do_timefx (*tsd);
 
        return 0;
 }