forked from mleeman/ogmtools
-
Notifications
You must be signed in to change notification settings - Fork 0
/
r_mp3.cpp
157 lines (130 loc) · 4.14 KB
/
r_mp3.cpp
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
/*
ogmmerge -- utility for splicing together ogg bitstreams
from component media subtypes
r_mp3.cpp
MP3 demultiplexer module
Written by Moritz Bunkus <[email protected]>
Based on Xiph.org's 'oggmerge' found in their CVS repository
See http://www.xiph.org
Distributed under the GPL
see the file COPYING for details
or visit http://www.gnu.org/copyleft/gpl.html
*/
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <ogg/ogg.h>
#include "ogmmerge.h"
#include "ogmstreams.h"
#include "queue.h"
#include "r_mp3.h"
#include "mp3_common.h"
#define PROBESIZE 8192
int mp3_reader_c::probe_file(FILE *file, off_t size) {
char buf[PROBESIZE];
int pos;
unsigned long header;
mp3_header_t mp3header;
if (size < PROBESIZE)
return 0;
if (fseeko(file, 0, SEEK_SET) != 0)
return 0;
if (fread(buf, 1, PROBESIZE, file) != PROBESIZE) {
fseeko(file, 0, SEEK_SET);
return 0;
}
fseeko(file, 0, SEEK_SET);
pos = find_mp3_header(buf, PROBESIZE, &header);
if (pos < 0)
return 0;
decode_mp3_header(header, &mp3header);
if ((4 - ((header >> 17) & 3)) != 3)
return 0;
pos = find_mp3_header(&buf[pos + mp3header.framesize + 4], PROBESIZE - pos -
4 - mp3header.framesize, &header);
if (pos != 0)
return 0;
return 1;
}
mp3_reader_c::mp3_reader_c(char *fname, audio_sync_t *nasync,
range_t *nrange, char **ncomments) throw (error_c) {
int pos;
unsigned long header;
mp3_header_t mp3header;
if ((file = fopen(fname, "r")) == NULL)
throw error_c("mp3_reader: Could not open source file.");
if (fseeko(file, 0, SEEK_END) != 0)
throw error_c("mp3_reader: Could not seek to end of file.");
size = ftello(file);
if (fseeko(file, 0, SEEK_SET) != 0)
throw error_c("mp3_reader: Could not seek to beginning of file.");
chunk = (unsigned char *)malloc(4096);
if (chunk == NULL)
die("malloc");
if (fread(chunk, 1, 4096, file) != 4096)
throw error_c("mp3_reader: Could not read 4096 bytes.");
if (fseeko(file, 0, SEEK_SET) != 0)
throw error_c("mp3_reader: Could not seek to beginning of file.");
pos = find_mp3_header((char *)chunk, 4096, &header);
if (pos < 0)
throw error_c("mp3_reader: No valid MP3 packet found in the first " \
"4096 bytes.\n");
decode_mp3_header(header, &mp3header);
bytes_processed = 0;
mp3packetizer = new mp3_packetizer_c(mp3_freqs[mp3header.sampling_frequency],
mp3header.stereo,
mp3_tabsel[mp3header.lsf]
[mp3header.bitrate_index],
nasync, nrange, ncomments);
if (verbose)
fprintf(stderr, "Using MP3 demultiplexer for %s.\n+-> Using " \
"MP3 output module for audio stream.\n", fname);
}
mp3_reader_c::~mp3_reader_c() {
if (file != NULL)
fclose(file);
if (chunk != NULL)
free(chunk);
if (mp3packetizer != NULL)
delete mp3packetizer;
}
int mp3_reader_c::read() {
int nread, last_frame;
do {
if (mp3packetizer->page_available())
return EMOREDATA;
nread = fread(chunk, 1, 4096, file);
if (nread <= 0) {
mp3packetizer->produce_eos_packet();
return 0;
}
last_frame = (nread == 4096 ? 0 : 1);
mp3packetizer->process((char *)chunk, nread, last_frame);
bytes_processed += nread;
if (last_frame)
return 0;
} while (1);
}
int mp3_reader_c::serial_in_use(int serial) {
return mp3packetizer->serial_in_use(serial);
}
ogmmerge_page_t *mp3_reader_c::get_header_page(int header_type) {
return mp3packetizer->get_header_page(header_type);
}
ogmmerge_page_t *mp3_reader_c::get_page() {
return mp3packetizer->get_page();
}
void mp3_reader_c::reset() {
if (mp3packetizer != NULL)
mp3packetizer->reset();
}
int mp3_reader_c::display_priority() {
return DISPLAYPRIORITY_HIGH - 1;
}
void mp3_reader_c::display_progress() {
fprintf(stdout, "progress: %lld/%lld bytes (%d%%)\r",
bytes_processed, size,
(int)(bytes_processed * 100L / size));
fflush(stdout);
}