312 lines
8.4 KiB
C++
312 lines
8.4 KiB
C++
/*
|
|
* Author: Harry van Haaren 2013
|
|
* harryhaaren@gmail.com
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef AVTK_WAVEFORM_H
|
|
#define AVTK_WAVEFORM_H
|
|
|
|
#include <FL/Fl_Widget.H>
|
|
#include <vector>
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <stdlib.h>
|
|
#include <math.h>
|
|
|
|
using namespace std;
|
|
|
|
namespace Avtk
|
|
{
|
|
|
|
class Waveform : public Fl_Widget
|
|
{
|
|
public:
|
|
Waveform(int _x, int _y, int _w, int _h, const char *_label=0 ):
|
|
Fl_Widget(_x, _y, _w, _h, _label)
|
|
{
|
|
x = _x;
|
|
y = _y;
|
|
w = _w;
|
|
h = _h;
|
|
|
|
label = _label;
|
|
|
|
highlight = false;
|
|
newWaveform = false;
|
|
|
|
waveformCr = 0;
|
|
waveformSurf = 0;
|
|
|
|
|
|
data = (float*)malloc( sizeof(float) * w );
|
|
|
|
srand (time(NULL));
|
|
|
|
for (int i = 0; i < _w; i++)
|
|
{
|
|
data[i] = rand() / RAND_MAX / 0.75;
|
|
}
|
|
|
|
newWaveform = true;
|
|
|
|
strokeRim = true;
|
|
}
|
|
|
|
bool strokeRim;
|
|
bool highlight;
|
|
int x, y, w, h;
|
|
const char* label;
|
|
|
|
cairo_t* waveformCr;
|
|
cairo_surface_t* waveformSurf;
|
|
|
|
bool newWaveform;
|
|
|
|
long dataSize;
|
|
float* data;
|
|
|
|
void setData( float* d, long size )
|
|
{
|
|
//cout << "AvtkWaveform: setDataPtr = " << data << endl;
|
|
dataSize = size;
|
|
data = d;
|
|
newWaveform = true;
|
|
|
|
|
|
damage(FL_DAMAGE_ALL);
|
|
}
|
|
|
|
void draw()
|
|
{
|
|
if (damage() & FL_DAMAGE_ALL)
|
|
{
|
|
cairo_t *cr = Fl::cairo_cc();
|
|
cairo_save(cr);
|
|
|
|
// clear the surface
|
|
cairo_rectangle(cr, x, y, w, h);
|
|
cairo_set_source_rgb (cr, 0.1,0.1,0.1);
|
|
cairo_fill( cr );
|
|
|
|
if ( newWaveform )
|
|
{
|
|
if ( !waveformCr )
|
|
{
|
|
// create the waveform surface and context
|
|
//cout << "waveform draw() creating new objects" << endl;
|
|
waveformSurf= cairo_image_surface_create ( CAIRO_FORMAT_ARGB32, w, h);
|
|
waveformCr = cairo_create ( waveformSurf );
|
|
//cout << "waveform draw() creating objects done" << endl;
|
|
}
|
|
|
|
// clear the surface
|
|
cairo_rectangle(waveformCr, 0, 0, w, h);
|
|
cairo_set_source_rgb (waveformCr, 0.1,0.1,0.1);
|
|
cairo_fill( waveformCr );
|
|
|
|
// set up dashed lines, 1 px off, 1 px on
|
|
double dashes[1];
|
|
dashes[0] = 2.0;
|
|
|
|
cairo_set_dash ( waveformCr, dashes, 1, 0.0);
|
|
cairo_set_line_width( waveformCr, 1.0);
|
|
|
|
// loop over each 2nd line, drawing dots
|
|
cairo_set_line_width(waveformCr, 1.0);
|
|
cairo_set_source_rgb(waveformCr, 0.4,0.4,0.4);
|
|
for ( int i = 1; i < 4; i++ )
|
|
{
|
|
cairo_move_to( waveformCr, ((w / 4.f)*i), 0);
|
|
cairo_line_to( waveformCr, ((w / 4.f)*i), h );
|
|
}
|
|
for ( int i = 1; i < 4; i++ )
|
|
{
|
|
cairo_move_to( waveformCr, 0, ((h / 4.f)*i) );
|
|
cairo_line_to( waveformCr, w, ((h / 4.f)*i) );
|
|
}
|
|
|
|
cairo_set_source_rgba( waveformCr, 66 / 255.f, 66 / 255.f , 66 / 255.f , 0.5 );
|
|
cairo_stroke(waveformCr);
|
|
cairo_set_dash ( waveformCr, dashes, 0, 0.0);
|
|
|
|
|
|
if ( !data )
|
|
{
|
|
// draw X
|
|
cairo_move_to( cr, 0 , 0 );
|
|
cairo_line_to( cr, 0 + w, 0 + h );
|
|
cairo_move_to( cr, 0 , 0 + h );
|
|
cairo_line_to( cr, 0 + w, 0 );
|
|
cairo_set_source_rgb ( cr, 0.2,0.2,0.2);
|
|
cairo_stroke(cr);
|
|
|
|
// draw text
|
|
cairo_move_to( cr, 0 + (w/2.f) - 65, 0 + (h/2.f) + 10 );
|
|
cairo_set_source_rgb ( cr, 0.6,0.6,0.6);
|
|
cairo_set_font_size( cr, 20 );
|
|
cairo_show_text( cr, "No data loaded" );
|
|
}
|
|
else
|
|
{
|
|
// don't draw every sample
|
|
int sampleCountForDrawing = -1;
|
|
|
|
float currentTop = 0.f;
|
|
float previousTop = 0.f;
|
|
float currentSample = 0.f;
|
|
|
|
// find how many samples per pixel
|
|
int samplesPerPix = int(dataSize / float(w));
|
|
//cout << "width = " << w << " sampsPerPx " << samplesPerPix << endl;
|
|
|
|
// loop over each pixel value we need
|
|
for( int p = 0; p < w; p++ )
|
|
{
|
|
float averageMax = 0.f;
|
|
float averageMin = 0.f;
|
|
|
|
// calc value for this pixel
|
|
for( int i = 0; i < samplesPerPix; i++ )
|
|
{
|
|
float tmp = data[i + (p * samplesPerPix)];
|
|
if ( tmp < 0 )
|
|
{
|
|
averageMin -= tmp;
|
|
}
|
|
else
|
|
{
|
|
averageMax += tmp;
|
|
}
|
|
}
|
|
|
|
// - 0.1 to keep in bounds of square
|
|
averageMin = (averageMin / samplesPerPix);
|
|
averageMax = (averageMax / samplesPerPix);
|
|
|
|
// don't use to the *very* top of the widget, only 0.8 (top & bottom)
|
|
float dH = h * 0.4;
|
|
|
|
// emulate log shape
|
|
float tmpMin = log( 0.1 + (averageMin*4.9) );
|
|
float tmpMax = log( 0.1 + (averageMax*4.9) );
|
|
|
|
// logMin range == -2.3 -> 1.6: add 2.3, then / 3.9
|
|
float logMin = -((tmpMin + 2.3) / 3.9);
|
|
float logMax = ((tmpMax + 2.3) / 3.9);
|
|
|
|
//printf("input %f, output %f\n", averageMin, logMin );
|
|
|
|
// draw lines
|
|
cairo_move_to( waveformCr, p, h * 0.5 + (logMin * dH ) );
|
|
cairo_line_to( waveformCr, p, h * 0.5 + (logMax * dH ) );
|
|
}
|
|
|
|
// stroke the waveform
|
|
cairo_set_source_rgb( waveformCr, 1.0,1.0,1.0);
|
|
cairo_stroke( waveformCr );
|
|
|
|
newWaveform = false;
|
|
}
|
|
}
|
|
|
|
cairo_set_source_surface(cr, waveformSurf, x, y);
|
|
cairo_rectangle( cr, x, y, w, h);
|
|
cairo_paint(cr);
|
|
|
|
if ( strokeRim )
|
|
{
|
|
// stroke rim
|
|
cairo_set_line_width(cr, 1);
|
|
cairo_rectangle(cr, x, y, w, h);
|
|
cairo_set_source_rgba( cr, 100 / 255.f, 100 / 255.f , 100 / 255.f , 1 );
|
|
cairo_stroke( cr );
|
|
}
|
|
|
|
//cout << "waveform draw() done" << endl;
|
|
|
|
cairo_restore(cr);
|
|
}
|
|
}
|
|
|
|
void resize(int X, int Y, int W, int H)
|
|
{
|
|
Fl_Widget::resize(X,Y,W,H);
|
|
|
|
// take the smaller value
|
|
long newSize = W > w ? W : w;
|
|
printf("Waveform new size %li, from %i\n", newSize, w );
|
|
|
|
x = X;
|
|
y = Y;
|
|
w = W;
|
|
h = H;
|
|
|
|
// FIXME: needs to be resampled, not clipped at end
|
|
// delete old data, and resize it
|
|
float* newData = (float*)malloc( sizeof(float) * w );
|
|
|
|
memcpy( newData, data, newSize );
|
|
free ( data );
|
|
data = newData;
|
|
|
|
newWaveform = true;
|
|
|
|
|
|
redraw();
|
|
}
|
|
|
|
int handle(int event)
|
|
{
|
|
return 0;
|
|
|
|
switch(event)
|
|
{
|
|
case FL_PUSH:
|
|
highlight = 0;
|
|
redraw();
|
|
return 1;
|
|
case FL_DRAG: {
|
|
int t = Fl::event_inside(this);
|
|
if (t != highlight) {
|
|
redraw();
|
|
}
|
|
}
|
|
return 1;
|
|
case FL_RELEASE:
|
|
if (highlight) {
|
|
highlight = 0;
|
|
redraw();
|
|
do_callback();
|
|
}
|
|
return 1;
|
|
case FL_SHORTCUT:
|
|
if ( test_shortcut() )
|
|
{
|
|
do_callback();
|
|
return 1;
|
|
}
|
|
return 0;
|
|
default:
|
|
return Fl_Widget::handle(event);
|
|
}
|
|
}
|
|
};
|
|
|
|
} // Avtk
|
|
|
|
#endif
|
|
|