cog/Plugins/CueSheet/CueSheetDecoder.m

205 lines
4.5 KiB
Objective-C

//
// CueSheetDecoder.m
// CueSheet
//
// Created by Zaphod Beeblebrox on 10/8/07.
// Copyright 2007 __MyCompanyName__. All rights reserved.
//
#import "CueSheetDecoder.h"
#import "CueSheet.h"
#import "CueSheetTrack.h"
#import "CueSheetContainer.h"
@implementation CueSheetDecoder
+ (NSArray *)fileTypes
{
return [CueSheetContainer fileTypes];
}
+ (NSArray *)mimeTypes
{
return [CueSheetContainer mimeTypes];
}
- (NSDictionary *)properties
{
NSMutableDictionary *properties = [[decoder properties] mutableCopy];
//Need to alter length
[properties setObject:[NSNumber numberWithDouble:((trackEnd - [track time]) * 1000)] forKey:@"length"];
return [properties autorelease];
}
- (BOOL)open:(id<CogSource>)s
{
if (![[s url] isFileURL]) {
return NO;
}
NSURL *url = [s url];
[s close];
cuesheet = [CueSheet cueSheetWithFile:[url path]];
[cuesheet retain];
NSArray *tracks = [cuesheet tracks];
int i;
for (i = 0; i < [tracks count]; i++)
{
if ([[[tracks objectAtIndex:i] track] isEqualToString:[url fragment]]){
track = [tracks objectAtIndex:i];
[track retain];
//Kind of a hackish way of accessing outside classes.
source = [NSClassFromString(@"AudioSource") audioSourceForURL:[track url]];
[source retain];
if (![source open:[track url]]) {
NSLog(@"Could not open cuesheet source");
return NO;
}
decoder = [NSClassFromString(@"AudioDecoder") audioDecoderForSource:source];
[decoder retain];
if (![decoder open:source]) {
NSLog(@"Could not open cuesheet decoder");
return NO;
}
CueSheetTrack *nextTrack = nil;
if (i + 1 < [tracks count]) {
nextTrack = [tracks objectAtIndex:i + 1];
}
NSDictionary *properties = [decoder properties];
int bitsPerSample = [[properties objectForKey:@"bitsPerSample"] intValue];
int channels = [[properties objectForKey:@"channels"] intValue];
float sampleRate = [[properties objectForKey:@"sampleRate"] floatValue];
bytesPerSecond = (int)((bitsPerSample/8) * channels * sampleRate);
[decoder seekToTime: [track time] * 1000.0];
if (nextTrack && [[[nextTrack url] absoluteString] isEqualToString:[[track url] absoluteString]]) {
trackEnd = [nextTrack time];
}
else {
trackEnd = [[properties objectForKey:@"length"] doubleValue]/1000.0;
}
//Note: Should register for observations of the decoder, but laziness consumes all.
[self willChangeValueForKey:@"properties"];
[self didChangeValueForKey:@"properties"];
return YES;
}
}
return NO;
}
- (void)close {
if (decoder) {
[decoder close];
[decoder release];
decoder = nil;
}
if (source) {
[source release];
source = nil;
}
if (cuesheet) {
[cuesheet release];
cuesheet = nil;
}
if (track) {
[track release];
track = nil;
}
}
- (BOOL)setTrack:(NSURL *)url
{
if ([[url fragment] intValue] == [[track track] intValue] + 1) {
NSArray *tracks = [cuesheet tracks];
int i;
for (i = 0; i < [tracks count]; i++) {
if ([[[tracks objectAtIndex:i] track] isEqualToString:[url fragment]]){
[track release];
track = [tracks objectAtIndex:i];
[track retain];
CueSheetTrack *nextTrack = nil;
if (i + 1 < [tracks count]) {
nextTrack = [tracks objectAtIndex:i + 1];
}
if (nextTrack && [[[nextTrack url] absoluteString] isEqualToString:[[track url] absoluteString]]) {
trackEnd = [nextTrack time];
}
else {
trackEnd = [[[decoder properties] objectForKey:@"length"] doubleValue]/1000.0;
}
NSLog(@"CHANGING TRACK!");
return YES;
}
}
}
return NO;
}
- (double)seekToTime:(double)time //milliseconds
{
double trackStartMs = [track time] * 1000.0;
double trackEndMs = trackEnd * 1000.0;
if (time > trackEndMs - trackStartMs) {
//need a better way of returning fail.
return -1.0;
}
time += trackStartMs;
bytePosition = (time/1000.0) * bytesPerSecond;
int bitsPerSample = [[[decoder properties] objectForKey:@"bitsPerSample"] intValue];
int channels = [[[decoder properties] objectForKey:@"channels"] intValue];
NSLog(@"Before: %li", bytePosition);
bytePosition -= bytePosition % (bitsPerSample/8 * channels);
NSLog(@"After: %li", bytePosition);
return [decoder seekToTime:time];
}
- (int)fillBuffer:(void *)buf ofSize:(UInt32)size
{
long trackByteEnd = trackEnd * bytesPerSecond;
if (bytePosition + size > trackByteEnd) {
size = trackByteEnd - bytePosition;
}
if (!size) {
return 0;
}
int n = [decoder fillBuffer:buf ofSize:size];
bytePosition += n;
return n;
}
@end