git.sesse.net Git - kdenlive/blob - src/vectorscope.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2010 by Simon Andreas Eugster (simon.eu@gmail.com)      *
   3  *   This file is part of kdenlive. See www.kdenlive.org.                  *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  ***************************************************************************/
  10
  11
  12 #include <QAction>
  13 #include <QColor>
  14 #include <QDebug>
  15 #include <QMouseEvent>
  16 #include <QMenu>
  17 #include <QPainter>
  18
  19 #include <QTime>
  20
  21 #include "colorplaneexport.h"
  22 #include "colortools.h"
  23 #include "renderer.h"
  24 #include "vectorscope.h"
  25 #include "vectorscopegenerator.h"
  26
  27 const float P75 = .75;
  28 const unsigned char DEFAULT_Y = 255;
  29
  30 const QPointF YUV_R(-.147,  .615);
  31 const QPointF YUV_G(-.289, -.515);
  32 const QPointF YUV_B(.437, -.100);
  33 const QPointF YUV_Cy(.147, -.615);
  34 const QPointF YUV_Mg(.289,  .515);
  35 const QPointF YUV_Yl(-.437,  .100);
  36
  37 const QPointF YPbPr_R(-.169, .5);
  38 const QPointF YPbPr_G(-.331, -.419);
  39 const QPointF YPbPr_B(.5, -.081);
  40 const QPointF YPbPr_Cy(.169, -.5);
  41 const QPointF YPbPr_Mg(.331, .419);
  42 const QPointF YPbPr_Yl(-.5, .081);
  43
  44
  45 Vectorscope::Vectorscope(Monitor *projMonitor, Monitor *clipMonitor, QWidget *parent) :
  46     AbstractScopeWidget(projMonitor, clipMonitor, parent),
  47     m_gain(1)
  48 {
  49     ui = new Ui::Vectorscope_UI();
  50     ui->setupUi(this);
  51
  52     m_colorTools = new ColorTools();
  53     m_colorPlaneExport = new ColorPlaneExport(this);
  54     m_vectorscopeGenerator = new VectorscopeGenerator();
  55
  56     ui->paintMode->addItem(i18n("Green 2"), QVariant(VectorscopeGenerator::PaintMode_Green2));
  57     ui->paintMode->addItem(i18n("Green"), QVariant(VectorscopeGenerator::PaintMode_Green));
  58     ui->paintMode->addItem(i18n("Black"), QVariant(VectorscopeGenerator::PaintMode_Black));
  59     ui->paintMode->addItem(i18n("Modified YUV (Chroma)"), QVariant(VectorscopeGenerator::PaintMode_Chroma));
  60     ui->paintMode->addItem(i18n("YUV"), QVariant(VectorscopeGenerator::PaintMode_YUV));
  61     ui->paintMode->addItem(i18n("Original Color"), QVariant(VectorscopeGenerator::PaintMode_Original));
  62
  63     ui->backgroundMode->addItem(i18n("None"), QVariant(BG_NONE));
  64     ui->backgroundMode->addItem(i18n("YUV"), QVariant(BG_YUV));
  65     ui->backgroundMode->addItem(i18n("Modified YUV (Chroma)"), QVariant(BG_CHROMA));
  66     ui->backgroundMode->addItem("YPbPr", QVariant(BG_YPbPr)); // TODO translate after 0.7.8 release
  67
  68     ui->sliderGain->setMinimum(0);
  69     ui->sliderGain->setMaximum(40);
  70
  71     bool b = true;
  72     b &= connect(ui->backgroundMode, SIGNAL(currentIndexChanged(int)), this, SLOT(slotBackgroundChanged()));
  73     b &= connect(ui->sliderGain, SIGNAL(valueChanged(int)), this, SLOT(slotGainChanged(int)));
  74     b &= connect(ui->paintMode, SIGNAL(currentIndexChanged(int)), this, SLOT(forceUpdateScope()));
  75     b &= connect(this, SIGNAL(signalMousePositionChanged()), this, SLOT(forceUpdateHUD()));
  76     ui->sliderGain->setValue(0);
  77
  78
  79     ///// Build context menu /////
  80
  81     m_menu->addSeparator();
  82
  83     m_aExportBackground = new QAction(i18n("Export background"), this);
  84     m_menu->addAction(m_aExportBackground);
  85     b &= connect(m_aExportBackground, SIGNAL(triggered()), this, SLOT(slotExportBackground()));
  86
  87     m_a75PBox = new QAction(i18n("75% box"), this);
  88     m_a75PBox->setCheckable(true);
  89     m_menu->addAction(m_a75PBox);
  90     b &= connect(m_a75PBox, SIGNAL(changed()), this, SLOT(forceUpdateBackground()));
  91
  92     m_aAxisEnabled = new QAction(i18n("Draw axis"), this);
  93     m_aAxisEnabled->setCheckable(true);
  94     m_menu->addAction(m_aAxisEnabled);
  95     b &= connect(m_aAxisEnabled, SIGNAL(changed()), this, SLOT(forceUpdateBackground()));
  96
  97     // TODO use i18n after the 0.7.8 release
  98     m_menu->addSeparator()->setText("Color Space");
  99     m_aColorSpace_YPbPr = new QAction("YPbPr", this);
 100     m_aColorSpace_YPbPr->setCheckable(true);
 101     m_aColorSpace_YUV = new QAction("YUV", this);
 102     m_aColorSpace_YUV->setCheckable(true);
 103     m_agColorSpace = new QActionGroup(this);
 104     m_agColorSpace->addAction(m_aColorSpace_YPbPr);
 105     m_agColorSpace->addAction(m_aColorSpace_YUV);
 106     m_menu->addAction(m_aColorSpace_YPbPr);
 107     m_menu->addAction(m_aColorSpace_YUV);
 108     b &= connect(m_aColorSpace_YPbPr, SIGNAL(toggled(bool)), this, SLOT(slotColorSpaceChanged()));
 109     b &= connect(m_aColorSpace_YUV, SIGNAL(toggled(bool)), this, SLOT(slotColorSpaceChanged()));
 110
 111     Q_ASSERT(b);
 112
 113     // To make the 1.0x text show
 114     slotGainChanged(ui->sliderGain->value());
 115
 116     init();
 117 }
 118
 119 Vectorscope::~Vectorscope()
 120 {
 121     writeConfig();
 122
 123     delete m_colorTools;
 124     delete m_colorPlaneExport;
 125     delete m_vectorscopeGenerator;
 126
 127     delete m_aColorSpace_YPbPr;
 128     delete m_aColorSpace_YUV;
 129     delete m_aExportBackground;
 130     delete m_aAxisEnabled;
 131     delete m_a75PBox;
 132     delete m_agColorSpace;
 133 }
 134
 135 QString Vectorscope::widgetName() const { return QString("Vectorscope"); }
 136
 137 void Vectorscope::readConfig()
 138 {
 139     AbstractScopeWidget::readConfig();
 140
 141     KSharedConfigPtr config = KGlobal::config();
 142     KConfigGroup scopeConfig(config, configName());
 143     m_a75PBox->setChecked(scopeConfig.readEntry("75PBox", false));
 144     m_aAxisEnabled->setChecked(scopeConfig.readEntry("axis", false));
 145     ui->backgroundMode->setCurrentIndex(scopeConfig.readEntry("backgroundmode").toInt());
 146     ui->paintMode->setCurrentIndex(scopeConfig.readEntry("paintmode").toInt());
 147     ui->sliderGain->setValue(scopeConfig.readEntry("gain", 1));
 148     m_aColorSpace_YPbPr->setChecked(scopeConfig.readEntry("colorspace_ypbpr", false));
 149     m_aColorSpace_YUV->setChecked(!m_aColorSpace_YPbPr->isChecked());
 150 }
 151
 152 void Vectorscope::writeConfig()
 153 {
 154     KSharedConfigPtr config = KGlobal::config();
 155     KConfigGroup scopeConfig(config, configName());
 156     scopeConfig.writeEntry("75PBox", m_a75PBox->isChecked());
 157     scopeConfig.writeEntry("axis", m_aAxisEnabled->isChecked());
 158     scopeConfig.writeEntry("backgroundmode", ui->backgroundMode->currentIndex());
 159     scopeConfig.writeEntry("paintmode", ui->paintMode->currentIndex());
 160     scopeConfig.writeEntry("gain", ui->sliderGain->value());
 161     scopeConfig.writeEntry("colorspace_ypbpr", m_aColorSpace_YPbPr->isChecked());
 162     scopeConfig.sync();
 163 }
 164
 165 QRect Vectorscope::scopeRect()
 166 {
 167     // Distance from top/left/right
 168     int offset = 6;
 169
 170     // We want to paint below the controls area. The line is the lowest element.
 171     QPoint topleft(offset, ui->verticalSpacer->geometry().y()+offset);
 172     QPoint bottomright(ui->horizontalSpacer->geometry().right()-offset, this->size().height()-offset);
 173
 174     QRect scopeRect(topleft, bottomright);
 175
 176     // Circle Width: min of width and height
 177     cw = (scopeRect.height() < scopeRect.width()) ? scopeRect.height() : scopeRect.width();
 178     scopeRect.setWidth(cw);
 179     scopeRect.setHeight(cw);
 180
 181
 182     m_centerPoint = m_vectorscopeGenerator->mapToCircle(scopeRect.size(), QPointF(0,0));
 183     pR75 = m_vectorscopeGenerator->mapToCircle(scopeRect.size(), P75*VectorscopeGenerator::scaling*YUV_R);
 184     pG75 = m_vectorscopeGenerator->mapToCircle(scopeRect.size(), P75*VectorscopeGenerator::scaling*YUV_G);
 185     pB75 = m_vectorscopeGenerator->mapToCircle(scopeRect.size(), P75*VectorscopeGenerator::scaling*YUV_B);
 186     pCy75 = m_vectorscopeGenerator->mapToCircle(scopeRect.size(), P75*VectorscopeGenerator::scaling*YUV_Cy);
 187     pMg75 = m_vectorscopeGenerator->mapToCircle(scopeRect.size(), P75*VectorscopeGenerator::scaling*YUV_Mg);
 188     pYl75 = m_vectorscopeGenerator->mapToCircle(scopeRect.size(), P75*VectorscopeGenerator::scaling*YUV_Yl);
 189
 190     return scopeRect;
 191 }
 192
 193 bool Vectorscope::isHUDDependingOnInput() const { return false; }
 194 bool Vectorscope::isScopeDependingOnInput() const { return true; }
 195 bool Vectorscope::isBackgroundDependingOnInput() const { return false; }
 196
 197 QImage Vectorscope::renderHUD(uint)
 198 {
 199
 200     QImage hud;
 201
 202     if (m_mouseWithinWidget) {
 203         // Mouse moved: Draw a circle over the scope
 204
 205         hud = QImage(m_scopeRect.size(), QImage::Format_ARGB32);
 206         hud.fill(qRgba(0, 0, 0, 0));
 207
 208         QPainter davinci(&hud);
 209         QPoint widgetCenterPoint = m_scopeRect.topLeft() + m_centerPoint;
 210
 211         int dx = -widgetCenterPoint.x()+m_mousePos.x();
 212         int dy =  widgetCenterPoint.y()-m_mousePos.y();
 213
 214         QPoint reference = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), QPointF(1,0));
 215
 216         float r = sqrt(dx*dx + dy*dy);
 217         float percent = (float) 100*r/VectorscopeGenerator::scaling/m_gain/(reference.x() - widgetCenterPoint.x());
 218
 219         switch (ui->backgroundMode->itemData(ui->backgroundMode->currentIndex()).toInt()) {
 220         case BG_NONE:
 221             davinci.setPen(penLight);
 222             break;
 223         default:
 224             if (r > cw/2) {
 225                 davinci.setPen(penLight);
 226             } else {
 227                 davinci.setPen(penDark);
 228             }
 229             break;
 230         }
 231         davinci.drawEllipse(m_centerPoint, (int)r, (int)r);
 232         davinci.setPen(penThin);
 233         davinci.drawText(QPoint(m_scopeRect.width()-40, m_scopeRect.height()), i18n("%1 \%", QString::number(percent, 'f', 0)));
 234
 235         float angle = copysign(acos(dx/r)*180/M_PI, dy);
 236         davinci.drawText(QPoint(10, m_scopeRect.height()), i18n("%1°", QString::number(angle, 'f', 1)));
 237
 238 //        m_circleEnabled = false;
 239     } else {
 240         hud = QImage(0, 0, QImage::Format_ARGB32);
 241     }
 242     emit signalHUDRenderingFinished(0, 1);
 243     return hud;
 244 }
 245
 246 QImage Vectorscope::renderScope(uint accelerationFactor, const QImage qimage)
 247 {
 248     QTime start = QTime::currentTime();
 249     QImage scope;
 250
 251     if (cw <= 0) {
 252         qDebug() << "Scope size not known yet. Aborting.";
 253     } else {
 254
 255         VectorscopeGenerator::ColorSpace colorSpace = m_aColorSpace_YPbPr->isChecked() ?
 256                                                       VectorscopeGenerator::ColorSpace_YPbPr : VectorscopeGenerator::ColorSpace_YUV;
 257         VectorscopeGenerator::PaintMode paintMode = (VectorscopeGenerator::PaintMode) ui->paintMode->itemData(ui->paintMode->currentIndex()).toInt();
 258         scope = m_vectorscopeGenerator->calculateVectorscope(m_scopeRect.size(),
 259                                                              qimage,
 260                                                              m_gain, paintMode, colorSpace,
 261                                                              m_aAxisEnabled->isChecked(), accelerationFactor);
 262
 263     }
 264
 265     unsigned int mseconds = start.msecsTo(QTime::currentTime());
 266     emit signalScopeRenderingFinished(mseconds, accelerationFactor);
 267     return scope;
 268 }
 269
 270 QImage Vectorscope::renderBackground(uint)
 271 {
 272     QTime start = QTime::currentTime();
 273     start.start();
 274
 275     QImage bg;
 276     switch (ui->backgroundMode->itemData(ui->backgroundMode->currentIndex()).toInt()) {
 277     case BG_YUV:
 278         qDebug() << "YUV background.";
 279         bg = m_colorTools->yuvColorWheel(m_scopeRect.size(), (unsigned char) 128, 1/VectorscopeGenerator::scaling, false, true);
 280         break;
 281     case BG_CHROMA:
 282         bg = m_colorTools->yuvColorWheel(m_scopeRect.size(), (unsigned char) 255, 1/VectorscopeGenerator::scaling, true, true);
 283         break;
 284     case BG_YPbPr:
 285         bg = m_colorTools->yPbPrColorWheel(m_scopeRect.size(), (unsigned char) 128, 1/VectorscopeGenerator::scaling, true);
 286         break;
 287     default:
 288         qDebug() << "No background.";
 289         bg = QImage(cw, cw, QImage::Format_ARGB32);
 290         bg.fill(qRgba(0,0,0,0));
 291         break;
 292     }
 293
 294
 295     // Draw the vectorscope circle
 296     QPainter davinci(&bg);
 297     QPoint vinciPoint;
 298
 299
 300     davinci.setRenderHint(QPainter::Antialiasing, true);
 301
 302     davinci.setPen(penThick);
 303     davinci.drawEllipse(0, 0, cw, cw);
 304
 305     // Draw RGB/CMY points with 100% chroma
 306     if (m_aColorSpace_YUV->isChecked()) {
 307         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YUV_R);
 308         davinci.drawEllipse(vinciPoint, 4,4);
 309         davinci.drawText(vinciPoint-QPoint(20, -10), "R");
 310
 311         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YUV_G);
 312         davinci.drawEllipse(vinciPoint, 4,4);
 313         davinci.drawText(vinciPoint-QPoint(20, 0), "G");
 314
 315         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YUV_B);
 316         davinci.drawEllipse(vinciPoint, 4,4);
 317         davinci.drawText(vinciPoint+QPoint(15, 10), "B");
 318
 319         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YUV_Cy);
 320         davinci.drawEllipse(vinciPoint, 4,4);
 321         davinci.drawText(vinciPoint+QPoint(15, -5), "Cy");
 322
 323         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YUV_Mg);
 324         davinci.drawEllipse(vinciPoint, 4,4);
 325         davinci.drawText(vinciPoint+QPoint(15, 10), "Mg");
 326
 327         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YUV_Yl);
 328         davinci.drawEllipse(vinciPoint, 4,4);
 329         davinci.drawText(vinciPoint-QPoint(25, 0), "Yl");
 330     } else {
 331         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YPbPr_R);
 332         davinci.drawEllipse(vinciPoint, 4,4);
 333         davinci.drawText(vinciPoint-QPoint(20, -10), "R");
 334
 335         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YPbPr_G);
 336         davinci.drawEllipse(vinciPoint, 4,4);
 337         davinci.drawText(vinciPoint-QPoint(20, 0), "G");
 338
 339         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YPbPr_B);
 340         davinci.drawEllipse(vinciPoint, 4,4);
 341         davinci.drawText(vinciPoint+QPoint(15, 10), "B");
 342
 343         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YPbPr_Cy);
 344         davinci.drawEllipse(vinciPoint, 4,4);
 345         davinci.drawText(vinciPoint+QPoint(15, -5), "Cy");
 346
 347         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YPbPr_Mg);
 348         davinci.drawEllipse(vinciPoint, 4,4);
 349         davinci.drawText(vinciPoint+QPoint(15, 10), "Mg");
 350
 351         vinciPoint = m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), VectorscopeGenerator::scaling*YPbPr_Yl);
 352         davinci.drawEllipse(vinciPoint, 4,4);
 353         davinci.drawText(vinciPoint-QPoint(25, 0), "Yl");
 354     }
 355
 356     switch (ui->backgroundMode->itemData(ui->backgroundMode->currentIndex()).toInt()) {
 357     case BG_NONE:
 358         davinci.setPen(penLight);
 359         break;
 360     default:
 361         davinci.setPen(penDark);
 362         break;
 363     }
 364
 365     // Draw axis
 366     if (m_aAxisEnabled->isChecked()) {
 367         davinci.drawLine(m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), QPointF(0,-.9)), m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), QPointF(0,.9)));
 368         davinci.drawLine(m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), QPointF(-.9,0)), m_vectorscopeGenerator->mapToCircle(m_scopeRect.size(), QPointF(.9,0)));
 369     }
 370
 371     // Draw center point
 372     switch (ui->backgroundMode->itemData(ui->backgroundMode->currentIndex()).toInt()) {
 373     case BG_CHROMA:
 374         davinci.setPen(penDark);
 375         break;
 376     default:
 377         davinci.setPen(penThin);
 378         break;
 379     }
 380     davinci.drawEllipse(m_centerPoint, 5,5);
 381
 382
 383     // Draw 75% box
 384     if (m_a75PBox->isChecked()) {
 385         davinci.drawLine(pR75, pYl75);
 386         davinci.drawLine(pYl75, pG75);
 387         davinci.drawLine(pG75, pCy75);
 388         davinci.drawLine(pCy75, pB75);
 389         davinci.drawLine(pB75, pMg75);
 390         davinci.drawLine(pMg75, pR75);
 391     }
 392
 393     // Draw RGB/CMY points with 75% chroma (for NTSC)
 394     davinci.setPen(penThin);
 395     davinci.drawEllipse(pR75, 3,3);
 396     davinci.drawEllipse(pG75, 3,3);
 397     davinci.drawEllipse(pB75, 3,3);
 398     davinci.drawEllipse(pCy75, 3,3);
 399     davinci.drawEllipse(pMg75, 3,3);
 400     davinci.drawEllipse(pYl75, 3,3);
 401
 402     // Draw realtime factor (number of skipped pixels)
 403     if (m_aRealtime->isChecked()) {
 404         davinci.setPen(penThin);
 405         davinci.drawText(QPoint(m_scopeRect.width()-40, m_scopeRect.height()-15), QVariant(m_accelFactorScope).toString().append("x"));
 406     }
 407
 408     emit signalBackgroundRenderingFinished(start.elapsed(), 1);
 409     return bg;
 410 }
 411
 412
 413
 414
 415 ///// Slots /////
 416
 417 void Vectorscope::slotGainChanged(int newval)
 418 {
 419     m_gain = 1 + (float)newval/10;
 420     ui->lblGain->setText(QString::number(m_gain, 'f', 1) + "x");
 421     forceUpdateScope();
 422 }
 423
 424 void Vectorscope::slotExportBackground()
 425 {
 426     qDebug() << "Exporting background";
 427     m_colorPlaneExport->show();
 428
 429 }
 430
 431 void Vectorscope::slotBackgroundChanged()
 432 {
 433     // Background changed, switch to a suitable color mode now
 434     int index;
 435     switch (ui->backgroundMode->itemData(ui->backgroundMode->currentIndex()).toInt()) {
 436     case BG_YUV:
 437         index = ui->paintMode->findData(QVariant(VectorscopeGenerator::PaintMode_Black));
 438         if (index >= 0) {
 439             ui->paintMode->setCurrentIndex(index);
 440         }
 441         break;
 442
 443     case BG_NONE:
 444         if (ui->paintMode->itemData(ui->paintMode->currentIndex()).toInt() == VectorscopeGenerator::PaintMode_Black) {
 445             index = ui->paintMode->findData(QVariant(VectorscopeGenerator::PaintMode_Green2));
 446             ui->paintMode->setCurrentIndex(index);
 447         }
 448         break;
 449     }
 450     forceUpdateBackground();
 451 }
 452
 453 void Vectorscope::slotColorSpaceChanged()
 454 {
 455     int index;
 456     if (m_aColorSpace_YPbPr->isChecked()) {
 457         if (ui->backgroundMode->itemData(ui->backgroundMode->currentIndex()).toInt() == BG_YUV) {
 458             index = ui->backgroundMode->findData(QVariant(BG_YPbPr));
 459             if (index >= 0) {
 460                 ui->backgroundMode->setCurrentIndex(index);
 461             }
 462         }
 463     } else {
 464         if (ui->backgroundMode->itemData(ui->backgroundMode->currentIndex()).toInt() == BG_YPbPr) {
 465             index = ui->backgroundMode->findData(QVariant(BG_YUV));
 466             if (index >= 0) {
 467                 ui->backgroundMode->setCurrentIndex(index);
 468             }
 469         }
 470     }
 471     forceUpdate();
 472 }