annotate layer/Colour3DPlotRenderer.cpp @ 1123:343887ac6766 spectrogram-minor-refactor

Some tidying
author Chris Cannam
date Fri, 22 Jul 2016 14:42:36 +0100
parents 94370157b265
children b71a0491d287
rev   line source
Chris@1071 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@1071 2
Chris@1071 3 /*
Chris@1071 4 Sonic Visualiser
Chris@1071 5 An audio file viewer and annotation editor.
Chris@1071 6 Centre for Digital Music, Queen Mary, University of London.
Chris@1071 7 This file copyright 2006-2016 Chris Cannam and QMUL.
Chris@1071 8
Chris@1071 9 This program is free software; you can redistribute it and/or
Chris@1071 10 modify it under the terms of the GNU General Public License as
Chris@1071 11 published by the Free Software Foundation; either version 2 of the
Chris@1071 12 License, or (at your option) any later version. See the file
Chris@1071 13 COPYING included with this distribution for more information.
Chris@1071 14 */
Chris@1071 15
Chris@1071 16 #include "Colour3DPlotRenderer.h"
Chris@1074 17 #include "RenderTimer.h"
Chris@1071 18
Chris@1117 19 #include "base/Profiler.h"
Chris@1117 20
Chris@1075 21 #include "data/model/DenseThreeDimensionalModel.h"
Chris@1075 22 #include "data/model/Dense3DModelPeakCache.h"
Chris@1075 23 #include "data/model/FFTModel.h"
Chris@1075 24
Chris@1077 25 #include "LayerGeometryProvider.h"
Chris@1082 26 #include "VerticalBinLayer.h"
Chris@1109 27 #include "PaintAssistant.h"
Chris@1109 28
Chris@1109 29 #include "view/ViewManager.h" // for main model sample rate. Pity
Chris@1075 30
Chris@1079 31 #include <vector>
Chris@1079 32
Chris@1123 33 #define DEBUG_COLOUR_PLOT_REPAINT 1
Chris@1094 34
Chris@1079 35 using namespace std;
Chris@1079 36
Chris@1073 37 Colour3DPlotRenderer::RenderResult
Chris@1113 38 Colour3DPlotRenderer::render(const LayerGeometryProvider *v, QPainter &paint, QRect rect)
Chris@1076 39 {
Chris@1090 40 return render(v, paint, rect, false);
Chris@1076 41 }
Chris@1076 42
Chris@1076 43 Colour3DPlotRenderer::RenderResult
Chris@1113 44 Colour3DPlotRenderer::renderTimeConstrained(const LayerGeometryProvider *v,
Chris@1090 45 QPainter &paint, QRect rect)
Chris@1076 46 {
Chris@1090 47 return render(v, paint, rect, true);
Chris@1076 48 }
Chris@1076 49
Chris@1096 50 QRect
Chris@1121 51 Colour3DPlotRenderer::getLargestUncachedRect(const LayerGeometryProvider *v)
Chris@1096 52 {
Chris@1121 53 RenderType renderType = decideRenderType(v);
Chris@1121 54
Chris@1121 55 if (renderType == DirectTranslucent) {
Chris@1121 56 return QRect(); // never cached
Chris@1121 57 }
Chris@1121 58
Chris@1096 59 int h = m_cache.getSize().height();
Chris@1096 60
Chris@1096 61 QRect areaLeft(0, 0, m_cache.getValidLeft(), h);
Chris@1096 62 QRect areaRight(m_cache.getValidRight(), 0,
Chris@1096 63 m_cache.getSize().width() - m_cache.getValidRight(), h);
Chris@1096 64
Chris@1096 65 if (areaRight.width() > areaLeft.width()) {
Chris@1096 66 return areaRight;
Chris@1096 67 } else {
Chris@1096 68 return areaLeft;
Chris@1096 69 }
Chris@1096 70 }
Chris@1096 71
Chris@1122 72 bool
Chris@1122 73 Colour3DPlotRenderer::geometryChanged(const LayerGeometryProvider *v)
Chris@1122 74 {
Chris@1122 75 RenderType renderType = decideRenderType(v);
Chris@1122 76
Chris@1122 77 if (renderType == DirectTranslucent) {
Chris@1122 78 return true; // never cached
Chris@1122 79 }
Chris@1122 80
Chris@1122 81 if (m_cache.getSize() == v->getPaintSize() &&
Chris@1122 82 m_cache.getZoomLevel() == v->getZoomLevel() &&
Chris@1122 83 m_cache.getStartFrame() == v->getStartFrame()) {
Chris@1122 84 return false;
Chris@1122 85 } else {
Chris@1122 86 return true;
Chris@1122 87 }
Chris@1122 88 }
Chris@1122 89
Chris@1076 90 Colour3DPlotRenderer::RenderResult
Chris@1113 91 Colour3DPlotRenderer::render(const LayerGeometryProvider *v,
Chris@1090 92 QPainter &paint, QRect rect, bool timeConstrained)
Chris@1073 93 {
Chris@1109 94 RenderType renderType = decideRenderType(v);
Chris@1109 95
Chris@1109 96 if (renderType != DrawBufferPixelResolution) {
Chris@1109 97 // Rendering should be fast in bin-resolution and direct draw
Chris@1109 98 // cases because we are quite well zoomed-in, and the sums are
Chris@1109 99 // easier this way. Calculating boundaries later will be
Chris@1109 100 // fiddly for partial paints otherwise.
Chris@1109 101 timeConstrained = false;
Chris@1109 102 }
Chris@1109 103
Chris@1079 104 int x0 = v->getXForViewX(rect.x());
Chris@1079 105 int x1 = v->getXForViewX(rect.x() + rect.width());
Chris@1079 106 if (x0 < 0) x0 = 0;
Chris@1079 107 if (x1 > v->getPaintWidth()) x1 = v->getPaintWidth();
Chris@1079 108
Chris@1120 109 sv_frame_t startFrame = v->getStartFrame();
Chris@1120 110
Chris@1079 111 m_cache.resize(v->getPaintSize());
Chris@1079 112 m_cache.setZoomLevel(v->getZoomLevel());
Chris@1079 113
Chris@1119 114 m_magCache.resize(v->getPaintSize().width());
Chris@1119 115 m_magCache.setZoomLevel(v->getZoomLevel());
Chris@1119 116
Chris@1120 117 if (renderType == DirectTranslucent) {
Chris@1121 118 MagnitudeRange range = renderDirectTranslucent(v, paint, rect);
Chris@1121 119 return { rect, range };
Chris@1120 120 }
Chris@1120 121
Chris@1123 122 #ifdef DEBUG_COLOUR_PLOT_REPAINT
Chris@1090 123 cerr << "cache start " << m_cache.getStartFrame()
Chris@1090 124 << " valid left " << m_cache.getValidLeft()
Chris@1090 125 << " valid right " << m_cache.getValidRight()
Chris@1094 126 << endl;
Chris@1094 127 cerr << " view start " << startFrame
Chris@1090 128 << " x0 " << x0
Chris@1090 129 << " x1 " << x1
Chris@1090 130 << endl;
Chris@1123 131 #endif
Chris@1090 132
Chris@1079 133 if (m_cache.isValid()) { // some part of the cache is valid
Chris@1079 134
Chris@1079 135 if (v->getXForFrame(m_cache.getStartFrame()) ==
Chris@1079 136 v->getXForFrame(startFrame) &&
Chris@1079 137 m_cache.getValidLeft() <= x0 &&
Chris@1079 138 m_cache.getValidRight() >= x1) {
Chris@1090 139
Chris@1123 140 #ifdef DEBUG_COLOUR_PLOT_REPAINT
Chris@1090 141 cerr << "cache hit" << endl;
Chris@1123 142 #endif
Chris@1090 143
Chris@1079 144 // cache is valid for the complete requested area
Chris@1079 145 paint.drawImage(rect, m_cache.getImage(), rect);
Chris@1119 146
Chris@1122 147 MagnitudeRange range = m_magCache.getRange(x0, x1 - x0);
Chris@1119 148
Chris@1119 149 return { rect, range };
Chris@1079 150
Chris@1079 151 } else {
Chris@1123 152 #ifdef DEBUG_COLOUR_PLOT_REPAINT
Chris@1090 153 cerr << "cache partial hit" << endl;
Chris@1123 154 #endif
Chris@1090 155
Chris@1079 156 // cache doesn't begin at the right frame or doesn't
Chris@1079 157 // contain the complete view, but might be scrollable or
Chris@1079 158 // partially usable
Chris@1090 159 m_cache.scrollTo(v, startFrame);
Chris@1119 160 m_magCache.scrollTo(v, startFrame);
Chris@1079 161
Chris@1079 162 // if we are not time-constrained, then we want to paint
Chris@1081 163 // the whole area in one go; we don't return a partial
Chris@1081 164 // paint. To avoid providing the more complex logic to
Chris@1081 165 // handle painting discontiguous areas, if the only valid
Chris@1079 166 // part of cache is in the middle, just make the whole
Chris@1079 167 // thing invalid and start again.
Chris@1079 168 if (!timeConstrained) {
Chris@1079 169 if (m_cache.getValidLeft() > x0 &&
Chris@1079 170 m_cache.getValidRight() < x1) {
Chris@1079 171 m_cache.invalidate();
Chris@1079 172 }
Chris@1079 173 }
Chris@1079 174 }
Chris@1090 175 } else {
Chris@1118 176 // cache is completely invalid
Chris@1090 177 m_cache.setStartFrame(startFrame);
Chris@1119 178 m_magCache.setStartFrame(startFrame);
Chris@1075 179 }
Chris@1075 180
Chris@1079 181 bool rightToLeft = false;
Chris@1079 182
Chris@1122 183 int reqx0 = x0;
Chris@1122 184 int reqx1 = x1;
Chris@1122 185
Chris@1079 186 if (!m_cache.isValid() && timeConstrained) {
Chris@1081 187 // When rendering the whole area, in a context where we might
Chris@1081 188 // not be able to complete the work, start from somewhere near
Chris@1081 189 // the middle so that the region of interest appears first
Chris@1079 190
Chris@1079 191 //!!! (perhaps we should avoid doing this if past repaints
Chris@1079 192 //!!! have been fast enough to do the whole in one shot)
Chris@1079 193 if (x0 == 0 && x1 == v->getPaintWidth()) {
Chris@1079 194 x0 = int(x1 * 0.3);
Chris@1079 195 }
Chris@1079 196 }
Chris@1079 197
Chris@1079 198 if (m_cache.isValid()) {
Chris@1090 199
Chris@1079 200 // When rendering only a part of the cache, we need to make
Chris@1079 201 // sure that the part we're rendering is adjacent to (or
Chris@1079 202 // overlapping) a valid area of cache, if we have one. The
Chris@1079 203 // alternative is to ditch the valid area of cache and render
Chris@1079 204 // only the requested area, but that's risky because this can
Chris@1079 205 // happen when just waving the pointer over a small part of
Chris@1079 206 // the view -- if we lose the partly-built cache every time
Chris@1079 207 // the user does that, we'll never finish building it.
Chris@1079 208 int left = x0;
Chris@1079 209 int width = x1 - x0;
Chris@1079 210 bool isLeftOfValidArea = false;
Chris@1079 211 m_cache.adjustToTouchValidArea(left, width, isLeftOfValidArea);
Chris@1079 212 x0 = left;
Chris@1079 213 x1 = x0 + width;
Chris@1079 214
Chris@1079 215 // That call also told us whether we should be painting
Chris@1079 216 // sub-regions of our target region in right-to-left order in
Chris@1079 217 // order to ensure contiguity
Chris@1079 218 rightToLeft = isLeftOfValidArea;
Chris@1079 219 }
Chris@1075 220
Chris@1109 221 // Note, we always paint the full height to cache. We want to
Chris@1109 222 // ensure the cache is coherent without having to worry about
Chris@1109 223 // vertical matching of required and valid areas as well as
Chris@1109 224 // horizontal.
Chris@1094 225
Chris@1109 226 if (renderType == DrawBufferBinResolution) {
Chris@1109 227
Chris@1094 228 renderToCacheBinResolution(v, x0, x1 - x0);
Chris@1109 229
Chris@1109 230 } else { // must be DrawBufferPixelResolution, handled DirectTranslucent earlier
Chris@1109 231
Chris@1094 232 renderToCachePixelResolution(v, x0, x1 - x0, rightToLeft, timeConstrained);
Chris@1094 233 }
Chris@1079 234
Chris@1079 235 QRect pr = rect & m_cache.getValidArea();
Chris@1079 236 paint.drawImage(pr.x(), pr.y(), m_cache.getImage(),
Chris@1079 237 pr.x(), pr.y(), pr.width(), pr.height());
Chris@1079 238
Chris@1079 239 if (!timeConstrained && (pr != rect)) {
Chris@1079 240 //!!! on a first cut, there is a risk that this will happen
Chris@1079 241 //!!! when we are at start/end of model -- trap, report, and
Chris@1079 242 //!!! then fix
Chris@1079 243 throw std::logic_error("internal error: failed to render entire requested rect even when not time-constrained");
Chris@1079 244 }
Chris@1120 245
Chris@1122 246 MagnitudeRange range = m_magCache.getRange(reqx0, reqx1 - reqx0);
Chris@1120 247
Chris@1120 248 return { pr, range };
Chris@1079 249
Chris@1079 250 //!!! todo, here or in caller: illuminateLocalFeatures
Chris@1079 251
Chris@1123 252 //!!! todo: handle vertical range other than full range of column
Chris@1110 253
Chris@1079 254 //!!! fft model scaling?
Chris@1073 255 }
Chris@1073 256
Chris@1109 257 Colour3DPlotRenderer::RenderType
Chris@1113 258 Colour3DPlotRenderer::decideRenderType(const LayerGeometryProvider *v) const
Chris@1094 259 {
Chris@1100 260 const DenseThreeDimensionalModel *model = m_sources.source;
Chris@1109 261 if (!model || !v || !(v->getViewManager())) {
Chris@1109 262 return DrawBufferPixelResolution; // or anything
Chris@1109 263 }
Chris@1109 264
Chris@1094 265 int binResolution = model->getResolution();
Chris@1094 266 int zoomLevel = v->getZoomLevel();
Chris@1109 267 sv_samplerate_t modelRate = model->getSampleRate();
Chris@1109 268
Chris@1109 269 double rateRatio = v->getViewManager()->getMainModelSampleRate() / modelRate;
Chris@1109 270 double relativeBinResolution = binResolution * rateRatio;
Chris@1109 271
Chris@1109 272 if (m_params.binDisplay == BinDisplay::PeakFrequencies) {
Chris@1109 273 // no alternative works here
Chris@1109 274 return DrawBufferPixelResolution;
Chris@1109 275 }
Chris@1109 276
Chris@1109 277 if (!m_params.alwaysOpaque && !m_params.interpolate) {
Chris@1109 278
Chris@1109 279 // consider translucent option -- only if not smoothing & not
Chris@1109 280 // explicitly requested opaque & sufficiently zoomed-in
Chris@1109 281
Chris@1117 282 if (model->getHeight() * 3 < v->getPaintHeight() &&
Chris@1117 283 relativeBinResolution >= 3 * zoomLevel) {
Chris@1109 284 return DirectTranslucent;
Chris@1109 285 }
Chris@1109 286 }
Chris@1109 287
Chris@1109 288 if (relativeBinResolution > zoomLevel) {
Chris@1109 289 return DrawBufferBinResolution;
Chris@1109 290 } else {
Chris@1109 291 return DrawBufferPixelResolution;
Chris@1109 292 }
Chris@1109 293 }
Chris@1109 294
Chris@1121 295 MagnitudeRange
Chris@1113 296 Colour3DPlotRenderer::renderDirectTranslucent(const LayerGeometryProvider *v,
Chris@1109 297 QPainter &paint,
Chris@1109 298 QRect rect)
Chris@1109 299 {
Chris@1117 300 Profiler profiler("Colour3DPlotRenderer::renderDirectTranslucent");
Chris@1117 301
Chris@1121 302 MagnitudeRange magRange;
Chris@1121 303
Chris@1115 304 QPoint illuminatePos;
Chris@1115 305 bool illuminate = v->shouldIlluminateLocalFeatures
Chris@1115 306 (m_sources.verticalBinLayer, illuminatePos);
Chris@1109 307
Chris@1109 308 const DenseThreeDimensionalModel *model = m_sources.source;
Chris@1109 309
Chris@1109 310 int x0 = rect.left();
Chris@1109 311 int x1 = rect.right() + 1;
Chris@1109 312
Chris@1109 313 int h = v->getPaintHeight();
Chris@1109 314
Chris@1109 315 sv_frame_t modelStart = model->getStartFrame();
Chris@1109 316 sv_frame_t modelEnd = model->getEndFrame();
Chris@1109 317 int modelResolution = model->getResolution();
Chris@1109 318
Chris@1109 319 double rateRatio =
Chris@1109 320 v->getViewManager()->getMainModelSampleRate() / model->getSampleRate();
Chris@1109 321
Chris@1109 322 // the s-prefix values are source, i.e. model, column and bin numbers
Chris@1109 323 int sx0 = int((double(v->getFrameForX(x0)) / rateRatio - double(modelStart))
Chris@1109 324 / modelResolution);
Chris@1109 325 int sx1 = int((double(v->getFrameForX(x1)) / rateRatio - double(modelStart))
Chris@1109 326 / modelResolution);
Chris@1109 327
Chris@1109 328 int sh = model->getHeight();
Chris@1109 329
Chris@1109 330 const int buflen = 40;
Chris@1109 331 char labelbuf[buflen];
Chris@1109 332
Chris@1115 333 int minbin = 0; //!!!
Chris@1109 334 int maxbin = sh - 1; //!!!
Chris@1109 335
Chris@1109 336 int psx = -1;
Chris@1109 337
Chris@1109 338 vector<float> preparedColumn;
Chris@1109 339
Chris@1109 340 int modelWidth = model->getWidth();
Chris@1109 341
Chris@1109 342 for (int sx = sx0; sx <= sx1; ++sx) {
Chris@1109 343
Chris@1109 344 if (sx < 0 || sx >= modelWidth) {
Chris@1109 345 continue;
Chris@1109 346 }
Chris@1109 347
Chris@1109 348 if (sx != psx) {
Chris@1109 349
Chris@1109 350 //!!! this is in common with renderDrawBuffer - pull it out
Chris@1109 351
Chris@1109 352 // order:
Chris@1109 353 // get column -> scale -> record extents ->
Chris@1109 354 // normalise -> peak pick -> apply display gain ->
Chris@1109 355 // distribute/interpolate
Chris@1109 356
Chris@1109 357 ColumnOp::Column fullColumn = model->getColumn(sx);
Chris@1109 358
Chris@1109 359 ColumnOp::Column column =
Chris@1109 360 vector<float>(fullColumn.data() + minbin,
Chris@1109 361 fullColumn.data() + maxbin + 1);
Chris@1109 362
Chris@1121 363 magRange.sample(column);
Chris@1121 364
Chris@1109 365 //!!! fft scale if (m_colourScale != ColourScaleType::Phase) {
Chris@1109 366 // column = ColumnOp::fftScale(column, m_fftSize);
Chris@1109 367 // }
Chris@1109 368
Chris@1109 369 // if (m_colourScale != ColourScaleType::Phase) {
Chris@1115 370 preparedColumn = ColumnOp::normalize(column, m_params.normalization);
Chris@1109 371 // }
Chris@1109 372
Chris@1109 373 if (m_params.binDisplay == BinDisplay::PeakBins) {
Chris@1115 374 preparedColumn = ColumnOp::peakPick(preparedColumn);
Chris@1109 375 }
Chris@1109 376
Chris@1109 377 psx = sx;
Chris@1109 378 }
Chris@1109 379
Chris@1109 380 sv_frame_t fx = sx * modelResolution + modelStart;
Chris@1109 381
Chris@1109 382 if (fx + modelResolution <= modelStart || fx > modelEnd) continue;
Chris@1109 383
Chris@1109 384 int rx0 = v->getXForFrame(int(double(fx) * rateRatio));
Chris@1109 385 int rx1 = v->getXForFrame(int(double(fx + modelResolution + 1) * rateRatio));
Chris@1109 386
Chris@1109 387 int rw = rx1 - rx0;
Chris@1109 388 if (rw < 1) rw = 1;
Chris@1109 389
Chris@1109 390 bool showLabel = (rw > 10 &&
Chris@1109 391 paint.fontMetrics().width("0.000000") < rw - 3 &&
Chris@1109 392 paint.fontMetrics().height() < (h / sh));
Chris@1109 393
Chris@1109 394 for (int sy = minbin; sy <= maxbin; ++sy) {
Chris@1109 395
Chris@1109 396 int ry0 = m_sources.verticalBinLayer->getIYForBin(v, sy);
Chris@1109 397 int ry1 = m_sources.verticalBinLayer->getIYForBin(v, sy + 1);
Chris@1116 398
Chris@1116 399 if (m_params.invertVertical) {
Chris@1116 400 ry0 = h - ry0 - 1;
Chris@1116 401 ry1 = h - ry1 - 1;
Chris@1116 402 }
Chris@1116 403
Chris@1109 404 QRect r(rx0, ry1, rw, ry0 - ry1);
Chris@1109 405
Chris@1109 406 float value = preparedColumn[sy - minbin];
Chris@1112 407 QColor colour = m_params.colourScale.getColour(value,
Chris@1112 408 m_params.colourRotation);
Chris@1109 409
Chris@1109 410 if (rw == 1) {
Chris@1109 411 paint.setPen(colour);
Chris@1109 412 paint.setBrush(Qt::NoBrush);
Chris@1109 413 paint.drawLine(r.x(), r.y(), r.x(), r.y() + r.height() - 1);
Chris@1109 414 continue;
Chris@1109 415 }
Chris@1109 416
Chris@1109 417 QColor pen(255, 255, 255, 80);
Chris@1109 418 QColor brush(colour);
Chris@1109 419
Chris@1109 420 if (rw > 3 && r.height() > 3) {
Chris@1109 421 brush.setAlpha(160);
Chris@1109 422 }
Chris@1109 423
Chris@1109 424 paint.setPen(Qt::NoPen);
Chris@1109 425 paint.setBrush(brush);
Chris@1109 426
Chris@1115 427 if (illuminate) {
Chris@1115 428 if (r.contains(illuminatePos)) {
Chris@1115 429 paint.setPen(v->getForeground());
Chris@1115 430 }
Chris@1115 431 }
Chris@1109 432
Chris@1109 433 #ifdef DEBUG_COLOUR_3D_PLOT_LAYER_PAINT
Chris@1109 434 // cerr << "rect " << r.x() << "," << r.y() << " "
Chris@1109 435 // << r.width() << "x" << r.height() << endl;
Chris@1109 436 #endif
Chris@1109 437
Chris@1109 438 paint.drawRect(r);
Chris@1109 439
Chris@1109 440 if (showLabel) {
Chris@1109 441 double value = model->getValueAt(sx, sy);
Chris@1109 442 snprintf(labelbuf, buflen, "%06f", value);
Chris@1109 443 QString text(labelbuf);
Chris@1109 444 PaintAssistant::drawVisibleText
Chris@1109 445 (v,
Chris@1109 446 paint,
Chris@1109 447 rx0 + 2,
Chris@1109 448 ry0 - h / sh - 1 + 2 + paint.fontMetrics().ascent(),
Chris@1109 449 text,
Chris@1109 450 PaintAssistant::OutlinedText);
Chris@1109 451 }
Chris@1109 452 }
Chris@1109 453 }
Chris@1121 454
Chris@1121 455 return magRange;
Chris@1094 456 }
Chris@1094 457
Chris@1080 458 void
Chris@1113 459 Colour3DPlotRenderer::renderToCachePixelResolution(const LayerGeometryProvider *v,
Chris@1094 460 int x0, int repaintWidth,
Chris@1094 461 bool rightToLeft,
Chris@1094 462 bool timeConstrained)
Chris@1079 463 {
Chris@1117 464 Profiler profiler("Colour3DPlotRenderer::renderToCachePixelResolution");
Chris@1094 465 cerr << "renderToCachePixelResolution" << endl;
Chris@1094 466
Chris@1094 467 // Draw to the draw buffer, and then copy from there. The draw
Chris@1094 468 // buffer is at the same resolution as the target in the cache, so
Chris@1094 469 // no extra scaling needed.
Chris@1079 470
Chris@1100 471 const DenseThreeDimensionalModel *model = m_sources.source;
Chris@1079 472 if (!model || !model->isOK() || !model->isReady()) {
Chris@1079 473 throw std::logic_error("no source model provided, or model not ready");
Chris@1079 474 }
Chris@1079 475
Chris@1079 476 int h = v->getPaintHeight();
Chris@1079 477
Chris@1094 478 clearDrawBuffer(repaintWidth, h);
Chris@1079 479
Chris@1094 480 vector<int> binforx(repaintWidth);
Chris@1079 481 vector<double> binfory(h);
Chris@1079 482
Chris@1079 483 bool usePeaksCache = false;
Chris@1079 484 int binsPerPeak = 1;
Chris@1094 485 int zoomLevel = v->getZoomLevel();
Chris@1094 486 int binResolution = model->getResolution();
Chris@1079 487
Chris@1094 488 for (int x = 0; x < repaintWidth; ++x) {
Chris@1094 489 sv_frame_t f0 = v->getFrameForX(x0 + x);
Chris@1094 490 double s0 = double(f0 - model->getStartFrame()) / binResolution;
Chris@1094 491 binforx[x] = int(s0 + 0.0001);
Chris@1094 492 }
Chris@1080 493
Chris@1094 494 if (m_sources.peaks) { // peaks cache exists
Chris@1080 495
Chris@1094 496 binsPerPeak = m_sources.peaks->getColumnsPerPeak();
Chris@1094 497 usePeaksCache = (binResolution * binsPerPeak) < zoomLevel;
Chris@1094 498
Chris@1094 499 if (m_params.colourScale.getScale() ==
Chris@1105 500 ColourScaleType::Phase) {
Chris@1094 501 usePeaksCache = false;
Chris@1079 502 }
Chris@1079 503 }
Chris@1082 504
Chris@1094 505 cerr << "[PIX] zoomLevel = " << zoomLevel
Chris@1094 506 << ", binResolution " << binResolution
Chris@1094 507 << ", binsPerPeak " << binsPerPeak
Chris@1094 508 << ", peak cache " << m_sources.peaks
Chris@1094 509 << ", usePeaksCache = " << usePeaksCache
Chris@1094 510 << endl;
Chris@1094 511
Chris@1080 512 for (int y = 0; y < h; ++y) {
Chris@1090 513 binfory[y] = m_sources.verticalBinLayer->getBinForY(v, h - y - 1);
Chris@1080 514 }
Chris@1079 515
Chris@1097 516 int attainedWidth;
Chris@1097 517
Chris@1103 518 if (m_params.binDisplay == BinDisplay::PeakFrequencies) {
Chris@1097 519 attainedWidth = renderDrawBufferPeakFrequencies(v,
Chris@1097 520 repaintWidth,
Chris@1097 521 h,
Chris@1097 522 binforx,
Chris@1097 523 binfory,
Chris@1097 524 rightToLeft,
Chris@1097 525 timeConstrained);
Chris@1097 526
Chris@1097 527 } else {
Chris@1097 528 attainedWidth = renderDrawBuffer(repaintWidth,
Chris@1080 529 h,
Chris@1080 530 binforx,
Chris@1080 531 binfory,
Chris@1080 532 usePeaksCache,
Chris@1080 533 rightToLeft,
Chris@1080 534 timeConstrained);
Chris@1097 535 }
Chris@1083 536
Chris@1094 537 if (attainedWidth == 0) return;
Chris@1084 538
Chris@1094 539 // draw buffer is pixel resolution, no scaling factors or padding involved
Chris@1084 540
Chris@1084 541 int paintedLeft = x0;
Chris@1084 542 if (rightToLeft) {
Chris@1084 543 paintedLeft += (repaintWidth - attainedWidth);
Chris@1084 544 }
Chris@1084 545
Chris@1094 546 m_cache.drawImage(paintedLeft, attainedWidth,
Chris@1094 547 m_drawBuffer,
Chris@1094 548 paintedLeft - x0, attainedWidth);
Chris@1121 549
Chris@1121 550 for (int i = 0; in_range_for(m_magRanges, i); ++i) {
Chris@1121 551 m_magCache.sampleColumn(i, m_magRanges.at(i));
Chris@1121 552 }
Chris@1094 553 }
Chris@1084 554
Chris@1094 555 void
Chris@1113 556 Colour3DPlotRenderer::renderToCacheBinResolution(const LayerGeometryProvider *v,
Chris@1094 557 int x0, int repaintWidth)
Chris@1094 558 {
Chris@1117 559 Profiler profiler("Colour3DPlotRenderer::renderToCacheBinResolution");
Chris@1094 560 cerr << "renderToCacheBinResolution" << endl;
Chris@1094 561
Chris@1094 562 // Draw to the draw buffer, and then scale-copy from there. Draw
Chris@1094 563 // buffer is at bin resolution, i.e. buffer x == source column
Chris@1094 564 // number. We use toolkit smooth scaling for interpolation.
Chris@1084 565
Chris@1100 566 const DenseThreeDimensionalModel *model = m_sources.source;
Chris@1094 567 if (!model || !model->isOK() || !model->isReady()) {
Chris@1094 568 throw std::logic_error("no source model provided, or model not ready");
Chris@1094 569 }
Chris@1094 570
Chris@1094 571 // The draw buffer will contain a fragment at bin resolution. We
Chris@1094 572 // need to ensure that it starts and ends at points where a
Chris@1094 573 // time-bin boundary occurs at an exact pixel boundary, and with a
Chris@1094 574 // certain amount of overlap across existing pixels so that we can
Chris@1094 575 // scale and draw from it without smoothing errors at the edges.
Chris@1094 576
Chris@1094 577 // If (getFrameForX(x) / increment) * increment ==
Chris@1094 578 // getFrameForX(x), then x is a time-bin boundary. We want two
Chris@1094 579 // such boundaries at either side of the draw buffer -- one which
Chris@1094 580 // we draw up to, and one which we subsequently crop at.
Chris@1094 581
Chris@1094 582 sv_frame_t leftBoundaryFrame = -1, leftCropFrame = -1;
Chris@1094 583 sv_frame_t rightBoundaryFrame = -1, rightCropFrame = -1;
Chris@1094 584
Chris@1094 585 int drawBufferWidth;
Chris@1094 586 int binResolution = model->getResolution();
Chris@1094 587
Chris@1094 588 for (int x = x0; ; --x) {
Chris@1094 589 sv_frame_t f = v->getFrameForX(x);
Chris@1094 590 if ((f / binResolution) * binResolution == f) {
Chris@1094 591 if (leftCropFrame == -1) leftCropFrame = f;
Chris@1094 592 else if (x < x0 - 2) {
Chris@1094 593 leftBoundaryFrame = f;
Chris@1094 594 break;
Chris@1094 595 }
Chris@1094 596 }
Chris@1094 597 }
Chris@1094 598 for (int x = x0 + repaintWidth; ; ++x) {
Chris@1094 599 sv_frame_t f = v->getFrameForX(x);
Chris@1094 600 if ((f / binResolution) * binResolution == f) {
Chris@1094 601 if (rightCropFrame == -1) rightCropFrame = f;
Chris@1094 602 else if (x > x0 + repaintWidth + 2) {
Chris@1094 603 rightBoundaryFrame = f;
Chris@1094 604 break;
Chris@1094 605 }
Chris@1094 606 }
Chris@1094 607 }
Chris@1094 608 drawBufferWidth = int
Chris@1094 609 ((rightBoundaryFrame - leftBoundaryFrame) / binResolution);
Chris@1094 610
Chris@1094 611 int h = v->getPaintHeight();
Chris@1094 612
Chris@1095 613 // For our purposes here, the draw buffer needs to be exactly our
Chris@1095 614 // target size (so we recreate always rather than just clear it)
Chris@1095 615
Chris@1095 616 recreateDrawBuffer(drawBufferWidth, h);
Chris@1094 617
Chris@1094 618 vector<int> binforx(drawBufferWidth);
Chris@1094 619 vector<double> binfory(h);
Chris@1094 620
Chris@1094 621 for (int x = 0; x < drawBufferWidth; ++x) {
Chris@1094 622 binforx[x] = int(leftBoundaryFrame / binResolution) + x;
Chris@1094 623 }
Chris@1094 624
Chris@1094 625 cerr << "[BIN] binResolution " << binResolution
Chris@1094 626 << endl;
Chris@1094 627
Chris@1094 628 for (int y = 0; y < h; ++y) {
Chris@1094 629 binfory[y] = m_sources.verticalBinLayer->getBinForY(v, h - y - 1);
Chris@1094 630 }
Chris@1094 631
Chris@1094 632 int attainedWidth = renderDrawBuffer(drawBufferWidth,
Chris@1094 633 h,
Chris@1094 634 binforx,
Chris@1094 635 binfory,
Chris@1094 636 false,
Chris@1094 637 false,
Chris@1094 638 false);
Chris@1094 639
Chris@1094 640 if (attainedWidth == 0) return;
Chris@1094 641
Chris@1094 642 int scaledLeft = v->getXForFrame(leftBoundaryFrame);
Chris@1094 643 int scaledRight = v->getXForFrame(rightBoundaryFrame);
Chris@1095 644
Chris@1095 645 cerr << "scaling draw buffer from width " << m_drawBuffer.width()
Chris@1095 646 << " to " << (scaledRight - scaledLeft) << " (nb drawBufferWidth = "
Chris@1095 647 << drawBufferWidth << ")" << endl;
Chris@1094 648
Chris@1094 649 QImage scaled = m_drawBuffer.scaled
Chris@1094 650 (scaledRight - scaledLeft, h,
Chris@1094 651 Qt::IgnoreAspectRatio, (m_params.interpolate ?
Chris@1094 652 Qt::SmoothTransformation :
Chris@1094 653 Qt::FastTransformation));
Chris@1084 654
Chris@1094 655 int scaledLeftCrop = v->getXForFrame(leftCropFrame);
Chris@1094 656 int scaledRightCrop = v->getXForFrame(rightCropFrame);
Chris@1094 657
Chris@1094 658 int targetLeft = scaledLeftCrop;
Chris@1094 659 if (targetLeft < 0) {
Chris@1094 660 targetLeft = 0;
Chris@1094 661 }
Chris@1094 662
Chris@1094 663 int targetWidth = scaledRightCrop - targetLeft;
Chris@1094 664 if (targetLeft + targetWidth > m_cache.getSize().width()) {
Chris@1094 665 targetWidth = m_cache.getSize().width() - targetLeft;
Chris@1094 666 }
Chris@1094 667
Chris@1094 668 int sourceLeft = targetLeft - scaledLeft;
Chris@1094 669 if (sourceLeft < 0) {
Chris@1094 670 sourceLeft = 0;
Chris@1094 671 }
Chris@1094 672
Chris@1094 673 int sourceWidth = targetWidth;
Chris@1094 674
Chris@1094 675 cerr << "repaintWidth = " << repaintWidth
Chris@1094 676 << ", targetWidth = " << targetWidth << endl;
Chris@1094 677
Chris@1094 678 if (targetWidth > 0) {
Chris@1094 679 m_cache.drawImage(targetLeft, targetWidth,
Chris@1094 680 scaled,
Chris@1094 681 sourceLeft, sourceWidth);
Chris@1084 682 }
Chris@1121 683
Chris@1121 684 for (int i = 0; i < targetWidth; ++i) {
Chris@1121 685 int sourceIx = int((double(i) / targetWidth) * sourceWidth);
Chris@1121 686 if (in_range_for(m_magRanges, sourceIx)) {
Chris@1121 687 m_magCache.sampleColumn(i, m_magRanges.at(sourceIx));
Chris@1121 688 }
Chris@1121 689 }
Chris@1079 690 }
Chris@1083 691
Chris@1083 692 int
Chris@1083 693 Colour3DPlotRenderer::renderDrawBuffer(int w, int h,
Chris@1083 694 const vector<int> &binforx,
Chris@1083 695 const vector<double> &binfory,
Chris@1083 696 bool usePeaksCache,
Chris@1083 697 bool rightToLeft,
Chris@1083 698 bool timeConstrained)
Chris@1083 699 {
Chris@1083 700 // Callers must have checked that the appropriate subset of
Chris@1083 701 // Sources data members are set for the supplied flags (e.g. that
Chris@1083 702 // peaks model exists if usePeaksCache)
Chris@1083 703
Chris@1083 704 RenderTimer timer(timeConstrained ?
Chris@1083 705 RenderTimer::FastRender :
Chris@1083 706 RenderTimer::NoTimeout);
Chris@1083 707
Chris@1083 708 int minbin = int(binfory[0] + 0.0001);
Chris@1083 709 int maxbin = int(binfory[h-1]);
Chris@1083 710 if (minbin < 0) minbin = 0;
Chris@1083 711 if (maxbin < 0) maxbin = minbin+1;
Chris@1083 712
Chris@1083 713 int divisor = 1;
Chris@1100 714 const DenseThreeDimensionalModel *sourceModel = m_sources.source;
Chris@1083 715 if (usePeaksCache) {
Chris@1083 716 divisor = m_sources.peaks->getColumnsPerPeak();
Chris@1083 717 sourceModel = m_sources.peaks;
Chris@1083 718 }
Chris@1083 719
Chris@1083 720 int psx = -1;
Chris@1083 721
Chris@1083 722 int start = 0;
Chris@1083 723 int finish = w;
Chris@1083 724 int step = 1;
Chris@1083 725
Chris@1083 726 if (rightToLeft) {
Chris@1083 727 start = w-1;
Chris@1083 728 finish = -1;
Chris@1083 729 step = -1;
Chris@1083 730 }
Chris@1083 731
Chris@1083 732 int columnCount = 0;
Chris@1083 733
Chris@1083 734 vector<float> preparedColumn;
Chris@1094 735
Chris@1094 736 int modelWidth = sourceModel->getWidth();
Chris@1121 737
Chris@1121 738 cerr << "modelWidth " << modelWidth << ", divisor " << divisor << endl;
Chris@1121 739
Chris@1083 740 for (int x = start; x != finish; x += step) {
Chris@1083 741
Chris@1083 742 // x is the on-canvas pixel coord; sx (later) will be the
Chris@1083 743 // source column index
Chris@1083 744
Chris@1083 745 ++columnCount;
Chris@1120 746
Chris@1123 747 #ifdef DEBUG_COLOUR_PLOT_REPAINT
Chris@1121 748 cerr << "x = " << x << ", binforx[x] = " << binforx[x] << endl;
Chris@1120 749 #endif
Chris@1083 750
Chris@1083 751 if (binforx[x] < 0) continue;
Chris@1083 752
Chris@1083 753 int sx0 = binforx[x] / divisor;
Chris@1083 754 int sx1 = sx0;
Chris@1083 755 if (x+1 < w) sx1 = binforx[x+1] / divisor;
Chris@1083 756 if (sx0 < 0) sx0 = sx1 - 1;
Chris@1083 757 if (sx0 < 0) continue;
Chris@1083 758 if (sx1 <= sx0) sx1 = sx0 + 1;
Chris@1083 759
Chris@1083 760 vector<float> pixelPeakColumn;
Chris@1121 761 MagnitudeRange magRange;
Chris@1083 762
Chris@1083 763 for (int sx = sx0; sx < sx1; ++sx) {
Chris@1083 764
Chris@1123 765 #ifdef DEBUG_COLOUR_PLOT_REPAINT
Chris@1094 766 cerr << "sx = " << sx << endl;
Chris@1083 767 #endif
Chris@1083 768
Chris@1094 769 if (sx < 0 || sx >= modelWidth) {
Chris@1083 770 continue;
Chris@1083 771 }
Chris@1083 772
Chris@1083 773 if (sx != psx) {
Chris@1083 774
Chris@1083 775 // order:
Chris@1083 776 // get column -> scale -> record extents ->
Chris@1083 777 // normalise -> peak pick -> apply display gain ->
Chris@1083 778 // distribute/interpolate
Chris@1083 779
Chris@1083 780 ColumnOp::Column fullColumn = sourceModel->getColumn(sx);
Chris@1090 781
Chris@1094 782 // cerr << "x " << x << ", sx " << sx << ", col height " << fullColumn.size()
Chris@1094 783 // << ", minbin " << minbin << ", maxbin " << maxbin << endl;
Chris@1090 784
Chris@1083 785 ColumnOp::Column column =
Chris@1083 786 vector<float>(fullColumn.data() + minbin,
Chris@1083 787 fullColumn.data() + maxbin + 1);
Chris@1083 788
Chris@1105 789 //!!! fft scale if (m_colourScale != ColourScaleType::Phase) {
Chris@1083 790 // column = ColumnOp::fftScale(column, m_fftSize);
Chris@1083 791 // }
Chris@1083 792
Chris@1121 793 magRange.sample(column);
Chris@1083 794
Chris@1105 795 // if (m_colourScale != ColourScaleType::Phase) {
Chris@1083 796 column = ColumnOp::normalize(column, m_params.normalization);
Chris@1083 797 // }
Chris@1083 798
Chris@1103 799 if (m_params.binDisplay == BinDisplay::PeakBins) {
Chris@1083 800 column = ColumnOp::peakPick(column);
Chris@1083 801 }
Chris@1083 802
Chris@1083 803 preparedColumn =
Chris@1083 804 ColumnOp::distribute(column, //!!! gain? ColumnOp::applyGain(column, m_gain),
Chris@1083 805 h,
Chris@1083 806 binfory,
Chris@1083 807 minbin,
Chris@1083 808 m_params.interpolate);
Chris@1083 809
Chris@1083 810 psx = sx;
Chris@1083 811 }
Chris@1083 812
Chris@1083 813 if (sx == sx0) {
Chris@1083 814 pixelPeakColumn = preparedColumn;
Chris@1083 815 } else {
Chris@1083 816 for (int i = 0; in_range_for(pixelPeakColumn, i); ++i) {
Chris@1083 817 pixelPeakColumn[i] = std::max(pixelPeakColumn[i],
Chris@1083 818 preparedColumn[i]);
Chris@1083 819 }
Chris@1083 820 }
Chris@1083 821 }
Chris@1083 822
Chris@1083 823 if (!pixelPeakColumn.empty()) {
Chris@1121 824
Chris@1083 825 for (int y = 0; y < h; ++y) {
Chris@1116 826 int py;
Chris@1116 827 if (m_params.invertVertical) {
Chris@1116 828 py = y;
Chris@1116 829 } else {
Chris@1116 830 py = h - y - 1;
Chris@1116 831 }
Chris@1083 832 m_drawBuffer.setPixel
Chris@1083 833 (x,
Chris@1116 834 py,
Chris@1083 835 m_params.colourScale.getPixel(pixelPeakColumn[y]));
Chris@1083 836 }
Chris@1121 837
Chris@1121 838 m_magRanges.push_back(magRange);
Chris@1083 839 }
Chris@1083 840
Chris@1083 841 double fractionComplete = double(columnCount) / double(w);
Chris@1083 842 if (timer.outOfTime(fractionComplete)) {
Chris@1121 843 cerr << "out of time" << endl;
Chris@1083 844 return columnCount;
Chris@1083 845 }
Chris@1083 846 }
Chris@1083 847
Chris@1083 848 return columnCount;
Chris@1083 849 }
Chris@1083 850
Chris@1097 851 int
Chris@1113 852 Colour3DPlotRenderer::renderDrawBufferPeakFrequencies(const LayerGeometryProvider *v,
Chris@1097 853 int w, int h,
Chris@1097 854 const vector<int> &binforx,
Chris@1097 855 const vector<double> &binfory,
Chris@1097 856 bool rightToLeft,
Chris@1097 857 bool timeConstrained)
Chris@1097 858 {
Chris@1097 859 // Callers must have checked that the appropriate subset of
Chris@1097 860 // Sources data members are set for the supplied flags (e.g. that
Chris@1097 861 // fft model exists)
Chris@1097 862
Chris@1097 863 RenderTimer timer(timeConstrained ?
Chris@1097 864 RenderTimer::FastRender :
Chris@1097 865 RenderTimer::NoTimeout);
Chris@1097 866
Chris@1097 867 int minbin = int(binfory[0] + 0.0001);
Chris@1097 868 int maxbin = int(binfory[h-1]);
Chris@1097 869 if (minbin < 0) minbin = 0;
Chris@1097 870 if (maxbin < 0) maxbin = minbin+1;
Chris@1097 871
Chris@1100 872 const FFTModel *fft = m_sources.fft;
Chris@1097 873
Chris@1097 874 FFTModel::PeakSet peakfreqs;
Chris@1097 875
Chris@1097 876 int psx = -1;
Chris@1097 877
Chris@1097 878 int start = 0;
Chris@1097 879 int finish = w;
Chris@1097 880 int step = 1;
Chris@1097 881
Chris@1097 882 if (rightToLeft) {
Chris@1097 883 start = w-1;
Chris@1097 884 finish = -1;
Chris@1097 885 step = -1;
Chris@1097 886 }
Chris@1097 887
Chris@1097 888 int columnCount = 0;
Chris@1097 889
Chris@1097 890 vector<float> preparedColumn;
Chris@1097 891
Chris@1097 892 int modelWidth = fft->getWidth();
Chris@1097 893 cerr << "modelWidth " << modelWidth << endl;
Chris@1097 894
Chris@1097 895 double minFreq = (double(minbin) * fft->getSampleRate()) / fft->getFFTSize();
Chris@1097 896 double maxFreq = (double(maxbin) * fft->getSampleRate()) / fft->getFFTSize();
Chris@1097 897
Chris@1103 898 bool logarithmic = (m_params.binScale == BinScale::Log);
Chris@1097 899
Chris@1097 900 for (int x = start; x != finish; x += step) {
Chris@1097 901
Chris@1097 902 // x is the on-canvas pixel coord; sx (later) will be the
Chris@1097 903 // source column index
Chris@1097 904
Chris@1097 905 ++columnCount;
Chris@1097 906
Chris@1097 907 if (binforx[x] < 0) continue;
Chris@1097 908
Chris@1097 909 int sx0 = binforx[x];
Chris@1097 910 int sx1 = sx0;
Chris@1097 911 if (x+1 < w) sx1 = binforx[x+1];
Chris@1097 912 if (sx0 < 0) sx0 = sx1 - 1;
Chris@1097 913 if (sx0 < 0) continue;
Chris@1097 914 if (sx1 <= sx0) sx1 = sx0 + 1;
Chris@1097 915
Chris@1097 916 vector<float> pixelPeakColumn;
Chris@1121 917 MagnitudeRange magRange;
Chris@1097 918
Chris@1097 919 for (int sx = sx0; sx < sx1; ++sx) {
Chris@1097 920
Chris@1097 921 if (sx < 0 || sx >= modelWidth) {
Chris@1097 922 continue;
Chris@1097 923 }
Chris@1097 924
Chris@1097 925 if (sx != psx) {
Chris@1097 926
Chris@1097 927 ColumnOp::Column fullColumn = fft->getColumn(sx);
Chris@1097 928
Chris@1097 929 ColumnOp::Column column =
Chris@1097 930 vector<float>(fullColumn.data() + minbin,
Chris@1097 931 fullColumn.data() + maxbin + 1);
Chris@1097 932
Chris@1121 933 magRange.sample(column);
Chris@1121 934
Chris@1105 935 //!!! fft scale if (m_colourScale != ColourScaleType::Phase) {
Chris@1097 936 // column = ColumnOp::fftScale(column, getFFTSize());
Chris@1097 937 // }
Chris@1097 938
Chris@1105 939 //!!! if (m_colourScale != ColourScaleType::Phase) {
Chris@1097 940 column = ColumnOp::normalize(column, m_params.normalization);
Chris@1097 941 //!!! }
Chris@1097 942
Chris@1097 943 preparedColumn = column;
Chris@1097 944 //!!! gain? preparedColumn = ColumnOp::applyGain(column, m_params.gain);
Chris@1097 945
Chris@1097 946 psx = sx;
Chris@1097 947 }
Chris@1097 948
Chris@1097 949 if (sx == sx0) {
Chris@1097 950 pixelPeakColumn = preparedColumn;
Chris@1097 951 peakfreqs = fft->getPeakFrequencies(FFTModel::AllPeaks, sx,
Chris@1097 952 minbin, maxbin - 1);
Chris@1097 953 } else {
Chris@1097 954 for (int i = 0; in_range_for(pixelPeakColumn, i); ++i) {
Chris@1097 955 pixelPeakColumn[i] = std::max(pixelPeakColumn[i],
Chris@1097 956 preparedColumn[i]);
Chris@1097 957 }
Chris@1097 958 }
Chris@1097 959 }
Chris@1097 960
Chris@1097 961 if (!pixelPeakColumn.empty()) {
Chris@1121 962
Chris@1097 963 for (FFTModel::PeakSet::const_iterator pi = peakfreqs.begin();
Chris@1097 964 pi != peakfreqs.end(); ++pi) {
Chris@1097 965
Chris@1097 966 int bin = pi->first;
Chris@1097 967 double freq = pi->second;
Chris@1097 968
Chris@1097 969 if (bin < minbin) continue;
Chris@1097 970 if (bin > maxbin) break;
Chris@1097 971
Chris@1097 972 double value = pixelPeakColumn[bin - minbin];
Chris@1097 973
Chris@1097 974 double y = v->getYForFrequency
Chris@1097 975 (freq, minFreq, maxFreq, logarithmic);
Chris@1097 976
Chris@1097 977 int iy = int(y + 0.5);
Chris@1097 978 if (iy < 0 || iy >= h) continue;
Chris@1097 979
Chris@1097 980 m_drawBuffer.setPixel
Chris@1097 981 (x,
Chris@1097 982 iy,
Chris@1097 983 m_params.colourScale.getPixel(value));
Chris@1097 984 }
Chris@1121 985
Chris@1121 986 m_magRanges.push_back(magRange);
Chris@1097 987 }
Chris@1097 988
Chris@1097 989 double fractionComplete = double(columnCount) / double(w);
Chris@1097 990 if (timer.outOfTime(fractionComplete)) {
Chris@1097 991 return columnCount;
Chris@1097 992 }
Chris@1097 993 }
Chris@1097 994
Chris@1097 995 return columnCount;
Chris@1097 996 }
Chris@1097 997
Chris@1079 998 void
Chris@1095 999 Colour3DPlotRenderer::recreateDrawBuffer(int w, int h)
Chris@1079 1000 {
Chris@1095 1001 m_drawBuffer = QImage(w, h, QImage::Format_Indexed8);
Chris@1079 1002
Chris@1095 1003 for (int pixel = 0; pixel < 256; ++pixel) {
Chris@1095 1004 m_drawBuffer.setColor
Chris@1095 1005 ((unsigned char)pixel,
Chris@1112 1006 m_params.colourScale.getColourForPixel
Chris@1112 1007 (pixel, m_params.colourRotation).rgb());
Chris@1079 1008 }
Chris@1079 1009
Chris@1079 1010 m_drawBuffer.fill(0);
Chris@1121 1011 m_magRanges.clear();
Chris@1079 1012 }
Chris@1079 1013
Chris@1095 1014 void
Chris@1095 1015 Colour3DPlotRenderer::clearDrawBuffer(int w, int h)
Chris@1095 1016 {
Chris@1095 1017 if (m_drawBuffer.width() < w || m_drawBuffer.height() != h) {
Chris@1095 1018 recreateDrawBuffer(w, h);
Chris@1095 1019 } else {
Chris@1095 1020 m_drawBuffer.fill(0);
Chris@1121 1021 m_magRanges.clear();
Chris@1095 1022 }
Chris@1095 1023 }
Chris@1079 1024
Chris@1095 1025