add more informative error info for some common errors; also, no not exit with error...
[flac.git] / src / flac / utils.c
1 /* flac - Command-line FLAC encoder/decoder
2  * Copyright (C) 2002,2003,2004  Josh Coalson
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
17  */
18
19 #if HAVE_CONFIG_H
20 #  include <config.h>
21 #endif
22
23 #include "utils.h"
24 #include "FLAC/assert.h"
25 #include <math.h>
26 #include <stdarg.h>
27 #include <stdlib.h>
28 #include <string.h>
29
30 int flac__utils_verbosity_ = 2;
31
32 static FLAC__bool local__parse_uint64_(const char *s, FLAC__uint64 *value)
33 {
34         FLAC__uint64 ret = 0;
35         char c;
36
37         if(*s == '\0')
38                 return false;
39
40         while('\0' != (c = *s++))
41                 if(c >= '0' && c <= '9')
42                         ret = ret * 10 + (c - '0');
43                 else
44                         return false;
45
46         *value = ret;
47         return true;
48 }
49
50 static FLAC__bool local__parse_timecode_(const char *s, double *value)
51 {
52         double ret;
53         unsigned i;
54         char c;
55
56         /* parse [0-9][0-9]*: */
57         c = *s++;
58         if(c >= '0' && c <= '9')
59                 i = (c - '0');
60         else
61                 return false;
62         while(':' != (c = *s++)) {
63                 if(c >= '0' && c <= '9')
64                         i = i * 10 + (c - '0');
65                 else
66                         return false;
67         }
68         ret = (double)i * 60.;
69
70         /* parse [0-9]*[.]?[0-9]* i.e. a sign-less rational number */
71         if(strspn(s, "1234567890.") != strlen(s))
72                 return false;
73         {
74                 const char *p = strchr(s, '.');
75                 if(p && 0 != strchr(++p, '.'))
76                         return false;
77         }
78         ret += atof(s);
79
80         *value = ret;
81         return true;
82 }
83
84 static FLAC__bool local__parse_cue_(const char *s, const char *end, unsigned *track, unsigned *index)
85 {
86         FLAC__bool got_track = false, got_index = false;
87         unsigned t = 0, i = 0;
88         char c;
89
90         while(end? s < end : *s != '\0') {
91                 c = *s++;
92                 if(c >= '0' && c <= '9') {
93                         t = t * 10 + (c - '0');
94                         got_track = true;
95                 }
96                 else if(c == '.')
97                         break;
98                 else
99                         return false;
100         }
101         while(end? s < end : *s != '\0') {
102                 c = *s++;
103                 if(c >= '0' && c <= '9') {
104                         i = i * 10 + (c - '0');
105                         got_index = true;
106                 }
107                 else
108                         return false;
109         }
110         *track = t;
111         *index = i;
112         return got_track && got_index;
113 }
114
115 /*
116  * @@@ this only works with sorted cuesheets (the spec strongly recommends but
117  * does not require sorted cuesheets).  but if it's not sorted, picking a
118  * nearest cue point has no significance.
119  */
120 static FLAC__uint64 local__find_closest_cue_(const FLAC__StreamMetadata_CueSheet *cuesheet, unsigned track, unsigned index, FLAC__uint64 total_samples, FLAC__bool look_forward)
121 {
122         int t, i;
123         if(look_forward) {
124                 for(t = 0; t < (int)cuesheet->num_tracks; t++)
125                         for(i = 0; i < (int)cuesheet->tracks[t].num_indices; i++)
126                                 if(cuesheet->tracks[t].number > track || (cuesheet->tracks[t].number == track && cuesheet->tracks[t].indices[i].number >= index))
127                                         return cuesheet->tracks[t].offset + cuesheet->tracks[t].indices[i].offset;
128                 return total_samples;
129         }
130         else {
131                 for(t = (int)cuesheet->num_tracks - 1; t >= 0; t--)
132                         for(i = (int)cuesheet->tracks[t].num_indices - 1; i >= 0; i--)
133                                 if(cuesheet->tracks[t].number < track || (cuesheet->tracks[t].number == track && cuesheet->tracks[t].indices[i].number <= index))
134                                         return cuesheet->tracks[t].offset + cuesheet->tracks[t].indices[i].offset;
135                 return 0;
136         }
137 }
138
139 void flac__utils_printf(FILE *stream, int level, const char *format, ...)
140 {
141         if(flac__utils_verbosity_ >= level) {
142                 va_list args;
143
144                 FLAC__ASSERT(0 != format);
145
146                 va_start(args, format);
147
148                 (void) vfprintf(stream, format, args);
149
150                 va_end(args);
151         }
152 }
153
154 #ifdef FLAC__VALGRIND_TESTING
155 size_t flac__utils_fwrite(const void *ptr, size_t size, size_t nmemb, FILE *stream)
156 {
157         size_t ret = fwrite(ptr, size, nmemb, stream);
158         if(!ferror(stream))
159                 fflush(stream);
160         return ret;
161 }
162 #endif
163
164 FLAC__bool flac__utils_parse_skip_until_specification(const char *s, utils__SkipUntilSpecification *spec)
165 {
166         FLAC__uint64 val;
167         FLAC__bool is_negative = false;
168
169         FLAC__ASSERT(0 != spec);
170
171         spec->is_relative = false;
172         spec->value_is_samples = true;
173         spec->value.samples = 0;
174
175         if(0 != s) {
176                 if(s[0] == '-') {
177                         is_negative = true;
178                         spec->is_relative = true;
179                         s++;
180                 }
181                 else if(s[0] == '+') {
182                         spec->is_relative = true;
183                         s++;
184                 }
185
186                 if(local__parse_uint64_(s, &val)) {
187                         spec->value_is_samples = true;
188                         spec->value.samples = (FLAC__int64)val;
189                         if(is_negative)
190                                 spec->value.samples = -(spec->value.samples);
191                 }
192                 else {
193                         double d;
194                         if(!local__parse_timecode_(s, &d))
195                                 return false;
196                         spec->value_is_samples = false;
197                         spec->value.seconds = d;
198                         if(is_negative)
199                                 spec->value.seconds = -(spec->value.seconds);
200                 }
201         }
202
203         return true;
204 }
205
206 void flac__utils_canonicalize_skip_until_specification(utils__SkipUntilSpecification *spec, unsigned sample_rate)
207 {
208         FLAC__ASSERT(0 != spec);
209         if(!spec->value_is_samples) {
210                 spec->value.samples = (FLAC__int64)(spec->value.seconds * (double)sample_rate);
211                 spec->value_is_samples = true;
212         }
213 }
214
215 FLAC__bool flac__utils_parse_cue_specification(const char *s, utils__CueSpecification *spec)
216 {
217         const char *start = s, *end = 0;
218
219         FLAC__ASSERT(0 != spec);
220
221         spec->has_start_point = spec->has_end_point = false;
222
223         s = strchr(s, '-');
224
225         if(0 != s) {
226                 if(s == start)
227                         start = 0;
228                 end = s+1;
229                 if(*end == '\0')
230                         end = 0;
231         }
232
233         if(start) {
234                 if(!local__parse_cue_(start, s, &spec->start_track, &spec->start_index))
235                         return false;
236                 spec->has_start_point = true;
237         }
238
239         if(end) {
240                 if(!local__parse_cue_(end, 0, &spec->end_track, &spec->end_index))
241                         return false;
242                 spec->has_end_point = true;
243         }
244
245         return true;
246 }
247
248 void flac__utils_canonicalize_cue_specification(const utils__CueSpecification *cue_spec, const FLAC__StreamMetadata_CueSheet *cuesheet, FLAC__uint64 total_samples, utils__SkipUntilSpecification *skip_spec, utils__SkipUntilSpecification *until_spec)
249 {
250         FLAC__ASSERT(0 != cue_spec);
251         FLAC__ASSERT(0 != cuesheet);
252         FLAC__ASSERT(0 != total_samples);
253         FLAC__ASSERT(0 != skip_spec);
254         FLAC__ASSERT(0 != until_spec);
255
256         skip_spec->is_relative = false;
257         skip_spec->value_is_samples = true;
258
259         until_spec->is_relative = false;
260         until_spec->value_is_samples = true;
261
262         if(cue_spec->has_start_point)
263                 skip_spec->value.samples = local__find_closest_cue_(cuesheet, cue_spec->start_track, cue_spec->start_index, total_samples, /*look_forward=*/false);
264         else
265                 skip_spec->value.samples = 0;
266
267         if(cue_spec->has_end_point)
268                 until_spec->value.samples = local__find_closest_cue_(cuesheet, cue_spec->end_track, cue_spec->end_index, total_samples, /*look_forward=*/true);
269         else
270                 until_spec->value.samples = total_samples;
271 }