File: time_scale_tests.c

package info (click to toggle)
spandsp 0.0.6%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye, buster
  • size: 15,796 kB
  • sloc: ansic: 123,211; xml: 13,229; sh: 12,189; cpp: 1,521; makefile: 1,097
file content (167 lines) | stat: -rw-r--r-- 4,937 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
/*
 * SpanDSP - a series of DSP components for telephony
 *
 * time_scale_tests.c
 *
 * Written by Steve Underwood <steveu@coppice.org>
 *
 * Copyright (C) 2004 Steve Underwood
 *
 * All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2, as
 * published by the Free Software Foundation.
 *
 * 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; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

/*! \page time_scale_tests_page Time scaling tests
\section time_scale_tests_page_sec_1 What does it do?
These tests run a speech file through the time scaling routines.

\section time_scale_tests_page_sec_2 How are the tests run?
These tests process a speech file called pre_time_scale.wav. This file should contain
8000 sample/second 16 bits/sample linear audio. The tests read this file, change the
time scale of its contents, and write the resulting audio to post_time_scale.wav.
This file also contains 8000 sample/second 16 bits/sample linear audio.
*/

#if defined(HAVE_CONFIG_H)
#include "config.h"
#endif

#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <sndfile.h>

#include "spandsp.h"

#include "spandsp/private/time_scale.h"

#define BLOCK_LEN       160

#define IN_FILE_NAME    "../test-data/local/short_nb_voice.wav"
#define OUT_FILE_NAME   "time_scale_result.wav"

int main(int argc, char *argv[])
{
    SNDFILE *inhandle;
    SNDFILE *outhandle;
    SF_INFO info;
    int16_t in[BLOCK_LEN];
    int16_t out[5*(BLOCK_LEN + TIME_SCALE_MAX_SAMPLE_RATE/TIME_SCALE_MIN_PITCH)];
    int frames;
    int new_frames;
    int out_frames;
    int count;
    int max;
    time_scale_state_t state;
    float rate;
    float sample_rate;
    const char *in_file_name;
    int sweep_rate;
    int opt;

    rate = 1.8f;
    sweep_rate = FALSE;
    in_file_name = IN_FILE_NAME;
    while ((opt = getopt(argc, argv, "i:r:s")) != -1)
    {
        switch (opt)
        {
        case 'i':
            in_file_name = optarg;
            break;
        case 'r':
            rate = atof(optarg);
            break;
        case 's':
            sweep_rate = TRUE;
            break;
        default:
            //usage();
            exit(2);
            break;
        }
    }
    if ((inhandle = sf_open(in_file_name, SFM_READ, &info)) == NULL)
    {
        printf("    Cannot open audio file '%s'\n", in_file_name);
        exit(2);
    }
    if (info.channels != 1)
    {
        printf("    Unexpected number of channels in audio file '%s'\n", in_file_name);
        exit(2);
    }
    sample_rate = info.samplerate;

    memset(&info, 0, sizeof(info));
    info.frames = 0;
    info.samplerate = sample_rate;
    info.channels = 1;
    info.format = SF_FORMAT_WAV | SF_FORMAT_PCM_16;
    info.sections = 1;
    info.seekable = 1;

    if ((outhandle = sf_open(OUT_FILE_NAME, SFM_WRITE, &info)) == NULL)
    {
        fprintf(stderr, "    Cannot create audio file '%s'\n", OUT_FILE_NAME);
        exit(2);
    }

    if ((time_scale_init(&state, (int) sample_rate, rate)) == NULL)
    {
        fprintf(stderr, "    Cannot start the time scaler\n");
        exit(2);
    }
    max = time_scale_max_output_len(&state, BLOCK_LEN);
    printf("Rate is %f, longest output block is %d\n", rate, max);
    count = 0;
    while ((frames = sf_readf_short(inhandle, in, BLOCK_LEN)))
    {
        new_frames = time_scale(&state, out, in, frames);
        out_frames = sf_writef_short(outhandle, out, new_frames);
        if (out_frames != new_frames)
        {
            fprintf(stderr, "    Error writing audio file\n");
            exit(2);
        }
        if (sweep_rate  &&  ++count > 100)
        {
            if (rate > 0.5f)
            {
                rate -= 0.1f;
                if (rate >= 0.99f  &&  rate <= 1.01f)
                    rate -= 0.1f;
                time_scale_init(&state, SAMPLE_RATE, rate);
                max = time_scale_max_output_len(&state, BLOCK_LEN);
                printf("Rate is %f, longest output block is %d\n", rate, max);
            }
            count = 0;
        }
    }
    if (sf_close(inhandle))
    {
        printf("    Cannot close audio file '%s'\n", in_file_name);
        exit(2);
    }
    if (sf_close(outhandle))
    {
        printf("    Cannot close audio file '%s'\n", OUT_FILE_NAME);
        exit(2);
    }
    return 0;
}
/*- End of function --------------------------------------------------------*/
/*- End of file ------------------------------------------------------------*/