add ActionManager::set_sensitive() for Gtk::ActionGroup
[ardour.git] / libs / evoral / src / SMF.cpp
index 047763723e213d6297405d1c99d809179260c37f..64dd4e6889a483f4043709ef7a1cce8247576d50 100644 (file)
  * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
+#include <algorithm>
 #include <cassert>
 #include <cmath>
 #include <iostream>
 #include <stdint.h>
+
+#include <glib/gstdio.h>
+
 #include "libsmf/smf.h"
+
 #include "evoral/Event.hpp"
 #include "evoral/SMF.hpp"
 #include "evoral/midi_util.h"
-#include "pbd/file_manager.h"
 
 #ifdef COMPILER_MSVC
 extern double round(double x);
@@ -35,6 +39,13 @@ using namespace std;
 
 namespace Evoral {
 
+SMF::SMF()
+       : _smf (0)
+       , _smf_track (0)
+       , _empty (true)
+       , _type0 (false)
+       {};
+
 SMF::~SMF()
 {
        close ();
@@ -44,7 +55,7 @@ uint16_t
 SMF::num_tracks() const
 {
        Glib::Threads::Mutex::Lock lm (_smf_lock);
-       return _smf->number_of_tracks;
+       return _smf ? _smf->number_of_tracks : 0;
 }
 
 uint16_t
@@ -78,17 +89,20 @@ SMF::seek_to_track(int track)
 bool
 SMF::test(const std::string& path)
 {
-       PBD::StdioFileDescriptor d (path, "r");
-       FILE* f = d.allocate ();
+       FILE* f = g_fopen(path.c_str(), "r");
        if (f == 0) {
                return false;
        }
 
-       smf_t* test_smf;
-       if ((test_smf = smf_load (f)) == NULL) {
+       smf_t* test_smf = smf_load(f);
+       fclose(f);
+
+       if (!test_smf) {
                return false;
        }
-       smf_delete (test_smf);
+       if (test_smf) {
+               smf_delete(test_smf);
+       }
        return true;
 }
 
@@ -103,24 +117,22 @@ SMF::open(const std::string& path, int track) THROW_FILE_ERROR
 {
        Glib::Threads::Mutex::Lock lm (_smf_lock);
 
+       _type0 = false;
+       _type0channels.clear ();
+
        assert(track >= 1);
        if (_smf) {
                smf_delete(_smf);
        }
 
-       _file_path = path;
-
-       PBD::StdioFileDescriptor d (_file_path, "r");
-       FILE* f = d.allocate ();
+       FILE* f = g_fopen(path.c_str(), "r");
        if (f == 0) {
                return -1;
-       }
-
-       if ((_smf = smf_load (f)) == 0) {
+       } else if ((_smf = smf_load(f)) == 0) {
+               fclose(f);
                return -1;
-       }
-
-       if ((_smf_track = smf_get_track_by_number(_smf, track)) == 0) {
+       } else if ((_smf_track = smf_get_track_by_number(_smf, track)) == 0) {
+               fclose(f);
                return -2;
        }
 
@@ -133,6 +145,34 @@ SMF::open(const std::string& path, int track) THROW_FILE_ERROR
                _empty = false;
        }
 
+       fclose(f);
+
+       lm.release ();
+       if (_smf->format == 0 && _smf->number_of_tracks == 1 && !_empty) {
+               // type-0 file: scan file for # of used channels.
+               int ret;
+               uint32_t delta_t = 0;
+               uint32_t size    = 0;
+               uint8_t* buf     = NULL;
+               event_id_t event_id = 0;
+               seek_to_start();
+               while ((ret = read_event (&delta_t, &size, &buf, &event_id)) >= 0) {
+                       if (ret == 0) {
+                               continue;
+                       }
+                       if (size == 0) {
+                               break;
+                       }
+                       uint8_t type = buf[0] & 0xf0;
+                       uint8_t chan = buf[0] & 0x0f;
+                       if (type < 0x80 || type > 0xE0) {
+                               continue;
+                       }
+                       _type0channels.insert(chan);
+               }
+               _type0 = true;
+               seek_to_start();
+       }
        return 0;
 }
 
@@ -153,8 +193,6 @@ SMF::create(const std::string& path, int track, uint16_t ppqn) THROW_FILE_ERROR
                smf_delete(_smf);
        }
 
-       _file_path = path;
-
        _smf = smf_new();
 
        if (_smf == NULL) {
@@ -167,7 +205,9 @@ SMF::create(const std::string& path, int track, uint16_t ppqn) THROW_FILE_ERROR
 
        for (int i = 0; i < track; ++i) {
                _smf_track = smf_track_new();
-               assert(_smf_track);
+               if (!_smf_track) {
+                       return -2;
+               }
                smf_add_track(_smf, _smf_track);
        }
 
@@ -180,18 +220,21 @@ SMF::create(const std::string& path, int track, uint16_t ppqn) THROW_FILE_ERROR
        {
                /* put a stub file on disk */
 
-               PBD::StdioFileDescriptor d (_file_path, "w+");
-               FILE* f = d.allocate ();
+               FILE* f = g_fopen (path.c_str(), "w+");
                if (f == 0) {
                        return -1;
                }
 
                if (smf_save (_smf, f)) {
+                       fclose (f);
                        return -1;
                }
+               fclose (f);
        }
 
        _empty = true;
+       _type0 = false;
+       _type0channels.clear ();
 
        return 0;
 }
@@ -205,6 +248,8 @@ SMF::close() THROW_FILE_ERROR
                smf_delete(_smf);
                _smf = 0;
                _smf_track = 0;
+               _type0 = false;
+               _type0channels.clear ();
        }
 }
 
@@ -212,7 +257,11 @@ void
 SMF::seek_to_start() const
 {
        Glib::Threads::Mutex::Lock lm (_smf_lock);
-       _smf_track->next_event_number = 1;
+       if (_smf_track) {
+               _smf_track->next_event_number = std::min(_smf_track->number_of_events, (size_t)1);
+       } else {
+               cerr << "WARNING: SMF seek_to_start() with no track" << endl;
+       }
 }
 
 /** Read an event from the current position in file.
@@ -280,10 +329,20 @@ SMF::read_event(uint32_t* delta_t, uint32_t* size, uint8_t** buf, event_id_t* no
                if (*size < (unsigned)event_size) {
                        *buf = (uint8_t*)realloc(*buf, event_size);
                }
+               assert (*buf);
                memcpy(*buf, event->midi_buffer, size_t(event_size));
                *size = event_size;
+               if (((*buf)[0] & 0xF0) == 0x90 && (*buf)[2] == 0) {
+                       /* normalize note on with velocity 0 to proper note off */
+                       (*buf)[0] = 0x80 | ((*buf)[0] & 0x0F);  /* note off */
+                       (*buf)[2] = 0x40;  /* default velocity */
+               }
 
-               assert(midi_event_is_valid(*buf, *size));
+               if (!midi_event_is_valid(*buf, *size)) {
+                       cerr << "WARNING: SMF ignoring illegal MIDI event" << endl;
+                       *size = 0;
+                       return -1;
+               }
 
                /* printf("SMF::read_event @ %u: ", *delta_t);
                   for (size_t i = 0; i < *size; ++i) {
@@ -310,6 +369,26 @@ SMF::append_event_delta(uint32_t delta_t, uint32_t size, const uint8_t* buf, eve
           printf("%X ", buf[i]);
           } printf("\n"); */
 
+       switch (buf[0]) {
+       case 0xf1:
+       case 0xf2:
+       case 0xf3:
+       case 0xf4:
+       case 0xf5:
+       case 0xf6:
+       case 0xf8:
+       case 0xf9:
+       case 0xfa:
+       case 0xfb:
+       case 0xfc:
+       case 0xfd:
+       case 0xfe:
+       case 0xff:
+               /* System Real Time or System Common event: not valid in SMF
+                */
+               return;
+       }
+
        if (!midi_event_is_valid(buf, size)) {
                cerr << "WARNING: SMF ignoring illegal MIDI event" << endl;
                return;
@@ -324,6 +403,7 @@ SMF::append_event_delta(uint32_t delta_t, uint32_t size, const uint8_t* buf, eve
        bool const store_id = (
                c == MIDI_CMD_NOTE_ON ||
                c == MIDI_CMD_NOTE_OFF ||
+               c == MIDI_CMD_NOTE_PRESSURE ||
                c == MIDI_CMD_PGM_CHANGE ||
                (c == MIDI_CMD_CONTROL && (buf[1] == MIDI_CTL_MSB_BANK || buf[1] == MIDI_CTL_LSB_BANK))
                               );
@@ -387,18 +467,25 @@ SMF::begin_write()
 }
 
 void
-SMF::end_write() THROW_FILE_ERROR
+SMF::end_write(string const & path) THROW_FILE_ERROR
 {
        Glib::Threads::Mutex::Lock lm (_smf_lock);
-       PBD::StdioFileDescriptor d (_file_path, "w+");
-       FILE* f = d.allocate ();
+
+       if (!_smf) {
+               return;
+       }
+
+       FILE* f = g_fopen (path.c_str(), "w+");
        if (f == 0) {
-               throw FileError (_file_path);
+               throw FileError (path);
        }
 
        if (smf_save(_smf, f) != 0) {
-               throw FileError (_file_path);
+               fclose(f);
+               throw FileError (path);
        }
+
+       fclose(f);
 }
 
 double
@@ -410,9 +497,104 @@ SMF::round_to_file_precision (double val) const
 }
 
 void
-SMF::set_path (const std::string& p)
+SMF::track_names(vector<string>& names) const
 {
-       _file_path = p;
+       if (!_smf) {
+               return;
+       }
+
+       names.clear ();
+
+       Glib::Threads::Mutex::Lock lm (_smf_lock);
+
+       for (uint16_t n = 0; n < _smf->number_of_tracks; ++n) {
+               smf_track_t* trk = smf_get_track_by_number (_smf, n+1);
+               if (!trk) {
+                       names.push_back (string());
+               } else {
+                       if (trk->name) {
+                               names.push_back (trk->name);
+                       } else {
+                               names.push_back (string());
+                       }
+               }
+       }
+}
+
+void
+SMF::instrument_names(vector<string>& names) const
+{
+       if (!_smf) {
+               return;
+       }
+
+       names.clear ();
+
+       Glib::Threads::Mutex::Lock lm (_smf_lock);
+
+       for (uint16_t n = 0; n < _smf->number_of_tracks; ++n) {
+               smf_track_t* trk = smf_get_track_by_number (_smf, n+1);
+               if (!trk) {
+                       names.push_back (string());
+               } else {
+                       if (trk->instrument) {
+                               names.push_back (trk->instrument);
+                       } else {
+                               names.push_back (string());
+                       }
+               }
+       }
+}
+
+SMF::Tempo::Tempo (smf_tempo_t* smft)
+       : time_pulses (smft->time_pulses)
+       , time_seconds (smft->time_seconds)
+       , microseconds_per_quarter_note (smft->microseconds_per_quarter_note)
+       , numerator (smft->numerator)
+       , denominator (smft->denominator)
+       , clocks_per_click (smft->clocks_per_click)
+       , notes_per_note (smft->notes_per_note)
+{
+}
+
+int
+SMF::num_tempos () const
+{
+       assert (_smf);
+       return smf_get_tempo_count (_smf);
+}
+
+SMF::Tempo*
+SMF::tempo_at_smf_pulse (size_t smf_pulse) const
+{
+       smf_tempo_t* t = smf_get_tempo_by_seconds (_smf, smf_pulse);
+       if (!t) {
+               return 0;
+       }
+       return new Tempo (t);
+}
+
+SMF::Tempo*
+SMF::tempo_at_seconds (double seconds) const
+{
+       smf_tempo_t* t = smf_get_tempo_by_seconds (_smf, seconds);
+       if (!t) {
+               return 0;
+       }
+       return new Tempo (t);
+}
+
+SMF::Tempo*
+SMF::nth_tempo (size_t n) const
+{
+       assert (_smf);
+
+       smf_tempo_t* t = smf_get_tempo_by_number (_smf, n);
+       if (!t) {
+               return 0;
+       }
+
+       return new Tempo (t);
 }
 
 } // namespace Evoral