non/timeline/src/Waveform.C

/*******************************************************************************/
/* Copyright (C) 2008 Jonathan Moore Liles                                     */
/*                                                                             */
/* 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 2 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; see the file COPYING.  If not,write to the Free Software */
/* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
/*******************************************************************************/

/* routines to draw a waveform from peak data */

#include <FL/Fl.H>
#include <FL/fl_draw.H>
#include "Waveform.H"

#include <math.h>
#include <algorithm>
using std::min;
using std::max;
#include <dsp.h>



bool Waveform::fill = true;
bool Waveform::outline = true;
bool Waveform::vary_color = true;
bool Waveform::logarithmic = true;



/* TODO: split the variations into separate functions. eg, plain,
 * outlined, filled, polygonal, rectified. */

void
Waveform::scale ( Peak *pbuf, int npeaks, float gain )
{
    /* scale it */
    for ( int j = npeaks; j--; )
    {
        pbuf[ j ].min *= gain;
        pbuf[ j ].max *= gain;
    }
}

/** draw a portion of /clip/'s waveform. coordinates are the portion to draw  */
void
Waveform::draw ( int X, int Y, int W, int H,
                 const Peak *pbuf, int peaks, int skip,
                 Fl_Color fg_color, Fl_Color bg_color )
{
    int j;

    int start = 0;

    const int halfheight = H / 2;
    const int mid = Y + halfheight;
    const int ty = Y + halfheight;

    W = min( peaks, W );

    if ( ! W )
        return;

    fg_color = fl_color_add_alpha( Fl::get_color( fg_color ), 200 );
    
    if ( Waveform::fill )
    {
        if ( Waveform::vary_color )
        {
            j = start;
            for ( int x = X; x < X + W; ++x, j += skip )
            {
                const Peak p = pbuf[ j ];

                const float diff = fabs( p.max - p.min );
 
                if ( diff > 2.0f )
                    fl_color( FL_RED );
                else
                    fl_color( fl_color_average( fg_color, bg_color, diff * 0.5f ) );
                
                const int ty = mid - ( halfheight * p.min );
                const int by = mid - ( halfheight * p.max );
                fl_line( x, ty, x, by );
            }
        }
        else
        {
            fl_color( fg_color );
   
            fl_begin_complex_polygon();
            
            j = start;

            for ( int x = X; x <= X + W; x++, j += skip  )
                fl_vertex( x, ty - ( halfheight * pbuf[ j ].min ) );

            j -= skip;

            for ( int x = X + W; x >= X; x--, j -= skip )
                fl_vertex( x, ty - ( halfheight * pbuf[ j ].max ) );
            
            fl_end_complex_polygon();
        }
    }

    if ( Waveform::outline )
    {
        fl_color( bg_color );

        fl_line_style( FL_SOLID, 0 );

        fl_begin_line();

        j = start;

        for ( int x = X; x < X + W; x++, j += skip  )
            fl_vertex( x, ty - ( halfheight * pbuf[ j ].min ) );

        fl_end_line();

        fl_begin_line();

        j = start;

        for ( int x = X; x < X + W; x++, j += skip )
            fl_vertex( x, ty - ( halfheight * pbuf[ j ].max ) );

        fl_end_line();

        fl_line_style( FL_SOLID, 0 );
    }
}