annotate layer/SliceLayer.cpp @ 280:3c402c6052f6

* Pull peak-picker out of SpectrumLayer and into FFTModel; use combined peak-picker and frequency estimator for SpectrogramLayer (makes the peak frequency spectrogram a bit quicker) * Add more information to spectrum and spectrogram crosshairs
author Chris Cannam
date Wed, 04 Jul 2007 15:29:16 +0000
parents b9380f679f70
children 1284955856ab
rev   line source
Chris@133 1
Chris@133 2 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@133 3
Chris@133 4 /*
Chris@133 5 Sonic Visualiser
Chris@133 6 An audio file viewer and annotation editor.
Chris@133 7 Centre for Digital Music, Queen Mary, University of London.
Chris@195 8 This file copyright 2006-2007 QMUL.
Chris@133 9
Chris@133 10 This program is free software; you can redistribute it and/or
Chris@133 11 modify it under the terms of the GNU General Public License as
Chris@133 12 published by the Free Software Foundation; either version 2 of the
Chris@133 13 License, or (at your option) any later version. See the file
Chris@133 14 COPYING included with this distribution for more information.
Chris@133 15 */
Chris@133 16
Chris@193 17 #include "SliceLayer.h"
Chris@133 18
Chris@133 19 #include "view/View.h"
Chris@153 20 #include "base/AudioLevel.h"
Chris@167 21 #include "base/RangeMapper.h"
Chris@195 22 #include "base/RealTime.h"
Chris@195 23
Chris@197 24 #include "ColourMapper.h"
Chris@195 25 #include "PaintAssistant.h"
Chris@133 26
Chris@133 27 #include <QPainter>
Chris@133 28 #include <QPainterPath>
Chris@133 29
Chris@193 30 SliceLayer::SliceLayer() :
Chris@193 31 m_sliceableModel(0),
Chris@153 32 m_colour(Qt::darkBlue),
Chris@197 33 m_colourMap(0),
Chris@153 34 m_energyScale(dBScale),
Chris@198 35 m_samplingMode(SampleMean),
Chris@195 36 m_plotStyle(PlotSteps),
Chris@193 37 m_binScale(LinearBins),
Chris@153 38 m_normalize(false),
Chris@198 39 m_gain(1.0),
Chris@198 40 m_currentf0(0),
Chris@198 41 m_currentf1(0)
Chris@133 42 {
Chris@133 43 }
Chris@133 44
Chris@193 45 SliceLayer::~SliceLayer()
Chris@133 46 {
Chris@193 47
Chris@133 48 }
Chris@133 49
Chris@133 50 void
Chris@193 51 SliceLayer::setSliceableModel(const Model *model)
Chris@133 52 {
Chris@193 53 const DenseThreeDimensionalModel *sliceable =
Chris@193 54 dynamic_cast<const DenseThreeDimensionalModel *>(model);
Chris@193 55
Chris@193 56 if (model && !sliceable) {
Chris@193 57 std::cerr << "WARNING: SliceLayer::setSliceableModel(" << model
Chris@193 58 << "): model is not a DenseThreeDimensionalModel" << std::endl;
Chris@193 59 }
Chris@193 60
Chris@193 61 if (m_sliceableModel == sliceable) return;
Chris@193 62
Chris@193 63 m_sliceableModel = sliceable;
Chris@193 64
Chris@193 65 connect(m_sliceableModel, SIGNAL(modelChanged()),
Chris@193 66 this, SIGNAL(modelChanged()));
Chris@193 67
Chris@193 68 connect(m_sliceableModel, SIGNAL(modelChanged(size_t, size_t)),
Chris@193 69 this, SIGNAL(modelChanged(size_t, size_t)));
Chris@193 70
Chris@193 71 connect(m_sliceableModel, SIGNAL(completionChanged()),
Chris@193 72 this, SIGNAL(modelCompletionChanged()));
Chris@193 73
Chris@193 74 emit modelReplaced();
Chris@153 75 }
Chris@153 76
Chris@153 77 void
Chris@193 78 SliceLayer::sliceableModelReplaced(const Model *orig, const Model *replacement)
Chris@153 79 {
Chris@193 80 std::cerr << "SliceLayer::sliceableModelReplaced(" << orig << ", " << replacement << ")" << std::endl;
Chris@153 81
Chris@193 82 if (orig == m_sliceableModel) {
Chris@193 83 setSliceableModel
Chris@193 84 (dynamic_cast<const DenseThreeDimensionalModel *>(replacement));
Chris@153 85 }
Chris@153 86 }
Chris@153 87
Chris@153 88 void
Chris@193 89 SliceLayer::modelAboutToBeDeleted(Model *m)
Chris@153 90 {
Chris@193 91 std::cerr << "SliceLayer::modelAboutToBeDeleted(" << m << ")" << std::endl;
Chris@153 92
Chris@193 93 if (m == m_sliceableModel) {
Chris@193 94 setSliceableModel(0);
Chris@153 95 }
Chris@133 96 }
Chris@133 97
Chris@198 98 QString
Chris@198 99 SliceLayer::getFeatureDescription(View *v, QPoint &p) const
Chris@198 100 {
Chris@199 101 int minbin, maxbin, range;
Chris@199 102 return getFeatureDescription(v, p, true, minbin, maxbin, range);
Chris@199 103 }
Chris@199 104
Chris@199 105 QString
Chris@199 106 SliceLayer::getFeatureDescription(View *v, QPoint &p,
Chris@199 107 bool includeBinDescription,
Chris@199 108 int &minbin, int &maxbin, int &range) const
Chris@199 109 {
Chris@199 110 minbin = 0;
Chris@199 111 maxbin = 0;
Chris@198 112 if (!m_sliceableModel) return "";
Chris@198 113
Chris@198 114 int xorigin = m_xorigins[v];
Chris@198 115 int w = v->width() - xorigin - 1;
Chris@198 116
Chris@198 117 int mh = m_sliceableModel->getHeight();
Chris@199 118 minbin = getBinForX(p.x() - xorigin, mh, w);
Chris@199 119 maxbin = getBinForX(p.x() - xorigin + 1, mh, w);
Chris@199 120
Chris@199 121 if (minbin >= mh) minbin = mh - 1;
Chris@199 122 if (maxbin >= mh) maxbin = mh - 1;
Chris@199 123 if (minbin < 0) minbin = 0;
Chris@199 124 if (maxbin < 0) maxbin = 0;
Chris@198 125
Chris@198 126 int sampleRate = m_sliceableModel->getSampleRate();
Chris@198 127
Chris@198 128 size_t f0 = m_currentf0;
Chris@198 129 size_t f1 = m_currentf1;
Chris@198 130
Chris@198 131 RealTime rt0 = RealTime::frame2RealTime(f0, sampleRate);
Chris@198 132 RealTime rt1 = RealTime::frame2RealTime(f1, sampleRate);
Chris@198 133
Chris@199 134 range = f1 - f0 + 1;
Chris@198 135
Chris@280 136 QString rtrangestr = QString("%1 s").arg((rt1 - rt0).toText().c_str());
Chris@280 137
Chris@199 138 if (includeBinDescription) {
Chris@198 139
Chris@199 140 float minvalue = 0.f;
Chris@248 141 if (minbin < int(m_values.size())) minvalue = m_values[minbin];
Chris@198 142
Chris@199 143 float maxvalue = minvalue;
Chris@248 144 if (maxbin < int(m_values.size())) maxvalue = m_values[maxbin];
Chris@199 145
Chris@199 146 if (minvalue > maxvalue) std::swap(minvalue, maxvalue);
Chris@199 147
Chris@199 148 QString binstr;
Chris@199 149 if (maxbin != minbin) {
Chris@199 150 binstr = tr("%1 - %2").arg(minbin+1).arg(maxbin+1);
Chris@199 151 } else {
Chris@199 152 binstr = QString("%1").arg(minbin+1);
Chris@199 153 }
Chris@199 154
Chris@199 155 QString valuestr;
Chris@199 156 if (maxvalue != minvalue) {
Chris@199 157 valuestr = tr("%1 - %2").arg(minvalue).arg(maxvalue);
Chris@199 158 } else {
Chris@199 159 valuestr = QString("%1").arg(minvalue);
Chris@199 160 }
Chris@199 161
Chris@280 162 QString description = tr("Time:\t%1 - %2\nRange:\t%3 samples (%4)\nBin:\t%5\n%6 value:\t%7")
Chris@199 163 .arg(QString::fromStdString(rt0.toText(true)))
Chris@199 164 .arg(QString::fromStdString(rt1.toText(true)))
Chris@199 165 .arg(range)
Chris@280 166 .arg(rtrangestr)
Chris@199 167 .arg(binstr)
Chris@199 168 .arg(m_samplingMode == NearestSample ? tr("First") :
Chris@199 169 m_samplingMode == SampleMean ? tr("Mean") : tr("Peak"))
Chris@199 170 .arg(valuestr);
Chris@199 171
Chris@199 172 return description;
Chris@199 173
Chris@199 174 } else {
Chris@199 175
Chris@280 176 QString description = tr("Time:\t%1 - %2\nRange:\t%3 samples (%4)")
Chris@199 177 .arg(QString::fromStdString(rt0.toText(true)))
Chris@199 178 .arg(QString::fromStdString(rt1.toText(true)))
Chris@280 179 .arg(range)
Chris@280 180 .arg(rtrangestr);
Chris@199 181
Chris@199 182 return description;
Chris@199 183 }
Chris@198 184 }
Chris@198 185
Chris@198 186 float
Chris@198 187 SliceLayer::getXForBin(int bin, int count, float w) const
Chris@198 188 {
Chris@198 189 float x = 0;
Chris@198 190
Chris@198 191 switch (m_binScale) {
Chris@198 192
Chris@198 193 case LinearBins:
Chris@198 194 x = (float(w) * bin) / count;
Chris@198 195 break;
Chris@198 196
Chris@198 197 case LogBins:
Chris@198 198 x = (float(w) * log10f(bin + 1)) / log10f(count + 1);
Chris@198 199 break;
Chris@198 200
Chris@198 201 case InvertedLogBins:
Chris@198 202 x = w - (float(w) * log10f(count - bin - 1)) / log10f(count);
Chris@198 203 break;
Chris@198 204 }
Chris@198 205
Chris@198 206 return x;
Chris@198 207 }
Chris@198 208
Chris@198 209 int
Chris@198 210 SliceLayer::getBinForX(float x, int count, float w) const
Chris@198 211 {
Chris@198 212 int bin = 0;
Chris@198 213
Chris@198 214 switch (m_binScale) {
Chris@198 215
Chris@198 216 case LinearBins:
Chris@198 217 bin = int((x * count) / w + 0.0001);
Chris@198 218 break;
Chris@198 219
Chris@198 220 case LogBins:
Chris@198 221 bin = int(powf(10.f, (x * log10f(count + 1)) / w) - 1 + 0.0001);
Chris@198 222 break;
Chris@198 223
Chris@198 224 case InvertedLogBins:
Chris@198 225 bin = count + 1 - int(powf(10.f, (log10f(count) * (w - x)) / float(w)) + 0.0001);
Chris@198 226 break;
Chris@198 227 }
Chris@198 228
Chris@198 229 return bin;
Chris@198 230 }
Chris@198 231
Chris@274 232 float
Chris@274 233 SliceLayer::getYForValue(float value, const View *v, float &norm) const
Chris@274 234 {
Chris@274 235 norm = 0.f;
Chris@274 236
Chris@274 237 if (m_yorigins.find(v) == m_yorigins.end()) return 0;
Chris@274 238
Chris@274 239 value *= m_gain;
Chris@274 240
Chris@274 241 int yorigin = m_yorigins[v];
Chris@274 242 int h = m_heights[v];
Chris@274 243 float thresh = -80.f;
Chris@274 244
Chris@274 245 float y = 0.f;
Chris@274 246
Chris@274 247 if (h <= 0) return y;
Chris@274 248
Chris@274 249 switch (m_energyScale) {
Chris@274 250
Chris@274 251 case dBScale:
Chris@274 252 {
Chris@274 253 float db = thresh;
Chris@274 254 if (value > 0.f) db = 10.f * log10f(value);
Chris@274 255 if (db < thresh) db = thresh;
Chris@274 256 norm = (db - thresh) / -thresh;
Chris@274 257 y = yorigin - (float(h) * norm);
Chris@274 258 break;
Chris@274 259 }
Chris@274 260
Chris@274 261 case MeterScale:
Chris@274 262 y = AudioLevel::multiplier_to_preview(value, h);
Chris@274 263 norm = float(y) / float(h);
Chris@274 264 y = yorigin - y;
Chris@274 265 break;
Chris@274 266
Chris@274 267 default:
Chris@274 268 norm = value;
Chris@274 269 y = yorigin - (float(h) * value);
Chris@274 270 break;
Chris@274 271 }
Chris@274 272
Chris@274 273 return y;
Chris@274 274 }
Chris@274 275
Chris@274 276 float
Chris@274 277 SliceLayer::getValueForY(float y, const View *v) const
Chris@274 278 {
Chris@274 279 float value = 0.f;
Chris@274 280
Chris@274 281 if (m_yorigins.find(v) == m_yorigins.end()) return value;
Chris@274 282
Chris@274 283 int yorigin = m_yorigins[v];
Chris@274 284 int h = m_heights[v];
Chris@274 285 float thresh = -80.f;
Chris@274 286
Chris@274 287 if (h <= 0) return value;
Chris@274 288
Chris@274 289 y = yorigin - y;
Chris@274 290
Chris@274 291 switch (m_energyScale) {
Chris@274 292
Chris@274 293 case dBScale:
Chris@274 294 {
Chris@274 295 float db = ((y / h) * -thresh) + thresh;
Chris@274 296 value = powf(10.f, db/10.f);
Chris@274 297 break;
Chris@274 298 }
Chris@274 299
Chris@274 300 case MeterScale:
Chris@274 301 value = AudioLevel::preview_to_multiplier(lrintf(y), h);
Chris@274 302 break;
Chris@274 303
Chris@274 304 default:
Chris@274 305 value = y / h;
Chris@274 306 }
Chris@274 307
Chris@274 308 return value / m_gain;
Chris@274 309 }
Chris@274 310
Chris@133 311 void
Chris@193 312 SliceLayer::paint(View *v, QPainter &paint, QRect rect) const
Chris@133 313 {
Chris@254 314 if (!m_sliceableModel || !m_sliceableModel->isOK() ||
Chris@254 315 !m_sliceableModel->isReady()) return;
Chris@133 316
Chris@195 317 paint.save();
Chris@195 318 paint.setRenderHint(QPainter::Antialiasing, false);
Chris@195 319
Chris@195 320 if (v->getViewManager() && v->getViewManager()->shouldShowScaleGuides()) {
Chris@195 321 if (!m_scalePoints.empty()) {
Chris@195 322 paint.setPen(QColor(240, 240, 240)); //!!! and dark background?
Chris@195 323 for (size_t i = 0; i < m_scalePoints.size(); ++i) {
Chris@195 324 paint.drawLine(0, m_scalePoints[i], rect.width(), m_scalePoints[i]);
Chris@195 325 }
Chris@195 326 }
Chris@195 327 }
Chris@195 328
Chris@195 329 paint.setPen(m_colour);
Chris@195 330
Chris@274 331 int xorigin = getVerticalScaleWidth(v, paint) + 1;
Chris@195 332 int w = v->width() - xorigin - 1;
Chris@198 333
Chris@198 334 m_xorigins[v] = xorigin; // for use in getFeatureDescription
Chris@133 335
Chris@195 336 int yorigin = v->height() - 20 - paint.fontMetrics().height() - 7;
Chris@195 337 int h = yorigin - paint.fontMetrics().height() - 8;
Chris@133 338
Chris@274 339 m_yorigins[v] = yorigin; // for getYForValue etc
Chris@274 340 m_heights[v] = h;
Chris@274 341
Chris@274 342 if (h <= 0) return;
Chris@274 343
Chris@133 344 QPainterPath path;
Chris@133 345
Chris@248 346 size_t mh = m_sliceableModel->getHeight();
Chris@133 347
Chris@193 348 int divisor = 0;
Chris@193 349
Chris@198 350 m_values.clear();
Chris@193 351 for (size_t bin = 0; bin < mh; ++bin) {
Chris@198 352 m_values.push_back(0.f);
Chris@193 353 }
Chris@193 354
Chris@193 355 size_t f0 = v->getCentreFrame();
Chris@193 356 int f0x = v->getXForFrame(f0);
Chris@195 357 f0 = v->getFrameForX(f0x);
Chris@193 358 size_t f1 = v->getFrameForX(f0x + 1);
Chris@195 359 if (f1 > f0) --f1;
Chris@193 360
Chris@274 361 std::cerr << "centre frame " << v->getCentreFrame() << ", x " << f0x << ", f0 " << f0 << ", f1 " << f1 << std::endl;
Chris@274 362
Chris@274 363 size_t res = m_sliceableModel->getResolution();
Chris@274 364 size_t col0 = f0 / res;
Chris@193 365 size_t col1 = col0;
Chris@274 366 if (m_samplingMode != NearestSample) col1 = f1 / res;
Chris@274 367 f0 = col0 * res;
Chris@274 368 f1 = (col1 + 1) * res - 1;
Chris@274 369
Chris@274 370 std::cerr << "resolution " << res << ", col0 " << col0 << ", col1 " << col1 << ", f0 " << f0 << ", f1 " << f1 << std::endl;
Chris@193 371
Chris@198 372 m_currentf0 = f0;
Chris@198 373 m_currentf1 = f1;
Chris@198 374
Chris@254 375 BiasCurve curve;
Chris@254 376 getBiasCurve(curve);
Chris@254 377 size_t cs = curve.size();
Chris@254 378
Chris@195 379 for (size_t col = col0; col <= col1; ++col) {
Chris@193 380 for (size_t bin = 0; bin < mh; ++bin) {
Chris@193 381 float value = m_sliceableModel->getValueAt(col, bin);
Chris@254 382 if (bin < cs) value *= curve[bin];
Chris@193 383 if (m_samplingMode == SamplePeak) {
Chris@198 384 if (value > m_values[bin]) m_values[bin] = value;
Chris@193 385 } else {
Chris@198 386 m_values[bin] += value;
Chris@193 387 }
Chris@153 388 }
Chris@193 389 ++divisor;
Chris@193 390 }
Chris@193 391
Chris@193 392 float max = 0.f;
Chris@193 393 for (size_t bin = 0; bin < mh; ++bin) {
Chris@198 394 if (m_samplingMode == SampleMean) m_values[bin] /= divisor;
Chris@198 395 if (m_values[bin] > max) max = m_values[bin];
Chris@193 396 }
Chris@193 397 if (max != 0.f && m_normalize) {
Chris@193 398 for (size_t bin = 0; bin < mh; ++bin) {
Chris@198 399 m_values[bin] /= max;
Chris@193 400 }
Chris@193 401 }
Chris@193 402
Chris@193 403 float py = 0;
Chris@193 404 float nx = xorigin;
Chris@193 405
Chris@197 406 ColourMapper mapper(m_colourMap, 0, 1);
Chris@197 407
Chris@193 408 for (size_t bin = 0; bin < mh; ++bin) {
Chris@193 409
Chris@198 410 float x = nx;
Chris@198 411 nx = xorigin + getXForBin(bin + 1, mh, w);
Chris@193 412
Chris@198 413 float value = m_values[bin];
Chris@197 414 float norm = 0.f;
Chris@274 415 float y = getYForValue(value, v, norm);
Chris@133 416
Chris@193 417 if (m_plotStyle == PlotLines) {
Chris@193 418
Chris@193 419 if (bin == 0) {
Chris@193 420 path.moveTo(x, y);
Chris@193 421 } else {
Chris@193 422 path.lineTo(x, y);
Chris@193 423 }
Chris@193 424
Chris@193 425 } else if (m_plotStyle == PlotSteps) {
Chris@193 426
Chris@193 427 if (bin == 0) {
Chris@193 428 path.moveTo(x, y);
Chris@193 429 } else {
Chris@193 430 path.lineTo(x, y);
Chris@193 431 }
Chris@193 432 path.lineTo(nx, y);
Chris@193 433
Chris@193 434 } else if (m_plotStyle == PlotBlocks) {
Chris@193 435
Chris@193 436 path.moveTo(x, yorigin);
Chris@133 437 path.lineTo(x, y);
Chris@193 438 path.lineTo(nx, y);
Chris@193 439 path.lineTo(nx, yorigin);
Chris@193 440 path.lineTo(x, yorigin);
Chris@197 441
Chris@197 442 } else if (m_plotStyle == PlotFilledBlocks) {
Chris@197 443
Chris@197 444 paint.fillRect(QRectF(x, y, nx - x, yorigin - y), mapper.map(norm));
Chris@133 445 }
Chris@193 446
Chris@193 447 py = y;
Chris@133 448 }
Chris@133 449
Chris@197 450 if (m_plotStyle != PlotFilledBlocks) {
Chris@197 451 paint.drawPath(path);
Chris@197 452 }
Chris@133 453 paint.restore();
Chris@198 454 /*
Chris@197 455 QPoint discard;
Chris@197 456
Chris@197 457 if (v->getViewManager() && v->getViewManager()->shouldShowFrameCount() &&
Chris@197 458 v->shouldIlluminateLocalFeatures(this, discard)) {
Chris@195 459
Chris@195 460 int sampleRate = m_sliceableModel->getSampleRate();
Chris@195 461
Chris@195 462 QString startText = QString("%1 / %2")
Chris@195 463 .arg(QString::fromStdString
Chris@195 464 (RealTime::frame2RealTime
Chris@195 465 (f0, sampleRate).toText(true)))
Chris@195 466 .arg(f0);
Chris@195 467
Chris@195 468 QString endText = QString(" %1 / %2")
Chris@195 469 .arg(QString::fromStdString
Chris@195 470 (RealTime::frame2RealTime
Chris@195 471 (f1, sampleRate).toText(true)))
Chris@195 472 .arg(f1);
Chris@195 473
Chris@195 474 QString durationText = QString("(%1 / %2) ")
Chris@195 475 .arg(QString::fromStdString
Chris@195 476 (RealTime::frame2RealTime
Chris@195 477 (f1 - f0 + 1, sampleRate).toText(true)))
Chris@195 478 .arg(f1 - f0 + 1);
Chris@195 479
Chris@195 480 v->drawVisibleText
Chris@195 481 (paint, xorigin + 5,
Chris@195 482 paint.fontMetrics().ascent() + 5,
Chris@195 483 startText, View::OutlinedText);
Chris@195 484
Chris@195 485 v->drawVisibleText
Chris@195 486 (paint, xorigin + 5,
Chris@195 487 paint.fontMetrics().ascent() + paint.fontMetrics().height() + 10,
Chris@195 488 endText, View::OutlinedText);
Chris@195 489
Chris@195 490 v->drawVisibleText
Chris@195 491 (paint, xorigin + 5,
Chris@195 492 paint.fontMetrics().ascent() + 2*paint.fontMetrics().height() + 15,
Chris@195 493 durationText, View::OutlinedText);
Chris@195 494 }
Chris@195 495 */
Chris@195 496 }
Chris@195 497
Chris@195 498 int
Chris@248 499 SliceLayer::getVerticalScaleWidth(View *, QPainter &paint) const
Chris@195 500 {
Chris@195 501 if (m_energyScale == LinearScale) {
Chris@220 502 return std::max(paint.fontMetrics().width("0.0") + 13,
Chris@220 503 paint.fontMetrics().width("x10-10"));
Chris@195 504 } else {
Chris@195 505 return std::max(paint.fontMetrics().width(tr("0dB")),
Chris@195 506 paint.fontMetrics().width(tr("-Inf"))) + 13;
Chris@195 507 }
Chris@195 508 }
Chris@195 509
Chris@195 510 void
Chris@195 511 SliceLayer::paintVerticalScale(View *v, QPainter &paint, QRect rect) const
Chris@195 512 {
Chris@195 513 float thresh = 0;
Chris@195 514 if (m_energyScale != LinearScale) {
Chris@195 515 thresh = AudioLevel::dB_to_multiplier(-80); //!!! thresh
Chris@195 516 }
Chris@195 517
Chris@195 518 // int h = (rect.height() * 3) / 4;
Chris@195 519 // int y = (rect.height() / 2) - (h / 2);
Chris@195 520
Chris@195 521 int yorigin = v->height() - 20 - paint.fontMetrics().height() - 6;
Chris@195 522 int h = yorigin - paint.fontMetrics().height() - 8;
Chris@195 523 if (h < 0) return;
Chris@195 524
Chris@195 525 QRect actual(rect.x(), rect.y() + yorigin - h, rect.width(), h);
Chris@195 526
Chris@220 527 int mult = 1;
Chris@220 528
Chris@195 529 PaintAssistant::paintVerticalLevelScale
Chris@195 530 (paint, actual, thresh, 1.0 / m_gain,
Chris@195 531 PaintAssistant::Scale(m_energyScale),
Chris@220 532 mult,
Chris@195 533 const_cast<std::vector<int> *>(&m_scalePoints));
Chris@220 534
Chris@220 535 if (mult != 1 && mult != 0) {
Chris@220 536 int log = lrintf(log10f(mult));
Chris@220 537 QString a = tr("x10");
Chris@220 538 QString b = QString("%1").arg(-log);
Chris@220 539 paint.drawText(3, 8 + paint.fontMetrics().ascent(), a);
Chris@220 540 paint.drawText(3 + paint.fontMetrics().width(a),
Chris@220 541 3 + paint.fontMetrics().ascent(), b);
Chris@220 542 }
Chris@133 543 }
Chris@133 544
Chris@153 545 Layer::PropertyList
Chris@193 546 SliceLayer::getProperties() const
Chris@153 547 {
Chris@153 548 PropertyList list;
Chris@153 549 list.push_back("Colour");
Chris@193 550 list.push_back("Plot Type");
Chris@198 551 // list.push_back("Sampling Mode");
Chris@153 552 list.push_back("Scale");
Chris@153 553 list.push_back("Normalize");
Chris@153 554 list.push_back("Gain");
Chris@193 555 list.push_back("Bin Scale");
Chris@153 556
Chris@153 557 return list;
Chris@153 558 }
Chris@153 559
Chris@153 560 QString
Chris@193 561 SliceLayer::getPropertyLabel(const PropertyName &name) const
Chris@153 562 {
Chris@153 563 if (name == "Colour") return tr("Colour");
Chris@193 564 if (name == "Plot Type") return tr("Plot Type");
Chris@153 565 if (name == "Energy Scale") return tr("Scale");
Chris@153 566 if (name == "Normalize") return tr("Normalize");
Chris@153 567 if (name == "Gain") return tr("Gain");
Chris@193 568 if (name == "Sampling Mode") return tr("Sampling Mode");
Chris@193 569 if (name == "Bin Scale") return tr("Plot X Scale");
Chris@153 570 return "";
Chris@153 571 }
Chris@153 572
Chris@153 573 Layer::PropertyType
Chris@193 574 SliceLayer::getPropertyType(const PropertyName &name) const
Chris@153 575 {
Chris@153 576 if (name == "Gain") return RangeProperty;
Chris@153 577 if (name == "Normalize") return ToggleProperty;
Chris@153 578 return ValueProperty;
Chris@153 579 }
Chris@153 580
Chris@153 581 QString
Chris@193 582 SliceLayer::getPropertyGroupName(const PropertyName &name) const
Chris@153 583 {
Chris@153 584 if (name == "Scale" ||
Chris@153 585 name == "Normalize" ||
Chris@193 586 name == "Sampling Mode" ||
Chris@193 587 name == "Gain") return tr("Scale");
Chris@193 588 if (name == "Plot Type" ||
Chris@193 589 name == "Bin Scale") return tr("Plot Type");
Chris@153 590 return QString();
Chris@153 591 }
Chris@153 592
Chris@153 593 int
Chris@193 594 SliceLayer::getPropertyRangeAndValue(const PropertyName &name,
Chris@216 595 int *min, int *max, int *deflt) const
Chris@153 596 {
Chris@216 597 int val = 0;
Chris@153 598
Chris@216 599 int garbage0, garbage1, garbage2;
Chris@153 600 if (!min) min = &garbage0;
Chris@153 601 if (!max) max = &garbage1;
Chris@248 602 if (!deflt) deflt = &garbage2;
Chris@153 603
Chris@153 604 if (name == "Gain") {
Chris@153 605
Chris@153 606 *min = -50;
Chris@153 607 *max = 50;
Chris@216 608 *deflt = 0;
Chris@153 609
Chris@193 610 std::cerr << "gain is " << m_gain << ", mode is " << m_samplingMode << std::endl;
Chris@193 611
Chris@216 612 val = lrint(log10(m_gain) * 20.0);
Chris@216 613 if (val < *min) val = *min;
Chris@216 614 if (val > *max) val = *max;
Chris@153 615
Chris@153 616 } else if (name == "Normalize") {
Chris@153 617
Chris@216 618 val = (m_normalize ? 1 : 0);
Chris@216 619 *deflt = 0;
Chris@153 620
Chris@153 621 } else if (name == "Colour") {
Chris@153 622
Chris@197 623 if (m_plotStyle == PlotFilledBlocks) {
Chris@197 624
Chris@197 625 *min = 0;
Chris@197 626 *max = ColourMapper::getColourMapCount() - 1;
Chris@216 627 *deflt = 0;
Chris@153 628
Chris@216 629 val = m_colourMap;
Chris@197 630
Chris@197 631 } else {
Chris@197 632
Chris@197 633 *min = 0;
Chris@197 634 *max = 5;
Chris@216 635 *deflt = 0;
Chris@197 636
Chris@216 637 if (m_colour == Qt::black) val = 0;
Chris@216 638 else if (m_colour == Qt::darkRed) val = 1;
Chris@216 639 else if (m_colour == Qt::darkBlue) val = 2;
Chris@216 640 else if (m_colour == Qt::darkGreen) val = 3;
Chris@216 641 else if (m_colour == QColor(200, 50, 255)) val = 4;
Chris@216 642 else if (m_colour == QColor(255, 150, 50)) val = 5;
Chris@197 643 }
Chris@153 644
Chris@153 645 } else if (name == "Scale") {
Chris@153 646
Chris@153 647 *min = 0;
Chris@153 648 *max = 2;
Chris@216 649 *deflt = (int)dBScale;
Chris@153 650
Chris@216 651 val = (int)m_energyScale;
Chris@153 652
Chris@193 653 } else if (name == "Sampling Mode") {
Chris@153 654
Chris@153 655 *min = 0;
Chris@193 656 *max = 2;
Chris@216 657 *deflt = (int)SampleMean;
Chris@193 658
Chris@216 659 val = (int)m_samplingMode;
Chris@153 660
Chris@193 661 } else if (name == "Plot Type") {
Chris@193 662
Chris@193 663 *min = 0;
Chris@197 664 *max = 3;
Chris@216 665 *deflt = (int)PlotSteps;
Chris@193 666
Chris@216 667 val = (int)m_plotStyle;
Chris@193 668
Chris@193 669 } else if (name == "Bin Scale") {
Chris@193 670
Chris@193 671 *min = 0;
Chris@198 672 *max = 2;
Chris@216 673 *deflt = (int)LinearBins;
Chris@198 674 // *max = 1; // I don't think we really do want to offer inverted log
Chris@193 675
Chris@216 676 val = (int)m_binScale;
Chris@193 677
Chris@153 678 } else {
Chris@216 679 val = Layer::getPropertyRangeAndValue(name, min, max, deflt);
Chris@153 680 }
Chris@153 681
Chris@216 682 return val;
Chris@153 683 }
Chris@153 684
Chris@153 685 QString
Chris@193 686 SliceLayer::getPropertyValueLabel(const PropertyName &name,
Chris@153 687 int value) const
Chris@153 688 {
Chris@153 689 if (name == "Colour") {
Chris@197 690 if (m_plotStyle == PlotFilledBlocks) {
Chris@197 691 return ColourMapper::getColourMapName(value);
Chris@197 692 } else {
Chris@197 693 switch (value) {
Chris@197 694 default:
Chris@197 695 case 0: return tr("Black");
Chris@197 696 case 1: return tr("Red");
Chris@197 697 case 2: return tr("Blue");
Chris@197 698 case 3: return tr("Green");
Chris@197 699 case 4: return tr("Purple");
Chris@197 700 case 5: return tr("Orange");
Chris@197 701 }
Chris@153 702 }
Chris@153 703 }
Chris@153 704 if (name == "Scale") {
Chris@153 705 switch (value) {
Chris@153 706 default:
Chris@153 707 case 0: return tr("Linear");
Chris@153 708 case 1: return tr("Meter");
Chris@153 709 case 2: return tr("dB");
Chris@153 710 }
Chris@153 711 }
Chris@193 712 if (name == "Sampling Mode") {
Chris@153 713 switch (value) {
Chris@153 714 default:
Chris@193 715 case 0: return tr("Any");
Chris@193 716 case 1: return tr("Mean");
Chris@193 717 case 2: return tr("Peak");
Chris@193 718 }
Chris@193 719 }
Chris@193 720 if (name == "Plot Type") {
Chris@193 721 switch (value) {
Chris@193 722 default:
Chris@193 723 case 0: return tr("Lines");
Chris@193 724 case 1: return tr("Steps");
Chris@193 725 case 2: return tr("Blocks");
Chris@197 726 case 3: return tr("Colours");
Chris@193 727 }
Chris@193 728 }
Chris@193 729 if (name == "Bin Scale") {
Chris@193 730 switch (value) {
Chris@193 731 default:
Chris@198 732 case 0: return tr("Linear Bins");
Chris@198 733 case 1: return tr("Log Bins");
Chris@198 734 case 2: return tr("Rev Log Bins");
Chris@153 735 }
Chris@153 736 }
Chris@153 737 return tr("<unknown>");
Chris@153 738 }
Chris@153 739
Chris@167 740 RangeMapper *
Chris@193 741 SliceLayer::getNewPropertyRangeMapper(const PropertyName &name) const
Chris@167 742 {
Chris@167 743 if (name == "Gain") {
Chris@167 744 return new LinearRangeMapper(-50, 50, -25, 25, tr("dB"));
Chris@167 745 }
Chris@167 746 return 0;
Chris@167 747 }
Chris@167 748
Chris@133 749 void
Chris@193 750 SliceLayer::setProperty(const PropertyName &name, int value)
Chris@133 751 {
Chris@153 752 if (name == "Gain") {
Chris@153 753 setGain(pow(10, float(value)/20.0));
Chris@153 754 } else if (name == "Colour") {
Chris@197 755 if (m_plotStyle == PlotFilledBlocks) {
Chris@197 756 setFillColourMap(value);
Chris@197 757 } else {
Chris@197 758 switch (value) {
Chris@197 759 default:
Chris@197 760 case 0: setBaseColour(Qt::black); break;
Chris@197 761 case 1: setBaseColour(Qt::darkRed); break;
Chris@197 762 case 2: setBaseColour(Qt::darkBlue); break;
Chris@197 763 case 3: setBaseColour(Qt::darkGreen); break;
Chris@197 764 case 4: setBaseColour(QColor(200, 50, 255)); break;
Chris@197 765 case 5: setBaseColour(QColor(255, 150, 50)); break;
Chris@197 766 }
Chris@153 767 }
Chris@153 768 } else if (name == "Scale") {
Chris@153 769 switch (value) {
Chris@153 770 default:
Chris@153 771 case 0: setEnergyScale(LinearScale); break;
Chris@153 772 case 1: setEnergyScale(MeterScale); break;
Chris@153 773 case 2: setEnergyScale(dBScale); break;
Chris@153 774 }
Chris@193 775 } else if (name == "Plot Type") {
Chris@193 776 setPlotStyle(PlotStyle(value));
Chris@193 777 } else if (name == "Sampling Mode") {
Chris@193 778 switch (value) {
Chris@193 779 default:
Chris@193 780 case 0: setSamplingMode(NearestSample); break;
Chris@193 781 case 1: setSamplingMode(SampleMean); break;
Chris@193 782 case 2: setSamplingMode(SamplePeak); break;
Chris@193 783 }
Chris@193 784 } else if (name == "Bin Scale") {
Chris@193 785 switch (value) {
Chris@193 786 default:
Chris@193 787 case 0: setBinScale(LinearBins); break;
Chris@193 788 case 1: setBinScale(LogBins); break;
Chris@193 789 case 2: setBinScale(InvertedLogBins); break;
Chris@193 790 }
Chris@153 791 } else if (name == "Normalize") {
Chris@153 792 setNormalize(value ? true : false);
Chris@153 793 }
Chris@153 794 }
Chris@153 795
Chris@153 796 void
Chris@193 797 SliceLayer::setBaseColour(QColor colour)
Chris@153 798 {
Chris@153 799 if (m_colour == colour) return;
Chris@153 800 m_colour = colour;
Chris@153 801 emit layerParametersChanged();
Chris@153 802 }
Chris@153 803
Chris@153 804 void
Chris@197 805 SliceLayer::setFillColourMap(int map)
Chris@197 806 {
Chris@197 807 if (m_colourMap == map) return;
Chris@197 808 m_colourMap = map;
Chris@197 809 emit layerParametersChanged();
Chris@197 810 }
Chris@197 811
Chris@197 812 void
Chris@193 813 SliceLayer::setEnergyScale(EnergyScale scale)
Chris@153 814 {
Chris@153 815 if (m_energyScale == scale) return;
Chris@153 816 m_energyScale = scale;
Chris@153 817 emit layerParametersChanged();
Chris@153 818 }
Chris@153 819
Chris@153 820 void
Chris@193 821 SliceLayer::setSamplingMode(SamplingMode mode)
Chris@153 822 {
Chris@193 823 if (m_samplingMode == mode) return;
Chris@193 824 m_samplingMode = mode;
Chris@153 825 emit layerParametersChanged();
Chris@153 826 }
Chris@153 827
Chris@153 828 void
Chris@193 829 SliceLayer::setPlotStyle(PlotStyle style)
Chris@153 830 {
Chris@193 831 if (m_plotStyle == style) return;
Chris@197 832 bool colourTypeChanged = (style == PlotFilledBlocks ||
Chris@197 833 m_plotStyle == PlotFilledBlocks);
Chris@193 834 m_plotStyle = style;
Chris@197 835 if (colourTypeChanged) {
Chris@197 836 emit layerParameterRangesChanged();
Chris@197 837 }
Chris@153 838 emit layerParametersChanged();
Chris@153 839 }
Chris@153 840
Chris@153 841 void
Chris@193 842 SliceLayer::setBinScale(BinScale scale)
Chris@153 843 {
Chris@193 844 if (m_binScale == scale) return;
Chris@193 845 m_binScale = scale;
Chris@153 846 emit layerParametersChanged();
Chris@153 847 }
Chris@153 848
Chris@153 849 void
Chris@193 850 SliceLayer::setNormalize(bool n)
Chris@153 851 {
Chris@153 852 if (m_normalize == n) return;
Chris@153 853 m_normalize = n;
Chris@153 854 emit layerParametersChanged();
Chris@153 855 }
Chris@153 856
Chris@153 857 void
Chris@193 858 SliceLayer::setGain(float gain)
Chris@153 859 {
Chris@153 860 if (m_gain == gain) return;
Chris@153 861 m_gain = gain;
Chris@153 862 emit layerParametersChanged();
Chris@153 863 }
Chris@153 864
Chris@153 865 QString
Chris@193 866 SliceLayer::toXmlString(QString indent, QString extraAttributes) const
Chris@153 867 {
Chris@153 868 QString s;
Chris@153 869
Chris@153 870 s += QString("colour=\"%1\" "
Chris@197 871 "colourScheme=\"%2\" "
Chris@197 872 "energyScale=\"%3\" "
Chris@197 873 "samplingMode=\"%4\" "
Chris@197 874 "gain=\"%5\" "
Chris@197 875 "normalize=\"%6\"")
Chris@153 876 .arg(encodeColour(m_colour))
Chris@197 877 .arg(m_colourMap)
Chris@153 878 .arg(m_energyScale)
Chris@193 879 .arg(m_samplingMode)
Chris@153 880 .arg(m_gain)
Chris@153 881 .arg(m_normalize ? "true" : "false");
Chris@153 882
Chris@153 883 return Layer::toXmlString(indent, extraAttributes + " " + s);
Chris@153 884 }
Chris@153 885
Chris@153 886 void
Chris@193 887 SliceLayer::setProperties(const QXmlAttributes &attributes)
Chris@153 888 {
Chris@153 889 bool ok = false;
Chris@153 890
Chris@153 891 QString colourSpec = attributes.value("colour");
Chris@153 892 if (colourSpec != "") {
Chris@153 893 QColor colour(colourSpec);
Chris@153 894 if (colour.isValid()) {
Chris@153 895 setBaseColour(QColor(colourSpec));
Chris@153 896 }
Chris@153 897 }
Chris@153 898
Chris@153 899 EnergyScale scale = (EnergyScale)
Chris@153 900 attributes.value("energyScale").toInt(&ok);
Chris@153 901 if (ok) setEnergyScale(scale);
Chris@153 902
Chris@193 903 SamplingMode mode = (SamplingMode)
Chris@193 904 attributes.value("samplingMode").toInt(&ok);
Chris@193 905 if (ok) setSamplingMode(mode);
Chris@153 906
Chris@197 907 int colourMap = attributes.value("colourScheme").toInt(&ok);
Chris@197 908 if (ok) setFillColourMap(colourMap);
Chris@197 909
Chris@153 910 float gain = attributes.value("gain").toFloat(&ok);
Chris@153 911 if (ok) setGain(gain);
Chris@153 912
Chris@153 913 bool normalize = (attributes.value("normalize").trimmed() == "true");
Chris@153 914 setNormalize(normalize);
Chris@133 915 }
Chris@133 916
Chris@133 917 bool
Chris@248 918 SliceLayer::getValueExtents(float &, float &, bool &, QString &) const
Chris@133 919 {
Chris@133 920 return false;
Chris@133 921 }
Chris@133 922