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 ***************************************************************************/
20 #include "waveformgenerator.h"
22 #define CHOP255(a) ((255) < (a) ? (255) : (a))
24 WaveformGenerator::WaveformGenerator()
28 WaveformGenerator::~WaveformGenerator()
32 QImage WaveformGenerator::calculateWaveform(const QSize &waveformSize, const QImage &image, WaveformGenerator::PaintMode paintMode,
33 const bool &drawAxis, WaveformGenerator::Rec rec, const uint &accelFactor)
35 Q_ASSERT(accelFactor >= 1);
40 QImage wave(waveformSize, QImage::Format_ARGB32);
42 if (waveformSize.width() <= 0 || waveformSize.height() <= 0 || image.width() <= 0 || image.height() <= 0) {
47 // Fill with transparent color
48 wave.fill(qRgba(0,0,0,0));
54 const uint ww = waveformSize.width();
55 const uint wh = waveformSize.height();
56 const uint iw = image.bytesPerLine();
57 const uint ih = image.height();
58 const uint byteCount = iw*ih;
60 uint waveValues[waveformSize.width()][waveformSize.height()];
61 for (int i = 0; i < waveformSize.width(); i++) {
62 for (int j = 0; j < waveformSize.height(); j++) {
67 // Number of input pixels that will fall on one scope pixel.
68 // Must be a float because the acceleration factor can be high, leading to <1 expected px per px.
69 const float pixelDepth = (float)((byteCount>>2) / accelFactor)/(ww*wh);
70 const float gain = 255/(8*pixelDepth);
71 qDebug() << "Pixel depth: expected " << pixelDepth << "; Gain: using " << gain << " (acceleration: " << accelFactor << "x)";
74 // Subtract 1 from sizes because we start counting from 0.
75 // Not doing it would result in attempts to paint outside of the image.
76 const float hPrediv = (float)(wh-1)/255;
77 const float wPrediv = (float)(ww-1)/(iw-1);
79 const uchar *bits = image.bits();
80 const uint stepsize = 4*accelFactor;
82 for (uint i = 0, x = 0; i < byteCount; i += stepsize) {
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]++;
101 x %= iw; // Modulo image width, to represent the current x position in the image
105 case PaintMode_Green:
106 for (int i = 0; i < waveformSize.width(); i++) {
107 for (int j = 0; j < waveformSize.height(); j++) {
108 // Logarithmic scale. Needs fine tuning by hand, but looks great.
109 wave.setPixel(i, waveformSize.height()-j-1, qRgba(CHOP255(52*log(0.1*gain*waveValues[i][j])),
110 CHOP255(52*log(gain*waveValues[i][j])),
111 CHOP255(52*log(.25*gain*waveValues[i][j])),
112 CHOP255(64*log(gain*waveValues[i][j]))));
116 case PaintMode_Yellow:
117 for (int i = 0; i < waveformSize.width(); i++) {
118 for (int j = 0; j < waveformSize.height(); j++) {
119 wave.setPixel(i, waveformSize.height()-j-1, qRgba(255,242,0, CHOP255(gain*waveValues[i][j])));
124 for (int i = 0; i < waveformSize.width(); i++) {
125 for (int j = 0; j < waveformSize.height(); j++) {
126 wave.setPixel(i, waveformSize.height()-j-1, qRgba(255,255,255, CHOP255(2*gain*waveValues[i][j])));
133 QPainter davinci(&wave);
135 davinci.setPen(qRgba(150,255,200,32));
136 davinci.setCompositionMode(QPainter::CompositionMode_Overlay);
137 for (uint i = 0; i <= 10; i++) {
138 dy = (float)i/10 * (wh-1);
139 for (uint x = 0; x < ww; x++) {
140 opx = wave.pixel(x, dy);
141 wave.setPixel(x,dy, qRgba(CHOP255(150+qRed(opx)), 255,
142 CHOP255(200+qBlue(opx)), CHOP255(32+qAlpha(opx))));
149 uint diff = time.elapsed();
150 emit signalCalculationFinished(wave, diff);