non/Peaks.C

306 lines
6.9 KiB
C++
Raw Normal View History

/*******************************************************************************/
/* 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. */
/*******************************************************************************/
#include "Peaks.H"
#include "Timeline.H"
#include <sys/mman.h>
#include <sys/types.h>
#include <sys/stat.h>
2008-02-17 00:28:48 +01:00
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sndfile.h>
#include "Clip.H"
#include "assert.h"
#include <math.h>
Peaks::peakbuffer Peaks::peakbuf;
/** Prepare a buffer of peaks from /s/ to /e/ for reading. Must be
* called before any calls to operator[] */
void
Peaks::fill_buffer ( int s, int e ) const
{
if ( timeline.fpp < _peaks->chunksize )
{
/* looks like we're going to have to switch to a higher resolution peak file
or read directly from the source */
2008-02-18 02:13:20 +01:00
read_peaks( s, e, (e - s) / timeline.fpp, timeline.fpp );
}
else
{
/* we'll just downsample on the fly in this case--no need for extra copying into
the buffer */
}
}
void
Peaks::downsample ( Peak *peaks, int s, int e, float *mhi, float *mlo ) const
{
*mhi = 0;
*mlo = 0;
if ( e > _len )
2008-02-17 03:59:45 +01:00
e = _len;
for ( int j = s; j < e; j++ )
{
const float lo = peaks[ j ].min;
const float hi = peaks[ j ].max;
if ( hi > *mhi )
*mhi = hi;
if ( lo < *mlo )
*mlo = lo;
}
}
int
Peaks::clip_read_peaks ( Peak *peaks, int npeaks, int chunksize ) const
{
2008-02-18 02:13:20 +01:00
sample_t *fbuf = new sample_t[ chunksize ];
size_t len;
int i;
for ( i = 0; i < npeaks; ++i )
{
/* read in a buffer */
len = _clip->read( fbuf, chunksize );
Peak &p = peaks[ i ];
p.min = 0;
p.max = 0;
for ( int j = len; j--; )
{
if ( fbuf[j] > p.max )
p.max = fbuf[j];
if ( fbuf[j] < p.min )
p.min = fbuf[j];
}
if ( len < chunksize )
break;
}
delete fbuf;
return i;
}
void
Peaks::read_peaks ( int s, int e, int npeaks, int chunksize ) const
{
2008-02-18 02:13:20 +01:00
printf( "reading peaks %d @ %d\n", npeaks, chunksize );
if ( peakbuf.size < npeaks )
{
peakbuf.size = npeaks;
// printf( "reallocating peak buffer %li\n", peakbuf.size );
peakbuf.buf = (peakdata*)realloc( peakbuf.buf, sizeof( peakdata ) + (peakbuf.size * sizeof( Peak )) );
}
_clip->open();
_clip->seek( s );
peakbuf.offset = s;
peakbuf.buf->chunksize = chunksize;
peakbuf.len = clip_read_peaks( peakbuf.buf->data, npeaks, chunksize );
_clip->close();
}
/** Return the peak for the range of samples */
Peak &
Peaks::peak ( nframes_t start, nframes_t end ) const
{
/* Is there a better way to return this? */
static Peak p;
2008-02-17 01:35:53 +01:00
if ( timeline.fpp < _peaks->chunksize )
{
assert( timeline.fpp == peakbuf.buf->chunksize );
start = (start - peakbuf.offset) / peakbuf.buf->chunksize;
end = (end - peakbuf.offset) / peakbuf.buf->chunksize;
if ( end > peakbuf.len )
end = peakbuf.len;
2008-02-17 01:35:53 +01:00
// assert( peakbuf.len > start );
downsample( peakbuf.buf->data, start, end, &p.max, &p.min );
}
else
{
start /= _peaks->chunksize;
end /= _peaks->chunksize;
downsample( _peaks->data, start, end, &p.max, &p.min );
}
return p;
}
/* virtual array. Index is a Pixel value, and it returns the
* (resampled) peaks for that pixel based on the current timeline
* zoom. */
Peak &
Peaks::operator[] ( int X ) const
{
return peak( timeline.x_to_ts( X ), timeline.x_to_ts( X + 1 ) );
}
static
const char *
peakname ( const char *filename )
{
static char file[512];
snprintf( file, 512, "%s.peak", filename );
return (const char*)&file;
}
bool
2008-02-18 00:33:59 +01:00
Peaks::open ( void )
{
2008-02-18 00:33:59 +01:00
const char *filename = _clip->name();
int fd;
2008-02-18 00:33:59 +01:00
make_peaks( 256 );
2008-02-17 03:59:45 +01:00
if ( ( fd = ::open( peakname( filename ), O_RDONLY ) ) < 0 )
2008-02-17 03:59:45 +01:00
return false;
{
struct stat st;
2008-02-17 03:59:45 +01:00
fstat( fd, &st );
2008-02-17 03:59:45 +01:00
_len = st.st_size;
}
_peaks = (peakdata*)mmap( NULL, _len, PROT_READ, MAP_SHARED, fd, 0 );
2008-02-17 00:28:48 +01:00
::close( fd );
if ( _peaks == MAP_FAILED )
2008-02-17 03:59:45 +01:00
printf( "failed to create mapping!\n" );
_len = (_len - sizeof( int )) / sizeof( Peak );
2008-02-17 00:28:48 +01:00
return true;
}
2008-02-17 03:59:45 +01:00
/** returns false if peak file for /filename/ is out of date */
bool
2008-02-18 00:33:59 +01:00
Peaks::current ( void ) const
{
2008-02-17 03:59:45 +01:00
int sfd, pfd;
2008-02-18 00:33:59 +01:00
if ( ( sfd = ::open( _clip->name(), O_RDONLY ) ) < 0 )
2008-02-17 03:59:45 +01:00
return true;
2008-02-18 00:33:59 +01:00
if ( ( pfd = ::open( peakname( _clip->name() ), O_RDONLY ) ) < 0 )
2008-02-17 03:59:45 +01:00
return false;
2008-02-17 03:59:45 +01:00
struct stat sst, pst;
2008-02-17 03:59:45 +01:00
fstat( sfd, &sst );
fstat( pfd, &pst );
2008-02-17 03:59:45 +01:00
close( sfd );
close( pfd );
return sst.st_mtime <= pst.st_mtime;
}
/** build peaks file for /filename/ if necessary */
bool
2008-02-18 00:33:59 +01:00
Peaks::make_peaks ( int chunksize )
2008-02-17 03:59:45 +01:00
{
2008-02-18 00:33:59 +01:00
const char *filename = _clip->name();
if ( current() )
2008-02-17 03:59:45 +01:00
return true;
if ( ! _clip->open() )
return false;
FILE *fp = fopen( peakname( filename ), "w" );
if ( fp == NULL )
{
_clip->close();
return false;
}
/* write chunksize first */
fwrite( &chunksize, sizeof( int ), 1, fp );
size_t len;
do {
Peak p;
len = clip_read_peaks( &p, 1, chunksize );
fwrite( &p, sizeof( Peak ), 1, fp );
}
while ( len );
_clip->close();
fclose( fp );
2008-02-18 00:33:59 +01:00
return true;
}
/** return normalization factor for range of samples from /start/ to
/end/ (uses known peak data if possible */
float
Peaks::normalization_factor( nframes_t start, nframes_t end ) const
{
float s;
fill_buffer( start, end );
Peak p = peak( start, end );
s = fabs( 1.0f / p.max );
if ( s * p.min < -1.0 )
s = 1 / fabs( p.max );
return s;
}