aboutsummaryrefslogtreecommitdiff
path: root/sound/rate.cpp
blob: 9b79e824bae7e2d8390ef317d456c1f7d353f2ba (plain)
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
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
/*
* August 21, 1998
* Copyright 1998 Fabrice Bellard.
*
* [Rewrote completly the code of Lance Norskog And Sundry
* Contributors with a more efficient algorithm.]
*
* This source code is freely redistributable and may be used for
* any purpose.	 This copyright notice must be maintained. 
* Lance Norskog And Sundry Contributors are not responsible for 
* the consequences of using this software.  
*/

/*
 * Sound Tools rate change effect file.
 */

#include "stdafx.h"
#include "rate.h"

#include <math.h>
/*
 * Linear Interpolation.
 *
 * The use of fractional increment allows us to use no buffer. It
 * avoid the problems at the end of the buffer we had with the old
 * method which stored a possibly big buffer of size
 * lcm(in_rate,out_rate).
 *
 * Limited to 16 bit samples and sampling frequency <= 65535 Hz. If
 * the input & output frequencies are equal, a delay of one sample is
 * introduced.	Limited to processing 32-bit count worth of samples.
 *
 * 1 << FRAC_BITS evaluating to zero in several places.	 Changed with
 * an (unsigned long) cast to make it safe.  MarkMLl 2/1/99
 *
 * Replaced all uses of floating point arithmetic by fixed point
 * calculations (Max Horn 2003-07-18).
 */

#define FRAC_BITS 16

/* Private data */

typedef struct ratestuff
{
	unsigned long opos_frac;  /* fractional position of the output stream in input stream unit */
	unsigned long opos;

	unsigned long opos_inc_frac;  /* fractional position increment in the output stream */
	unsigned long opos_inc;

	unsigned long ipos;	 /* position in the input stream (integer) */

	st_sample_t ilast[2]; /* last sample(s) in the input stream (left/right channel) */
} *rate_t;

/*
 * Prepare processing.
 */
int st_rate_start(eff_t effp, st_rate_t inrate, st_rate_t outrate)
{
	rate_t rate = (rate_t) effp->priv;
	unsigned long incr;

	if (inrate == outrate) {
		st_fail("Input and Output rates must be different to use rate effect");
		return (ST_EOF);
	}

	if (inrate >= 65536 || outrate >= 65536) {
		st_fail("rate effect can only handle rates < 65536");
		return (ST_EOF);
	}

	rate->opos_frac = 0;
	rate->opos = 0;

	/* increment */
	incr = (inrate << FRAC_BITS) / outrate;

	rate->opos_inc_frac = incr & ((1UL << FRAC_BITS) - 1);
	rate->opos_inc = incr >> FRAC_BITS;

	rate->ipos = 0;

	rate->ilast[0] = 0;
	rate->ilast[1] = 0;
	return (ST_SUCCESS);
}

/*
 * Processed signed long samples from ibuf to obuf.
 * Return number of samples processed.
 */
template<bool stereo>
int st_rate_flow(eff_t effp, AudioInputStream &input, st_sample_t *obuf, st_size_t *osamp, st_volume_t vol)
{
	rate_t rate = (rate_t) effp->priv;
	st_sample_t *ostart, *oend;
	st_sample_t ilast[2], icur[2], out;
	unsigned long tmp;

	ilast[0] = rate->ilast[0];
	if (stereo)
		ilast[1] = rate->ilast[1];

	ostart = obuf;
	oend = obuf + *osamp * 2;

	// If the input position exceeds the output position, then we aborted the
	// previous conversion run because the output buffer was full. Resume!
	if (rate->ipos > rate->opos)
		goto resume;

	while (obuf < oend && !input.eof()) {

		/* read enough input samples so that ipos > opos */
		while (rate->ipos <= rate->opos) {
			ilast[0] = input.read();
			if (stereo)
				ilast[1] = input.read();
			rate->ipos++;
			/* See if we finished the input buffer yet */

			if (input.eof())
				goto the_end;
		}

		// read the input sample(s)
		icur[0] = input.read();
		if (stereo)
			icur[1] = input.read();

resume:
		// Loop as long as the outpos trails behind, and as long as there is
		// still space in the output buffer.
		while (rate->ipos > rate->opos && obuf < oend) {

			// interpolate
			out = (st_sample_t) (ilast[0] + (((icur[0] - ilast[0]) * rate->opos_frac + (1UL << (FRAC_BITS-1))) >> FRAC_BITS));
			// adjust volume
			out = out * vol / 256;
	
			// output left channel sample
			clampedAdd(*obuf++, out);
			
			if (stereo) {
				// interpolate
				out = (st_sample_t) (ilast[1] + (((icur[1] - ilast[1]) * rate->opos_frac + (1UL << (FRAC_BITS-1))) >> FRAC_BITS));
				// adjust volume
				out = out * vol / 256;
			}
	
			// output right channel sample
			clampedAdd(*obuf++, out);
	
			// Increment output position
			tmp = rate->opos_frac + rate->opos_inc_frac;
			rate->opos = rate->opos + rate->opos_inc + (tmp >> FRAC_BITS);
			rate->opos_frac = tmp & ((1UL << FRAC_BITS) - 1);
		}

		// Increment input position again (for the sample we read now)
		rate->ipos++;
		ilast[0] = icur[0];
		if (stereo)
			ilast[1] = icur[1];
	}

the_end:
	*osamp = (obuf - ostart) / 2;
	rate->ilast[0] = ilast[0];
	if (stereo)
		rate->ilast[1] = ilast[1];
	return (ST_SUCCESS);
}


#ifndef _MSC_VER
#pragma mark -
#endif


LinearRateConverter::LinearRateConverter(st_rate_t inrate, st_rate_t outrate) {
	st_rate_start(&effp, inrate, outrate);
}

int LinearRateConverter::flow(AudioInputStream &input, st_sample_t *obuf, st_size_t *osamp, st_volume_t vol) {
	if (input.isStereo())
		return st_rate_flow<true>(&effp, input, obuf, osamp, vol);
	else
		return st_rate_flow<false>(&effp, input, obuf, osamp, vol);
}

int LinearRateConverter::drain(st_sample_t *obuf, st_size_t *osamp, st_volume_t vol) {
	return (ST_SUCCESS);
}