Chris@133: 
Chris@133: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@133: 
Chris@133: /*
Chris@133:     Sonic Visualiser
Chris@133:     An audio file viewer and annotation editor.
Chris@133:     Centre for Digital Music, Queen Mary, University of London.
Chris@182:     This file copyright 2006 QMUL.
Chris@133:     
Chris@133:     This program is free software; you can redistribute it and/or
Chris@133:     modify it under the terms of the GNU General Public License as
Chris@133:     published by the Free Software Foundation; either version 2 of the
Chris@133:     License, or (at your option) any later version.  See the file
Chris@133:     COPYING included with this distribution for more information.
Chris@133: */
Chris@133: 
Chris@1238: #ifndef SV_SPECTRUM_LAYER_H
Chris@1238: #define SV_SPECTRUM_LAYER_H
Chris@133: 
Chris@193: #include "SliceLayer.h"
Chris@133: 
Chris@153: #include "base/Window.h"
Chris@153: 
Chris@133: #include "data/model/DenseTimeValueModel.h"
Chris@133: 
Chris@1238: #include "PianoScale.h"
Chris@1238: 
Chris@133: #include <QColor>
Chris@349: #include <QMutex>
Chris@133: 
Chris@133: class FFTModel;
Chris@133: 
Chris@1238: class SpectrumLayer : public SliceLayer,
Chris@1238:                       public PianoScale::HorizontalScaleProvider
Chris@133: {
Chris@133:     Q_OBJECT
Chris@133: 
Chris@133: public:
Chris@133:     SpectrumLayer();
Chris@133:     ~SpectrumLayer();
Chris@133:     
Chris@133:     void setModel(DenseTimeValueModel *model);
cannam@1241:     virtual const Model *getModel() const override { return m_originModel; }
Chris@133: 
Chris@918:     virtual bool getCrosshairExtents(LayerGeometryProvider *, QPainter &, QPoint cursorPos,
cannam@1241:                                      std::vector<QRect> &extents) const override;
cannam@1241:     virtual void paintCrosshairs(LayerGeometryProvider *, QPainter &, QPoint) const override;
Chris@254: 
cannam@1241:     virtual QString getFeatureDescription(LayerGeometryProvider *v, QPoint &) const override;
Chris@199: 
cannam@1241:     virtual void paint(LayerGeometryProvider *v, QPainter &paint, QRect rect) const override;
Chris@275: 
cannam@1241:     virtual VerticalPosition getPreferredFrameCountPosition() const override {
Chris@1266:         return PositionTop;
Chris@277:     }
Chris@277: 
cannam@1241:     virtual PropertyList getProperties() const override;
cannam@1241:     virtual QString getPropertyLabel(const PropertyName &) const override;
cannam@1241:     virtual QString getPropertyIconName(const PropertyName &) const override;
cannam@1241:     virtual PropertyType getPropertyType(const PropertyName &) const override;
cannam@1241:     virtual QString getPropertyGroupName(const PropertyName &) const override;
Chris@153:     virtual int getPropertyRangeAndValue(const PropertyName &,
cannam@1241:                                          int *min, int *max, int *deflt) const override;
Chris@153:     virtual QString getPropertyValueLabel(const PropertyName &,
Chris@1266:                                           int value) const override;
cannam@1241:     virtual RangeMapper *getNewPropertyRangeMapper(const PropertyName &) const override;
cannam@1241:     virtual void setProperty(const PropertyName &, int value) override;
cannam@1241:     virtual void setProperties(const QXmlAttributes &) override;
Chris@133: 
Chris@918:     virtual bool getXScaleValue(const LayerGeometryProvider *v, int x,
cannam@1241:                                 double &value, QString &unit) const override;
Chris@260: 
Chris@918:     virtual bool getYScaleValue(const LayerGeometryProvider *, int y,
cannam@1241:                                 double &value, QString &unit) const override;
Chris@274: 
Chris@918:     virtual bool getYScaleDifference(const LayerGeometryProvider *, int y0, int y1,
cannam@1241:                                      double &diff, QString &unit) const override;
Chris@274: 
cannam@1241:     virtual bool isLayerScrollable(const LayerGeometryProvider *) const override { return false; }
Chris@133: 
Chris@153:     void setChannel(int);
Chris@153:     int getChannel() const { return m_channel; }
Chris@153: 
Chris@805:     void setWindowSize(int);
Chris@805:     int getWindowSize() const { return m_windowSize; }
Chris@153:     
Chris@805:     void setWindowHopLevel(int level);
Chris@805:     int getWindowHopLevel() const { return m_windowHopLevel; }
Chris@153: 
Chris@153:     void setWindowType(WindowType type);
Chris@153:     WindowType getWindowType() const { return m_windowType; }
Chris@153: 
Chris@284:     void setShowPeaks(bool);
Chris@284:     bool getShowPeaks() const { return m_showPeaks; }
Chris@284: 
Chris@316:     virtual void toXml(QTextStream &stream, QString indent = "",
cannam@1241:                        QString extraAttributes = "") const override;
Chris@153: 
Chris@1238:     virtual double getFrequencyForX(const LayerGeometryProvider *,
Chris@1238:                                     double x) const override;
Chris@1238:     virtual double getXForFrequency(const LayerGeometryProvider *,
Chris@1238:                                     double freq) const override;
Chris@1238: 
Chris@153: protected slots:
Chris@153:     void preferenceChanged(PropertyContainer::PropertyName name);
Chris@133: 
Chris@133: protected:
Chris@193:     DenseTimeValueModel    *m_originModel;
Chris@153:     int                     m_channel;
Chris@153:     bool                    m_channelSet;
Chris@805:     int                  m_windowSize;
Chris@153:     WindowType              m_windowType;
Chris@805:     int                  m_windowHopLevel;
Chris@284:     bool                    m_showPeaks;
Chris@275:     mutable bool            m_newFFTNeeded;
Chris@153: 
Chris@349:     mutable QMutex m_fftMutex;
Chris@349: 
Chris@193:     void setupFFT();
Chris@153: 
cannam@1241:     virtual void getBiasCurve(BiasCurve &) const override;
Chris@254:     BiasCurve m_biasCurve;
Chris@254: 
Chris@805:     int getWindowIncrement() const {
Chris@153:         if (m_windowHopLevel == 0) return m_windowSize;
Chris@153:         else if (m_windowHopLevel == 1) return (m_windowSize * 3) / 4;
Chris@153:         else return m_windowSize / (1 << (m_windowHopLevel - 1));
Chris@153:     }
Chris@133: };
Chris@133: 
Chris@133: #endif