Chris@147: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@147: 
Chris@147: /*
Chris@147:     Sonic Visualiser
Chris@147:     An audio file viewer and annotation editor.
Chris@147:     Centre for Digital Music, Queen Mary, University of London.
Chris@147:     This file copyright 2006 Chris Cannam.
Chris@147:     
Chris@147:     This program is free software; you can redistribute it and/or
Chris@147:     modify it under the terms of the GNU General Public License as
Chris@147:     published by the Free Software Foundation; either version 2 of the
Chris@147:     License, or (at your option) any later version.  See the file
Chris@147:     COPYING included with this distribution for more information.
Chris@147: */
Chris@147: 
Chris@147: #ifndef _NOTE_MODEL_H_
Chris@147: #define _NOTE_MODEL_H_
Chris@147: 
Chris@437: #include "IntervalModel.h"
Chris@852: #include "NoteData.h"
Chris@391: #include "base/RealTime.h"
Chris@150: #include "base/PlayParameterRepository.h"
Chris@852: #include "base/Pitch.h"
Chris@147: 
Chris@147: /**
Chris@441:  * NoteModel -- a concrete IntervalModel for notes.
Chris@441:  */
Chris@441: 
Chris@441: /**
Chris@441:  * Note type for use in a sparse model.  All we mean by a "note" is
Chris@441:  * something that has an onset time, a single value, a duration, and a
Chris@441:  * level.  Like other points, it can also have a label.  With this
Chris@441:  * point type, the model can be thought of as representing a simple
Chris@441:  * MIDI-type piano roll, except that the y coordinates (values) do not
Chris@441:  * have to be discrete integers.
Chris@147:  */
Chris@147: 
Chris@147: struct Note
Chris@147: {
Chris@147: public:
Chris@340:     Note(long _frame) : frame(_frame), value(0.0f), duration(0), level(1.f) { }
Chris@929:     Note(long _frame, float _value, int _duration, float _level, QString _label) :
Chris@340: 	frame(_frame), value(_value), duration(_duration), level(_level), label(_label) { }
Chris@147: 
Chris@147:     int getDimensions() const { return 3; }
Chris@147: 
Chris@147:     long frame;
Chris@147:     float value;
Chris@929:     int duration;
Chris@340:     float level;
Chris@147:     QString label;
Chris@147: 
Chris@338:     QString getLabel() const { return label; }
Chris@338:     
Chris@314:     void toXml(QTextStream &stream,
Chris@314:                QString indent = "",
Chris@314:                QString extraAttributes = "") const
Chris@147:     {
Chris@314: 	stream <<
Chris@340:             QString("%1<point frame=\"%2\" value=\"%3\" duration=\"%4\" level=\"%5\" label=\"%6\" %7/>\n")
Chris@627: 	    .arg(indent).arg(frame).arg(value).arg(duration).arg(level)
Chris@627:             .arg(XmlExportable::encodeEntities(label)).arg(extraAttributes);
Chris@147:     }
Chris@147: 
Chris@929:     QString toDelimitedDataString(QString delimiter, int sampleRate) const
Chris@147:     {
Chris@147:         QStringList list;
Chris@147:         list << RealTime::frame2RealTime(frame, sampleRate).toString().c_str();
Chris@147:         list << QString("%1").arg(value);
Chris@340:         list << RealTime::frame2RealTime(duration, sampleRate).toString().c_str();
Chris@340:         list << QString("%1").arg(level);
Chris@318:         if (label != "") list << label;
Chris@147:         return list.join(delimiter);
Chris@147:     }
Chris@147: 
Chris@147:     struct Comparator {
Chris@147: 	bool operator()(const Note &p1,
Chris@147: 			const Note &p2) const {
Chris@147: 	    if (p1.frame != p2.frame) return p1.frame < p2.frame;
Chris@147: 	    if (p1.value != p2.value) return p1.value < p2.value;
Chris@147: 	    if (p1.duration != p2.duration) return p1.duration < p2.duration;
Chris@340:             if (p1.level != p2.level) return p1.level < p2.level;
Chris@147: 	    return p1.label < p2.label;
Chris@147: 	}
Chris@147:     };
Chris@147:     
Chris@147:     struct OrderComparator {
Chris@147: 	bool operator()(const Note &p1,
Chris@147: 			const Note &p2) const {
Chris@147: 	    return p1.frame < p2.frame;
Chris@147: 	}
Chris@147:     };
Chris@147: };
Chris@147: 
Chris@147: 
Chris@852: class NoteModel : public IntervalModel<Note>, public NoteExportable
Chris@147: {
Chris@423:     Q_OBJECT
Chris@423:     
Chris@147: public:
Chris@929:     NoteModel(int sampleRate, int resolution,
Chris@256: 	      bool notifyOnAdd = true) :
Chris@437: 	IntervalModel<Note>(sampleRate, resolution, notifyOnAdd),
Chris@256: 	m_valueQuantization(0)
Chris@256:     {
Chris@391: 	PlayParameterRepository::getInstance()->addPlayable(this);
Chris@256:     }
Chris@256: 
Chris@929:     NoteModel(int sampleRate, int resolution,
Chris@147: 	      float valueMinimum, float valueMaximum,
Chris@147: 	      bool notifyOnAdd = true) :
Chris@437: 	IntervalModel<Note>(sampleRate, resolution,
Chris@437:                             valueMinimum, valueMaximum,
Chris@437:                             notifyOnAdd),
Chris@147: 	m_valueQuantization(0)
Chris@147:     {
Chris@391: 	PlayParameterRepository::getInstance()->addPlayable(this);
Chris@391:     }
Chris@391: 
Chris@391:     virtual ~NoteModel()
Chris@391:     {
Chris@391:         PlayParameterRepository::getInstance()->removePlayable(this);
Chris@147:     }
Chris@147: 
Chris@147:     float getValueQuantization() const { return m_valueQuantization; }
Chris@147:     void setValueQuantization(float q) { m_valueQuantization = q; }
Chris@147: 
Chris@345:     QString getTypeName() const { return tr("Note"); }
Chris@345: 
Chris@391:     virtual bool canPlay() const { return true; }
Chris@391: 
Chris@866:     virtual QString getDefaultPlayClipId() const
Chris@391:     {
Chris@909:         return "elecpiano";
Chris@391:     }
Chris@391: 
Chris@288:     virtual void toXml(QTextStream &out,
Chris@288:                        QString indent = "",
Chris@288:                        QString extraAttributes = "") const
Chris@147:     {
Chris@318:         std::cerr << "NoteModel::toXml: extraAttributes = \"" 
Chris@318:                   << extraAttributes.toStdString() << std::endl;
Chris@318: 
Chris@437:         IntervalModel<Note>::toXml
Chris@288: 	    (out,
Chris@288:              indent,
Chris@452: 	     QString("%1 subtype=\"note\" valueQuantization=\"%2\"")
Chris@147: 	     .arg(extraAttributes).arg(m_valueQuantization));
Chris@147:     }
Chris@147: 
Chris@424:     /**
Chris@424:      * TabularModel methods.  
Chris@424:      */
Chris@424:     
Chris@424:     virtual int getColumnCount() const
Chris@424:     {
Chris@424:         return 6;
Chris@424:     }
Chris@424: 
Chris@424:     virtual QString getHeading(int column) const
Chris@424:     {
Chris@424:         switch (column) {
Chris@424:         case 0: return tr("Time");
Chris@424:         case 1: return tr("Frame");
Chris@424:         case 2: return tr("Pitch");
Chris@424:         case 3: return tr("Duration");
Chris@424:         case 4: return tr("Level");
Chris@424:         case 5: return tr("Label");
Chris@424:         default: return tr("Unknown");
Chris@424:         }
Chris@424:     }
Chris@424: 
Chris@424:     virtual QVariant getData(int row, int column, int role) const
Chris@424:     {
Chris@437:         if (column < 4) {
Chris@437:             return IntervalModel<Note>::getData(row, column, role);
Chris@425:         }
Chris@425: 
Chris@606:         PointListConstIterator i = getPointListIteratorForRow(row);
Chris@424:         if (i == m_points.end()) return QVariant();
Chris@424: 
Chris@424:         switch (column) {
Chris@424:         case 4: return i->level;
Chris@424:         case 5: return i->label;
Chris@424:         default: return QVariant();
Chris@424:         }
Chris@424:     }
Chris@424: 
Chris@424:     virtual Command *getSetDataCommand(int row, int column, const QVariant &value, int role)
Chris@424:     {
Chris@437:         if (column < 4) {
Chris@437:             return IntervalModel<Note>::getSetDataCommand
Chris@425:                 (row, column, value, role);
Chris@425:         }
Chris@425: 
Chris@740:         if (role != Qt::EditRole) return 0;
Chris@606:         PointListConstIterator i = getPointListIteratorForRow(row);
Chris@740:         if (i == m_points.end()) return 0;
Chris@424:         EditCommand *command = new EditCommand(this, tr("Edit Data"));
Chris@424: 
Chris@424:         Point point(*i);
Chris@424:         command->deletePoint(point);
Chris@424: 
Chris@424:         switch (column) {
Chris@424:         case 4: point.level = value.toDouble(); break;
Chris@424:         case 5: point.label = value.toString(); break;
Chris@424:         }
Chris@424: 
Chris@424:         command->addPoint(point);
Chris@424:         return command->finish();
Chris@424:     }
Chris@424: 
Chris@424:     virtual SortType getSortType(int column) const
Chris@424:     {
Chris@424:         if (column == 5) return SortAlphabetical;
Chris@424:         return SortNumeric;
Chris@424:     }
Chris@424: 
Chris@852:     /**
Chris@852:      * NoteExportable methods.
Chris@852:      */
Chris@852: 
Chris@852:     NoteList getNotes() const {
Chris@929:         return getNotesWithin(getStartFrame(), getEndFrame());
Chris@852:     }
Chris@852: 
Chris@929:     NoteList getNotesWithin(int startFrame, int endFrame) const {
Chris@852:         
Chris@852: 	PointList points = getPoints(startFrame, endFrame);
Chris@852:         NoteList notes;
Chris@852: 
Chris@852:         for (PointList::iterator pli =
Chris@852: 		 points.begin(); pli != points.end(); ++pli) {
Chris@852: 
Chris@929: 	    int duration = pli->duration;
Chris@852:             if (duration == 0 || duration == 1) {
Chris@852:                 duration = getSampleRate() / 20;
Chris@852:             }
Chris@852: 
Chris@852:             int pitch = lrintf(pli->value);
Chris@852:             
Chris@852:             int velocity = 100;
Chris@852:             if (pli->level > 0.f && pli->level <= 1.f) {
Chris@852:                 velocity = lrintf(pli->level * 127);
Chris@852:             }
Chris@852: 
Chris@852:             NoteData note(pli->frame, duration, pitch, velocity);
Chris@852: 
Chris@852:             if (getScaleUnits() == "Hz") {
Chris@852:                 note.frequency = pli->value;
Chris@852:                 note.midiPitch = Pitch::getPitchForFrequency(note.frequency);
Chris@852:                 note.isMidiPitchQuantized = false;
Chris@852:             }
Chris@852:         
Chris@852:             notes.push_back(note);
Chris@852:         }
Chris@852:         
Chris@852:         return notes;
Chris@852:     }
Chris@852: 
Chris@147: protected:
Chris@147:     float m_valueQuantization;
Chris@147: };
Chris@147: 
Chris@147: #endif