ref: baaa888c574e075222ad70b002fe6a140a19098b
dir: /src/speech/cst_track_io.c/
/*************************************************************************/
/* */
/* Language Technologies Institute */
/* Carnegie Mellon University */
/* Copyright (c) 1999 */
/* 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. */
/* 3. Original authors' names are not deleted. */
/* 4. The authors' names are not used to endorse or promote products */
/* derived from this software without specific prior written */
/* permission. */
/* */
/* CARNEGIE MELLON UNIVERSITY 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 CARNEGIE MELLON UNIVERSITY 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. */
/* */
/*************************************************************************/
/* Author: Alan W Black (awb@cs.cmu.edu) */
/* Date: August 2000 */
/*************************************************************************/
/* */
/* Track i/o */
/* */
/*************************************************************************/
#include "cst_string.h"
#include "cst_endian.h"
#include "cst_tokenstream.h"
#include "cst_track.h"
int cst_track_save_est(cst_track *t, const char *filename)
{
cst_file fd;
int i,j;
if ((fd = cst_fopen(filename,CST_OPEN_WRITE|CST_OPEN_BINARY)) == NULL)
{
cst_errmsg("cst_track_save_est: can't open file \"%s\"\n",
filename);
return -1;
}
cst_fprintf(fd,"EST_File Track\n");
cst_fprintf(fd,"DataType ascii\n");
cst_fprintf(fd,"NumFrames %d\n",t->num_frames);
cst_fprintf(fd,"NumChannels %d\n",t->num_channels);
cst_fprintf(fd,"BreaksPresent true\n");
cst_fprintf(fd,"EST_Header_End\n");
for (i=0; i < t->num_frames; i++)
{
cst_fprintf(fd,"%f\t1 \t",t->times[i]);
for (j=0; j < t->num_channels; j++)
cst_fprintf(fd,"%f ",t->frames[i][j]);
cst_fprintf(fd,"\n");
}
cst_fclose(fd);
return 0;
}
int cst_track_save_est_binary(cst_track *t, const char *filename)
{
cst_file fd;
float foo;
int i,j;
if ((fd = cst_fopen(filename,CST_OPEN_WRITE|CST_OPEN_BINARY)) == NULL)
{
cst_errmsg("cst_track_save_est_binary: can't open file \"%s\"\n",
filename);
return -1;
}
cst_fprintf(fd,"EST_File Track\n");
cst_fprintf(fd,"DataType binary\n");
cst_fprintf(fd,"ByteOrder %s\n",
CST_LITTLE_ENDIAN ? BYTE_ORDER_LITTLE : BYTE_ORDER_BIG);
cst_fprintf(fd,"NumFrames %d\n",t->num_frames);
cst_fprintf(fd,"NumChannels %d\n",t->num_channels);
cst_fprintf(fd,"BreaksPresent true\n");
cst_fprintf(fd,"EST_Header_End\n");
foo = 1.0; /* put a bogus 'breaks' value in for now */
for (i=0; i < t->num_frames; i++)
{
cst_fwrite(fd, t->times + i, sizeof(float), 1);
cst_fwrite(fd, &foo, sizeof(float), 1);
for (j=0; j < t->num_channels; j++)
cst_fwrite(fd, &(t->frames[i][j]), sizeof(float), 1);
}
cst_fclose(fd);
return 0;
}
static int load_frame_ascii(cst_track *t, int i, cst_tokenstream *ts)
{
int j;
t->times[i] = cst_atof(ts_get(ts));
ts_get(ts); /* the can be only 1 */
for (j=0; j < t->num_channels; j++)
t->frames[i][j] = cst_atof(ts_get(ts));
if ((i+1 < t->num_frames) && (ts_eof(ts)))
{
return -1;
}
return 0;
}
static int load_frame_binary(cst_track *t, int i, cst_tokenstream *ts, int swap)
{
float val;
int j;
if (cst_fread(ts->fd, &val, sizeof(float), 1) != 1)
return -1;
if (swap)
swapfloat(&val);
t->times[i] = val;
/* Ignore the 'breaks' field */
if (cst_fread(ts->fd, &val, sizeof(float), 1) != 1)
return -1;
for (j=0; j < t->num_channels; j++)
{
if (cst_fread(ts->fd, &val, sizeof(float), 1) != 1)
return -1;
if (swap)
swapfloat(&val);
t->frames[i][j] = val;
}
return 0;
}
int cst_track_load_est(cst_track *t, const char *filename)
{
cst_tokenstream *ts;
const char *tok;
int num_frames,num_channels;
int i,ascii = 1,swap = 0,rv;
num_frames=0;
num_channels=0;
ts = ts_open(filename,NULL,NULL,NULL,NULL);
if (ts == NULL)
{
cst_errmsg("cst_track_load: can't open file \"%s\"\n",
filename);
return -1;
}
if (!cst_streq(ts_get(ts),"EST_File"))
{
cst_errmsg("cst_track_load: not an EST file \"%s\"\n",
filename);
ts_close(ts); return -1;
}
if (!cst_streq(ts_get(ts),"Track"))
{
cst_errmsg("cst_track_load: not an track file \"%s\"\n",
filename);
ts_close(ts); return -1;
}
while (!cst_streq("EST_Header_End",(tok=ts_get(ts))))
{
if (cst_streq("DataType",tok))
{
tok = ts_get(ts);
if (cst_streq("ascii",tok))
{
ascii = 1;
}
else if (cst_streq("binary",tok))
{
ascii = 0;
}
else
{
cst_errmsg("cst_track_load: don't know how to deal "
"with type \"%s\"\n", tok);
ts_close(ts);
return -1;
}
}
else if (cst_streq("ByteOrder",tok))
{
tok = ts_get(ts);
swap = (cst_streq(tok, BYTE_ORDER_BIG) && CST_LITTLE_ENDIAN)
|| (cst_streq(tok, BYTE_ORDER_LITTLE) && CST_BIG_ENDIAN);
}
else if (cst_streq("NumFrames",tok))
num_frames = atoi(ts_get(ts));
else if (cst_streq("NumChannels",tok))
num_channels = atoi(ts_get(ts));
else
ts_get(ts);
if (ts_eof(ts))
{
cst_errmsg("cst_track_load: EOF in header \"%s\"\n",
filename);
ts_close(ts); return -1;
}
}
cst_track_resize(t,num_frames,num_channels);
for (i=0; i < t->num_frames; i++)
{
if (ascii)
rv = load_frame_ascii(t, i, ts);
else
rv = load_frame_binary(t, i, ts, swap);
if (rv < 0)
{
ts_close(ts);
cst_errmsg("cst_track_load: EOF in data \"%s\"\n",
filename);
return rv;
}
}
ts_get(ts);
if (!ts_eof(ts))
{
cst_errmsg("cst_track_load: not EOF when expected \"%s\"\n",
filename);
ts_close(ts); return -1;
}
ts_close(ts);
return 0;
}