Chris@0: /* -*- c-basic-offset: 4 -*-  vi:set ts=8 sts=4 sw=4: */
Chris@0: 
Chris@0: /*
Chris@0:     A waveform viewer and audio annotation editor.
Chris@5:     Chris Cannam, Queen Mary University of London, 2005-2006
Chris@0:     
Chris@0:     This is experimental software.  Not for distribution.
Chris@0: */
Chris@0: 
Chris@0: #include "Colour3DPlotLayer.h"
Chris@0: 
Chris@0: #include "base/View.h"
Chris@0: #include "base/Profiler.h"
Chris@0: 
Chris@0: #include <QPainter>
Chris@0: #include <QImage>
Chris@0: #include <QRect>
Chris@0: 
Chris@0: #include <iostream>
Chris@0: 
Chris@0: #include <cassert>
Chris@0: 
Chris@0: 
Chris@0: Colour3DPlotLayer::Colour3DPlotLayer(View *w) :
Chris@0:     Layer(w),
Chris@0:     m_model(0),
Chris@0:     m_cache(0)
Chris@0: {
Chris@0:     m_view->addLayer(this);
Chris@0: }
Chris@0: 
Chris@0: Colour3DPlotLayer::~Colour3DPlotLayer()
Chris@0: {
Chris@0: }
Chris@0: 
Chris@0: void
Chris@0: Colour3DPlotLayer::setModel(const DenseThreeDimensionalModel *model)
Chris@0: {
Chris@0:     m_model = model;
Chris@0:     if (!m_model || !m_model->isOK()) return;
Chris@0: 
Chris@0:     connect(m_model, SIGNAL(modelChanged()), this, SIGNAL(modelChanged()));
Chris@0:     connect(m_model, SIGNAL(modelChanged(size_t, size_t)),
Chris@0: 	    this, SIGNAL(modelChanged(size_t, size_t)));
Chris@0: 
Chris@0:     connect(m_model, SIGNAL(completionChanged()),
Chris@0: 	    this, SIGNAL(modelCompletionChanged()));
Chris@0: 
Chris@0:     connect(m_model, SIGNAL(modelChanged()), this, SLOT(cacheInvalid()));
Chris@0:     connect(m_model, SIGNAL(modelChanged(size_t, size_t)),
Chris@0: 	    this, SLOT(cacheInvalid(size_t, size_t)));
Chris@0: 
Chris@0:     emit modelReplaced();
Chris@0: }
Chris@0: 
Chris@0: void
Chris@0: Colour3DPlotLayer::cacheInvalid()
Chris@0: {
Chris@0:     delete m_cache; 
Chris@0:     m_cache = 0;
Chris@0: }
Chris@0: 
Chris@0: void
Chris@0: Colour3DPlotLayer::cacheInvalid(size_t, size_t)
Chris@0: {
Chris@0:     cacheInvalid();
Chris@0: }
Chris@0: 
Chris@25: bool
Chris@25: Colour3DPlotLayer::isLayerScrollable() const
Chris@25: {
Chris@25:     QPoint discard;
Chris@25:     return !m_view->shouldIlluminateLocalFeatures(this, discard);
Chris@25: }
Chris@25: 
Chris@25: QString
Chris@25: Colour3DPlotLayer::getFeatureDescription(QPoint &pos) const
Chris@25: {
Chris@25:     if (!m_model) return "";
Chris@25: 
Chris@25:     int x = pos.x();
Chris@25:     int y = pos.y();
Chris@25: 
Chris@25:     size_t modelStart = m_model->getStartFrame();
Chris@25:     size_t modelWindow = m_model->getWindowSize();
Chris@25: 
Chris@25:     int sx0 = modelWindow *
Chris@25: 	int((getFrameForX(x) - long(modelStart)) / long(modelWindow));
Chris@25:     int sx1 = sx0 + modelWindow;
Chris@25: 
Chris@25:     float binHeight = float(m_view->height()) / m_model->getYBinCount();
Chris@25:     int sy = (m_view->height() - y) / binHeight;
Chris@25: 
Chris@25:     float value = m_model->getBinValue(sx0, sy);
Chris@25:     
Chris@25:     QString binName = m_model->getBinName(sy);
Chris@25:     if (binName == "") binName = QString("[%1]").arg(sy + 1);
Chris@25:     else binName = QString("%1 [%2]").arg(binName).arg(sy + 1);
Chris@25: 
Chris@25:     QString text = tr("Time:\t%1 - %2\nBin:\t%3\nValue:\t%4")
Chris@25: 	.arg(RealTime::frame2RealTime(sx0, m_model->getSampleRate())
Chris@25: 	     .toText(true).c_str())
Chris@25: 	.arg(RealTime::frame2RealTime(sx1, m_model->getSampleRate())
Chris@25: 	     .toText(true).c_str())
Chris@25: 	.arg(binName)
Chris@25: 	.arg(value);
Chris@25: 
Chris@25:     return text;
Chris@25: }
Chris@25: 
Chris@25: int
Chris@25: Colour3DPlotLayer::getVerticalScaleWidth(QPainter &paint) const
Chris@25: {
Chris@25:     if (!m_model) return 0;
Chris@25: 
Chris@25:     QString sampleText("123");
Chris@25:     int tw = paint.fontMetrics().width(sampleText);
Chris@25: 
Chris@25:     for (size_t i = 0; i < m_model->getYBinCount(); ++i) {
Chris@25: 	if (m_model->getBinName(i).length() > sampleText.length()) {
Chris@25: 	    sampleText = m_model->getBinName(i);
Chris@25: 	}
Chris@25:     }
Chris@25:     if (sampleText != "123") {
Chris@25: 	tw = std::max(tw, paint.fontMetrics().width(sampleText));
Chris@25:     }
Chris@25: 
Chris@25:     return tw + 13;
Chris@25: }
Chris@25: 
Chris@25: void
Chris@25: Colour3DPlotLayer::paintVerticalScale(QPainter &paint, QRect rect) const
Chris@25: {
Chris@25:     if (!m_model) return;
Chris@25: 
Chris@25:     int h = rect.height(), w = rect.width();
Chris@25:     float binHeight = float(m_view->height()) / m_model->getYBinCount();
Chris@25: 
Chris@25: //    int textHeight = paint.fontMetrics().height();
Chris@25: //    int toff = -textHeight + paint.fontMetrics().ascent() + 2;
Chris@25: 
Chris@25:     for (size_t i = 0; i < m_model->getYBinCount(); ++i) {
Chris@25: 
Chris@25: 	int y0 = m_view->height() - (i * binHeight) - 1;
Chris@25: 	
Chris@25: 	QString text = m_model->getBinName(i);
Chris@25: 	if (text == "") text = QString("[%1]").arg(i + 1);
Chris@25: 
Chris@25: 	paint.drawLine(0, y0, w, y0);
Chris@25: 
Chris@25: 	int cy = y0 - binHeight/2;
Chris@25: 	int ty = cy + paint.fontMetrics().ascent()/2;
Chris@25: 
Chris@25: //	int tx = w - 10 - paint.fontMetrics().width(text);
Chris@25: 	paint.drawText(10, ty, text);
Chris@25:     }
Chris@25: }
Chris@25: 
Chris@0: void
Chris@0: Colour3DPlotLayer::paint(QPainter &paint, QRect rect) const
Chris@0: {
Chris@0: //    Profiler profiler("Colour3DPlotLayer::paint");
Chris@0: //    std::cerr << "Colour3DPlotLayer::paint(): m_model is " << m_model << ", zoom level is " << m_view->getZoomLevel() << std::endl;
Chris@0: 
Chris@0:     //!!! This doesn't yet accommodate the fact that the model may
Chris@0:     //have a different sample rate from an underlying model.  At the
Chris@0:     //moment our paint mechanism assumes all models have the same
Chris@0:     //sample rate.  If that isn't the case, they won't align and the
Chris@0:     //time ruler will match whichever model was used to construct it.
Chris@0:     //Obviously it is not going to be the case in general that models
Chris@0:     //will have the same samplerate, so we need a pane samplerate as
Chris@0:     //well which we trivially realign to.  (We can probably require
Chris@0:     //the waveform and spectrogram layers to display at the pane
Chris@0:     //samplerate.)
Chris@0: 
Chris@0:     int completion = 0;
Chris@0:     if (!m_model || !m_model->isOK() || !m_model->isReady(&completion)) {
Chris@0: 	if (completion > 0) {
Chris@0: 	    paint.fillRect(0, 10, m_view->width() * completion / 100,
Chris@0: 			   10, QColor(120, 120, 120));
Chris@0: 	}
Chris@0: 	return;
Chris@0:     }
Chris@0: 
Chris@0:     long startFrame = m_view->getStartFrame();
Chris@0:     int zoomLevel = m_view->getZoomLevel();
Chris@0: 
Chris@0:     size_t modelStart = m_model->getStartFrame();
Chris@0:     size_t modelEnd = m_model->getEndFrame();
Chris@0:     size_t modelWindow = m_model->getWindowSize();
Chris@0: 
Chris@12:     size_t cacheWidth = (modelEnd - modelStart) / modelWindow + 1;
Chris@12:     size_t cacheHeight = m_model->getYBinCount();
Chris@12: 
Chris@12:     if (m_cache &&
Chris@12: 	(m_cache->width() != cacheWidth ||
Chris@12: 	 m_cache->height() != cacheHeight)) {
Chris@12: 
Chris@12: 	delete m_cache;
Chris@12: 	m_cache = 0;
Chris@12:     }
Chris@12: 
Chris@0:     if (!m_cache) {
Chris@0: 
Chris@12: 	m_cache = new QImage(cacheWidth, cacheHeight, QImage::Format_Indexed8);
Chris@0: 
Chris@0: 	m_cache->setNumColors(256);
Chris@0: 	DenseThreeDimensionalModel::BinValueSet values;
Chris@0: /*
Chris@0: 	for (int pixel = 0; pixel < 256; ++pixel) {
Chris@0: 	    int hue = 256 - pixel;
Chris@0: //	    int hue = 220 - pixel;
Chris@0: //	    if (hue < 0) hue += 360;
Chris@0: 	    QColor color = QColor::fromHsv(hue, pixel/2 + 128, pixel);
Chris@0: 	    m_cache->setColor(pixel, qRgb(color.red(), color.green(), color.blue()));
Chris@0: 	}
Chris@0: */
Chris@0: 
Chris@0: 	float min = m_model->getMinimumLevel();
Chris@0: 	float max = m_model->getMaximumLevel();
Chris@0: 
Chris@0: 	if (max == min) max = min + 1.0;
Chris@0: 
Chris@0: //	int min = lrintf(m_model->getMinimumLevel());
Chris@0: //	int max = lrintf(m_model->getMaximumLevel());
Chris@0: 	for (int value = 0; value < 256; ++value) {
Chris@0: //	    int spread = ((value - min) * 256) / (max - min);
Chris@0: //	    int hue = 256 - spread;
Chris@0: //	    QColor color = QColor::fromHsv(hue, spread/2 + 128, spread);
Chris@0: 	    int hue = 256 - value;
Chris@26: 	    QColor colour = QColor::fromHsv(hue, value/2 + 128, value);
Chris@26: 	    m_cache->setColor(value, qRgba(colour.red(), colour.green(), colour.blue(), 80));
Chris@0: //	    std::cerr << "Colour3DPlotLayer: Index " << value << ": hue " << hue << std::endl;
Chris@0: 	}
Chris@0: 
Chris@0: 	m_cache->fill(min);
Chris@0: 
Chris@0: 	for (size_t f = modelStart; f <= modelEnd; f += modelWindow) {
Chris@0: 	
Chris@0: 	    values.clear();
Chris@0: 	    m_model->getBinValues(f, values);
Chris@0: 	    
Chris@0: 	    for (size_t y = 0; y < m_model->getYBinCount(); ++y) {
Chris@0: 
Chris@0: 		float value = min;
Chris@0: 		if (y < values.size()) value = values[y];
Chris@0: 
Chris@0: 		//!!! divide-by-zero!
Chris@0: 		int pixel = int(((value - min) * 256) / (max - min));
Chris@0: 
Chris@24: 		if (pixel == 256) pixel = 255;
Chris@24: 
Chris@0: 		m_cache->setPixel(f / modelWindow, y, pixel);
Chris@0: 	    }
Chris@0: 	}
Chris@0:     }
Chris@0: 
Chris@0:     int x0 = rect.left();
Chris@0:     int x1 = rect.right() + 1;
Chris@0: 
Chris@0: //    int y0 = rect.top();
Chris@0: //    int y1 = rect.bottom();
Chris@0:     int w = x1 - x0;
Chris@0:     int h = m_view->height();
Chris@0: 
Chris@0:     // The cache is from the model's start frame to the model's end
Chris@0:     // frame at the model's window increment frames per pixel.  We
Chris@0:     // want to draw from our start frame + x0 * zoomLevel to our start
Chris@0:     // frame + x1 * zoomLevel at zoomLevel frames per pixel.
Chris@0: 
Chris@0:     //!!! Strictly speaking we want quite different paint mechanisms
Chris@0:     //for models that have more than one bin per pixel in either
Chris@0:     //direction.  This one is only really appropriate for models with
Chris@0:     //far fewer bins in both directions.
Chris@0: 
Chris@20:     int sx0 = int((getFrameForX(x0) - long(modelStart)) / long(modelWindow));
Chris@20:     int sx1 = int((getFrameForX(x1) - long(modelStart)) / long(modelWindow));
Chris@0:     int sw = sx1 - sx0;
Chris@0:     int sh = m_model->getYBinCount();
Chris@0: 
Chris@0: /*
Chris@0:     std::cerr << "Colour3DPlotLayer::paint: w " << w << ", h " << h << ", sx0 " << sx0 << ", sx1 " << sx1 << ", sw " << sw << ", sh " << sh << std::endl;
Chris@0:     std::cerr << "Colour3DPlotLayer: sample rate is " << m_model->getSampleRate() << ", window size " << m_model->getWindowSize() << std::endl;
Chris@0: */
Chris@0: 
Chris@25:     QPoint illuminatePos;
Chris@25:     bool illuminate = m_view->shouldIlluminateLocalFeatures(this, illuminatePos);
Chris@25: 
Chris@0:     for (int sx = sx0 - 1; sx <= sx1; ++sx) {
Chris@0: 
Chris@0: 	int fx = sx * int(modelWindow);
Chris@0: 
Chris@0: 	if (fx + modelWindow < int(modelStart) ||
Chris@0: 	    fx > int(modelEnd)) continue;
Chris@0: 
Chris@0: 	for (int sy = 0; sy < sh; ++sy) {
Chris@0: 
Chris@20: 	    int rx0 = getXForFrame(fx + int(modelStart));
Chris@20: 	    int rx1 = getXForFrame(fx + int(modelStart) + int(modelWindow));
Chris@0: 
Chris@0: 	    int ry0 = h - (sy * h) / sh - 1;
Chris@0: 	    int ry1 = h - ((sy + 1) * h) / sh - 2;
Chris@0: 	    QRgb pixel = qRgb(255, 255, 255);
Chris@0: 	    if (sx >= 0 && sx < m_cache->width() &&
Chris@0: 		sy >= 0 && sy < m_cache->height()) {
Chris@0: 		pixel = m_cache->pixel(sx, sy);
Chris@0: 	    }
Chris@0: 
Chris@0: 	    QColor pen(255, 255, 255, 80);
Chris@0: //	    QColor pen(pixel);
Chris@0: 	    QColor brush(pixel);
Chris@0: 	    brush.setAlpha(160);
Chris@0: //	    paint.setPen(pen);
Chris@0: 	    paint.setPen(Qt::NoPen);
Chris@0: 	    paint.setBrush(brush);
Chris@0: 
Chris@0: 	    int w = rx1 - rx0;
Chris@0: 	    if (w < 1) w = 1;
Chris@25: 
Chris@25: 	    QRect r(rx0, ry0 - h / sh - 1, w, h / sh + 1);
Chris@25: 
Chris@25: 	    if (illuminate) {
Chris@25: 		if (r.contains(illuminatePos)) {
Chris@25: 		    paint.setPen(Qt::black);//!!!
Chris@25: 		}
Chris@25: 	    }
Chris@25: 
Chris@25: 	    paint.drawRect(r);
Chris@0: 
Chris@0: 	    if (sx >= 0 && sx < m_cache->width() &&
Chris@0: 		sy >= 0 && sy < m_cache->height()) {
Chris@24: 		if (w > 10) {
Chris@24: 		    int dv = m_cache->pixelIndex(sx, sy);
Chris@24: //		    if (dv != 0 && paint.fontMetrics().height() < (h / sh)) {
Chris@24: 		    if (paint.fontMetrics().height() < (h / sh)) {
Chris@24: 			float value = m_model->getBinValue(fx, sy);
Chris@24: 			QString text = QString("%1").arg(value); //dv);
Chris@24: 			if (paint.fontMetrics().width(text) < w - 3) {
Chris@24: 			    paint.setPen(Qt::white);
Chris@24: 			    paint.drawText(rx0 + 2,
Chris@24: 					   ry0 - h / sh - 1 + 2 + paint.fontMetrics().ascent(),
Chris@24: 					   QString("%1").arg(value));
Chris@24: 			}
Chris@0: 		    }
Chris@0: 		}
Chris@0: 	    }
Chris@0: 	}
Chris@0:     }
Chris@0:     
Chris@0: /*
Chris@0:     QRect targetRect(x0, 0, w, h);
Chris@0:     QRect sourceRect(sx0, 0, sw, sh);
Chris@0: 
Chris@0:     QImage scaled(w, h, QImage::Format_RGB32);
Chris@0: 
Chris@0:     for (int x = 0; x < w; ++x) {
Chris@0: 	for (int y = 0; y < h; ++y) {
Chris@0: 
Chris@0: 	    
Chris@0: 
Chris@0: 	    int sx = sx0 + (x * sw) / w;
Chris@0: 	    int sy = sh - (y * sh) / h - 1;
Chris@0: //	    std::cerr << "Colour3DPlotLayer::paint: sx " << sx << ", sy " << sy << ", cache w " << m_cache->width() << ", cache h " << m_cache->height() << std::endl;
Chris@0: 	    if (sx >= 0 && sy >= 0 &&
Chris@0: 		sx < m_cache->width() && sy < m_cache->height()) {
Chris@0: 		scaled.setPixel(x, y, m_cache->pixel(sx, sy));
Chris@0: 	    } else {
Chris@0: 		scaled.setPixel(x, y, qRgba(255, 255, 255, 80));
Chris@0: 	    }
Chris@0: 	}
Chris@0:     }
Chris@0: 
Chris@0:     paint.drawImage(x0, 0, scaled);
Chris@0: */
Chris@0: }
Chris@0: 
Chris@28: bool
Chris@28: Colour3DPlotLayer::snapToFeatureFrame(int &frame,
Chris@28: 				      size_t &resolution,
Chris@28: 				      SnapType snap) const
Chris@24: {
Chris@24:     if (!m_model) {
Chris@28: 	return Layer::snapToFeatureFrame(frame, resolution, snap);
Chris@24:     }
Chris@24: 
Chris@24:     resolution = m_model->getWindowSize();
Chris@28:     int left = (frame / resolution) * resolution;
Chris@28:     int right = left + resolution;
Chris@28: 
Chris@28:     switch (snap) {
Chris@28:     case SnapLeft:  frame = left;  break;
Chris@28:     case SnapRight: frame = right; break;
Chris@28:     case SnapNearest:
Chris@28:     case SnapNeighbouring:
Chris@28: 	if (frame - left > right - frame) frame = right;
Chris@28: 	else frame = left;
Chris@28: 	break;
Chris@28:     }
Chris@24:     
Chris@28:     return true;
Chris@24: }
Chris@24: 
Chris@0: #ifdef INCLUDE_MOCFILES
Chris@0: #include "Colour3DPlotLayer.moc.cpp"
Chris@0: #endif
Chris@0: 
Chris@0: