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.
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 "TimeRulerLayer.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include "data/model/Model.h"
lbajardsilogic@0: #include "base/RealTime.h"
lbajardsilogic@0: #include "view/View.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <QPainter>
lbajardsilogic@0: 
lbajardsilogic@0: #include <iostream>
lbajardsilogic@0: 
lbajardsilogic@0: using std::cerr;
lbajardsilogic@0: using std::endl;
lbajardsilogic@0: 
lbajardsilogic@0: TimeRulerLayer::TimeRulerLayer() :
lbajardsilogic@0:     Layer(),
lbajardsilogic@0:     m_model(0),
lbajardsilogic@0:     m_colour(Qt::black),
lbajardsilogic@0:     m_labelHeight(LabelTop)
lbajardsilogic@0: {
lbajardsilogic@0:     
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: TimeRulerLayer::setModel(Model *model)
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_model == model) return;
lbajardsilogic@0:     m_model = model;
lbajardsilogic@0:     emit modelReplaced();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: TimeRulerLayer::setBaseColour(QColor colour)
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_colour == colour) return;
lbajardsilogic@0:     m_colour = colour;
lbajardsilogic@0:     emit layerParametersChanged();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: Layer::PropertyList
lbajardsilogic@0: TimeRulerLayer::getProperties() const
lbajardsilogic@0: {
lbajardsilogic@0:     PropertyList list;
lbajardsilogic@0:     list.push_back("Colour");
lbajardsilogic@0:     return list;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: QString
lbajardsilogic@0: TimeRulerLayer::getPropertyLabel(const PropertyName &name) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (name == "Colour") return tr("Colour");
lbajardsilogic@0:     return "";
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: Layer::PropertyType
lbajardsilogic@0: TimeRulerLayer::getPropertyType(const PropertyName &) const
lbajardsilogic@0: {
lbajardsilogic@0:     return ValueProperty;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: int
lbajardsilogic@0: TimeRulerLayer::getPropertyRangeAndValue(const PropertyName &name,
lbajardsilogic@0: 					 int *min, int *max, int *deflt) const
lbajardsilogic@0: {
lbajardsilogic@0:     int val = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     if (name == "Colour") {
lbajardsilogic@0: 
lbajardsilogic@0: 	if (min) *min = 0;
lbajardsilogic@0: 	if (max) *max = 5;
lbajardsilogic@0:         if (deflt) *deflt = 0;
lbajardsilogic@0: 
lbajardsilogic@0: 	if (m_colour == Qt::black) val = 0;
lbajardsilogic@0: 	else if (m_colour == Qt::darkRed) val = 1;
lbajardsilogic@0: 	else if (m_colour == Qt::darkBlue) val = 2;
lbajardsilogic@0: 	else if (m_colour == Qt::darkGreen) val = 3;
lbajardsilogic@0: 	else if (m_colour == QColor(200, 50, 255)) val = 4;
lbajardsilogic@0: 	else if (m_colour == QColor(255, 150, 50)) val = 5;
lbajardsilogic@0: 
lbajardsilogic@0:     } else {
lbajardsilogic@0: 	
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: TimeRulerLayer::getPropertyValueLabel(const PropertyName &name,
lbajardsilogic@0: 				    int value) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (name == "Colour") {
lbajardsilogic@0: 	switch (value) {
lbajardsilogic@0: 	default:
lbajardsilogic@0: 	case 0: return tr("Black");
lbajardsilogic@0: 	case 1: return tr("Red");
lbajardsilogic@0: 	case 2: return tr("Blue");
lbajardsilogic@0: 	case 3: return tr("Green");
lbajardsilogic@0: 	case 4: return tr("Purple");
lbajardsilogic@0: 	case 5: return tr("Orange");
lbajardsilogic@0: 	}
lbajardsilogic@0:     }
lbajardsilogic@0:     return tr("<unknown>");
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: TimeRulerLayer::setProperty(const PropertyName &name, int value)
lbajardsilogic@0: {
lbajardsilogic@0:     if (name == "Colour") {
lbajardsilogic@0: 	switch (value) {
lbajardsilogic@0: 	default:
lbajardsilogic@0: 	case 0:	setBaseColour(Qt::black); break;
lbajardsilogic@0: 	case 1: setBaseColour(Qt::darkRed); break;
lbajardsilogic@0: 	case 2: setBaseColour(Qt::darkBlue); break;
lbajardsilogic@0: 	case 3: setBaseColour(Qt::darkGreen); break;
lbajardsilogic@0: 	case 4: setBaseColour(QColor(200, 50, 255)); break;
lbajardsilogic@0: 	case 5: setBaseColour(QColor(255, 150, 50)); break;
lbajardsilogic@0: 	}
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: TimeRulerLayer::paint(View *v, QPainter &paint, QRect rect) const
lbajardsilogic@0: {
lbajardsilogic@0: //    std::cerr << "TimeRulerLayer::paint (" << rect.x() << "," << rect.y()
lbajardsilogic@0: //	      << ") [" << rect.width() << "x" << rect.height() << "]" << std::endl;
lbajardsilogic@0:     
lbajardsilogic@0:     if (!m_model || !m_model->isOK()) return;
lbajardsilogic@0: 
lbajardsilogic@0:     int sampleRate = m_model->getSampleRate();
lbajardsilogic@0:     if (!sampleRate) return;
lbajardsilogic@0: 
lbajardsilogic@0:     long startFrame = v->getStartFrame();
lbajardsilogic@0:     long endFrame = v->getEndFrame();
lbajardsilogic@0: 
lbajardsilogic@0:     int zoomLevel = v->getZoomLevel();
lbajardsilogic@0: 
lbajardsilogic@0:     long rectStart = startFrame + (rect.x() - 100) * zoomLevel;
lbajardsilogic@0:     long rectEnd = startFrame + (rect.x() + rect.width() + 100) * zoomLevel;
lbajardsilogic@0: //    if (rectStart < startFrame) rectStart = startFrame;
lbajardsilogic@0: //    if (rectEnd > endFrame) rectEnd = endFrame;
lbajardsilogic@0: 
lbajardsilogic@0: //    std::cerr << "TimeRulerLayer::paint: calling paint.save()" << std::endl;
lbajardsilogic@0:     paint.save();
lbajardsilogic@0: //!!!    paint.setClipRect(v->rect());
lbajardsilogic@0: 
lbajardsilogic@0:     int minPixelSpacing = 50;
lbajardsilogic@0: 
lbajardsilogic@0:     RealTime rtStart = RealTime::frame2RealTime(startFrame, sampleRate);
lbajardsilogic@0:     RealTime rtEnd = RealTime::frame2RealTime(endFrame, sampleRate);
lbajardsilogic@0: //    cerr << "startFrame " << startFrame << ", endFrame " << v->getEndFrame() << ", rtStart " << rtStart << ", rtEnd " << rtEnd << endl;
lbajardsilogic@0:     int count = v->width() / minPixelSpacing;
lbajardsilogic@0:     if (count < 1) count = 1;
lbajardsilogic@0:     RealTime rtGap = (rtEnd - rtStart) / count;
lbajardsilogic@0: //    cerr << "rtGap is " << rtGap << endl;
lbajardsilogic@0: 
lbajardsilogic@0:     int incms;
lbajardsilogic@0:     bool quarter = false;
lbajardsilogic@0: 
lbajardsilogic@0:     if (rtGap.sec > 0) {
lbajardsilogic@0: 	incms = 1000;
lbajardsilogic@0: 	int s = rtGap.sec;
lbajardsilogic@0: 	if (s > 0) { incms *= 5; s /= 5; }
lbajardsilogic@0: 	if (s > 0) { incms *= 2; s /= 2; }
lbajardsilogic@0: 	if (s > 0) { incms *= 6; s /= 6; quarter = true; }
lbajardsilogic@0: 	if (s > 0) { incms *= 5; s /= 5; quarter = false; }
lbajardsilogic@0: 	if (s > 0) { incms *= 2; s /= 2; }
lbajardsilogic@0: 	if (s > 0) { incms *= 6; s /= 6; quarter = true; }
lbajardsilogic@0: 	while (s > 0) {
lbajardsilogic@0: 	    incms *= 10;
lbajardsilogic@0: 	    s /= 10;
lbajardsilogic@0: 	    quarter = false;
lbajardsilogic@0: 	}
lbajardsilogic@0:     } else {
lbajardsilogic@0: 	incms = 1;
lbajardsilogic@0: 	int ms = rtGap.msec();
lbajardsilogic@0: 	if (ms > 0) { incms *= 10; ms /= 10; }
lbajardsilogic@0: 	if (ms > 0) { incms *= 10; ms /= 10; }
lbajardsilogic@0: 	if (ms > 0) { incms *= 5; ms /= 5; }
lbajardsilogic@0: 	if (ms > 0) { incms *= 2; ms /= 2; }
lbajardsilogic@0:     }
lbajardsilogic@0: //    cerr << "incms is " << incms << endl;
lbajardsilogic@0: 
lbajardsilogic@0:     RealTime rt = RealTime::frame2RealTime(rectStart, sampleRate);
lbajardsilogic@0:     long ms = rt.sec * 1000 + rt.msec();
lbajardsilogic@0:     ms = (ms / incms) * incms - incms;
lbajardsilogic@0: 
lbajardsilogic@0:     RealTime incRt = RealTime::fromMilliseconds(incms);
lbajardsilogic@0:     long incFrame = RealTime::realTime2Frame(incRt, sampleRate);
lbajardsilogic@0:     int incX = incFrame / zoomLevel;
lbajardsilogic@0:     int ticks = 10;
lbajardsilogic@0:     if (incX < minPixelSpacing * 2) {
lbajardsilogic@0: 	ticks = quarter ? 4 : 5;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     QRect oldClipRect = rect;
lbajardsilogic@0:     QRect newClipRect(oldClipRect.x() - 25, oldClipRect.y(),
lbajardsilogic@0: 		      oldClipRect.width() + 50, oldClipRect.height());
lbajardsilogic@0:     paint.setClipRect(newClipRect);
lbajardsilogic@0:     paint.setClipRect(rect);
lbajardsilogic@0: 
lbajardsilogic@0:     QColor greyColour(m_colour);
lbajardsilogic@0:     if (m_colour == Qt::black) {
lbajardsilogic@0: 	greyColour = QColor(200,200,200);
lbajardsilogic@0:     } else {
lbajardsilogic@0: 	greyColour = m_colour.light(150);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     while (1) {
lbajardsilogic@0: 
lbajardsilogic@0: 	rt = RealTime::fromMilliseconds(ms);
lbajardsilogic@0: 	ms += incms;
lbajardsilogic@0: 
lbajardsilogic@0: 	long frame = RealTime::realTime2Frame(rt, sampleRate);
lbajardsilogic@0: 	if (frame >= rectEnd) break;
lbajardsilogic@0: 
lbajardsilogic@0: 	int x = (frame - startFrame) / zoomLevel;
lbajardsilogic@0: 	if (x < rect.x() || x >= rect.x() + rect.width()) continue;
lbajardsilogic@0: 
lbajardsilogic@0: 	paint.setPen(greyColour);
lbajardsilogic@0: 	paint.drawLine(x, 0, x, v->height());
lbajardsilogic@0: 
lbajardsilogic@0: 	paint.setPen(m_colour);
lbajardsilogic@0: 	paint.drawLine(x, 0, x, 5);
lbajardsilogic@0: 	paint.drawLine(x, v->height() - 6, x, v->height() - 1);
lbajardsilogic@0: 
lbajardsilogic@0: 	QString text(QString::fromStdString(rt.toText()));
lbajardsilogic@0: 
lbajardsilogic@0: 	int y;
lbajardsilogic@0: 	QFontMetrics metrics = paint.fontMetrics();
lbajardsilogic@0: 	switch (m_labelHeight) {
lbajardsilogic@0: 	default:
lbajardsilogic@0: 	case LabelTop:
lbajardsilogic@0: 	    y = 6 + metrics.ascent();
lbajardsilogic@0: 	    break;
lbajardsilogic@0: 	case LabelMiddle:
lbajardsilogic@0: 	    y = v->height() / 2 - metrics.height() / 2 + metrics.ascent();
lbajardsilogic@0: 	    break;
lbajardsilogic@0: 	case LabelBottom:
lbajardsilogic@0: 	    y = v->height() - metrics.height() + metrics.ascent() - 6;
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: 	int tw = metrics.width(text);
lbajardsilogic@0: 
lbajardsilogic@0:         if (v->getViewManager() && v->getViewManager()->getOverlayMode() !=
lbajardsilogic@0:             ViewManager::NoOverlays) {
lbajardsilogic@0: 
lbajardsilogic@0:             if (v->getLayer(0) == this) {
lbajardsilogic@0:                 // backmost layer, don't worry about outlining the text
lbajardsilogic@0:                 paint.drawText(x+2 - tw/2, y, text);
lbajardsilogic@0:             } else {
lbajardsilogic@0:                 v->drawVisibleText(paint, x+2 - tw/2, y, text, View::OutlinedText);
lbajardsilogic@0:             }
lbajardsilogic@0:         }
lbajardsilogic@0: 
lbajardsilogic@0: 	paint.setPen(greyColour);
lbajardsilogic@0: 
lbajardsilogic@0: 	for (int i = 1; i < ticks; ++i) {
lbajardsilogic@0: 	    rt = rt + (incRt / ticks);
lbajardsilogic@0: 	    frame = RealTime::realTime2Frame(rt, sampleRate);
lbajardsilogic@0: 	    x = (frame - startFrame) / zoomLevel;
lbajardsilogic@0: 	    int sz = 5;
lbajardsilogic@0: 	    if (ticks == 10) {
lbajardsilogic@0: 		if ((i % 2) == 1) {
lbajardsilogic@0: 		    if (i == 5) {
lbajardsilogic@0: 			paint.drawLine(x, 0, x, v->height());
lbajardsilogic@0: 		    } else sz = 3;
lbajardsilogic@0: 		} else {
lbajardsilogic@0: 		    sz = 7;
lbajardsilogic@0: 		}
lbajardsilogic@0: 	    }
lbajardsilogic@0: 	    paint.drawLine(x, 0, x, sz);
lbajardsilogic@0: 	    paint.drawLine(x, v->height() - sz - 1, x, v->height() - 1);
lbajardsilogic@0: 	}
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     paint.restore();
lbajardsilogic@0: }
lbajardsilogic@0:     
lbajardsilogic@0: QString
lbajardsilogic@0: TimeRulerLayer::toXmlString(QString indent, QString extraAttributes) const
lbajardsilogic@0: {
lbajardsilogic@0:     return Layer::toXmlString(indent, extraAttributes +
lbajardsilogic@0: 			      QString(" colour=\"%1\"").arg(encodeColour(m_colour)));
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@18: QString TimeRulerLayer::toEasaierXmlString(QString indent, QString extraAttributes) const
lbajardsilogic@18: {
lbajardsilogic@18:     return Layer::toEasaierXmlString(indent, extraAttributes +
lbajardsilogic@18: 			      QString(" colour=\"%1\"").arg(encodeColour(m_colour)));
lbajardsilogic@18: }
lbajardsilogic@18: 
lbajardsilogic@0: void
lbajardsilogic@0: TimeRulerLayer::setProperties(const QXmlAttributes &attributes)
lbajardsilogic@0: {
lbajardsilogic@0:     QString colourSpec = attributes.value("colour");
lbajardsilogic@0:     if (colourSpec != "") {
lbajardsilogic@0: 	QColor colour(colourSpec);
lbajardsilogic@0: 	if (colour.isValid()) {
lbajardsilogic@0: 	    setBaseColour(QColor(colourSpec));
lbajardsilogic@0: 	}
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: