Merge with trunk R2978.
[ardour.git] / libs / ardour / smf_source.cc
index b046ad6fca78247fe5f14f761b4fdd0854ebe7a2..29f36cc2cd3be54c2d6d655414f6d847a6762aca 100644 (file)
@@ -55,7 +55,6 @@ SMFSource::SMFSource (Session& s, std::string path, Flag flags)
        , _channel(0)
        , _flags (Flag(flags | Writable)) // FIXME: this needs to be writable for now
        , _allow_remove_if_empty(true)
-       , _timeline_position (0)
        , _fd (0)
        , _last_ev_time(0)
        , _track_size(4) // 4 bytes for the ever-present EOT event
@@ -70,6 +69,8 @@ SMFSource::SMFSource (Session& s, std::string path, Flag flags)
        if (open()) {
                throw failed_constructor ();
        }
+
+       //cerr << "SMF Source path: " << path << endl;
        
        assert(_name.find("/") == string::npos);
 }
@@ -79,7 +80,6 @@ SMFSource::SMFSource (Session& s, const XMLNode& node)
        , _channel(0)
        , _flags (Flag (Writable|CanRename))
        , _allow_remove_if_empty(true)
-       , _timeline_position (0)
        , _fd (0)
        , _last_ev_time(0)
        , _track_size(4) // 4 bytes for the ever-present EOT event
@@ -99,6 +99,8 @@ SMFSource::SMFSource (Session& s, const XMLNode& node)
                throw failed_constructor ();
        }
        
+       //cerr << "SMF Source name: " << _name << endl;
+       
        assert(_name.find("/") == string::npos);
 }
 
@@ -113,7 +115,7 @@ bool
 SMFSource::removable () const
 {
        return (_flags & Removable) && ((_flags & RemoveAtDestroy) || 
-                                     ((_flags & RemovableIfEmpty) && is_empty (_path)));
+                                     ((_flags & RemovableIfEmpty) && is_empty()));
 }
 
 int
@@ -137,7 +139,7 @@ SMFSource::init (string pathstr, bool must_exist)
 int
 SMFSource::open()
 {
-       cerr << "Opening SMF file " << path() << " writeable: " << writable() << endl;
+       //cerr << "Opening SMF file " << path() << " writeable: " << writable() << endl;
 
        assert(writable()); // FIXME;
 
@@ -149,26 +151,26 @@ SMFSource::open()
                uint32_t track_size_be = 0;
                fread(&track_size_be, 4, 1, _fd);
                _track_size = GUINT32_FROM_BE(track_size_be);
-               cerr << "SMF - read track size " << _track_size << endl;
+               //cerr << "SMF - read track size " << _track_size << endl;
 
        // We're making a new file
        } else {
                _fd = fopen(path().c_str(), "w+");
-               _track_size = 0;
+               _track_size = 4;
 
-               // write a tentative header just to pad things out so writing happens in the right spot
+               // Write a tentative header just to pad things out so writing happens in the right spot
                flush_header();
-               // FIXME: write the footer here too so it's a valid SMF (screw up writing ATM though)
+               write_footer();
+               seek_to_end();
        }
 
        return (_fd == 0) ? -1 : 0;
 }
 
-int
-SMFSource::update_header (nframes_t when, struct tm&, time_t)
+void
+SMFSource::seek_to_end()
 {
-       _timeline_position = when;
-       return flush_header();
+       fseek(_fd, -4, SEEK_END);
 }
 
 int
@@ -176,7 +178,7 @@ SMFSource::flush_header ()
 {
        // FIXME: write timeline position somehow?
        
-       cerr << "SMF Flushing header\n";
+       //cerr << "SMF Flushing header\n";
 
        assert(_fd);
 
@@ -205,16 +207,24 @@ SMFSource::flush_header ()
 int
 SMFSource::flush_footer()
 {
-       cerr << "SMF - Writing EOT\n";
+       seek_to_end();
+       write_footer();
+       seek_to_end();
 
-       fseek(_fd, 0, SEEK_END);
-       write_var_len(1); // whatever...
-       char eot[4] = { 0xFF, 0x2F, 0x00 }; // end-of-track meta-event
-       fwrite(eot, 1, 4, _fd);
-       fflush(_fd);
        return 0;
 }
 
+void
+SMFSource::write_footer()
+{
+       //cerr << "SMF " << name() << " writing EOT at byte " << ftell(_fd) << endl;
+       
+       write_var_len(0);
+       char eot[3] = { 0xFF, 0x2F, 0x00 }; // end-of-track meta-event
+       fwrite(eot, 1, 3, _fd);
+       fflush(_fd);
+}
+
 /** Returns the offset of the first event in the file with a time past @a start,
  * relative to the start of the source.
  *
@@ -246,84 +256,102 @@ SMFSource::find_first_event_after(nframes_t start)
  * will have it's time field set to it's delta time, in SMF tempo-based ticks, using the
  * rate given by ppqn() (it is the caller's responsibility to calculate a real time).
  *
+ * \a size should be the capacity of \a buf.  If it is not large enough, \a buf will
+ * be freed and a new buffer allocated in its place, the size of which will be placed
+ * in size.
+ *
  * Returns event length (including status byte) on success, 0 if event was
  * skipped (eg a meta event), or -1 on EOF (or end of track).
  */
 int
-SMFSource::read_event(MidiEvent& ev) const
+SMFSource::read_event(uint32_t* delta_t, uint32_t* size, Byte** buf) const
 {
-       // - 4 is for the EOT event, which we don't actually want to read
-       //if (feof(_fd) || ftell(_fd) >= _header_size + _track_size - 4) {
        if (feof(_fd)) {
                return -1;
        }
 
-       uint32_t delta_time = read_var_len();
+       assert(delta_t);
+       assert(size);
+       assert(buf);
+
+       *delta_t = read_var_len();
        assert(!feof(_fd));
-       int status = fgetc(_fd);
+
+       const int status = fgetc(_fd);
        assert(status != EOF); // FIXME die gracefully
+
+       //printf("Status @ %X = %X\n", (unsigned)ftell(_fd) - 1, status);
+
        if (status == 0xFF) {
                assert(!feof(_fd));
-               int type = fgetc(_fd);
+               const int type = fgetc(_fd);
                if ((unsigned char)type == 0x2F) {
-                       //cerr << "SMF - hit EOT" << endl;
-                       return -1; // we hit the logical EOF anyway...
+                       //cerr << _name << " hit EOT" << endl;
+                       return -1;
                } else {
-                       ev.size = 0;
-                       ev.time = delta_time; // this is needed regardless
+                       *size = 0;
                        return 0;
                }
        }
        
-       ev.time = delta_time;
-       ev.size = midi_event_size((unsigned char)status) + 1;
+       const int event_size = midi_event_size((unsigned char)status) + 1;
+       if (event_size <= 0) {
+               *size = 0;
+               return 0;
+       }
+       
+       // Make sure we have enough scratch buffer
+       if (*size < (unsigned)event_size)
+               *buf = (Byte*)realloc(*buf, event_size);
+       
+       *size = event_size;
 
-       if (ev.buffer == NULL)
-               ev.buffer = (Byte*)malloc(sizeof(Byte) * ev.size);
+       /*if (ev.buffer == NULL)
+               ev.buffer = (Byte*)malloc(sizeof(Byte) * ev.size);*/
 
-       ev.buffer[0] = (unsigned char)status;
-       fread(ev.buffer+1, 1, ev.size - 1, _fd);
+       (*buf)[0] = (unsigned char)status;
+       if (event_size > 1)
+               fread((*buf) + 1, 1, *size - 1, _fd);
 
-       /*printf("SMF - read event, delta = %u, size = %zu, data = ",
-               delta_time, ev.size);
-       for (size_t i=0; i < ev.size; ++i) {
-               printf("%X ", ev.buffer[i]);
+       /*printf("%s read event: delta = %u, size = %u, data = ", _name.c_str(), *delta_t, *size);
+       for (size_t i=0; i < *size; ++i) {
+               printf("%X ", (*buf)[i]);
        }
        printf("\n");*/
        
-       return ev.size;
+       return (int)*size;
 }
 
 /** All stamps in audio frames */
 nframes_t
 SMFSource::read_unlocked (MidiRingBuffer& dst, nframes_t start, nframes_t cnt, nframes_t stamp_offset) const
 {
-       //cerr << "SMF - read " << start << ", count=" << cnt << ", offset=" << stamp_offset << endl;
+       //cerr << "SMF " << name() << " read " << start << ", count=" << cnt << ", offset=" << stamp_offset << endl;
 
        // 64 bits ought to be enough for anybody
        uint64_t time = 0; // in SMF ticks, 1 tick per _ppqn
 
        _read_data_count = 0;
 
-       // FIXME: ugh
-       unsigned char ev_buf[MidiBuffer::max_event_size()];
-       MidiEvent ev;
-       ev.time = 0;
-       ev.size = MidiBuffer::max_event_size();
-       ev.buffer = ev_buf;
+       // Output parameters for read_event (which will allocate scratch in buffer as needed)
+       uint32_t ev_delta_t = 0;
+       uint32_t ev_size = 0;
+       Byte*    ev_buffer = 0;
 
-       // FIXME: it would be an impressive feat to actually make this any slower :)
-       
+       size_t scratch_size = 0; // keep track of scratch to minimize reallocs
+
+       // FIXME: don't seek to start and search every read (brutal!)
        fseek(_fd, _header_size, 0);
        
        // FIXME: assumes tempo never changes after start
        const double frames_per_beat = _session.tempo_map().tempo_at(_timeline_position).frames_per_beat(
-                       _session.engine().frame_rate());
+                       _session.engine().frame_rate(),
+                       _session.tempo_map().meter_at(_timeline_position));
        
-       uint64_t start_ticks = (uint64_t)((start / frames_per_beat) * _ppqn);
+       const uint64_t start_ticks = (uint64_t)((start / frames_per_beat) * _ppqn);
 
        while (!feof(_fd)) {
-               int ret = read_event(ev);
+               int ret = read_event(&ev_delta_t, &ev_size, &ev_buffer);
                if (ret == -1) { // EOF
                        //cerr << "SMF - EOF\n";
                        break;
@@ -331,24 +359,28 @@ SMFSource::read_unlocked (MidiRingBuffer& dst, nframes_t start, nframes_t cnt, n
 
                if (ret == 0) { // meta-event (skipped)
                        //cerr << "SMF - META\n";
-                       time += ev.time; // just accumulate delta time and ignore event
+                       time += ev_delta_t; // just accumulate delta time and ignore event
                        continue;
                }
 
-               time += ev.time; // accumulate delta time
-               ev.time = time; // set ev.time to actual time (relative to source start)
+               time += ev_delta_t; // accumulate delta time
 
-               if (ev.time >= start_ticks) {
-                       if (ev.time < start_ticks + (cnt / frames_per_beat)) {
+               if (time >= start_ticks) {
+                       const nframes_t ev_frame_time = (nframes_t)(
+                                       ((time / (double)_ppqn) * frames_per_beat)) + stamp_offset;
+
+                       if (ev_frame_time <= start + cnt)
+                               dst.write(ev_frame_time, ev_size, ev_buffer);
+                       else
                                break;
-                       } else {
-                               ev.time = (nframes_t)(((ev.time / (double)_ppqn) * frames_per_beat)) + stamp_offset;
-                               // write event time in absolute frames
-                               dst.write(ev.time, ev.size, ev.buffer);
-                       }
                }
 
-               _read_data_count += ev.size;
+               _read_data_count += ev_size;
+
+               if (ev_size > scratch_size)
+                       scratch_size = ev_size;
+               else
+                       ev_size = scratch_size; // minimize realloc in read_event
        }
        
        return cnt;
@@ -359,51 +391,87 @@ nframes_t
 SMFSource::write_unlocked (MidiRingBuffer& src, nframes_t cnt)
 {
        _write_data_count = 0;
+               
+       double time;
+       size_t size;
 
-       boost::shared_ptr<MidiBuffer> buf_ptr(new MidiBuffer(1024)); // FIXME: size?
-       MidiBuffer& buf = *buf_ptr.get();
-       src.read(buf, /*_length*/0, _length + cnt); // FIXME?
+       size_t buf_capacity = 4;
+       Byte* buf = (Byte*)malloc(buf_capacity);
+       
+       if (_model && ! _model->writing())
+               _model->start_write();
 
-       fseek(_fd, 0, SEEK_END);
+       while (true) {
+               bool ret = src.full_peek(sizeof(double), (Byte*)&time);
+               if (!ret || time - _timeline_position > _length + cnt)
+                       break;
 
-       // FIXME: assumes tempo never changes after start
-       const double frames_per_beat = _session.tempo_map().tempo_at(_timeline_position).frames_per_beat(
-                       _session.engine().frame_rate());
-       
-       for (size_t i=0; i < buf.size(); ++i) {
-               MidiEvent& ev = buf[i];
-               assert(ev.time >= _timeline_position);
-               ev.time -= _timeline_position;
-               assert(ev.time >= _last_ev_time);
-               const uint32_t delta_time = (uint32_t)((ev.time - _last_ev_time) / frames_per_beat * _ppqn);
-               
-               /*printf("SMF - writing event, delta = %u, size = %zu, data = ",
-                       delta_time, ev.size);
-               for (size_t i=0; i < ev.size; ++i) {
-                       printf("%X ", ev.buffer[i]);
+               ret = src.read_prefix(&time, &size);
+               if (!ret)
+                       break;
+
+               if (size > buf_capacity) {
+                       buf_capacity = size;
+                       buf = (Byte*)realloc(buf, size);
                }
-               printf("\n");
-               */
-               size_t stamp_size = write_var_len(delta_time);
-               fwrite(ev.buffer, 1, ev.size, _fd);
 
-               _track_size += stamp_size + ev.size;
-               _write_data_count += ev.size;
+               ret = src.read_contents(size, buf);
+               if (!ret) {
+                       cerr << "ERROR: Read time/size but not buffer, corrupt MIDI ring buffer" << endl;
+                       break;
+               }
                
-               _last_ev_time = ev.time;
+               assert(time >= _timeline_position);
+               time -= _timeline_position;
+
+               const MidiEvent ev(time, size, buf);
+               append_event_unlocked(ev);
+
+               if (_model)
+                       _model->append(ev);
        }
 
        fflush(_fd);
+       free(buf);
 
        const nframes_t oldlen = _length;
        update_length(oldlen, cnt);
 
-       _model->append(buf);
-
-       ViewDataRangeReady (buf_ptr, oldlen, cnt); /* EMIT SIGNAL */
+       ViewDataRangeReady (_timeline_position + oldlen, cnt); /* EMIT SIGNAL */
        
        return cnt;
 }
+               
+
+void
+SMFSource::append_event_unlocked(const MidiEvent& ev)
+{
+       /*printf("SMF - writing event, time = %lf, size = %u, data = ", ev.time(), ev.size());
+       for (size_t i=0; i < ev.size(); ++i) {
+               printf("%X ", ev.buffer()[i]);
+       }
+       printf("\n");*/
+
+       assert(ev.time() >= 0);
+
+       assert(ev.time() >= _last_ev_time);
+       
+       // FIXME: assumes tempo never changes after start
+       const double frames_per_beat = _session.tempo_map().tempo_at(_timeline_position).frames_per_beat(
+                       _session.engine().frame_rate(),
+                       _session.tempo_map().meter_at(_timeline_position));
+       
+       const uint32_t delta_time = (uint32_t)((ev.time() - _last_ev_time) / frames_per_beat * _ppqn);
+
+       const size_t stamp_size = write_var_len(delta_time);
+       fwrite(ev.buffer(), 1, ev.size(), _fd);
+
+       _track_size += stamp_size + ev.size();
+       _write_data_count += ev.size();
+
+       _last_ev_time = ev.time();
+}
+
 
 XMLNode&
 SMFSource::get_state ()
@@ -450,13 +518,23 @@ SMFSource::mark_for_remove ()
        _flags = Flag (_flags | RemoveAtDestroy);
 }
 
+void
+SMFSource::mark_streaming_midi_write_started (NoteMode mode, nframes_t start_frame)
+{
+       MidiSource::mark_streaming_midi_write_started (mode, start_frame);
+       _last_ev_time = 0;
+}
+
 void
 SMFSource::mark_streaming_write_completed ()
 {
+       MidiSource::mark_streaming_write_completed();
+
        if (!writable()) {
                return;
        }
        
+       flush_header();
        flush_footer();
 
 #if 0
@@ -680,7 +758,7 @@ SMFSource::set_allow_remove_if_empty (bool yn)
 }
 
 int
-SMFSource::set_name (string newname, bool destructive)
+SMFSource::set_source_name (string newname, bool destructive)
 {
        //Glib::Mutex::Lock lm (_lock); FIXME
        string oldpath = _path;
@@ -703,16 +781,18 @@ SMFSource::set_name (string newname, bool destructive)
 }
 
 bool
-SMFSource::is_empty (string path)
+SMFSource::is_empty () const
 {
-       /* XXX fix me */
+       bool ret = (_track_size > 4);
+
+       //cerr << name() << " IS EMPTY: " << ret << endl;
 
-       return false;
+       return ret;
 }
 
 
 void
-SMFSource::write_chunk_header(char id[4], uint32_t length)
+SMFSource::write_chunk_header(const char id[4], uint32_t length)
 {
        const uint32_t length_be = GUINT32_TO_BE(length);
 
@@ -721,7 +801,7 @@ SMFSource::write_chunk_header(char id[4], uint32_t length)
 }
 
 void
-SMFSource::write_chunk(char id[4], uint32_t length, void* data)
+SMFSource::write_chunk(const char id[4], uint32_t length, void* data)
 {
        write_chunk_header(id, length);
        
@@ -759,9 +839,6 @@ SMFSource::read_var_len() const
 {
        assert(!feof(_fd));
 
-       //int offset = ftell(_fd);
-       //cerr << "SMF - reading var len at " << offset << endl;
-
        uint32_t value;
        unsigned char c;
 
@@ -777,42 +854,74 @@ SMFSource::read_var_len() const
 }
 
 void
-SMFSource::load_model(bool lock)
+SMFSource::load_model(bool lock, bool force_reload)
 {
+       if (_writing)
+               return;
+
        if (lock)
                Glib::Mutex::Lock lm (_lock);
 
-       destroy_model();
-       _model = new MidiModel();
+       if (_model && !force_reload && !_model->empty()) {
+               //cerr << _name << " NOT reloading model " << _model.get() << " (" << _model->n_notes()
+               //      << " notes)" << endl;
+               return;
+       } else {
+               cerr << _name << " loading model" << endl;
+       }
+
+       if (! _model) {
+               _model = boost::shared_ptr<MidiModel>(new MidiModel(_session));
+               cerr << _name << " loaded new model " << _model.get() << endl;
+       } else {
+               cerr << _name << " reloading model " << _model.get()
+                       << " (" << _model->n_notes() << " notes)" <<endl;
+               _model->clear();
+       }
+
+       _model->start_write();
 
        fseek(_fd, _header_size, 0);
 
-       nframes_t time = 0;
+       uint64_t time = 0; /* in SMF ticks */
        MidiEvent ev;
        
+       size_t scratch_size = 0; // keep track of scratch and minimize reallocs
+       
        // FIXME: assumes tempo never changes after start
        const double frames_per_beat = _session.tempo_map().tempo_at(_timeline_position).frames_per_beat(
-                       _session.engine().frame_rate());
+                       _session.engine().frame_rate(),
+                       _session.tempo_map().meter_at(_timeline_position));
        
+       uint32_t delta_t = 0;
        int ret;
-       while ((ret = read_event(ev)) >= 0) {
-               time += ev.time;
-               ev.time = time;
-
-               ev.time = (nframes_t)(ev.time * frames_per_beat / (double)_ppqn);
-
+       while ((ret = read_event(&delta_t, &ev.size(), &ev.buffer())) >= 0) {
+               
+               time += delta_t;
+               
                if (ret > 0) { // didn't skip (meta) event
-                       //cerr << "ADDING EVENT TO MODEL: " << ev.time << endl;
+                       // make ev.time absolute time in frames
+                       ev.time() = (double)time * frames_per_beat / (double)_ppqn;
+
                        _model->append(ev);
                }
+
+               if (ev.size() > scratch_size)
+                       scratch_size = ev.size();
+               else
+                       ev.size() = scratch_size;
        }
+       
+       _model->end_write(false);
+
+       free(ev.buffer());
 }
 
 
 void
 SMFSource::destroy_model()
 {
-       delete _model;
-       _model = NULL;
+       //cerr << _name << " destroying model " << _model.get() << endl;
+       _model.reset();
 }