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