git.sesse.net Git - kdenlive/blob - src/colorscopes/histogram.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 #include <QMenu>
  12 #include <QTime>
  13 #include "histogramgenerator.h"
  14 #include "histogram.h"
  15 #include "renderer.h"
  16
  17 Histogram::Histogram(MonitorManager *manager, QWidget *parent) :
  18     AbstractGfxScopeWidget(manager, false, parent)
  19 {
  20     ui = new Ui::Histogram_UI();
  21     ui->setupUi(this);
  22
  23
  24     m_aUnscaled = new QAction(i18n("Unscaled"), this);
  25     m_aUnscaled->setCheckable(true);
  26
  27     m_aRec601 = new QAction(i18n("Rec. 601"), this);
  28     m_aRec601->setCheckable(true);
  29     m_aRec709 = new QAction(i18n("Rec. 709"), this);
  30     m_aRec709->setCheckable(true);
  31     m_agRec = new QActionGroup(this);
  32     m_agRec->addAction(m_aRec601);
  33     m_agRec->addAction(m_aRec709);
  34
  35     m_menu->addSeparator();
  36     m_menu->addAction(m_aUnscaled);
  37     m_menu->addSeparator()->setText(i18n("Luma mode"));
  38     m_menu->addAction(m_aRec601);
  39     m_menu->addAction(m_aRec709);
  40
  41     bool b = true;
  42     b &= connect(ui->cbY, SIGNAL(toggled(bool)), this, SLOT(forceUpdateScope()));
  43     b &= connect(ui->cbS, SIGNAL(toggled(bool)), this, SLOT(forceUpdateScope()));
  44     b &= connect(ui->cbR, SIGNAL(toggled(bool)), this, SLOT(forceUpdateScope()));
  45     b &= connect(ui->cbG, SIGNAL(toggled(bool)), this, SLOT(forceUpdateScope()));
  46     b &= connect(ui->cbB, SIGNAL(toggled(bool)), this, SLOT(forceUpdateScope()));
  47     b &= connect(m_aUnscaled, SIGNAL(toggled(bool)), this, SLOT(forceUpdateScope()));
  48     b &= connect(m_aRec601, SIGNAL(toggled(bool)), this, SLOT(forceUpdateScope()));
  49     b &= connect(m_aRec709, SIGNAL(toggled(bool)), this, SLOT(forceUpdateScope()));
  50     Q_ASSERT(b);
  51
  52     init();
  53     m_histogramGenerator = new HistogramGenerator();
  54 }
  55
  56 Histogram::~Histogram()
  57 {
  58     writeConfig();
  59     delete m_histogramGenerator;
  60     delete ui;
  61     delete m_aUnscaled;
  62     delete m_aRec601;
  63     delete m_aRec709;
  64     delete m_agRec;
  65 }
  66
  67 void Histogram::readConfig()
  68 {
  69     AbstractGfxScopeWidget::readConfig();
  70
  71     KSharedConfigPtr config = KGlobal::config();
  72     KConfigGroup scopeConfig(config, configName());
  73     ui->cbY->setChecked(scopeConfig.readEntry("yEnabled", true));
  74     ui->cbS->setChecked(scopeConfig.readEntry("sEnabled", false));
  75     ui->cbR->setChecked(scopeConfig.readEntry("rEnabled", true));
  76     ui->cbG->setChecked(scopeConfig.readEntry("gEnabled", true));
  77     ui->cbB->setChecked(scopeConfig.readEntry("bEnabled", true));
  78     m_aRec601->setChecked(scopeConfig.readEntry("rec601", false));
  79     m_aRec709->setChecked(!m_aRec601->isChecked());
  80 }
  81
  82 void Histogram::writeConfig()
  83 {
  84     KSharedConfigPtr config = KGlobal::config();
  85     KConfigGroup scopeConfig(config, configName());
  86     scopeConfig.writeEntry("yEnabled", ui->cbY->isChecked());
  87     scopeConfig.writeEntry("sEnabled", ui->cbS->isChecked());
  88     scopeConfig.writeEntry("rEnabled", ui->cbR->isChecked());
  89     scopeConfig.writeEntry("gEnabled", ui->cbG->isChecked());
  90     scopeConfig.writeEntry("bEnabled", ui->cbB->isChecked());
  91     scopeConfig.writeEntry("rec601", m_aRec601->isChecked());
  92     scopeConfig.sync();
  93 }
  94
  95 QString Histogram::widgetName() const { return QString("Histogram"); }
  96
  97 bool Histogram::isHUDDependingOnInput() const { return false; }
  98 bool Histogram::isScopeDependingOnInput() const { return true; }
  99 bool Histogram::isBackgroundDependingOnInput() const { return false; }
 100
 101 QRect Histogram::scopeRect()
 102 {
 103     //qDebug() << "According to the spacer, the top left point is " << ui->verticalSpacer->geometry().x() << "/" << ui->verticalSpacer->geometry().y();
 104     QPoint topleft(offset, offset+ ui->verticalSpacer->geometry().y());
 105     return QRect(topleft, this->rect().size() - QSize(topleft.x() + offset, topleft.y() + offset));
 106 }
 107
 108 QImage Histogram::renderHUD(uint)
 109 {
 110     emit signalHUDRenderingFinished(0, 1);
 111     return QImage();
 112 }
 113 QImage Histogram::renderGfxScope(uint accelFactor, const QImage qimage)
 114 {
 115     QTime start = QTime::currentTime();
 116     start.start();
 117
 118     const int componentFlags =   (ui->cbY->isChecked() ? 1 : 0) * HistogramGenerator::ComponentY
 119                                | (ui->cbS->isChecked() ? 1 : 0) * HistogramGenerator::ComponentSum
 120                                | (ui->cbR->isChecked() ? 1 : 0) * HistogramGenerator::ComponentR
 121                                | (ui->cbG->isChecked() ? 1 : 0) * HistogramGenerator::ComponentG
 122                                | (ui->cbB->isChecked() ? 1 : 0) * HistogramGenerator::ComponentB;
 123
 124     HistogramGenerator::Rec rec = m_aRec601->isChecked() ? HistogramGenerator::Rec_601 : HistogramGenerator::Rec_709;
 125
 126     QImage histogram = m_histogramGenerator->calculateHistogram(m_scopeRect.size(), qimage, componentFlags,
 127                                                                 rec, m_aUnscaled->isChecked(), accelFactor);
 128
 129     emit signalScopeRenderingFinished(start.elapsed(), accelFactor);
 130     return histogram;
 131 }
 132 QImage Histogram::renderBackground(uint)
 133 {
 134     emit signalBackgroundRenderingFinished(0, 1);
 135     return QImage();
 136 }