Chris@152: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@152: 
Chris@152: /*
Chris@152:     Sonic Visualiser
Chris@152:     An audio file viewer and annotation editor.
Chris@152:     Centre for Digital Music, Queen Mary, University of London.
Chris@202:     This file copyright 2006 Chris Cannam and QMUL.
Chris@152:     
Chris@152:     This program is free software; you can redistribute it and/or
Chris@152:     modify it under the terms of the GNU General Public License as
Chris@152:     published by the Free Software Foundation; either version 2 of the
Chris@152:     License, or (at your option) any later version.  See the file
Chris@152:     COPYING included with this distribution for more information.
Chris@152: */
Chris@152: 
Chris@152: #ifndef _EDITABLE_DENSE_THREE_DIMENSIONAL_MODEL_H_
Chris@152: #define _EDITABLE_DENSE_THREE_DIMENSIONAL_MODEL_H_
Chris@152: 
Chris@152: #include "DenseThreeDimensionalModel.h"
Chris@152: 
Chris@536: #include <QReadWriteLock>
Chris@536: 
Chris@533: #include <vector>
Chris@533: 
Chris@152: class EditableDenseThreeDimensionalModel : public DenseThreeDimensionalModel
Chris@152: {
Chris@152:     Q_OBJECT
Chris@152: 
Chris@152: public:
Chris@535: 
Chris@535:     // EditableDenseThreeDimensionalModel supports a basic compression
Chris@535:     // method that reduces the size of multirate data (e.g. wavelet
Chris@535:     // transform outputs) that are stored as plain 3d grids by about
Chris@535:     // 60% or thereabouts.  However, it can only be used for models
Chris@535:     // whose columns are set in order from 0 and never subsequently
Chris@535:     // changed.  If the model is going to be actually edited, it must
Chris@535:     // have NoCompression.
Chris@535: 
Chris@535:     enum CompressionType
Chris@535:     {
Chris@535:         NoCompression,
Chris@535:         BasicMultirateCompression
Chris@535:     };
Chris@535: 
Chris@152:     EditableDenseThreeDimensionalModel(size_t sampleRate,
Chris@152: 				       size_t resolution,
Chris@152: 				       size_t yBinCount,
Chris@535:                                        CompressionType compression,
Chris@152: 				       bool notifyOnAdd = true);
Chris@152: 
Chris@152:     virtual bool isOK() const;
Chris@152: 
Chris@152:     virtual size_t getSampleRate() const;
Chris@152:     virtual size_t getStartFrame() const;
Chris@152:     virtual size_t getEndFrame() const;
Chris@152: 
Chris@152:     virtual Model *clone() const;
Chris@152:     
Chris@152: 
Chris@152:     /**
Chris@611:      * Set the frame offset of the first column.
Chris@611:      */
Chris@611:     virtual void setStartFrame(size_t);
Chris@611: 
Chris@611:     /**
Chris@152:      * Return the number of sample frames covered by each set of bins.
Chris@152:      */
Chris@152:     virtual size_t getResolution() const;
Chris@152: 
Chris@152:     /**
Chris@152:      * Set the number of sample frames covered by each set of bins.
Chris@152:      */
Chris@152:     virtual void setResolution(size_t sz);
Chris@152: 
Chris@152:     /**
Chris@182:      * Return the number of columns.
Chris@182:      */
Chris@182:     virtual size_t getWidth() const;
Chris@182: 
Chris@182:     /**
Chris@152:      * Return the number of bins in each set of bins.
Chris@152:      */
Chris@182:     virtual size_t getHeight() const; 
Chris@152: 
Chris@152:     /**
Chris@152:      * Set the number of bins in each set of bins.
Chris@152:      */
Chris@182:     virtual void setHeight(size_t sz);
Chris@152: 
Chris@152:     /**
Chris@152:      * Return the minimum value of the value in each bin.
Chris@152:      */
Chris@152:     virtual float getMinimumLevel() const;
Chris@152: 
Chris@152:     /**
Chris@152:      * Set the minimum value of the value in a bin.
Chris@152:      */
Chris@152:     virtual void setMinimumLevel(float sz);
Chris@152: 
Chris@152:     /**
Chris@152:      * Return the maximum value of the value in each bin.
Chris@152:      */
Chris@152:     virtual float getMaximumLevel() const;
Chris@152: 
Chris@152:     /**
Chris@152:      * Set the maximum value of the value in a bin.
Chris@152:      */
Chris@152:     virtual void setMaximumLevel(float sz);
Chris@152: 
Chris@182:     /**
Chris@182:      * Return true if there are data available for the given column.
Chris@182:      */
Chris@182:     virtual bool isColumnAvailable(size_t x) const { return x < getWidth(); }
Chris@152: 
Chris@152:     /**
Chris@182:      * Get the set of bin values at the given column.
Chris@152:      */
Chris@533:     virtual Column getColumn(size_t x) const;
Chris@152: 
Chris@152:     /**
Chris@182:      * Get a single value, from the n'th bin of the given column.
Chris@152:      */
Chris@182:     virtual float getValueAt(size_t x, size_t n) const;
Chris@152: 
Chris@152:     /**
Chris@182:      * Set the entire set of bin values at the given column.
Chris@152:      */
Chris@182:     virtual void setColumn(size_t x, const Column &values);
Chris@152: 
Chris@152:     virtual QString getBinName(size_t n) const;
Chris@152:     virtual void setBinName(size_t n, QString);
Chris@152:     virtual void setBinNames(std::vector<QString> names);
Chris@152: 
Chris@478:     bool shouldUseLogValueScale() const;
Chris@478: 
Chris@333:     virtual void setCompletion(int completion, bool update = true);
Chris@152:     virtual int getCompletion() const { return m_completion; }
Chris@152: 
Chris@345:     QString getTypeName() const { return tr("Editable Dense 3-D"); }
Chris@345: 
Chris@318:     virtual QString toDelimitedDataString(QString delimiter) const;
Chris@318: 
Chris@152:     virtual void toXml(QTextStream &out,
Chris@152:                        QString indent = "",
Chris@152:                        QString extraAttributes = "") const;
Chris@152: 
Chris@152: protected:
Chris@533:     typedef QVector<Column> ValueMatrix;
Chris@152:     ValueMatrix m_data;
Chris@152: 
Chris@535:     // m_trunc is used for simple compression.  If at least the top N
Chris@535:     // elements of column x (for N = some proportion of the column
Chris@535:     // height) are equal to those of an earlier column x', then
Chris@535:     // m_trunc[x] will contain x-x' and column x will be truncated so
Chris@535:     // as to remove the duplicate elements.  If the equal elements are
Chris@535:     // at the bottom, then m_trunc[x] will contain x'-x (a negative
Chris@535:     // value).  If m_trunc[x] is 0 then the whole of column x is
Chris@535:     // stored.
Chris@535:     std::vector<signed char> m_trunc;
Chris@534:     void truncateAndStore(size_t index, const Column & values);
Chris@534:     Column expandAndRetrieve(size_t index) const;
Chris@534: 
Chris@152:     std::vector<QString> m_binNames;
Chris@152: 
Chris@611:     size_t m_startFrame;
Chris@152:     size_t m_sampleRate;
Chris@152:     size_t m_resolution;
Chris@152:     size_t m_yBinCount;
Chris@535:     CompressionType m_compression;
Chris@152:     float m_minimum;
Chris@152:     float m_maximum;
Chris@256:     bool m_haveExtents;
Chris@152:     bool m_notifyOnAdd;
Chris@152:     long m_sinceLastNotifyMin;
Chris@152:     long m_sinceLastNotifyMax;
Chris@152:     int m_completion;
Chris@152: 
Chris@536:     mutable QReadWriteLock m_lock;
Chris@152: };
Chris@152: 
Chris@152: #endif