psp readme, some adjustments
[libpicofe.git] / psp / mp3.c
CommitLineData
2951214e 1#include <stdio.h>
a6df06b7 2#include <string.h>
2951214e 3
a6df06b7 4#include <pspkernel.h>
5#include <pspsdk.h>
6#include <pspaudiocodec.h>
7
8#include "../../Pico/PicoInt.h"
9#include "../../Pico/sound/mix.h"
10#include "../common/lprintf.h"
11
12int mp3_last_error = 0;
13
14static SceUID thread_job_sem = -1;
15static SceUID thread_busy_sem = -1;
16static int thread_exit = 0;
17
18// MPEG-1, layer 3
19static int bitrates[] = { 0, 32, 40, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, 0 };
20//static int samplerates[] = { 44100, 48000, 32000, 0 };
21
22#define MIN_INFRAME_SIZE 96
23#define IN_BUFFER_SIZE (2*1024)
24
25static unsigned long mp3_codec_struct[65] __attribute__((aligned(64)));
26
27static unsigned char mp3_src_buffer[2][IN_BUFFER_SIZE] __attribute__((aligned(64)));
28static short mp3_mix_buffer[2][1152*2] __attribute__((aligned(64)));
29static int working_buf = 0;
30
31static const char *mp3_fname = NULL;
32static SceUID mp3_handle = -1;
33static int mp3_src_pos = 0, mp3_src_size = 0;
34
35static int decode_thread(SceSize args, void *argp);
36
37
38static void psp_sem_lock(SceUID sem)
2951214e 39{
a6df06b7 40 int ret = sceKernelWaitSema(sem, 1, 0);
41 if (ret < 0) lprintf("sceKernelWaitSema(%08x) failed with %08x\n", sem, ret);
42}
43
44static void psp_sem_unlock(SceUID sem)
45{
46 int ret = sceKernelSignalSema(sem, 1);
47 if (ret < 0) lprintf("sceKernelSignalSema(%08x) failed with %08x\n", sem, ret);
48}
49
50// only accepts MPEG-1, layer3
51static int find_sync_word(unsigned char *data, int len)
52{
53 int i;
54 for (i = 0; i < len-1; i++)
55 {
56 if ( data[i+0] != 0xff) continue;
57 if ((data[i+1] & 0xfe) == 0xfa) return i;
58 i++;
59 }
60 return -1;
61}
62
63static int read_next_frame(int which_buffer)
64{
65 int i, bytes_read, frame_offset;
66 int bitrate, padding, frame_size = 0;
67
68 for (i = 0; i < 32; i++)
69 {
70 bytes_read = sceIoRead(mp3_handle, mp3_src_buffer[which_buffer], sizeof(mp3_src_buffer[which_buffer]));
71 mp3_src_pos += bytes_read;
72 if (bytes_read < MIN_INFRAME_SIZE) {
73 mp3_src_pos = mp3_src_size;
74 return 0; // EOF/IO failure
75 }
76 frame_offset = find_sync_word(mp3_src_buffer[which_buffer], bytes_read);
77 if (frame_offset < 0) {
78 lprintf("missing syncword, foffs=%i\n", mp3_src_pos - bytes_read);
79 mp3_src_pos--;
80 sceIoLseek32(mp3_handle, mp3_src_pos, PSP_SEEK_SET);
81 continue;
82 }
83 if (bytes_read - frame_offset < 4) {
84 lprintf("syncword @ EOB, foffs=%i\n", mp3_src_pos - bytes_read);
85 mp3_src_pos--;
86 sceIoLseek32(mp3_handle, mp3_src_pos, PSP_SEEK_SET);
87 continue;
88 }
89
90 bitrate = mp3_src_buffer[which_buffer][frame_offset+2] >> 4;
91 padding = (mp3_src_buffer[which_buffer][frame_offset+2] & 2) >> 1;
92
93 frame_size = 144000*bitrates[bitrate]/44100 + padding;
94 if (frame_size <= 0) {
95 lprintf("bad frame, foffs=%i\n", mp3_src_pos - bytes_read);
96 continue; // bad frame
97 }
98
99 if (bytes_read - frame_offset < frame_size) {
100 lprintf("unfit, foffs=%i\n", mp3_src_pos - bytes_read);
101 mp3_src_pos -= bytes_read - frame_offset;
102 sceIoLseek32(mp3_handle, mp3_src_pos, PSP_SEEK_SET);
103 continue; // didn't fit, re-read..
104 }
105
106 if (frame_offset) {
60a10527 107 //lprintf("unaligned, foffs=%i, offs=%i\n", mp3_src_pos - bytes_read, frame_offset);
a6df06b7 108 memmove(mp3_src_buffer[which_buffer], mp3_src_buffer[which_buffer] + frame_offset, frame_size);
109 }
110
111 // align for next frame read
112 mp3_src_pos -= bytes_read - (frame_offset + frame_size);
113 sceIoLseek32(mp3_handle, mp3_src_pos, PSP_SEEK_SET);
114
115 break;
116 }
117
118 return frame_size > 0 ? frame_size : -1;
119}
120
121
122static SceUID load_start_module(const char *prxname)
123{
124 SceUID mod = pspSdkLoadStartModule(prxname, PSP_MEMORY_PARTITION_KERNEL);
125 if (mod < 0)
126 lprintf("failed to load %s: %08x\n", prxname, mod);
127 return mod;
128}
129
130
131int mp3_init(void)
132{
133 SceUID thid, mod;
134 int ret;
135
136 /* load modules */
137 /* <= 1.5 (and probably some other, not sure which) fw need this to for audiocodec to work,
138 * so if it fails, assume we are just on new enough firmware and continue.. */
139 load_start_module("flash0:/kd/me_for_vsh.prx");
140
141 if (sceKernelDevkitVersion() < 0x02070010)
142 mod = load_start_module("flash0:/kd/audiocodec.prx");
143 else mod = load_start_module("flash0:/kd/avcodec.prx");
144 if (mod < 0) {
145 ret = mod = load_start_module("flash0:/kd/audiocodec_260.prx"); // last chance..
146 if (mod < 0) goto fail;
147 }
148
149 /* audiocodec init */
150 memset(mp3_codec_struct, 0, sizeof(mp3_codec_struct));
151 ret = sceAudiocodecCheckNeedMem(mp3_codec_struct, 0x1002);
152 if (ret < 0) {
153 lprintf("sceAudiocodecCheckNeedMem failed with %08x\n", ret);
154 goto fail;
155 }
156
157 ret = sceAudiocodecGetEDRAM(mp3_codec_struct, 0x1002);
158 if (ret < 0) {
159 lprintf("sceAudiocodecGetEDRAM failed with %08x\n", ret);
160 goto fail;
161 }
162
163 ret = sceAudiocodecInit(mp3_codec_struct, 0x1002);
164 if (ret < 0) {
165 lprintf("sceAudiocodecInit failed with %08x\n", ret);
166 goto fail1;
167 }
168
169 /* thread and stuff */
170 thread_job_sem = sceKernelCreateSema("p_mp3job_sem", 0, 0, 1, NULL);
171 if (thread_job_sem < 0) {
172 lprintf("sceKernelCreateSema() failed: %08x\n", thread_job_sem);
173 ret = thread_job_sem;
174 goto fail1;
175 }
176
177 thread_busy_sem = sceKernelCreateSema("p_mp3busy_sem", 0, 1, 1, NULL);
178 if (thread_busy_sem < 0) {
179 lprintf("sceKernelCreateSema() failed: %08x\n", thread_busy_sem);
180 ret = thread_busy_sem;
181 goto fail2;
182 }
183
a6df06b7 184 thread_exit = 0;
185 thid = sceKernelCreateThread("mp3decode_thread", decode_thread, 30, 0x2000, 0, 0); /* use slightly higher prio then main */
186 if (thid < 0) {
187 lprintf("failed to create decode thread: %08x\n", thid);
188 ret = thid;
189 goto fail3;
190 }
191 ret = sceKernelStartThread(thid, 0, 0);
192 if (ret < 0) {
193 lprintf("failed to start decode thread: %08x\n", ret);
194 goto fail3;
195 }
196
197 mp3_last_error = 0;
2951214e 198 return 0;
a6df06b7 199
200fail3:
201 sceKernelDeleteSema(thread_busy_sem);
202 thread_busy_sem = -1;
203fail2:
204 sceKernelDeleteSema(thread_job_sem);
205 thread_job_sem = -1;
206fail1:
207 sceAudiocodecReleaseEDRAM(mp3_codec_struct);
208fail:
209 mp3_last_error = ret;
210 return 1;
2951214e 211}
212
a6df06b7 213void mp3_deinit(void)
2951214e 214{
a6df06b7 215 lprintf("deinit\n");
216 thread_exit = 1;
217 psp_sem_lock(thread_busy_sem);
218 psp_sem_unlock(thread_busy_sem);
219
220 sceKernelSignalSema(thread_job_sem, 1);
221 sceKernelDelayThread(100*1000);
222
223 if (mp3_handle >= 0) sceIoClose(mp3_handle);
224 mp3_handle = -1;
225 mp3_fname = NULL;
226
227 sceKernelDeleteSema(thread_busy_sem);
228 thread_busy_sem = -1;
229 sceKernelDeleteSema(thread_job_sem);
230 thread_job_sem = -1;
231 sceAudiocodecReleaseEDRAM(mp3_codec_struct);
2951214e 232}
233
a6df06b7 234// may overflow stack?
235static int decode_thread(SceSize args, void *argp)
2951214e 236{
a6df06b7 237 int ret, frame_size;
238
239 lprintf("decode_thread started with id %08x, priority %i\n",
240 sceKernelGetThreadId(), sceKernelGetThreadCurrentPriority());
241
242 while (!thread_exit)
243 {
244 psp_sem_lock(thread_job_sem);
245 if (thread_exit) break;
246
247 psp_sem_lock(thread_busy_sem);
248 //lprintf("{ job\n");
249
250 frame_size = read_next_frame(working_buf);
251 if (frame_size > 0)
252 {
253 mp3_codec_struct[6] = (unsigned long)mp3_src_buffer[working_buf];
254 mp3_codec_struct[8] = (unsigned long)mp3_mix_buffer[working_buf];
255 mp3_codec_struct[7] = mp3_codec_struct[10] = frame_size;
256 mp3_codec_struct[9] = 1152 * 4;
257
258 ret = sceAudiocodecDecode(mp3_codec_struct, 0x1002);
259 if (ret < 0) lprintf("sceAudiocodecDecode failed with %08x\n", ret);
260 }
261
262 //lprintf("} job\n");
263 psp_sem_unlock(thread_busy_sem);
264 }
265
266 lprintf("leaving decode thread\n");
267 sceKernelExitDeleteThread(0);
2951214e 268 return 0;
269}
270
a6df06b7 271
272int mp3_get_bitrate(FILE *f, int size)
273{
5ecedd0c 274 int ret, retval = -1, sample_rate, bitrate;
a6df06b7 275 // filenames are stored instead handles in PSP, due to stupid max open file limit
276 char *fname = (char *)f;
277
278 /* make sure thread is not busy.. */
279 psp_sem_lock(thread_busy_sem);
280
281 if (mp3_handle >= 0) sceIoClose(mp3_handle);
282 mp3_handle = sceIoOpen(fname, PSP_O_RDONLY, 0777);
283 if (mp3_handle < 0) {
284 lprintf("sceIoOpen(%s) failed\n", fname);
285 goto end;
286 }
287
288 mp3_src_pos = 0;
289 ret = read_next_frame(0);
290 if (ret <= 0) {
291 lprintf("read_next_frame() failed\n");
292 goto end;
293 }
294 sample_rate = (mp3_src_buffer[0][2] & 0x0c) >> 2;
295 bitrate = mp3_src_buffer[0][2] >> 4;
296
297 if (sample_rate != 0) {
298 lprintf("unsupported samplerate\n");
299 goto end; // only 44kHz supported..
300 }
301 bitrate = bitrates[bitrate];
302 if (bitrate == 0) {
303 lprintf("unsupported bitrate\n");
304 goto end;
305 }
306
307 /* looking good.. */
5ecedd0c 308 retval = bitrate;
a6df06b7 309end:
310 if (mp3_handle >= 0) sceIoClose(mp3_handle);
311 mp3_handle = -1;
312 mp3_fname = NULL;
313 psp_sem_unlock(thread_busy_sem);
5ecedd0c 314 if (retval < 0) mp3_last_error = -1; // remember we had a problem..
315 return retval;
a6df06b7 316}
317
318
319static int mp3_job_started = 0, mp3_samples_ready = 0, mp3_buffer_offs = 0, mp3_play_bufsel = 0;
320
321void mp3_start_play(FILE *f, int pos)
322{
323 char *fname = (char *)f;
324
325 lprintf("mp3_start_play(%s) @ %i\n", fname, pos);
326 psp_sem_lock(thread_busy_sem);
327
60a10527 328 if (mp3_fname != fname || mp3_handle < 0)
a6df06b7 329 {
330 if (mp3_handle >= 0) sceIoClose(mp3_handle);
331 mp3_handle = sceIoOpen(fname, PSP_O_RDONLY, 0777);
332 if (mp3_handle < 0) {
333 lprintf("sceIoOpen(%s) failed\n", fname);
334 psp_sem_unlock(thread_busy_sem);
335 return;
336 }
337 mp3_src_size = sceIoLseek32(mp3_handle, 0, PSP_SEEK_END);
338 mp3_fname = fname;
339 }
340
341 // seek..
342 mp3_src_pos = (int) (((float)pos / 1023.0f) * (float)mp3_src_size);
343 sceIoLseek32(mp3_handle, mp3_src_pos, PSP_SEEK_SET);
344 lprintf("seek %i: %i/%i\n", pos, mp3_src_pos, mp3_src_size);
345
346 mp3_job_started = 1;
347 mp3_samples_ready = mp3_buffer_offs = mp3_play_bufsel = 0;
348 working_buf = 0;
349
350 /* send a request to decode first frame */
351 psp_sem_unlock(thread_busy_sem);
352 psp_sem_unlock(thread_job_sem);
353 sceKernelDelayThread(1); // reschedule
354}
355
356
2951214e 357void mp3_update(int *buffer, int length, int stereo)
358{
a6df06b7 359 int length_mp3;
360
361 // playback was started, track not ended
362 if (mp3_handle < 0 || mp3_src_pos >= mp3_src_size) return;
363
364 length_mp3 = length;
365 if (PsndRate == 22050) length_mp3 <<= 1; // mp3s are locked to 44100Hz stereo
366 else if (PsndRate == 11025) length_mp3 <<= 2; // so make length 44100ish
367
368 /* do we have to wait? */
369 if (mp3_job_started && mp3_samples_ready < length_mp3)
370 {
371 psp_sem_lock(thread_busy_sem);
372 psp_sem_unlock(thread_busy_sem);
373 mp3_job_started = 0;
374 mp3_samples_ready += 1152;
375 }
376
377 /* mix mp3 data, only stereo */
378 if (mp3_samples_ready >= length_mp3)
379 {
380 int shr = 0;
381 void (*mix_samples)(int *dest_buf, short *mp3_buf, int count) = mix_16h_to_32;
382 if (PsndRate == 22050) { mix_samples = mix_16h_to_32_s1; shr = 1; }
383 else if (PsndRate == 11025) { mix_samples = mix_16h_to_32_s2; shr = 2; }
384
385 if (1152 - mp3_buffer_offs >= length_mp3) {
386 mix_samples(buffer, mp3_mix_buffer[mp3_play_bufsel] + mp3_buffer_offs*2, length<<1);
387
388 mp3_buffer_offs += length_mp3;
389 } else {
390 // collect samples from both buffers..
391 int left = 1152 - mp3_buffer_offs;
392 if (mp3_play_bufsel == 0)
393 {
394 mix_samples(buffer, mp3_mix_buffer[0] + mp3_buffer_offs*2, length<<1);
395 mp3_buffer_offs = length_mp3 - left;
396 mp3_play_bufsel = 1;
397 } else {
398 mix_samples(buffer, mp3_mix_buffer[1] + mp3_buffer_offs*2, (left>>shr)<<1);
399 mp3_buffer_offs = length_mp3 - left;
400 mix_samples(buffer + ((left>>shr)<<1),
401 mp3_mix_buffer[0], (mp3_buffer_offs>>shr)<<1);
402 mp3_play_bufsel = 0;
403 }
404 }
405 mp3_samples_ready -= length_mp3;
406 }
407
408 // ask to decode more if we already can
409 if (!mp3_job_started)
410 {
411 mp3_job_started = 1;
412 working_buf ^= 1;
413
414 /* next job.. */
415 psp_sem_lock(thread_busy_sem); // just in case
416 psp_sem_unlock(thread_busy_sem);
417 psp_sem_unlock(thread_job_sem);
418 sceKernelDelayThread(1);
419 }
420}
421
422
423int mp3_get_offset(void) // 0-1023
424{
425 unsigned int offs1024 = 0;
426 int cdda_on;
427
428 cdda_on = (PicoMCD & 1) && (PicoOpt&0x800) && !(Pico_mcd->s68k_regs[0x36] & 1) &&
429 (Pico_mcd->scd.Status_CDC & 1) && mp3_handle >= 0;
430
431 if (cdda_on) {
432 offs1024 = mp3_src_pos << 7;
433 offs1024 /= mp3_src_size >> 3;
434 }
435 lprintf("offs1024=%u (%i/%i)\n", offs1024, mp3_src_pos, mp3_src_size);
436
437 return offs1024;
2951214e 438}
439
a6df06b7 440