lbajardsilogic@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
lbajardsilogic@0: 
lbajardsilogic@0: /*
lbajardsilogic@0:     Sonic Visualiser
lbajardsilogic@0:     An audio file viewer and annotation editor.
lbajardsilogic@0:     Centre for Digital Music, Queen Mary, University of London.
lbajardsilogic@0:     This file copyright 2006 Chris Cannam and QMUL.
lbajardsilogic@0:     
lbajardsilogic@0:     This program is free software; you can redistribute it and/or
lbajardsilogic@0:     modify it under the terms of the GNU General Public License as
lbajardsilogic@0:     published by the Free Software Foundation; either version 2 of the
lbajardsilogic@0:     License, or (at your option) any later version.  See the file
lbajardsilogic@0:     COPYING included with this distribution for more information.
lbajardsilogic@0: */
lbajardsilogic@0: 
lbajardsilogic@0: #include "Colour3DPlotLayer.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include "view/View.h"
lbajardsilogic@0: #include "base/Profiler.h"
lbajardsilogic@0: #include "base/LogRange.h"
lbajardsilogic@0: #include "ColourMapper.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <QPainter>
lbajardsilogic@0: #include <QImage>
lbajardsilogic@0: #include <QRect>
lbajardsilogic@0: 
lbajardsilogic@0: #include <iostream>
lbajardsilogic@0: 
lbajardsilogic@0: #include <cassert>
lbajardsilogic@0: 
lbajardsilogic@0: //#define DEBUG_COLOUR_3D_PLOT_LAYER_PAINT 1
lbajardsilogic@0: 
lbajardsilogic@0: 
lbajardsilogic@0: Colour3DPlotLayer::Colour3DPlotLayer() :
lbajardsilogic@0:     m_model(0),
lbajardsilogic@0:     m_cache(0),
lbajardsilogic@0:     m_cacheStart(0),
lbajardsilogic@0:     m_colourScale(LinearScale),
lbajardsilogic@0:     m_colourMap(0),
lbajardsilogic@0:     m_normalizeColumns(false),
lbajardsilogic@0:     m_normalizeVisibleArea(false)
lbajardsilogic@0: {
lbajardsilogic@0:     
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: Colour3DPlotLayer::~Colour3DPlotLayer()
lbajardsilogic@0: {
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::setModel(const DenseThreeDimensionalModel *model)
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_model == model) return;
lbajardsilogic@0:     const DenseThreeDimensionalModel *oldModel = m_model;
lbajardsilogic@0:     m_model = model;
lbajardsilogic@0:     if (!m_model || !m_model->isOK()) return;
lbajardsilogic@0: 
lbajardsilogic@0:     connect(m_model, SIGNAL(modelChanged()), this, SIGNAL(modelChanged()));
lbajardsilogic@0:     connect(m_model, SIGNAL(modelChanged(size_t, size_t)),
lbajardsilogic@0: 	    this, SIGNAL(modelChanged(size_t, size_t)));
lbajardsilogic@0: 
lbajardsilogic@0:     connect(m_model, SIGNAL(completionChanged()),
lbajardsilogic@0: 	    this, SIGNAL(modelCompletionChanged()));
lbajardsilogic@0: 
lbajardsilogic@0:     connect(m_model, SIGNAL(modelChanged()), this, SLOT(cacheInvalid()));
lbajardsilogic@0:     connect(m_model, SIGNAL(modelChanged(size_t, size_t)),
lbajardsilogic@0: 	    this, SLOT(cacheInvalid(size_t, size_t)));
lbajardsilogic@0: 
lbajardsilogic@0:     emit modelReplaced();
lbajardsilogic@0:     emit sliceableModelReplaced(oldModel, model);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::cacheInvalid()
lbajardsilogic@0: {
lbajardsilogic@0:     delete m_cache; 
lbajardsilogic@0:     m_cache = 0;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::cacheInvalid(size_t, size_t)
lbajardsilogic@0: {
lbajardsilogic@0:     cacheInvalid();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: Layer::PropertyList
lbajardsilogic@0: Colour3DPlotLayer::getProperties() const
lbajardsilogic@0: {
lbajardsilogic@0:     PropertyList list;
lbajardsilogic@0:     list.push_back("Colour");
lbajardsilogic@0:     list.push_back("Colour Scale");
lbajardsilogic@0:     list.push_back("Normalize Columns");
lbajardsilogic@0:     list.push_back("Normalize Visible Area");
lbajardsilogic@0:     return list;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: QString
lbajardsilogic@0: Colour3DPlotLayer::getPropertyLabel(const PropertyName &name) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (name == "Colour") return tr("Colour");
lbajardsilogic@0:     if (name == "Colour Scale") return tr("Scale");
lbajardsilogic@0:     if (name == "Normalize Columns") return tr("Normalize Columns");
lbajardsilogic@0:     if (name == "Normalize Visible Area") return tr("Normalize Visible Area");
lbajardsilogic@0:     return "";
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: Layer::PropertyType
lbajardsilogic@0: Colour3DPlotLayer::getPropertyType(const PropertyName &name) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (name == "Normalize Columns") return ToggleProperty;
lbajardsilogic@0:     if (name == "Normalize Visible Area") return ToggleProperty;
lbajardsilogic@0:     return ValueProperty;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: QString
lbajardsilogic@0: Colour3DPlotLayer::getPropertyGroupName(const PropertyName &name) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (name == "Normalize Columns" ||
lbajardsilogic@0:         name == "Normalize Visible Area" ||
lbajardsilogic@0: 	name == "Colour Scale") return tr("Scale");
lbajardsilogic@0:     return QString();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: int
lbajardsilogic@0: Colour3DPlotLayer::getPropertyRangeAndValue(const PropertyName &name,
lbajardsilogic@0:                                             int *min, int *max, int *deflt) const
lbajardsilogic@0: {
lbajardsilogic@0:     int val = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     int garbage0, garbage1, garbage2;
lbajardsilogic@0:     if (!min) min = &garbage0;
lbajardsilogic@0:     if (!max) max = &garbage1;
lbajardsilogic@0:     if (!deflt) deflt = &garbage2;
lbajardsilogic@0: 
lbajardsilogic@0:     if (name == "Colour Scale") {
lbajardsilogic@0: 
lbajardsilogic@0: 	*min = 0;
lbajardsilogic@0: 	*max = 2;
lbajardsilogic@0:         *deflt = (int)LinearScale;
lbajardsilogic@0: 
lbajardsilogic@0: 	val = (int)m_colourScale;
lbajardsilogic@0: 
lbajardsilogic@0:     } else if (name == "Colour") {
lbajardsilogic@0: 
lbajardsilogic@0: 	*min = 0;
lbajardsilogic@0: 	*max = ColourMapper::getColourMapCount() - 1;
lbajardsilogic@0:         *deflt = 0;
lbajardsilogic@0: 
lbajardsilogic@0: 	val = m_colourMap;
lbajardsilogic@0: 
lbajardsilogic@0:     } else if (name == "Normalize Columns") {
lbajardsilogic@0: 	
lbajardsilogic@0:         *deflt = 0;
lbajardsilogic@0: 	val = (m_normalizeColumns ? 1 : 0);
lbajardsilogic@0: 
lbajardsilogic@0:     } else if (name == "Normalize Visible Area") {
lbajardsilogic@0: 	
lbajardsilogic@0:         *deflt = 0;
lbajardsilogic@0: 	val = (m_normalizeVisibleArea ? 1 : 0);
lbajardsilogic@0: 
lbajardsilogic@0:     } else {
lbajardsilogic@0: 	val = Layer::getPropertyRangeAndValue(name, min, max, deflt);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     return val;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: QString
lbajardsilogic@0: Colour3DPlotLayer::getPropertyValueLabel(const PropertyName &name,
lbajardsilogic@0: 				    int value) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (name == "Colour") {
lbajardsilogic@0:         return ColourMapper::getColourMapName(value);
lbajardsilogic@0:     }
lbajardsilogic@0:     if (name == "Colour Scale") {
lbajardsilogic@0: 	switch (value) {
lbajardsilogic@0: 	default:
lbajardsilogic@0: 	case 0: return tr("Linear");
lbajardsilogic@0: 	case 1: return tr("Log");
lbajardsilogic@0: 	case 2: return tr("+/-1");
lbajardsilogic@0: 	}
lbajardsilogic@0:     }
lbajardsilogic@0:     return tr("<unknown>");
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::setProperty(const PropertyName &name, int value)
lbajardsilogic@0: {
lbajardsilogic@0:     if (name == "Colour Scale") {
lbajardsilogic@0: 	switch (value) {
lbajardsilogic@0: 	default:
lbajardsilogic@0: 	case 0: setColourScale(LinearScale); break;
lbajardsilogic@0: 	case 1: setColourScale(LogScale); break;
lbajardsilogic@0: 	case 2: setColourScale(PlusMinusOneScale); break;
lbajardsilogic@0: 	}
lbajardsilogic@0:     } else if (name == "Colour") {
lbajardsilogic@0:         setColourMap(value);
lbajardsilogic@0:     } else if (name == "Normalize Columns") {
lbajardsilogic@0: 	setNormalizeColumns(value ? true : false);
lbajardsilogic@0:     } else if (name == "Normalize Visible Area") {
lbajardsilogic@0: 	setNormalizeVisibleArea(value ? true : false);
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::setColourScale(ColourScale scale)
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_colourScale == scale) return;
lbajardsilogic@0:     m_colourScale = scale;
lbajardsilogic@0:     cacheInvalid();
lbajardsilogic@0:     emit layerParametersChanged();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::setColourMap(int map)
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_colourMap == map) return;
lbajardsilogic@0:     m_colourMap = map;
lbajardsilogic@0:     cacheInvalid();
lbajardsilogic@0:     emit layerParametersChanged();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::setNormalizeColumns(bool n)
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_normalizeColumns == n) return;
lbajardsilogic@0:     m_normalizeColumns = n;
lbajardsilogic@0:     cacheInvalid();
lbajardsilogic@0:     emit layerParametersChanged();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: bool
lbajardsilogic@0: Colour3DPlotLayer::getNormalizeColumns() const
lbajardsilogic@0: {
lbajardsilogic@0:     return m_normalizeColumns;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::setNormalizeVisibleArea(bool n)
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_normalizeVisibleArea == n) return;
lbajardsilogic@0:     m_normalizeVisibleArea = n;
lbajardsilogic@0:     cacheInvalid();
lbajardsilogic@0:     emit layerParametersChanged();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: bool
lbajardsilogic@0: Colour3DPlotLayer::getNormalizeVisibleArea() const
lbajardsilogic@0: {
lbajardsilogic@0:     return m_normalizeVisibleArea;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: bool
lbajardsilogic@0: Colour3DPlotLayer::isLayerScrollable(const View *v) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_normalizeVisibleArea) return false;
lbajardsilogic@0:     QPoint discard;
lbajardsilogic@0:     return !v->shouldIlluminateLocalFeatures(this, discard);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: QString
lbajardsilogic@0: Colour3DPlotLayer::getFeatureDescription(View *v, QPoint &pos) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (!m_model) return "";
lbajardsilogic@0: 
lbajardsilogic@0:     int x = pos.x();
lbajardsilogic@0:     int y = pos.y();
lbajardsilogic@0: 
lbajardsilogic@0:     size_t modelStart = m_model->getStartFrame();
lbajardsilogic@0:     size_t modelResolution = m_model->getResolution();
lbajardsilogic@0: 
lbajardsilogic@0:     float srRatio =
lbajardsilogic@0:         float(v->getViewManager()->getMainModelSampleRate()) /
lbajardsilogic@0:         float(m_model->getSampleRate());
lbajardsilogic@0: 
lbajardsilogic@0:     int sx0 = int((v->getFrameForX(x) / srRatio - long(modelStart)) /
lbajardsilogic@0:                   long(modelResolution));
lbajardsilogic@0: 
lbajardsilogic@0:     int f0 = sx0 * modelResolution;
lbajardsilogic@0:     int f1 =  f0 + modelResolution;
lbajardsilogic@0: 
lbajardsilogic@0:     float binHeight = float(v->height()) / m_model->getHeight();
lbajardsilogic@0:     int sy = int((v->height() - y) / binHeight);
lbajardsilogic@0: 
lbajardsilogic@0:     float value = m_model->getValueAt(sx0, sy);
lbajardsilogic@0: 
lbajardsilogic@0: //    std::cerr << "bin value (" << sx0 << "," << sy << ") is " << value << std::endl;
lbajardsilogic@0:     
lbajardsilogic@0:     QString binName = m_model->getBinName(sy);
lbajardsilogic@0:     if (binName == "") binName = QString("[%1]").arg(sy + 1);
lbajardsilogic@0:     else binName = QString("%1 [%2]").arg(binName).arg(sy + 1);
lbajardsilogic@0: 
lbajardsilogic@0:     QString text = tr("Time:\t%1 - %2\nBin:\t%3\nValue:\t%4")
lbajardsilogic@0: 	.arg(RealTime::frame2RealTime(f0, m_model->getSampleRate())
lbajardsilogic@0: 	     .toText(true).c_str())
lbajardsilogic@0: 	.arg(RealTime::frame2RealTime(f1, m_model->getSampleRate())
lbajardsilogic@0: 	     .toText(true).c_str())
lbajardsilogic@0: 	.arg(binName)
lbajardsilogic@0: 	.arg(value);
lbajardsilogic@0: 
lbajardsilogic@0:     return text;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: int
lbajardsilogic@0: Colour3DPlotLayer::getColourScaleWidth(QPainter &) const
lbajardsilogic@0: {
lbajardsilogic@0:     int cw = 20;
lbajardsilogic@0:     return cw;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: int
lbajardsilogic@0: Colour3DPlotLayer::getVerticalScaleWidth(View *, QPainter &paint) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (!m_model) return 0;
lbajardsilogic@0: 
lbajardsilogic@0:     QString sampleText = QString("[%1]").arg(m_model->getHeight());
lbajardsilogic@0:     int tw = paint.fontMetrics().width(sampleText);
lbajardsilogic@0:     bool another = false;
lbajardsilogic@0: 
lbajardsilogic@0:     for (size_t i = 0; i < m_model->getHeight(); ++i) {
lbajardsilogic@0: 	if (m_model->getBinName(i).length() > sampleText.length()) {
lbajardsilogic@0: 	    sampleText = m_model->getBinName(i);
lbajardsilogic@0:             another = true;
lbajardsilogic@0: 	}
lbajardsilogic@0:     }
lbajardsilogic@0:     if (another) {
lbajardsilogic@191: 	tw = MAX(tw, paint.fontMetrics().width(sampleText));
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     return tw + 13 + getColourScaleWidth(paint);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::paintVerticalScale(View *v, QPainter &paint, QRect rect) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (!m_model) return;
lbajardsilogic@0: 
lbajardsilogic@0:     int h = rect.height(), w = rect.width();
lbajardsilogic@0:     float binHeight = float(v->height()) / m_model->getHeight();
lbajardsilogic@0: 
lbajardsilogic@0:     int cw = getColourScaleWidth(paint);
lbajardsilogic@0:     
lbajardsilogic@0:     int ch = h - 20;
lbajardsilogic@0:     if (ch > 20 && m_cache) {
lbajardsilogic@0: 
lbajardsilogic@0:         paint.setPen(Qt::black);
lbajardsilogic@0:         paint.drawRect(4, 10, cw - 8, ch - 19);
lbajardsilogic@0: 
lbajardsilogic@0:         for (int y = 0; y < ch - 20; ++y) {
lbajardsilogic@0:             QRgb c = m_cache->color(((ch - 20 - y) * 255) / (ch - 20));
lbajardsilogic@0: //            std::cerr << "y = " << y << ": rgb " << qRed(c) << "," << qGreen(c) << "," << qBlue(c) << std::endl;
lbajardsilogic@0:             paint.setPen(QColor(qRed(c), qGreen(c), qBlue(c)));
lbajardsilogic@0:             paint.drawLine(5, 11 + y, cw - 5, 11 + y);
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     paint.setPen(Qt::black);
lbajardsilogic@0: 
lbajardsilogic@0:     int count = v->height() / paint.fontMetrics().height();
lbajardsilogic@0:     int step = m_model->getHeight() / count;
lbajardsilogic@0:     if (step == 0) step = 1;
lbajardsilogic@0: 
lbajardsilogic@0:     for (size_t i = 0; i < m_model->getHeight(); ++i) {
lbajardsilogic@0: 
lbajardsilogic@0:         if ((i % step) != 0) continue;
lbajardsilogic@0: 
lbajardsilogic@0: 	int y0 = int(v->height() - (i * binHeight) - 1);
lbajardsilogic@0: 	
lbajardsilogic@0: 	QString text = m_model->getBinName(i);
lbajardsilogic@0: 	if (text == "") text = QString("[%1]").arg(i + 1);
lbajardsilogic@0: 
lbajardsilogic@0: 	paint.drawLine(cw, y0, w, y0);
lbajardsilogic@0: 
lbajardsilogic@0: 	int cy = int(y0 - (step * binHeight)/2);
lbajardsilogic@0: 	int ty = cy + paint.fontMetrics().ascent()/2;
lbajardsilogic@0: 
lbajardsilogic@0: 	paint.drawText(cw + 5, ty, text);
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::getColumn(size_t col,
lbajardsilogic@0:                              DenseThreeDimensionalModel::Column &values) const
lbajardsilogic@0: {
lbajardsilogic@0:     m_model->getColumn(col, values);
lbajardsilogic@0: 
lbajardsilogic@0:     float colMax = 0.f, colMin = 0.f;
lbajardsilogic@0: 
lbajardsilogic@0:     float min = 0.f, max = 0.f;
lbajardsilogic@0:     if (m_normalizeColumns) {
lbajardsilogic@0:         min = m_model->getMinimumLevel();
lbajardsilogic@0:         max = m_model->getMaximumLevel();
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_normalizeColumns) {
lbajardsilogic@0:         for (size_t y = 0; y < values.size(); ++y) {
lbajardsilogic@0:             if (y == 0 || values[y] > colMax) colMax = values[y];
lbajardsilogic@0:             if (y == 0 || values[y] < colMin) colMin = values[y];
lbajardsilogic@0:         }
lbajardsilogic@0:         if (colMin == colMax) colMax = colMin + 1;
lbajardsilogic@0:     }
lbajardsilogic@0:     
lbajardsilogic@0:     for (size_t y = 0; y < values.size(); ++y) {
lbajardsilogic@0:         
lbajardsilogic@0:         float value = min;
lbajardsilogic@0: 
lbajardsilogic@0:         value = values[y];
lbajardsilogic@0: 
lbajardsilogic@0:         if (m_normalizeColumns) {
lbajardsilogic@0:             float norm = (value - colMin) / (colMax - colMin);
lbajardsilogic@0:             value = min + (max - min) * norm;
lbajardsilogic@0:         }
lbajardsilogic@0: 
lbajardsilogic@0:         values[y] = value;
lbajardsilogic@0:     }    
lbajardsilogic@0: }
lbajardsilogic@0:     
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::fillCache(size_t firstBin, size_t lastBin) const
lbajardsilogic@0: {
lbajardsilogic@0:     size_t modelStart = m_model->getStartFrame();
lbajardsilogic@0:     size_t modelEnd = m_model->getEndFrame();
lbajardsilogic@0:     size_t modelResolution = m_model->getResolution();
lbajardsilogic@0: 
lbajardsilogic@0: //    std::cerr << "Colour3DPlotLayer::fillCache: " << firstBin << " -> " << lastBin << std::endl;
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_normalizeVisibleArea || m_normalizeColumns) {
lbajardsilogic@0:         firstBin = modelStart / modelResolution;
lbajardsilogic@0:         lastBin = modelEnd / modelResolution;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     size_t cacheWidth = lastBin - firstBin + 1;
lbajardsilogic@0:     size_t cacheHeight = m_model->getHeight();
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_cache &&
lbajardsilogic@0: 	(m_cacheStart != firstBin ||
lbajardsilogic@0:          m_cache->width() != int(cacheWidth) ||
lbajardsilogic@0: 	 m_cache->height() != int(cacheHeight))) {
lbajardsilogic@0: 
lbajardsilogic@0: 	delete m_cache;
lbajardsilogic@0: 	m_cache = 0;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_cache) return;
lbajardsilogic@0: 
lbajardsilogic@0:     m_cache = new QImage(cacheWidth, cacheHeight, QImage::Format_Indexed8);
lbajardsilogic@0:     m_cacheStart = firstBin;
lbajardsilogic@0: 
lbajardsilogic@0: //    std::cerr << "Cache size " << cacheWidth << "x" << cacheHeight << " starting " << m_cacheStart << std::endl;
lbajardsilogic@0: 
lbajardsilogic@0:     m_cache->setNumColors(256);
lbajardsilogic@0:     DenseThreeDimensionalModel::Column values;
lbajardsilogic@0: 
lbajardsilogic@0:     float min = m_model->getMinimumLevel();
lbajardsilogic@0:     float max = m_model->getMaximumLevel();
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_colourScale == LogScale) {
lbajardsilogic@0:         LogRange::mapRange(min, max);
lbajardsilogic@0:     } else if (m_colourScale == PlusMinusOneScale) {
lbajardsilogic@0:         min = -1.f;
lbajardsilogic@0:         max = 1.f;
lbajardsilogic@0:     }
lbajardsilogic@0:     
lbajardsilogic@0:     if (max == min) max = min + 1.0;
lbajardsilogic@0:     
lbajardsilogic@0:     ColourMapper mapper(m_colourMap, 0.f, 255.f);
lbajardsilogic@0:     
lbajardsilogic@0:     for (int index = 0; index < 256; ++index) {
lbajardsilogic@0:         
lbajardsilogic@0:         QColor colour = mapper.map(index);
lbajardsilogic@0:         m_cache->setColor(index, qRgb(colour.red(), colour.green(), colour.blue()));
lbajardsilogic@0:     }
lbajardsilogic@0:     
lbajardsilogic@0:     m_cache->fill(0);
lbajardsilogic@0: 
lbajardsilogic@0:     float visibleMax = 0.f, visibleMin = 0.f;
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_normalizeVisibleArea && !m_normalizeColumns) {
lbajardsilogic@0:         
lbajardsilogic@0:         for (size_t c = firstBin; c <= lastBin; ++c) {
lbajardsilogic@0: 	
lbajardsilogic@0:             values.clear();
lbajardsilogic@0:             getColumn(c, values);
lbajardsilogic@0: 
lbajardsilogic@0:             float colMax = 0.f, colMin = 0.f;
lbajardsilogic@0: 
lbajardsilogic@0:             for (size_t y = 0; y < m_model->getHeight(); ++y) {
lbajardsilogic@0:                 if (y >= values.size()) break;
lbajardsilogic@0:                 if (y == 0 || values[y] > colMax) colMax = values[y];
lbajardsilogic@0:                 if (y == 0 || values[y] < colMin) colMin = values[y];
lbajardsilogic@0:             }
lbajardsilogic@0: 
lbajardsilogic@0:             if (c == firstBin || colMax > visibleMax) visibleMax = colMax;
lbajardsilogic@0:             if (c == firstBin || colMin < visibleMin) visibleMin = colMin;
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0:     
lbajardsilogic@0:     if (visibleMin == visibleMax) visibleMax = visibleMin + 1;
lbajardsilogic@0: 
lbajardsilogic@0:     for (size_t c = firstBin; c <= lastBin; ++c) {
lbajardsilogic@0: 	
lbajardsilogic@0:         values.clear();
lbajardsilogic@0:         getColumn(c, values);
lbajardsilogic@0: 
lbajardsilogic@0:         for (size_t y = 0; y < m_model->getHeight(); ++y) {
lbajardsilogic@0: 
lbajardsilogic@0:             float value = min;
lbajardsilogic@0:             if (y < values.size()) {
lbajardsilogic@0:                 value = values[y];
lbajardsilogic@0:             }
lbajardsilogic@0:             
lbajardsilogic@0:             if (m_normalizeVisibleArea && !m_normalizeColumns) {
lbajardsilogic@0:                 float norm = (value - visibleMin) / (visibleMax - visibleMin);
lbajardsilogic@0:                 value = min + (max - min) * norm;
lbajardsilogic@0:             }
lbajardsilogic@0: 
lbajardsilogic@0:             if (m_colourScale == LogScale) {
lbajardsilogic@0:                 value = LogRange::map(value);
lbajardsilogic@0:             }
lbajardsilogic@0: 
lbajardsilogic@0:             int pixel = int(((value - min) * 256) / (max - min));
lbajardsilogic@0:             if (pixel < 0) pixel = 0;
lbajardsilogic@0:             if (pixel > 255) pixel = 255;
lbajardsilogic@0: 
lbajardsilogic@0:             m_cache->setPixel(c - firstBin, y, pixel);
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::paint(View *v, QPainter &paint, QRect rect) const
lbajardsilogic@0: {
lbajardsilogic@0: //    Profiler profiler("Colour3DPlotLayer::paint");
lbajardsilogic@0: #ifdef DEBUG_COLOUR_3D_PLOT_LAYER_PAINT
lbajardsilogic@0:     std::cerr << "Colour3DPlotLayer::paint(): m_model is " << m_model << ", zoom level is " << v->getZoomLevel() << std::endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: 
lbajardsilogic@0:     int completion = 0;
lbajardsilogic@0:     if (!m_model || !m_model->isOK() || !m_model->isReady(&completion)) {
lbajardsilogic@0: 	if (completion > 0) {
lbajardsilogic@0: 	    paint.fillRect(0, 10, v->width() * completion / 100,
lbajardsilogic@0: 			   10, QColor(120, 120, 120));
lbajardsilogic@0: 	}
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_normalizeVisibleArea && !m_normalizeColumns) rect = v->rect();
lbajardsilogic@0: 
lbajardsilogic@0:     size_t modelStart = m_model->getStartFrame();
lbajardsilogic@0:     size_t modelEnd = m_model->getEndFrame();
lbajardsilogic@0:     size_t modelResolution = m_model->getResolution();
lbajardsilogic@0: 
lbajardsilogic@0:     // The cache is from the model's start frame to the model's end
lbajardsilogic@0:     // frame at the model's window increment frames per pixel.  We
lbajardsilogic@0:     // want to draw from our start frame + x0 * zoomLevel to our start
lbajardsilogic@0:     // frame + x1 * zoomLevel at zoomLevel frames per pixel.
lbajardsilogic@0: 
lbajardsilogic@0:     //  We have quite different paint mechanisms for rendering "large"
lbajardsilogic@0:     //  bins (more than one bin per pixel in both directions) and
lbajardsilogic@0:     //  "small".  This is "large"; see paintDense below for "small".
lbajardsilogic@0: 
lbajardsilogic@0:     int x0 = rect.left();
lbajardsilogic@0:     int x1 = rect.right() + 1;
lbajardsilogic@0: 
lbajardsilogic@0:     int h = v->height();
lbajardsilogic@0: 
lbajardsilogic@0:     float srRatio =
lbajardsilogic@0:         float(v->getViewManager()->getMainModelSampleRate()) /
lbajardsilogic@0:         float(m_model->getSampleRate());
lbajardsilogic@0: 
lbajardsilogic@0:     int sx0 = int((v->getFrameForX(x0) / srRatio - long(modelStart))
lbajardsilogic@0:                   / long(modelResolution));
lbajardsilogic@0:     int sx1 = int((v->getFrameForX(x1) / srRatio - long(modelStart))
lbajardsilogic@0:                   / long(modelResolution));
lbajardsilogic@0:     int sh = m_model->getHeight();
lbajardsilogic@0: 
lbajardsilogic@0:     if (sx0 > 0) --sx0;
lbajardsilogic@0:     fillCache(sx0 < 0 ? 0 : sx0,
lbajardsilogic@0:               sx1 < 0 ? 0 : sx1);
lbajardsilogic@0: 
lbajardsilogic@0:     if (int(m_model->getHeight()) >= v->height() ||
lbajardsilogic@0:         int(modelResolution) < v->getZoomLevel() / 2) {
lbajardsilogic@0:         paintDense(v, paint, rect);
lbajardsilogic@0:         return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef DEBUG_COLOUR_3D_PLOT_LAYER_PAINT
lbajardsilogic@0:     std::cerr << "Colour3DPlotLayer::paint: w " << w << ", h " << h << ", sx0 " << sx0 << ", sx1 " << sx1 << ", sw " << sw << ", sh " << sh << std::endl;
lbajardsilogic@0:     std::cerr << "Colour3DPlotLayer: sample rate is " << m_model->getSampleRate() << ", resolution " << m_model->getResolution() << std::endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: 
lbajardsilogic@0:     QPoint illuminatePos;
lbajardsilogic@0:     bool illuminate = v->shouldIlluminateLocalFeatures(this, illuminatePos);
lbajardsilogic@0:     char labelbuf[10];
lbajardsilogic@0: 
lbajardsilogic@0:     for (int sx = sx0; sx <= sx1; ++sx) {
lbajardsilogic@0: 
lbajardsilogic@0:         int scx = 0;
lbajardsilogic@0:         if (sx > int(m_cacheStart)) scx = sx - m_cacheStart;
lbajardsilogic@0:         
lbajardsilogic@0: 	int fx = sx * int(modelResolution);
lbajardsilogic@0: 
lbajardsilogic@0: 	if (fx + int(modelResolution) < int(modelStart) ||
lbajardsilogic@0: 	    fx > int(modelEnd)) continue;
lbajardsilogic@0: 
lbajardsilogic@0:         int rx0 = v->getXForFrame(int((fx + int(modelStart)) * srRatio));
lbajardsilogic@0: 	int rx1 = v->getXForFrame(int((fx + int(modelStart) + int(modelResolution) + 1) * srRatio));
lbajardsilogic@0: 
lbajardsilogic@0: 	int rw = rx1 - rx0;
lbajardsilogic@0: 	if (rw < 1) rw = 1;
lbajardsilogic@0: 
lbajardsilogic@0: 	bool showLabel = (rw > 10 &&
lbajardsilogic@0: 			  paint.fontMetrics().width("0.000000") < rw - 3 &&
lbajardsilogic@0: 			  paint.fontMetrics().height() < (h / sh));
lbajardsilogic@0:         
lbajardsilogic@0: 	for (int sy = 0; sy < sh; ++sy) {
lbajardsilogic@0: 
lbajardsilogic@0: 	    int ry0 = h - (sy * h) / sh - 1;
lbajardsilogic@0: 	    QRgb pixel = qRgb(255, 255, 255);
lbajardsilogic@0: 	    if (scx >= 0 && scx < m_cache->width() &&
lbajardsilogic@0: 		sy >= 0 && sy < m_cache->height()) {
lbajardsilogic@0: 		pixel = m_cache->pixel(scx, sy);
lbajardsilogic@0: 	    }
lbajardsilogic@0: 
lbajardsilogic@0: 	    QRect r(rx0, ry0 - h / sh - 1, rw, h / sh + 1);
lbajardsilogic@0: 
lbajardsilogic@0:             if (rw == 1) {
lbajardsilogic@0:                 paint.setPen(pixel);
lbajardsilogic@0:                 paint.setBrush(Qt::NoBrush);
lbajardsilogic@0:                 paint.drawLine(r.x(), r.y(), r.x(), r.y() + r.height() - 1);
lbajardsilogic@0:                 continue;
lbajardsilogic@0:             }
lbajardsilogic@0: 
lbajardsilogic@0: 	    QColor pen(255, 255, 255, 80);
lbajardsilogic@0: 	    QColor brush(pixel);
lbajardsilogic@0: 
lbajardsilogic@0:             if (rw > 3 && r.height() > 3) {
lbajardsilogic@0:                 brush.setAlpha(160);
lbajardsilogic@0:             }
lbajardsilogic@0: 
lbajardsilogic@0: 	    paint.setPen(Qt::NoPen);
lbajardsilogic@0: 	    paint.setBrush(brush);
lbajardsilogic@0: 
lbajardsilogic@0: 	    if (illuminate) {
lbajardsilogic@0: 		if (r.contains(illuminatePos)) {
lbajardsilogic@0: 		    paint.setPen(Qt::black);//!!!
lbajardsilogic@0: 		}
lbajardsilogic@0: 	    }
lbajardsilogic@0:             
lbajardsilogic@0: #ifdef DEBUG_COLOUR_3D_PLOT_LAYER_PAINT
lbajardsilogic@0:             std::cerr << "rect " << r.x() << "," << r.y() << " "
lbajardsilogic@0:                       << r.width() << "x" << r.height() << std::endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: 
lbajardsilogic@0: 	    paint.drawRect(r);
lbajardsilogic@0: 
lbajardsilogic@0: 	    if (showLabel) {
lbajardsilogic@0: 		if (scx >= 0 && scx < m_cache->width() &&
lbajardsilogic@0: 		    sy >= 0 && sy < m_cache->height()) {
lbajardsilogic@0: 		    float value = m_model->getValueAt(scx, sy);
lbajardsilogic@0: 		    sprintf(labelbuf, "%06f", value);
lbajardsilogic@0: 		    QString text(labelbuf);
lbajardsilogic@0: 		    paint.setPen(Qt::white);
lbajardsilogic@0: 		    paint.drawText(rx0 + 2,
lbajardsilogic@0: 				   ry0 - h / sh - 1 + 2 + paint.fontMetrics().ascent(),
lbajardsilogic@0: 				   text);
lbajardsilogic@0: 		}
lbajardsilogic@0: 	    }
lbajardsilogic@0: 	}
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::paintDense(View *v, QPainter &paint, QRect rect) const
lbajardsilogic@0: {
lbajardsilogic@0:     size_t modelStart = m_model->getStartFrame();
lbajardsilogic@0:     size_t modelResolution = m_model->getResolution();
lbajardsilogic@0: 
lbajardsilogic@0:     float srRatio =
lbajardsilogic@0:         float(v->getViewManager()->getMainModelSampleRate()) /
lbajardsilogic@0:         float(m_model->getSampleRate());
lbajardsilogic@0: 
lbajardsilogic@0:     int x0 = rect.left();
lbajardsilogic@0:     int x1 = rect.right() + 1;
lbajardsilogic@0: 
lbajardsilogic@0:     int w = x1 - x0;
lbajardsilogic@0:     int h = v->height();
lbajardsilogic@0:     int sh = m_model->getHeight();
lbajardsilogic@0: 
lbajardsilogic@0:     QImage img(w, h, QImage::Format_RGB32);
lbajardsilogic@0: 
lbajardsilogic@0:     for (int x = x0; x < x1; ++x) {
lbajardsilogic@0: 
lbajardsilogic@0:         long xf = long(v->getFrameForX(x) / srRatio);
lbajardsilogic@0:         if (xf < 0) {
lbajardsilogic@0:             for (int y = 0; y < h; ++y) {
lbajardsilogic@0:                 img.setPixel(x - x0, y, m_cache->color(0));
lbajardsilogic@0:             }
lbajardsilogic@0:             continue;
lbajardsilogic@0:         }
lbajardsilogic@0: 
lbajardsilogic@0:         float sx0 = (float(xf) - modelStart) / modelResolution;
lbajardsilogic@0:         float sx1 = (float(v->getFrameForX(x+1) / srRatio) - modelStart) / modelResolution;
lbajardsilogic@0:             
lbajardsilogic@0:         int sx0i = int(sx0 + 0.001);
lbajardsilogic@0:         int sx1i = int(sx1);
lbajardsilogic@0: 
lbajardsilogic@0:         for (int y = 0; y < h; ++y) {
lbajardsilogic@0: 
lbajardsilogic@0:             float sy0 = (float(h - y - 1) * sh) / h;
lbajardsilogic@0:             float sy1 = (float(h - y) * sh) / h;
lbajardsilogic@0:             
lbajardsilogic@0:             int sy0i = int(sy0 + 0.001);
lbajardsilogic@0:             int sy1i = int(sy1);
lbajardsilogic@0: 
lbajardsilogic@0:             float mag = 0.0, div = 0.0;
lbajardsilogic@0:             int max = 0;
lbajardsilogic@0: 
lbajardsilogic@0:             for (int sx = sx0i; sx <= sx1i; ++sx) {
lbajardsilogic@0: 
lbajardsilogic@0:                 int scx = 0;
lbajardsilogic@0:                 if (sx > int(m_cacheStart)) scx = sx - int(m_cacheStart);
lbajardsilogic@0:         
lbajardsilogic@0:                 if (scx < 0 || scx >= m_cache->width()) continue;
lbajardsilogic@0: 
lbajardsilogic@0:                 for (int sy = sy0i; sy <= sy1i; ++sy) {
lbajardsilogic@0: 
lbajardsilogic@0:                     if (sy < 0 || sy >= m_cache->height()) continue;
lbajardsilogic@0: 
lbajardsilogic@0:                     float prop = 1.0;
lbajardsilogic@0:                     if (sx == sx0i) prop *= (sx + 1) - sx0;
lbajardsilogic@0:                     if (sx == sx1i) prop *= sx1 - sx;
lbajardsilogic@0:                     if (sy == sy0i) prop *= (sy + 1) - sy0;
lbajardsilogic@0:                     if (sy == sy1i) prop *= sy1 - sy;
lbajardsilogic@0: 
lbajardsilogic@0:                     mag += prop * m_cache->pixelIndex(scx, sy);
lbajardsilogic@191:                     max = MAX(max, m_cache->pixelIndex(scx, sy));
lbajardsilogic@0:                     div += prop;
lbajardsilogic@0:                 }
lbajardsilogic@0:             }
lbajardsilogic@0: 
lbajardsilogic@0:             if (div != 0) mag /= div;
lbajardsilogic@0:             if (mag < 0) mag = 0;
lbajardsilogic@0:             if (mag > 255) mag = 255;
lbajardsilogic@0:             if (max < 0) max = 0;
lbajardsilogic@0:             if (max > 255) max = 255;
lbajardsilogic@0: 
lbajardsilogic@0:             img.setPixel(x - x0, y, m_cache->color(int(mag + 0.001)));
lbajardsilogic@0: //            img.setPixel(x - x0, y, m_cache->color(max));
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     paint.drawImage(x0, 0, img);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: bool
lbajardsilogic@0: Colour3DPlotLayer::snapToFeatureFrame(View *v, int &frame,
lbajardsilogic@0: 				      size_t &resolution,
lbajardsilogic@0: 				      SnapType snap) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (!m_model) {
lbajardsilogic@0: 	return Layer::snapToFeatureFrame(v, frame, resolution, snap);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     resolution = m_model->getResolution();
lbajardsilogic@0:     int left = (frame / resolution) * resolution;
lbajardsilogic@0:     int right = left + resolution;
lbajardsilogic@0: 
lbajardsilogic@0:     switch (snap) {
lbajardsilogic@0:     case SnapLeft:  frame = left;  break;
lbajardsilogic@0:     case SnapRight: frame = right; break;
lbajardsilogic@0:     case SnapNearest:
lbajardsilogic@0:     case SnapNeighbouring:
lbajardsilogic@0: 	if (frame - left > right - frame) frame = right;
lbajardsilogic@0: 	else frame = left;
lbajardsilogic@0: 	break;
lbajardsilogic@0:     }
lbajardsilogic@0:     
lbajardsilogic@0:     return true;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: QString
lbajardsilogic@0: Colour3DPlotLayer::toXmlString(QString indent, QString extraAttributes) const
lbajardsilogic@0: {
lbajardsilogic@0:     QString s;
lbajardsilogic@0:     
lbajardsilogic@0:     s += QString("scale=\"%1\" "
lbajardsilogic@0:                  "colourScheme=\"%2\" "
lbajardsilogic@0: 		 "normalizeColumns=\"%3\" "
lbajardsilogic@0:                  "normalizeVisibleArea=\"%4\"")
lbajardsilogic@0: 	.arg((int)m_colourScale)
lbajardsilogic@0:         .arg(m_colourMap)
lbajardsilogic@0:         .arg(m_normalizeColumns ? "true" : "false")
lbajardsilogic@0:         .arg(m_normalizeVisibleArea ? "true" : "false");
lbajardsilogic@0: 
lbajardsilogic@0:     return Layer::toXmlString(indent, extraAttributes + " " + s);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: Colour3DPlotLayer::setProperties(const QXmlAttributes &attributes)
lbajardsilogic@0: {
lbajardsilogic@0:     bool ok = false;
lbajardsilogic@0: 
lbajardsilogic@0:     ColourScale scale = (ColourScale)attributes.value("scale").toInt(&ok);
lbajardsilogic@0:     if (ok) setColourScale(scale);
lbajardsilogic@0: 
lbajardsilogic@0:     int colourMap = attributes.value("colourScheme").toInt(&ok);
lbajardsilogic@0:     if (ok) setColourMap(colourMap);
lbajardsilogic@0: 
lbajardsilogic@0:     bool normalizeColumns =
lbajardsilogic@0:         (attributes.value("normalizeColumns").trimmed() == "true");
lbajardsilogic@0:     setNormalizeColumns(normalizeColumns);
lbajardsilogic@0: 
lbajardsilogic@0:     bool normalizeVisibleArea =
lbajardsilogic@0:         (attributes.value("normalizeVisibleArea").trimmed() == "true");
lbajardsilogic@0:     setNormalizeVisibleArea(normalizeVisibleArea);
lbajardsilogic@0: }
lbajardsilogic@0: