vstat.c

语音信号处理工具包,里面提供了很多有用的操作。这是最新版本

/*
  ----------------------------------------------------------------
	Speech Signal Processing Toolkit (SPTK): version 3.0
			 SPTK Working Group

		   Department of Computer Science
		   Nagoya Institute of Technology
				and
    Interdisciplinary Graduate School of Science and Engineering
		   Tokyo Institute of Technology
		      Copyright (c) 1984-2000
			All Rights Reserved.

  Permission is hereby granted, free of charge, to use and
  distribute this software and its documentation without
  restriction, including without limitation the rights to use,
  copy, modify, merge, publish, distribute, sublicense, and/or
  sell copies of this work, and to permit persons to whom this
  work is furnished to do so, subject to the following conditions:

    1. The code must retain the above copyright notice, this list
       of conditions and the following disclaimer.

    2. Any modifications must be clearly marked as such.

  NAGOYA INSTITUTE OF TECHNOLOGY, TOKYO INSITITUTE OF TECHNOLOGY,
  SPTK WORKING GROUP, AND THE CONTRIBUTORS TO THIS WORK DISCLAIM
  ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL
  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO EVENT
  SHALL NAGOYA INSTITUTE OF TECHNOLOGY, TOKYO INSITITUTE OF
  TECHNOLOGY, SPTK WORKING GROUP, NOR THE CONTRIBUTORS BE LIABLE
  FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY
  DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
  WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
  ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
  PERFORMANCE OF THIS SOFTWARE.
 ----------------------------------------------------------------
*/

/************************************************************************
*									*
*    Obtain Means and Covariances of Input Vectors			*
*									*
*					1998.12 T.Masuko		*
*	usage:								*
*		vstat [ options ] [ infile ] > stdout			*
*	options:							*
*		-l l     :  length of vector			[1]	*
*		-n n     :  order of vector			[1-1]	*
*		-t t     :  number of vector			[all]	*
*		-d       :  diagonal covariance			[FALSE]	*
*		-r       :  output correlation instead of cov.	[FALSE]	*
*		-o       :  output format  (see stdout)		[0]	*
*	infile:								*
*		data sequence						*
*		    x_1(1), ..., x_1(L), x_2(1) ...			*
*	stdout:								*
*		output format						*
*			0	mean vector and covariance matrix	*
*			1	mean vector				*
*			2	covariance matrix			*
*		mean vector						*
*		    m(1), ..., m(L),					*
*		covariance matrix					*
*		    U(11), ..., m(1L),					*
*		    ...............,					*
*		    U(L1), ..., m(LL), ...				*
*	note:								*
*		if '-d' is specified,					*
*			off-diagonal elements are suppressed.		*
*		'-d' and '-r' are exclusive				*
*			('-r' has priority over '-d').			*
*	require:							*
*									*
************************************************************************/

static char *rcs_id = "$Id: vstat.c,v 1.2 2002/12/25 05:33:14 sako Exp $";

/*  Standard C Libralies  */
#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include <SPTK.h>

typedef enum _Boolean {FA, TR} Boolean;
char *BOOL[] = {"FALSE", "TRUE"};


/*  Default Values  */
#define LENG		1
#define OUTMEAN		TR
#define OUTCOV		TR
#define	DIAGC		FA
#define	CORR		FA


/*  Command Name  */
char	*cmnd;

void usage(int status)
{
    fprintf(stderr, "\n");
    fprintf(stderr, " %s - vector statistics calculation\n",cmnd);
    fprintf(stderr, "\n");
    fprintf(stderr, "  usage:\n", cmnd);
    fprintf(stderr, "       %s [ options ] [ infile ] > stdout\n", cmnd);
    fprintf(stderr, "  options:\n");
    fprintf(stderr, "       -l l   : length of vector                    [%d]\n", LENG);
    fprintf(stderr, "       -n n   : order of vector                     [l-1]\n");
    fprintf(stderr, "       -t t   : number of vector                    [N/A]\n");
    fprintf(stderr, "       -o o   : output format                       [0]\n");
    fprintf(stderr, "                  0 mean & covariance\n");
    fprintf(stderr, "                  1 mean\n");
    fprintf(stderr, "                  2 covariance\n");
    fprintf(stderr, "       -d     : diagonal covariance                 [%s]\n", BOOL[DIAGC]);
    fprintf(stderr, "       -r     : output correlation instead of cov.  [%s]\n", BOOL[CORR]);
    fprintf(stderr, "       -h     : print this message\n");
    fprintf(stderr, "  infile:\n");
    fprintf(stderr, "       vectors (float)                [stdin]\n");
    fprintf(stderr, "  stdout:\n");
    fprintf(stderr, "       mean(s) and covariance(s) of input vectors (float)\n");
    fprintf(stderr, "  note:\n");
    fprintf(stderr, "       if '-d' is specified, off-diagonal elements are suppressed.\n");
    fprintf(stderr, "       '-d' and '-r' are exclusive ('-r' has priority over '-d').\n");
    fprintf(stderr, "\n");
    exit(status);
}


main(argc,argv)
int	argc;
char	*argv[];
{
    FILE	*fp = stdin;
    double      *x, *mean, **cov, *var;
    int	        leng = LENG, nv = -1, i, j, k, lp, outtype = 0;
    Boolean	outmean = OUTMEAN, outcov = OUTCOV, diagc = DIAGC, corr = CORR;
    double	sqrt();
    
    if ((cmnd = strrchr(argv[0], '/')) == NULL)
	cmnd = argv[0];
    else
	cmnd++;
    while (--argc)
	if (**++argv == '-') {
	    switch (*(*argv+1)) {
		case 'l':
		    leng = atoi(*++argv);
		    --argc;
		    break;
		case 'n':
		    leng = atoi(*++argv)+1;
		    --argc;
		    break;
		case 't':
		    nv = atoi(*++argv);
		    --argc;
		    break;
		case 'o':
		    outtype = atoi(*++argv);
		    --argc;
		    break;
		case 'd':
		    diagc = 1 - diagc;
		    break;
		case 'r':
		    corr = 1 - corr;
		    break;
		case 'h':
		    usage(0);
		default:
		    fprintf(stderr, "%s : Invalid option '%c' !\n", cmnd, *(*argv+1));
		    usage(1);
		}
	}
	else 
	    fp = getfp(*argv, "r");

    switch (outtype){
	case 1:
	    outcov = FA;
	    break;
	case 2:
	    outmean = FA;
	    break;
    }
    if (diagc && corr)
	diagc = FA;

    mean = dgetmem(leng + leng);
    x = mean + leng;
    if (outcov) {
        if (!diagc) {
	    cov = (double **)getmem(leng, sizeof(*cov));
	    cov[0] = dgetmem(leng*leng);
	    for (i = 1; i < leng; i++)
	        cov[i] = cov[i-1] + leng;
        }
        else
	    var = dgetmem(leng);
    }

    while (!feof(fp)) {
	for (i = 0; i < leng; i++) {
	    mean[i] = 0.0;
	    if (outcov) {
	        if (!diagc)
	            for (j = 0; j < leng; j++)
		        cov[i][j] = 0.0;
	        else
		    var[i] = 0.0;
	    }
	}
	for(lp = nv; lp; ) {
	    if(freadf(x, sizeof(*x), leng, fp) != leng)
		break;
	    for(i = 0; i < leng; i++) {
		mean[i] += x[i];
		if (outcov) {
		    if (!diagc)
		        for (j = i; j < leng; j++)
			    cov[i][j] += x[i] * x[j];
		    else
		        var[i] += x[i] * x[i];
		}
	    }
	    --lp;
	}
	if (lp == 0 || nv == -1) {
	    if (nv > 0)
		k = nv;
	    else
		k = -lp - 1;
	    for (i = 0; i < leng; i++)
		mean[i] /= k;
	    if (outcov) {
	        if (!diagc)
		for (i = 0; i < leng; i++)
		    for (j = i; j < leng; j++)
			cov[j][i] = cov[i][j] = cov[i][j]/k - mean[i]*mean[j];
	        else
		    for (i = 0; i < leng; i++)
		        var[i] = var[i]/k - mean[i]*mean[i];
	    }
	    if (corr) {
		for (i = 0; i < leng; i++)
		    for(j = i+1; j < leng; j++)
			cov[j][i] = cov[i][j] = cov[i][j] / sqrt(cov[i][i]*cov[j][j]);
		for (i = 0; i < leng; i++)
		    cov[i][i] = 1.0;
	    }
	    if (outmean)
	        fwritef(mean, sizeof(*mean), leng, stdout);
	    if (outcov)
		if (!diagc)
		    fwritef(cov[0], sizeof(*cov[0]), leng*leng, stdout);
		else
		    fwritef(var, sizeof(*var), leng, stdout);
	}
    }
    exit(0);
}