annotate layer/Colour3DPlotLayer.cpp @ 445:4a14499fb184

* Save & reload vertical scale extents from note, time/value & colour 3d plot layers * Fix display of vertical scale in zoomed colour 3d plot layer
author Chris Cannam
date Wed, 12 Nov 2008 15:51:41 +0000
parents e5800f4490c4
children ae51d4b73860
rev   line source
Chris@58 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@0 2
Chris@0 3 /*
Chris@59 4 Sonic Visualiser
Chris@59 5 An audio file viewer and annotation editor.
Chris@59 6 Centre for Digital Music, Queen Mary, University of London.
Chris@182 7 This file copyright 2006 Chris Cannam and QMUL.
Chris@0 8
Chris@59 9 This program is free software; you can redistribute it and/or
Chris@59 10 modify it under the terms of the GNU General Public License as
Chris@59 11 published by the Free Software Foundation; either version 2 of the
Chris@59 12 License, or (at your option) any later version. See the file
Chris@59 13 COPYING included with this distribution for more information.
Chris@0 14 */
Chris@0 15
Chris@0 16 #include "Colour3DPlotLayer.h"
Chris@0 17
Chris@128 18 #include "view/View.h"
Chris@0 19 #include "base/Profiler.h"
Chris@197 20 #include "base/LogRange.h"
Chris@444 21 #include "base/RangeMapper.h"
Chris@376 22 #include "ColourMapper.h"
Chris@0 23
Chris@0 24 #include <QPainter>
Chris@0 25 #include <QImage>
Chris@0 26 #include <QRect>
Chris@316 27 #include <QTextStream>
Chris@0 28
Chris@0 29 #include <iostream>
Chris@0 30
Chris@0 31 #include <cassert>
Chris@0 32
Chris@353 33 //#define DEBUG_COLOUR_3D_PLOT_LAYER_PAINT 1
Chris@125 34
Chris@0 35
Chris@44 36 Colour3DPlotLayer::Colour3DPlotLayer() :
Chris@0 37 m_model(0),
Chris@159 38 m_cache(0),
Chris@197 39 m_cacheStart(0),
Chris@197 40 m_colourScale(LinearScale),
Chris@197 41 m_colourMap(0),
Chris@197 42 m_normalizeColumns(false),
Chris@357 43 m_normalizeVisibleArea(false),
Chris@444 44 m_invertVertical(false),
Chris@444 45 m_miny(0),
Chris@444 46 m_maxy(0)
Chris@0 47 {
Chris@44 48
Chris@0 49 }
Chris@0 50
Chris@0 51 Colour3DPlotLayer::~Colour3DPlotLayer()
Chris@0 52 {
Chris@0 53 }
Chris@0 54
Chris@0 55 void
Chris@0 56 Colour3DPlotLayer::setModel(const DenseThreeDimensionalModel *model)
Chris@0 57 {
Chris@193 58 if (m_model == model) return;
Chris@193 59 const DenseThreeDimensionalModel *oldModel = m_model;
Chris@0 60 m_model = model;
Chris@0 61 if (!m_model || !m_model->isOK()) return;
Chris@0 62
Chris@320 63 connectSignals(m_model);
Chris@0 64
Chris@0 65 connect(m_model, SIGNAL(modelChanged()), this, SLOT(cacheInvalid()));
Chris@0 66 connect(m_model, SIGNAL(modelChanged(size_t, size_t)),
Chris@0 67 this, SLOT(cacheInvalid(size_t, size_t)));
Chris@0 68
Chris@0 69 emit modelReplaced();
Chris@193 70 emit sliceableModelReplaced(oldModel, model);
Chris@0 71 }
Chris@0 72
Chris@0 73 void
Chris@0 74 Colour3DPlotLayer::cacheInvalid()
Chris@0 75 {
Chris@0 76 delete m_cache;
Chris@0 77 m_cache = 0;
Chris@0 78 }
Chris@0 79
Chris@0 80 void
Chris@0 81 Colour3DPlotLayer::cacheInvalid(size_t, size_t)
Chris@0 82 {
Chris@0 83 cacheInvalid();
Chris@0 84 }
Chris@0 85
Chris@159 86 Layer::PropertyList
Chris@159 87 Colour3DPlotLayer::getProperties() const
Chris@159 88 {
Chris@159 89 PropertyList list;
Chris@197 90 list.push_back("Colour");
Chris@159 91 list.push_back("Colour Scale");
Chris@197 92 list.push_back("Normalize Columns");
Chris@197 93 list.push_back("Normalize Visible Area");
Chris@357 94 list.push_back("Invert Vertical Scale");
Chris@159 95 return list;
Chris@159 96 }
Chris@159 97
Chris@159 98 QString
Chris@159 99 Colour3DPlotLayer::getPropertyLabel(const PropertyName &name) const
Chris@159 100 {
Chris@197 101 if (name == "Colour") return tr("Colour");
Chris@197 102 if (name == "Colour Scale") return tr("Scale");
Chris@197 103 if (name == "Normalize Columns") return tr("Normalize Columns");
Chris@197 104 if (name == "Normalize Visible Area") return tr("Normalize Visible Area");
Chris@357 105 if (name == "Invert Vertical Scale") return tr("Invert Vertical Scale");
Chris@159 106 return "";
Chris@159 107 }
Chris@159 108
Chris@346 109 QString
Chris@346 110 Colour3DPlotLayer::getPropertyIconName(const PropertyName &name) const
Chris@346 111 {
Chris@346 112 if (name == "Normalize Columns") return "normalise-columns";
Chris@346 113 if (name == "Normalize Visible Area") return "normalise";
Chris@357 114 if (name == "Invert Vertical Scale") return "invert-vertical";
Chris@346 115 return "";
Chris@346 116 }
Chris@346 117
Chris@159 118 Layer::PropertyType
Chris@159 119 Colour3DPlotLayer::getPropertyType(const PropertyName &name) const
Chris@159 120 {
Chris@197 121 if (name == "Normalize Columns") return ToggleProperty;
Chris@197 122 if (name == "Normalize Visible Area") return ToggleProperty;
Chris@357 123 if (name == "Invert Vertical Scale") return ToggleProperty;
Chris@159 124 return ValueProperty;
Chris@159 125 }
Chris@159 126
Chris@159 127 QString
Chris@159 128 Colour3DPlotLayer::getPropertyGroupName(const PropertyName &name) const
Chris@159 129 {
Chris@197 130 if (name == "Normalize Columns" ||
Chris@197 131 name == "Normalize Visible Area" ||
Chris@357 132 name == "Invert Vertical Scale" ||
Chris@197 133 name == "Colour Scale") return tr("Scale");
Chris@159 134 return QString();
Chris@159 135 }
Chris@159 136
Chris@159 137 int
Chris@159 138 Colour3DPlotLayer::getPropertyRangeAndValue(const PropertyName &name,
Chris@216 139 int *min, int *max, int *deflt) const
Chris@159 140 {
Chris@216 141 int val = 0;
Chris@159 142
Chris@216 143 int garbage0, garbage1, garbage2;
Chris@159 144 if (!min) min = &garbage0;
Chris@159 145 if (!max) max = &garbage1;
Chris@216 146 if (!deflt) deflt = &garbage2;
Chris@159 147
Chris@159 148 if (name == "Colour Scale") {
Chris@159 149
Chris@159 150 *min = 0;
Chris@197 151 *max = 2;
Chris@216 152 *deflt = (int)LinearScale;
Chris@159 153
Chris@216 154 val = (int)m_colourScale;
Chris@159 155
Chris@197 156 } else if (name == "Colour") {
Chris@197 157
Chris@197 158 *min = 0;
Chris@197 159 *max = ColourMapper::getColourMapCount() - 1;
Chris@216 160 *deflt = 0;
Chris@197 161
Chris@216 162 val = m_colourMap;
Chris@197 163
Chris@197 164 } else if (name == "Normalize Columns") {
Chris@197 165
Chris@216 166 *deflt = 0;
Chris@216 167 val = (m_normalizeColumns ? 1 : 0);
Chris@197 168
Chris@197 169 } else if (name == "Normalize Visible Area") {
Chris@197 170
Chris@216 171 *deflt = 0;
Chris@216 172 val = (m_normalizeVisibleArea ? 1 : 0);
Chris@197 173
Chris@357 174 } else if (name == "Invert Vertical Scale") {
Chris@357 175
Chris@357 176 *deflt = 0;
Chris@357 177 val = (m_invertVertical ? 1 : 0);
Chris@357 178
Chris@159 179 } else {
Chris@216 180 val = Layer::getPropertyRangeAndValue(name, min, max, deflt);
Chris@159 181 }
Chris@159 182
Chris@216 183 return val;
Chris@159 184 }
Chris@159 185
Chris@159 186 QString
Chris@159 187 Colour3DPlotLayer::getPropertyValueLabel(const PropertyName &name,
Chris@159 188 int value) const
Chris@159 189 {
Chris@197 190 if (name == "Colour") {
Chris@197 191 return ColourMapper::getColourMapName(value);
Chris@197 192 }
Chris@159 193 if (name == "Colour Scale") {
Chris@159 194 switch (value) {
Chris@159 195 default:
Chris@198 196 case 0: return tr("Linear");
Chris@198 197 case 1: return tr("Log");
Chris@198 198 case 2: return tr("+/-1");
Chris@159 199 }
Chris@159 200 }
Chris@159 201 return tr("<unknown>");
Chris@159 202 }
Chris@159 203
Chris@159 204 void
Chris@159 205 Colour3DPlotLayer::setProperty(const PropertyName &name, int value)
Chris@159 206 {
Chris@159 207 if (name == "Colour Scale") {
Chris@159 208 switch (value) {
Chris@159 209 default:
Chris@159 210 case 0: setColourScale(LinearScale); break;
Chris@197 211 case 1: setColourScale(LogScale); break;
Chris@197 212 case 2: setColourScale(PlusMinusOneScale); break;
Chris@159 213 }
Chris@197 214 } else if (name == "Colour") {
Chris@197 215 setColourMap(value);
Chris@197 216 } else if (name == "Normalize Columns") {
Chris@197 217 setNormalizeColumns(value ? true : false);
Chris@197 218 } else if (name == "Normalize Visible Area") {
Chris@197 219 setNormalizeVisibleArea(value ? true : false);
Chris@357 220 } else if (name == "Invert Vertical Scale") {
Chris@357 221 setInvertVertical(value ? true : false);
Chris@159 222 }
Chris@159 223 }
Chris@159 224
Chris@159 225 void
Chris@159 226 Colour3DPlotLayer::setColourScale(ColourScale scale)
Chris@159 227 {
Chris@159 228 if (m_colourScale == scale) return;
Chris@159 229 m_colourScale = scale;
Chris@159 230 cacheInvalid();
Chris@159 231 emit layerParametersChanged();
Chris@159 232 }
Chris@159 233
Chris@197 234 void
Chris@197 235 Colour3DPlotLayer::setColourMap(int map)
Chris@197 236 {
Chris@197 237 if (m_colourMap == map) return;
Chris@197 238 m_colourMap = map;
Chris@197 239 cacheInvalid();
Chris@197 240 emit layerParametersChanged();
Chris@197 241 }
Chris@197 242
Chris@197 243 void
Chris@197 244 Colour3DPlotLayer::setNormalizeColumns(bool n)
Chris@197 245 {
Chris@197 246 if (m_normalizeColumns == n) return;
Chris@197 247 m_normalizeColumns = n;
Chris@197 248 cacheInvalid();
Chris@197 249 emit layerParametersChanged();
Chris@197 250 }
Chris@197 251
Chris@197 252 bool
Chris@197 253 Colour3DPlotLayer::getNormalizeColumns() const
Chris@197 254 {
Chris@197 255 return m_normalizeColumns;
Chris@197 256 }
Chris@197 257
Chris@197 258 void
Chris@197 259 Colour3DPlotLayer::setNormalizeVisibleArea(bool n)
Chris@197 260 {
Chris@197 261 if (m_normalizeVisibleArea == n) return;
Chris@197 262 m_normalizeVisibleArea = n;
Chris@197 263 cacheInvalid();
Chris@197 264 emit layerParametersChanged();
Chris@197 265 }
Chris@197 266
Chris@197 267 bool
Chris@197 268 Colour3DPlotLayer::getNormalizeVisibleArea() const
Chris@197 269 {
Chris@197 270 return m_normalizeVisibleArea;
Chris@197 271 }
Chris@197 272
Chris@357 273 void
Chris@357 274 Colour3DPlotLayer::setInvertVertical(bool n)
Chris@357 275 {
Chris@357 276 if (m_invertVertical == n) return;
Chris@357 277 m_invertVertical = n;
Chris@357 278 cacheInvalid();
Chris@357 279 emit layerParametersChanged();
Chris@357 280 }
Chris@357 281
Chris@357 282 bool
Chris@357 283 Colour3DPlotLayer::getInvertVertical() const
Chris@357 284 {
Chris@357 285 return m_invertVertical;
Chris@357 286 }
Chris@357 287
Chris@25 288 bool
Chris@44 289 Colour3DPlotLayer::isLayerScrollable(const View *v) const
Chris@25 290 {
Chris@197 291 if (m_normalizeVisibleArea) return false;
Chris@25 292 QPoint discard;
Chris@44 293 return !v->shouldIlluminateLocalFeatures(this, discard);
Chris@25 294 }
Chris@25 295
Chris@444 296 bool
Chris@444 297 Colour3DPlotLayer::getValueExtents(float &min, float &max,
Chris@444 298 bool &logarithmic, QString &unit) const
Chris@444 299 {
Chris@444 300 if (!m_model) return false;
Chris@444 301
Chris@444 302 min = 0;
Chris@444 303 max = m_model->getHeight();
Chris@444 304
Chris@444 305 logarithmic = false;
Chris@444 306 unit = "";
Chris@444 307
Chris@444 308 return true;
Chris@444 309 }
Chris@444 310
Chris@444 311 bool
Chris@444 312 Colour3DPlotLayer::getDisplayExtents(float &min, float &max) const
Chris@444 313 {
Chris@444 314 if (!m_model) return false;
Chris@444 315
Chris@444 316 min = m_miny;
Chris@444 317 max = m_maxy;
Chris@444 318 if (max <= min) {
Chris@444 319 min = 0;
Chris@444 320 max = m_model->getHeight();
Chris@444 321 }
Chris@444 322 if (min < 0) min = 0;
Chris@444 323 if (max > m_model->getHeight()) max = m_model->getHeight();
Chris@444 324
Chris@444 325 return true;
Chris@444 326 }
Chris@444 327
Chris@444 328 bool
Chris@444 329 Colour3DPlotLayer::setDisplayExtents(float min, float max)
Chris@444 330 {
Chris@444 331 if (!m_model) return false;
Chris@444 332
Chris@444 333 m_miny = lrintf(min);
Chris@444 334 m_maxy = lrintf(max);
Chris@444 335
Chris@444 336 emit layerParametersChanged();
Chris@444 337 return true;
Chris@444 338 }
Chris@444 339
Chris@444 340 int
Chris@444 341 Colour3DPlotLayer::getVerticalZoomSteps(int &defaultStep) const
Chris@444 342 {
Chris@444 343 if (!m_model) return 0;
Chris@444 344
Chris@444 345 defaultStep = 0;
Chris@444 346 int h = m_model->getHeight();
Chris@444 347 return h;
Chris@444 348 }
Chris@444 349
Chris@444 350 int
Chris@444 351 Colour3DPlotLayer::getCurrentVerticalZoomStep() const
Chris@444 352 {
Chris@444 353 if (!m_model) return 0;
Chris@444 354
Chris@444 355 float min, max;
Chris@444 356 getDisplayExtents(min, max);
Chris@444 357 return m_model->getHeight() - lrintf(max - min);
Chris@444 358 }
Chris@444 359
Chris@444 360 void
Chris@444 361 Colour3DPlotLayer::setVerticalZoomStep(int step)
Chris@444 362 {
Chris@444 363 if (!m_model) return;
Chris@444 364
Chris@445 365 // std::cerr << "Colour3DPlotLayer::setVerticalZoomStep(" <<step <<"): before: miny = " << m_miny << ", maxy = " << m_maxy << std::endl;
Chris@444 366
Chris@444 367 int dist = m_model->getHeight() - step;
Chris@444 368 if (dist < 1) dist = 1;
Chris@444 369 float centre = m_miny + (float(m_maxy) - float(m_miny)) / 2.f;
Chris@444 370 m_miny = lrintf(centre - float(dist)/2);
Chris@444 371 if (m_miny < 0) m_miny = 0;
Chris@444 372 m_maxy = m_miny + dist;
Chris@444 373 if (m_maxy > m_model->getHeight()) m_maxy = m_model->getHeight();
Chris@444 374
Chris@445 375 // std::cerr << "Colour3DPlotLayer::setVerticalZoomStep(" <<step <<"): after: miny = " << m_miny << ", maxy = " << m_maxy << std::endl;
Chris@444 376
Chris@444 377 emit layerParametersChanged();
Chris@444 378 }
Chris@444 379
Chris@444 380 RangeMapper *
Chris@444 381 Colour3DPlotLayer::getNewVerticalZoomRangeMapper() const
Chris@444 382 {
Chris@444 383 if (!m_model) return 0;
Chris@444 384
Chris@444 385 return new LinearRangeMapper(0, m_model->getHeight(),
Chris@444 386 0, m_model->getHeight(), "");
Chris@444 387 }
Chris@444 388
Chris@25 389 QString
Chris@44 390 Colour3DPlotLayer::getFeatureDescription(View *v, QPoint &pos) const
Chris@25 391 {
Chris@25 392 if (!m_model) return "";
Chris@25 393
Chris@25 394 int x = pos.x();
Chris@25 395 int y = pos.y();
Chris@25 396
Chris@25 397 size_t modelStart = m_model->getStartFrame();
Chris@130 398 size_t modelResolution = m_model->getResolution();
Chris@25 399
Chris@159 400 float srRatio =
Chris@159 401 float(v->getViewManager()->getMainModelSampleRate()) /
Chris@159 402 float(m_model->getSampleRate());
Chris@159 403
Chris@160 404 int sx0 = int((v->getFrameForX(x) / srRatio - long(modelStart)) /
Chris@160 405 long(modelResolution));
Chris@25 406
Chris@160 407 int f0 = sx0 * modelResolution;
Chris@160 408 int f1 = f0 + modelResolution;
Chris@160 409
Chris@160 410 float binHeight = float(v->height()) / m_model->getHeight();
Chris@248 411 int sy = int((v->height() - y) / binHeight);
Chris@25 412
Chris@357 413 if (m_invertVertical) sy = m_model->getHeight() - sy - 1;
Chris@357 414
Chris@160 415 float value = m_model->getValueAt(sx0, sy);
Chris@159 416
Chris@159 417 // std::cerr << "bin value (" << sx0 << "," << sy << ") is " << value << std::endl;
Chris@25 418
Chris@25 419 QString binName = m_model->getBinName(sy);
Chris@25 420 if (binName == "") binName = QString("[%1]").arg(sy + 1);
Chris@25 421 else binName = QString("%1 [%2]").arg(binName).arg(sy + 1);
Chris@25 422
Chris@25 423 QString text = tr("Time:\t%1 - %2\nBin:\t%3\nValue:\t%4")
Chris@160 424 .arg(RealTime::frame2RealTime(f0, m_model->getSampleRate())
Chris@25 425 .toText(true).c_str())
Chris@160 426 .arg(RealTime::frame2RealTime(f1, m_model->getSampleRate())
Chris@25 427 .toText(true).c_str())
Chris@25 428 .arg(binName)
Chris@25 429 .arg(value);
Chris@25 430
Chris@25 431 return text;
Chris@25 432 }
Chris@25 433
Chris@25 434 int
Chris@248 435 Colour3DPlotLayer::getColourScaleWidth(QPainter &) const
Chris@159 436 {
Chris@159 437 int cw = 20;
Chris@159 438 return cw;
Chris@159 439 }
Chris@159 440
Chris@159 441 int
Chris@248 442 Colour3DPlotLayer::getVerticalScaleWidth(View *, QPainter &paint) const
Chris@25 443 {
Chris@25 444 if (!m_model) return 0;
Chris@25 445
Chris@160 446 QString sampleText = QString("[%1]").arg(m_model->getHeight());
Chris@25 447 int tw = paint.fontMetrics().width(sampleText);
Chris@98 448 bool another = false;
Chris@25 449
Chris@160 450 for (size_t i = 0; i < m_model->getHeight(); ++i) {
Chris@25 451 if (m_model->getBinName(i).length() > sampleText.length()) {
Chris@25 452 sampleText = m_model->getBinName(i);
Chris@98 453 another = true;
Chris@25 454 }
Chris@25 455 }
Chris@98 456 if (another) {
Chris@25 457 tw = std::max(tw, paint.fontMetrics().width(sampleText));
Chris@25 458 }
Chris@25 459
Chris@159 460 return tw + 13 + getColourScaleWidth(paint);
Chris@25 461 }
Chris@25 462
Chris@25 463 void
Chris@44 464 Colour3DPlotLayer::paintVerticalScale(View *v, QPainter &paint, QRect rect) const
Chris@25 465 {
Chris@25 466 if (!m_model) return;
Chris@25 467
Chris@25 468 int h = rect.height(), w = rect.width();
Chris@25 469
Chris@159 470 int cw = getColourScaleWidth(paint);
Chris@159 471
Chris@159 472 int ch = h - 20;
Chris@159 473 if (ch > 20 && m_cache) {
Chris@159 474
Chris@287 475 paint.setPen(v->getForeground());
Chris@159 476 paint.drawRect(4, 10, cw - 8, ch - 19);
Chris@159 477
Chris@159 478 for (int y = 0; y < ch - 20; ++y) {
Chris@159 479 QRgb c = m_cache->color(((ch - 20 - y) * 255) / (ch - 20));
Chris@159 480 // std::cerr << "y = " << y << ": rgb " << qRed(c) << "," << qGreen(c) << "," << qBlue(c) << std::endl;
Chris@159 481 paint.setPen(QColor(qRed(c), qGreen(c), qBlue(c)));
Chris@159 482 paint.drawLine(5, 11 + y, cw - 5, 11 + y);
Chris@159 483 }
Chris@159 484 }
Chris@159 485
Chris@287 486 paint.setPen(v->getForeground());
Chris@159 487
Chris@445 488 int sh = m_model->getHeight();
Chris@445 489
Chris@445 490 int symin = m_miny;
Chris@445 491 int symax = m_maxy;
Chris@445 492 if (symax <= symin) {
Chris@445 493 symin = 0;
Chris@445 494 symax = sh;
Chris@445 495 }
Chris@445 496 if (symin < 0) symin = 0;
Chris@445 497 if (symax > sh) symax = sh;
Chris@445 498
Chris@98 499 int count = v->height() / paint.fontMetrics().height();
Chris@445 500 int step = (symax - symin) / count;
Chris@98 501 if (step == 0) step = 1;
Chris@25 502
Chris@445 503 float binHeight = float(v->height()) / (symax - symin);
Chris@445 504
Chris@445 505 for (size_t i = symin; i < symax; ++i) {
Chris@25 506
Chris@357 507 size_t idx = i;
Chris@357 508 if (m_invertVertical) idx = m_model->getHeight() - idx - 1;
Chris@357 509
Chris@357 510 if ((idx % step) != 0) continue;
Chris@98 511
Chris@445 512 int y0 = int(v->height() - ((i - symin) * binHeight) - 1);
Chris@25 513
Chris@357 514 QString text = m_model->getBinName(idx);
Chris@357 515 if (text == "") text = QString("[%1]").arg(idx + 1);
Chris@25 516
Chris@159 517 paint.drawLine(cw, y0, w, y0);
Chris@25 518
Chris@248 519 int cy = int(y0 - (step * binHeight)/2);
Chris@25 520 int ty = cy + paint.fontMetrics().ascent()/2;
Chris@25 521
Chris@159 522 paint.drawText(cw + 5, ty, text);
Chris@25 523 }
Chris@25 524 }
Chris@25 525
Chris@0 526 void
Chris@197 527 Colour3DPlotLayer::getColumn(size_t col,
Chris@197 528 DenseThreeDimensionalModel::Column &values) const
Chris@197 529 {
Chris@197 530 m_model->getColumn(col, values);
Chris@197 531
Chris@224 532 float colMax = 0.f, colMin = 0.f;
Chris@197 533
Chris@197 534 float min = 0.f, max = 0.f;
Chris@197 535 if (m_normalizeColumns) {
Chris@197 536 min = m_model->getMinimumLevel();
Chris@197 537 max = m_model->getMaximumLevel();
Chris@197 538 }
Chris@197 539
Chris@197 540 if (m_normalizeColumns) {
Chris@197 541 for (size_t y = 0; y < values.size(); ++y) {
Chris@224 542 if (y == 0 || values[y] > colMax) colMax = values[y];
Chris@224 543 if (y == 0 || values[y] < colMin) colMin = values[y];
Chris@197 544 }
Chris@224 545 if (colMin == colMax) colMax = colMin + 1;
Chris@197 546 }
Chris@197 547
Chris@197 548 for (size_t y = 0; y < values.size(); ++y) {
Chris@197 549
Chris@197 550 float value = min;
Chris@197 551
Chris@197 552 value = values[y];
Chris@197 553
Chris@197 554 if (m_normalizeColumns) {
Chris@224 555 float norm = (value - colMin) / (colMax - colMin);
Chris@224 556 value = min + (max - min) * norm;
Chris@197 557 }
Chris@197 558
Chris@197 559 values[y] = value;
Chris@197 560 }
Chris@197 561 }
Chris@197 562
Chris@197 563 void
Chris@197 564 Colour3DPlotLayer::fillCache(size_t firstBin, size_t lastBin) const
Chris@197 565 {
Chris@197 566 size_t modelStart = m_model->getStartFrame();
Chris@197 567 size_t modelEnd = m_model->getEndFrame();
Chris@197 568 size_t modelResolution = m_model->getResolution();
Chris@197 569
Chris@224 570 // std::cerr << "Colour3DPlotLayer::fillCache: " << firstBin << " -> " << lastBin << std::endl;
Chris@197 571
Chris@197 572 if (!m_normalizeVisibleArea || m_normalizeColumns) {
Chris@197 573 firstBin = modelStart / modelResolution;
Chris@197 574 lastBin = modelEnd / modelResolution;
Chris@197 575 }
Chris@197 576
Chris@197 577 size_t cacheWidth = lastBin - firstBin + 1;
Chris@197 578 size_t cacheHeight = m_model->getHeight();
Chris@197 579
Chris@197 580 if (m_cache &&
Chris@197 581 (m_cacheStart != firstBin ||
Chris@248 582 m_cache->width() != int(cacheWidth) ||
Chris@248 583 m_cache->height() != int(cacheHeight))) {
Chris@197 584
Chris@197 585 delete m_cache;
Chris@197 586 m_cache = 0;
Chris@197 587 }
Chris@197 588
Chris@197 589 if (m_cache) return;
Chris@197 590
Chris@197 591 m_cache = new QImage(cacheWidth, cacheHeight, QImage::Format_Indexed8);
Chris@197 592 m_cacheStart = firstBin;
Chris@197 593
Chris@224 594 // std::cerr << "Cache size " << cacheWidth << "x" << cacheHeight << " starting " << m_cacheStart << std::endl;
Chris@197 595
Chris@197 596 m_cache->setNumColors(256);
Chris@197 597 DenseThreeDimensionalModel::Column values;
Chris@197 598
Chris@197 599 float min = m_model->getMinimumLevel();
Chris@197 600 float max = m_model->getMaximumLevel();
Chris@197 601
Chris@197 602 if (m_colourScale == LogScale) {
Chris@197 603 LogRange::mapRange(min, max);
Chris@197 604 } else if (m_colourScale == PlusMinusOneScale) {
Chris@197 605 min = -1.f;
Chris@197 606 max = 1.f;
Chris@197 607 }
Chris@197 608
Chris@197 609 if (max == min) max = min + 1.0;
Chris@197 610
Chris@197 611 ColourMapper mapper(m_colourMap, 0.f, 255.f);
Chris@197 612
Chris@197 613 for (int index = 0; index < 256; ++index) {
Chris@197 614
Chris@197 615 QColor colour = mapper.map(index);
Chris@197 616 m_cache->setColor(index, qRgb(colour.red(), colour.green(), colour.blue()));
Chris@197 617 }
Chris@197 618
Chris@197 619 m_cache->fill(0);
Chris@197 620
Chris@224 621 float visibleMax = 0.f, visibleMin = 0.f;
Chris@197 622
Chris@197 623 if (m_normalizeVisibleArea && !m_normalizeColumns) {
Chris@197 624
Chris@197 625 for (size_t c = firstBin; c <= lastBin; ++c) {
Chris@197 626
Chris@197 627 values.clear();
Chris@197 628 getColumn(c, values);
Chris@197 629
Chris@224 630 float colMax = 0.f, colMin = 0.f;
Chris@197 631
Chris@197 632 for (size_t y = 0; y < m_model->getHeight(); ++y) {
Chris@197 633 if (y >= values.size()) break;
Chris@197 634 if (y == 0 || values[y] > colMax) colMax = values[y];
Chris@224 635 if (y == 0 || values[y] < colMin) colMin = values[y];
Chris@197 636 }
Chris@197 637
Chris@197 638 if (c == firstBin || colMax > visibleMax) visibleMax = colMax;
Chris@224 639 if (c == firstBin || colMin < visibleMin) visibleMin = colMin;
Chris@197 640 }
Chris@197 641 }
Chris@197 642
Chris@224 643 if (visibleMin == visibleMax) visibleMax = visibleMin + 1;
Chris@224 644
Chris@197 645 for (size_t c = firstBin; c <= lastBin; ++c) {
Chris@197 646
Chris@197 647 values.clear();
Chris@197 648 getColumn(c, values);
Chris@197 649
Chris@197 650 for (size_t y = 0; y < m_model->getHeight(); ++y) {
Chris@197 651
Chris@197 652 float value = min;
Chris@197 653 if (y < values.size()) {
Chris@197 654 value = values[y];
Chris@197 655 }
Chris@197 656
Chris@197 657 if (m_normalizeVisibleArea && !m_normalizeColumns) {
Chris@224 658 float norm = (value - visibleMin) / (visibleMax - visibleMin);
Chris@224 659 value = min + (max - min) * norm;
Chris@224 660 }
Chris@224 661
Chris@224 662 if (m_colourScale == LogScale) {
Chris@224 663 value = LogRange::map(value);
Chris@197 664 }
Chris@197 665
Chris@197 666 int pixel = int(((value - min) * 256) / (max - min));
Chris@197 667 if (pixel < 0) pixel = 0;
Chris@197 668 if (pixel > 255) pixel = 255;
Chris@197 669
Chris@357 670 if (m_invertVertical) {
Chris@357 671 m_cache->setPixel(c - firstBin, m_model->getHeight() - y - 1,
Chris@357 672 pixel);
Chris@357 673 } else {
Chris@357 674 m_cache->setPixel(c - firstBin, y, pixel);
Chris@357 675 }
Chris@197 676 }
Chris@197 677 }
Chris@197 678 }
Chris@197 679
Chris@197 680 void
Chris@44 681 Colour3DPlotLayer::paint(View *v, QPainter &paint, QRect rect) const
Chris@0 682 {
Chris@443 683 /*!!!
Chris@443 684 if (m_model) {
Chris@443 685 std::cerr << "Colour3DPlotLayer::paint: model says shouldUseLogValueScale = " << m_model->shouldUseLogValueScale() << std::endl;
Chris@443 686 }
Chris@443 687 */
Chris@0 688 // Profiler profiler("Colour3DPlotLayer::paint");
Chris@125 689 #ifdef DEBUG_COLOUR_3D_PLOT_LAYER_PAINT
Chris@125 690 std::cerr << "Colour3DPlotLayer::paint(): m_model is " << m_model << ", zoom level is " << v->getZoomLevel() << std::endl;
Chris@125 691 #endif
Chris@0 692
Chris@0 693 int completion = 0;
Chris@0 694 if (!m_model || !m_model->isOK() || !m_model->isReady(&completion)) {
Chris@0 695 if (completion > 0) {
Chris@44 696 paint.fillRect(0, 10, v->width() * completion / 100,
Chris@0 697 10, QColor(120, 120, 120));
Chris@0 698 }
Chris@0 699 return;
Chris@0 700 }
Chris@0 701
Chris@197 702 if (m_normalizeVisibleArea && !m_normalizeColumns) rect = v->rect();
Chris@197 703
Chris@0 704 size_t modelStart = m_model->getStartFrame();
Chris@0 705 size_t modelEnd = m_model->getEndFrame();
Chris@130 706 size_t modelResolution = m_model->getResolution();
Chris@0 707
Chris@197 708 // The cache is from the model's start frame to the model's end
Chris@197 709 // frame at the model's window increment frames per pixel. We
Chris@197 710 // want to draw from our start frame + x0 * zoomLevel to our start
Chris@197 711 // frame + x1 * zoomLevel at zoomLevel frames per pixel.
Chris@12 712
Chris@197 713 // We have quite different paint mechanisms for rendering "large"
Chris@197 714 // bins (more than one bin per pixel in both directions) and
Chris@197 715 // "small". This is "large"; see paintDense below for "small".
Chris@12 716
Chris@197 717 int x0 = rect.left();
Chris@197 718 int x1 = rect.right() + 1;
Chris@12 719
Chris@197 720 int h = v->height();
Chris@0 721
Chris@197 722 float srRatio =
Chris@197 723 float(v->getViewManager()->getMainModelSampleRate()) /
Chris@197 724 float(m_model->getSampleRate());
Chris@0 725
Chris@197 726 int sx0 = int((v->getFrameForX(x0) / srRatio - long(modelStart))
Chris@197 727 / long(modelResolution));
Chris@197 728 int sx1 = int((v->getFrameForX(x1) / srRatio - long(modelStart))
Chris@197 729 / long(modelResolution));
Chris@197 730 int sh = m_model->getHeight();
Chris@159 731
Chris@444 732 int symin = m_miny;
Chris@444 733 int symax = m_maxy;
Chris@444 734 if (symax <= symin) {
Chris@444 735 symin = 0;
Chris@444 736 symax = sh;
Chris@444 737 }
Chris@444 738 if (symin < 0) symin = 0;
Chris@444 739 if (symax > sh) symax = sh;
Chris@444 740
Chris@197 741 if (sx0 > 0) --sx0;
Chris@197 742 fillCache(sx0 < 0 ? 0 : sx0,
Chris@197 743 sx1 < 0 ? 0 : sx1);
Chris@0 744
Chris@351 745 #ifdef DEBUG_COLOUR_3D_PLOT_LAYER_PAINT
Chris@351 746 std::cerr << "Colour3DPlotLayer::paint: height = "<< m_model->getHeight() << ", modelStart = " << modelStart << ", resolution = " << modelResolution << ", model rate = " << m_model->getSampleRate() << std::endl;
Chris@351 747 #endif
Chris@351 748
Chris@248 749 if (int(m_model->getHeight()) >= v->height() ||
Chris@348 750 int(modelResolution * m_model->getSampleRate()) < v->getZoomLevel() / 2) {
Chris@347 751 #ifdef DEBUG_COLOUR_3D_PLOT_LAYER_PAINT
Chris@351 752 std::cerr << "calling paintDense" << std::endl;
Chris@347 753 #endif
Chris@98 754 paintDense(v, paint, rect);
Chris@98 755 return;
Chris@98 756 }
Chris@98 757
Chris@125 758 #ifdef DEBUG_COLOUR_3D_PLOT_LAYER_PAINT
Chris@347 759 std::cerr << "Colour3DPlotLayer::paint: w " << x1-x0 << ", h " << h << ", sx0 " << sx0 << ", sx1 " << sx1 << ", sw " << sx1-sx0 << ", sh " << sh << std::endl;
Chris@130 760 std::cerr << "Colour3DPlotLayer: sample rate is " << m_model->getSampleRate() << ", resolution " << m_model->getResolution() << std::endl;
Chris@125 761 #endif
Chris@0 762
Chris@25 763 QPoint illuminatePos;
Chris@44 764 bool illuminate = v->shouldIlluminateLocalFeatures(this, illuminatePos);
Chris@54 765 char labelbuf[10];
Chris@25 766
Chris@197 767 for (int sx = sx0; sx <= sx1; ++sx) {
Chris@0 768
Chris@197 769 int scx = 0;
Chris@248 770 if (sx > int(m_cacheStart)) scx = sx - m_cacheStart;
Chris@197 771
Chris@130 772 int fx = sx * int(modelResolution);
Chris@0 773
Chris@351 774 if (fx + int(modelResolution) <= int(modelStart) ||
Chris@0 775 fx > int(modelEnd)) continue;
Chris@0 776
Chris@248 777 int rx0 = v->getXForFrame(int((fx + int(modelStart)) * srRatio));
Chris@248 778 int rx1 = v->getXForFrame(int((fx + int(modelStart) + int(modelResolution) + 1) * srRatio));
Chris@54 779
Chris@159 780 int rw = rx1 - rx0;
Chris@159 781 if (rw < 1) rw = 1;
Chris@54 782
Chris@159 783 bool showLabel = (rw > 10 &&
Chris@159 784 paint.fontMetrics().width("0.000000") < rw - 3 &&
Chris@54 785 paint.fontMetrics().height() < (h / sh));
Chris@98 786
Chris@444 787 for (int sy = symin; sy < symax; ++sy) {
Chris@0 788
Chris@444 789 int ry0 = h - ((sy - symin) * h) / (symax - symin) - 1;
Chris@0 790 QRgb pixel = qRgb(255, 255, 255);
Chris@197 791 if (scx >= 0 && scx < m_cache->width() &&
Chris@0 792 sy >= 0 && sy < m_cache->height()) {
Chris@197 793 pixel = m_cache->pixel(scx, sy);
Chris@0 794 }
Chris@0 795
Chris@444 796 QRect r(rx0, ry0 - h / (symax - symin) - 1,
Chris@444 797 rw, h / (symax - symin) + 1);
Chris@159 798
Chris@159 799 if (rw == 1) {
Chris@159 800 paint.setPen(pixel);
Chris@159 801 paint.setBrush(Qt::NoBrush);
Chris@159 802 paint.drawLine(r.x(), r.y(), r.x(), r.y() + r.height() - 1);
Chris@159 803 continue;
Chris@159 804 }
Chris@159 805
Chris@0 806 QColor pen(255, 255, 255, 80);
Chris@0 807 QColor brush(pixel);
Chris@159 808
Chris@159 809 if (rw > 3 && r.height() > 3) {
Chris@159 810 brush.setAlpha(160);
Chris@159 811 }
Chris@159 812
Chris@0 813 paint.setPen(Qt::NoPen);
Chris@0 814 paint.setBrush(brush);
Chris@0 815
Chris@25 816 if (illuminate) {
Chris@25 817 if (r.contains(illuminatePos)) {
Chris@287 818 paint.setPen(v->getForeground());
Chris@25 819 }
Chris@25 820 }
Chris@76 821
Chris@125 822 #ifdef DEBUG_COLOUR_3D_PLOT_LAYER_PAINT
Chris@159 823 std::cerr << "rect " << r.x() << "," << r.y() << " "
Chris@159 824 << r.width() << "x" << r.height() << std::endl;
Chris@125 825 #endif
Chris@25 826
Chris@25 827 paint.drawRect(r);
Chris@0 828
Chris@54 829 if (showLabel) {
Chris@197 830 if (scx >= 0 && scx < m_cache->width() &&
Chris@54 831 sy >= 0 && sy < m_cache->height()) {
Chris@197 832 float value = m_model->getValueAt(scx, sy);
Chris@54 833 sprintf(labelbuf, "%06f", value);
Chris@54 834 QString text(labelbuf);
Chris@287 835 paint.setPen(v->getBackground());
Chris@54 836 paint.drawText(rx0 + 2,
Chris@54 837 ry0 - h / sh - 1 + 2 + paint.fontMetrics().ascent(),
Chris@54 838 text);
Chris@0 839 }
Chris@0 840 }
Chris@0 841 }
Chris@0 842 }
Chris@98 843 }
Chris@0 844
Chris@98 845 void
Chris@98 846 Colour3DPlotLayer::paintDense(View *v, QPainter &paint, QRect rect) const
Chris@98 847 {
Chris@98 848 size_t modelStart = m_model->getStartFrame();
Chris@130 849 size_t modelResolution = m_model->getResolution();
Chris@0 850
Chris@159 851 float srRatio =
Chris@159 852 float(v->getViewManager()->getMainModelSampleRate()) /
Chris@159 853 float(m_model->getSampleRate());
Chris@159 854
Chris@98 855 int x0 = rect.left();
Chris@98 856 int x1 = rect.right() + 1;
Chris@0 857
Chris@98 858 int w = x1 - x0;
Chris@98 859 int h = v->height();
Chris@160 860 int sh = m_model->getHeight();
Chris@98 861
Chris@444 862 int symin = m_miny;
Chris@444 863 int symax = m_maxy;
Chris@444 864 if (symax <= symin) {
Chris@444 865 symin = 0;
Chris@444 866 symax = sh;
Chris@444 867 }
Chris@444 868 if (symin < 0) symin = 0;
Chris@444 869 if (symax > sh) symax = sh;
Chris@444 870
Chris@98 871 QImage img(w, h, QImage::Format_RGB32);
Chris@98 872
Chris@98 873 for (int x = x0; x < x1; ++x) {
Chris@98 874
Chris@347 875 long xf = long(v->getFrameForX(x));
Chris@105 876 if (xf < 0) {
Chris@105 877 for (int y = 0; y < h; ++y) {
Chris@105 878 img.setPixel(x - x0, y, m_cache->color(0));
Chris@105 879 }
Chris@105 880 continue;
Chris@105 881 }
Chris@105 882
Chris@347 883 xf /= srRatio;
Chris@347 884
Chris@130 885 float sx0 = (float(xf) - modelStart) / modelResolution;
Chris@159 886 float sx1 = (float(v->getFrameForX(x+1) / srRatio) - modelStart) / modelResolution;
Chris@98 887
Chris@98 888 int sx0i = int(sx0 + 0.001);
Chris@98 889 int sx1i = int(sx1);
Chris@98 890
Chris@98 891 for (int y = 0; y < h; ++y) {
Chris@98 892
Chris@444 893 float sy0 = symin + (float(h - y - 1) * (symax - symin)) / h;
Chris@444 894 float sy1 = symin + (float(h - y) * (symax - symin)) / h;
Chris@98 895
Chris@98 896 int sy0i = int(sy0 + 0.001);
Chris@98 897 int sy1i = int(sy1);
Chris@98 898
Chris@98 899 float mag = 0.0, div = 0.0;
Chris@159 900 int max = 0;
Chris@98 901
Chris@98 902 for (int sx = sx0i; sx <= sx1i; ++sx) {
Chris@98 903
Chris@197 904 int scx = 0;
Chris@248 905 if (sx > int(m_cacheStart)) scx = sx - int(m_cacheStart);
Chris@197 906
Chris@197 907 if (scx < 0 || scx >= m_cache->width()) continue;
Chris@98 908
Chris@98 909 for (int sy = sy0i; sy <= sy1i; ++sy) {
Chris@98 910
Chris@98 911 if (sy < 0 || sy >= m_cache->height()) continue;
Chris@98 912
Chris@98 913 float prop = 1.0;
Chris@98 914 if (sx == sx0i) prop *= (sx + 1) - sx0;
Chris@98 915 if (sx == sx1i) prop *= sx1 - sx;
Chris@98 916 if (sy == sy0i) prop *= (sy + 1) - sy0;
Chris@98 917 if (sy == sy1i) prop *= sy1 - sy;
Chris@98 918
Chris@197 919 mag += prop * m_cache->pixelIndex(scx, sy);
Chris@197 920 max = std::max(max, m_cache->pixelIndex(scx, sy));
Chris@98 921 div += prop;
Chris@98 922 }
Chris@98 923 }
Chris@98 924
Chris@98 925 if (div != 0) mag /= div;
Chris@98 926 if (mag < 0) mag = 0;
Chris@98 927 if (mag > 255) mag = 255;
Chris@159 928 if (max < 0) max = 0;
Chris@159 929 if (max > 255) max = 255;
Chris@98 930
Chris@98 931 img.setPixel(x - x0, y, m_cache->color(int(mag + 0.001)));
Chris@159 932 // img.setPixel(x - x0, y, m_cache->color(max));
Chris@98 933 }
Chris@0 934 }
Chris@0 935
Chris@98 936 paint.drawImage(x0, 0, img);
Chris@0 937 }
Chris@0 938
Chris@28 939 bool
Chris@44 940 Colour3DPlotLayer::snapToFeatureFrame(View *v, int &frame,
Chris@28 941 size_t &resolution,
Chris@28 942 SnapType snap) const
Chris@24 943 {
Chris@24 944 if (!m_model) {
Chris@44 945 return Layer::snapToFeatureFrame(v, frame, resolution, snap);
Chris@24 946 }
Chris@24 947
Chris@130 948 resolution = m_model->getResolution();
Chris@28 949 int left = (frame / resolution) * resolution;
Chris@28 950 int right = left + resolution;
Chris@28 951
Chris@28 952 switch (snap) {
Chris@28 953 case SnapLeft: frame = left; break;
Chris@28 954 case SnapRight: frame = right; break;
Chris@28 955 case SnapNearest:
Chris@28 956 case SnapNeighbouring:
Chris@28 957 if (frame - left > right - frame) frame = right;
Chris@28 958 else frame = left;
Chris@28 959 break;
Chris@28 960 }
Chris@24 961
Chris@28 962 return true;
Chris@24 963 }
Chris@24 964
Chris@316 965 void
Chris@316 966 Colour3DPlotLayer::toXml(QTextStream &stream,
Chris@316 967 QString indent, QString extraAttributes) const
Chris@197 968 {
Chris@316 969 QString s = QString("scale=\"%1\" "
Chris@316 970 "colourScheme=\"%2\" "
Chris@316 971 "normalizeColumns=\"%3\" "
Chris@445 972 "normalizeVisibleArea=\"%4\" "
Chris@445 973 "minY=\"%5\" "
Chris@445 974 "maxY=\"%6\" ")
Chris@197 975 .arg((int)m_colourScale)
Chris@197 976 .arg(m_colourMap)
Chris@197 977 .arg(m_normalizeColumns ? "true" : "false")
Chris@445 978 .arg(m_normalizeVisibleArea ? "true" : "false")
Chris@445 979 .arg(m_miny)
Chris@445 980 .arg(m_maxy);
Chris@197 981
Chris@316 982 Layer::toXml(stream, indent, extraAttributes + " " + s);
Chris@197 983 }
Chris@197 984
Chris@197 985 void
Chris@197 986 Colour3DPlotLayer::setProperties(const QXmlAttributes &attributes)
Chris@197 987 {
Chris@445 988 bool ok = false, alsoOk = false;
Chris@197 989
Chris@197 990 ColourScale scale = (ColourScale)attributes.value("scale").toInt(&ok);
Chris@197 991 if (ok) setColourScale(scale);
Chris@197 992
Chris@197 993 int colourMap = attributes.value("colourScheme").toInt(&ok);
Chris@197 994 if (ok) setColourMap(colourMap);
Chris@197 995
Chris@197 996 bool normalizeColumns =
Chris@197 997 (attributes.value("normalizeColumns").trimmed() == "true");
Chris@197 998 setNormalizeColumns(normalizeColumns);
Chris@197 999
Chris@197 1000 bool normalizeVisibleArea =
Chris@197 1001 (attributes.value("normalizeVisibleArea").trimmed() == "true");
Chris@197 1002 setNormalizeVisibleArea(normalizeVisibleArea);
Chris@445 1003
Chris@445 1004 float min = attributes.value("minY").toFloat(&ok);
Chris@445 1005 float max = attributes.value("maxY").toFloat(&alsoOk);
Chris@445 1006 if (ok && alsoOk) setDisplayExtents(min, max);
Chris@197 1007 }
Chris@197 1008