1 /***************************************************************************
2 * Copyright (C) 2010 by Simon Andreas Eugster (simon.eu@gmail.com) *
3 * This file is part of kdenlive. See www.kdenlive.org. *
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 ***************************************************************************/
19 #include "waveformgenerator.h"
21 #define CHOP255(a) ((255) < (a) ? (255) : (a))
23 WaveformGenerator::WaveformGenerator()
27 WaveformGenerator::~WaveformGenerator()
31 QImage WaveformGenerator::calculateWaveform(const QSize &waveformSize, const QImage &image, WaveformGenerator::PaintMode paintMode,
32 const bool &drawAxis, WaveformGenerator::Rec rec, const uint &accelFactor)
34 Q_ASSERT(accelFactor >= 1);
39 QImage wave(waveformSize, QImage::Format_ARGB32);
41 if (waveformSize.width() <= 0 || waveformSize.height() <= 0 || image.width() <= 0 || image.height() <= 0) {
46 // Fill with transparent color
47 wave.fill(qRgba(0,0,0,0));
53 const uint ww = waveformSize.width();
54 const uint wh = waveformSize.height();
55 const uint iw = image.bytesPerLine();
56 const uint ih = image.height();
57 const uint byteCount = iw*ih;
59 uint waveValues[waveformSize.width()][waveformSize.height()];
60 for (int i = 0; i < waveformSize.width(); i++) {
61 for (int j = 0; j < waveformSize.height(); j++) {
66 // Number of input pixels that will fall on one scope pixel.
67 // Must be a float because the acceleration factor can be high, leading to <1 expected px per px.
68 const float pixelDepth = (float)((byteCount>>2) / accelFactor)/(ww*wh);
69 const float gain = 255/(8*pixelDepth);
70 //qDebug() << "Pixel depth: expected " << pixelDepth << "; Gain: using " << gain << " (acceleration: " << accelFactor << "x)";
73 // Subtract 1 from sizes because we start counting from 0.
74 // Not doing it would result in attempts to paint outside of the image.
75 const float hPrediv = (float)(wh-1)/255;
76 const float wPrediv = (float)(ww-1)/(iw-1);
78 const uchar *bits = image.bits();
80 for (uint i = 0, x = 0; i < byteCount; i += 4) {
82 Q_ASSERT(bits < image.bits() + byteCount);
86 if (rec == WaveformGenerator::Rec_601) {
88 dY = .299*qRed(*col) + .587*qGreen(*col) + .114*qBlue(*col);
91 dY = .2125*qRed(*col) + .7154*qGreen(*col) + .0721*qBlue(*col);
93 // dY is on [0,255] now.
97 waveValues[(int)dx][(int)dy]++;
103 if (accelFactor > 1) {
104 bits += 4*iw*(accelFactor-1);
105 i += 4*iw*(accelFactor-1);
111 case PaintMode_Green:
112 for (int i = 0; i < waveformSize.width(); i++) {
113 for (int j = 0; j < waveformSize.height(); j++) {
114 // Logarithmic scale. Needs fine tuning by hand, but looks great.
115 wave.setPixel(i, waveformSize.height()-j-1, qRgba(CHOP255(52*log(0.1*gain*waveValues[i][j])),
116 CHOP255(52*log(gain*waveValues[i][j])),
117 CHOP255(52*log(.25*gain*waveValues[i][j])),
118 CHOP255(64*log(gain*waveValues[i][j]))));
122 case PaintMode_Yellow:
123 for (int i = 0; i < waveformSize.width(); i++) {
124 for (int j = 0; j < waveformSize.height(); j++) {
125 wave.setPixel(i, waveformSize.height()-j-1, qRgba(255,242,0, CHOP255(gain*waveValues[i][j])));
130 for (int i = 0; i < waveformSize.width(); i++) {
131 for (int j = 0; j < waveformSize.height(); j++) {
132 wave.setPixel(i, waveformSize.height()-j-1, qRgba(255,255,255, CHOP255(2*gain*waveValues[i][j])));
139 QPainter davinci(&wave);
141 davinci.setPen(qRgba(150,255,200,32));
142 davinci.setCompositionMode(QPainter::CompositionMode_Overlay);
143 for (uint i = 0; i <= 10; i++) {
144 dy = (float)i/10 * (wh-1);
145 for (uint x = 0; x < ww; x++) {
146 opx = wave.pixel(x, dy);
147 wave.setPixel(x,dy, qRgba(CHOP255(150+qRed(opx)), 255,
148 CHOP255(200+qBlue(opx)), CHOP255(32+qAlpha(opx))));
155 //uint diff = time.elapsed();
156 //emit signalCalculationFinished(wave, diff);