ref: 39fa1f39d257a6317141c6a3efb32ca103e47748
dir: /src/noiseprof.c/
/* * noiseprof - Noise Profiling Effect. * * Written by Ian Turner (vectro@vectro.org) * * Copyright 1999 Ian Turner and others * This file is part of SoX. * SoX is free software; you can redistribute it and/or modify * it under the terms of the GNU Lesser General Public License as published * by the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * */ #include "noisered.h" #include <assert.h> #include <string.h> #include <errno.h> typedef struct chandata { float *sum; int *profilecount; float *window; } chandata_t; typedef struct profdata { char* output_filename; FILE* output_file; chandata_t *chandata; st_size_t bufdata; } * profdata_t; /* * Get the filename, if any. We don't open it until st_noiseprof_start. */ int st_noiseprof_getopts(eff_t effp, int n, char **argv) { profdata_t data = (profdata_t) effp->priv; if (n == 1) { data->output_filename = argv[0]; } else if (n > 1) { st_fail("Usage: noiseprof [filename]"); return (ST_EOF); } return (ST_SUCCESS); } /* * Prepare processing. * Do all initializations. */ int st_noiseprof_start(eff_t effp) { profdata_t data = (profdata_t) effp->priv; int channels = effp->ininfo.channels; int i; if (data->output_filename != NULL) { data->output_file = fopen(data->output_filename, "w"); if (data->output_file == NULL) { st_fail("Couldn't open output file %s: %s", data->output_filename, strerror(errno)); } } else { /* FIXME: We should detect output to stdout, and redirect to stderr. */ data->output_file = stderr; } data->chandata = (chandata_t*)calloc(channels, sizeof(*(data->chandata))); for (i = 0; i < channels; i ++) { data->chandata[i].sum = (float*)calloc(FREQCOUNT, sizeof(float)); data->chandata[i].profilecount = (int*)calloc(FREQCOUNT, sizeof(int)); data->chandata[i].window = (float*)calloc(WINDOWSIZE, sizeof(float)); } data->bufdata = 0; return (ST_SUCCESS); } /* Collect statistics from the complete window on channel chan. */ static void collect_data(profdata_t data, chandata_t* chan) { float *out = (float*)calloc(FREQCOUNT, sizeof(float)); int i; PowerSpectrum(WINDOWSIZE, chan->window, out); for (i = 0; i < FREQCOUNT; i ++) { float value = log(out[i]); if (finite(value)) { chan->sum[i] += value; chan->profilecount[i] ++; } } free(out); } /* * Grab what we can from ibuf, and process if we have a whole window. */ int st_noiseprof_flow(eff_t effp, st_sample_t *ibuf, st_sample_t *obuf, st_size_t *isamp, st_size_t *osamp) { profdata_t data = (profdata_t) effp->priv; int samp = min(*isamp, *osamp); int tracks = effp->ininfo.channels; int track_samples = samp / tracks; int ncopy = 0; int i; assert(effp->ininfo.channels == effp->outinfo.channels); /* How many samples per track to analyze? */ ncopy = min(track_samples, WINDOWSIZE-data->bufdata); /* Collect data for every channel. */ for (i = 0; i < tracks; i ++) { chandata_t* chan = &(data->chandata[i]); int j; for (j = 0; j < ncopy; j ++) { chan->window[j+data->bufdata] = ST_SAMPLE_TO_FLOAT_DWORD(ibuf[i+j*tracks]); } if (ncopy + data->bufdata == WINDOWSIZE) { collect_data(data, chan); } } data->bufdata += ncopy; assert(data->bufdata <= WINDOWSIZE); if (data->bufdata == WINDOWSIZE) data->bufdata = 0; memcpy(obuf, ibuf, ncopy*tracks); *isamp = *osamp = ncopy*tracks; return (ST_SUCCESS); } /* * Finish off the last window. */ int st_noiseprof_drain(eff_t effp, st_sample_t *obuf, st_size_t *osamp) { profdata_t data = (profdata_t) effp->priv; int tracks = effp->ininfo.channels; int i; *osamp = 0; if (data->bufdata == 0) { return ST_SUCCESS; } for (i = 0; i < tracks; i ++) { int j; for (j = data->bufdata+1; j < WINDOWSIZE; j ++) { data->chandata[i].window[j] = 0; } collect_data(data, &(data->chandata[i])); } return (ST_SUCCESS); } /* * Print profile and clean up. */ int st_noiseprof_stop(eff_t effp) { profdata_t data = (profdata_t) effp->priv; int i; for (i = 0; i < effp->ininfo.channels; i ++) { int j; chandata_t* chan = &(data->chandata[i]); fprintf(data->output_file, "Channel %d: ", i); for (j = 0; j < FREQCOUNT; j ++) { fprintf(data->output_file, "%s%f", j == 0 ? "" : ", ", chan->sum[j] / chan->profilecount[j]); } fprintf(data->output_file, "\n"); free(chan->sum); free(chan->profilecount); } free(data->chandata); if (data->output_file != stderr && data->output_file != stdout) { fclose(data->output_file); } return (ST_SUCCESS); } /* For Emacs: Local Variables: c-basic-offset: 4 */