202 lines
4.0 KiB
Objective-C
202 lines
4.0 KiB
Objective-C
//
|
|
// WMADecoder.m
|
|
// WMA
|
|
//
|
|
// Created by Andre Reffhaug on 2/26/08.
|
|
// Copyright 2008 __MyCompanyName__. All rights reserved.
|
|
//
|
|
|
|
// test
|
|
#import "WMADecoder.h"
|
|
|
|
#define ST_BUFF 2048
|
|
|
|
@implementation WMADecoder
|
|
|
|
|
|
|
|
|
|
- (BOOL)open:(id<CogSource>)s
|
|
{
|
|
source = [s retain];
|
|
|
|
|
|
int err, i;
|
|
const char *filename = [[[source url] path] UTF8String];
|
|
|
|
ic = NULL;
|
|
numSamples = 0;
|
|
samplePos = 0;
|
|
sampleBuffer = av_malloc(AVCODEC_MAX_AUDIO_FRAME_SIZE);
|
|
// register all available codecs
|
|
av_register_all();
|
|
|
|
|
|
NSLog(@"lolbots: %s", filename);
|
|
|
|
|
|
err = av_open_input_file(&ic, filename, NULL, 0, NULL);
|
|
|
|
if (err < 0)
|
|
NSLog(@"Opening .WMA file failed horribly: %d", err);
|
|
|
|
for(i = 0; i < ic->nb_streams; i++) {
|
|
c = &ic->streams[i]->codec;
|
|
if(c->codec_type == CODEC_TYPE_AUDIO)
|
|
{
|
|
NSLog(@"audio codec found");
|
|
break;
|
|
}
|
|
}
|
|
|
|
av_find_stream_info(ic);
|
|
|
|
codec = avcodec_find_decoder(c->codec_id);
|
|
if (!codec) {
|
|
NSLog(@"codec not found");
|
|
return NO;
|
|
}
|
|
|
|
if (avcodec_open(c, codec) < 0) {
|
|
NSLog(@"could not open codec");
|
|
return NO;
|
|
}
|
|
|
|
dump_format(ic, 0, filename, 0);
|
|
|
|
if (ic->title[0] != '\0')
|
|
NSLog(@"Title: %s", ic->title);
|
|
if (ic->author[0] != '\0')
|
|
NSLog(@"Author: %s", ic->author);
|
|
if (ic->album[0] != '\0')
|
|
NSLog(@"Album: %s", ic->album);
|
|
if (ic->year != 0)
|
|
NSLog(@"Year: %d", ic->year);
|
|
if (ic->track != 0)
|
|
NSLog(@"Track: %d", ic->track);
|
|
if (ic->genre[0] != '\0')
|
|
NSLog(@"Genre: %s", ic->genre);
|
|
if (ic->copyright[0] != '\0')
|
|
NSLog(@"Copyright: %s", ic->copyright);
|
|
if (ic->comment[0] != '\0')
|
|
NSLog(@"Comments: %s", ic->comment);
|
|
|
|
NSLog(@"bitrate: %d", ic->bit_rate);
|
|
NSLog(@"sample rate: %d", c->sample_rate);
|
|
NSLog(@"channels: %d", c->channels);
|
|
|
|
channels = c->channels;
|
|
bitrate = ic->bit_rate;
|
|
bitsPerSample = c->channels * 8;
|
|
totalFrames = c->sample_rate * (ic->duration/1000000LL);
|
|
frequency = 0;
|
|
|
|
return YES;
|
|
|
|
}
|
|
|
|
- (void)close
|
|
{
|
|
avcodec_close(c);
|
|
av_close_input_file(ic);
|
|
av_free(sampleBuffer);
|
|
|
|
[source close];
|
|
[source release];
|
|
}
|
|
|
|
- (int)readAudio:(void *)buf frames:(UInt32)frames
|
|
{
|
|
uint8_t *inbuf_ptr;
|
|
int size, out_size, len;
|
|
AVPacket framePacket;
|
|
int framesRead = 0;
|
|
|
|
int bytesPerFrame = (bitsPerSample/8) * channels;
|
|
|
|
|
|
while (frames > 0)
|
|
{
|
|
if (samplePos < numSamples)
|
|
{
|
|
int samplesLeft;
|
|
samplesLeft = numSamples - samplePos;
|
|
|
|
if (samplesLeft > frames)
|
|
samplesLeft = frames;
|
|
|
|
memcpy(buf, sampleBuffer + (samplePos * bytesPerFrame), samplesLeft * bytesPerFrame);
|
|
buf += samplesLeft * bytesPerFrame;
|
|
framesRead += samplesLeft;
|
|
frames -= samplesLeft;
|
|
samplePos += samplesLeft;
|
|
}
|
|
if (frames > 0)
|
|
{
|
|
if (av_read_frame(ic, &framePacket) < 0)
|
|
{
|
|
NSLog(@"Uh oh...");
|
|
break;
|
|
}
|
|
|
|
size = framePacket.size;
|
|
inbuf_ptr = framePacket.data;
|
|
|
|
len = avcodec_decode_audio(c, (void *)sampleBuffer, &numSamples,
|
|
inbuf_ptr, size);
|
|
|
|
if (len < 0)
|
|
break;
|
|
|
|
if (out_size <= 0)
|
|
continue;
|
|
|
|
numSamples /= bytesPerFrame;
|
|
samplePos = 0;
|
|
|
|
// the frame packet needs to be freed before we av_read_frame a new one
|
|
if (framePacket.data)
|
|
av_free_packet(&framePacket);
|
|
|
|
}
|
|
}
|
|
|
|
return framesRead;
|
|
|
|
}
|
|
|
|
- (long)seek:(long)frame
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
|
|
- (NSDictionary *)properties
|
|
{
|
|
return [NSDictionary dictionaryWithObjectsAndKeys:
|
|
[NSNumber numberWithInt:channels], @"channels",
|
|
[NSNumber numberWithInt:bitsPerSample], @"bitsPerSample",
|
|
[NSNumber numberWithFloat:frequency], @"sampleRate",
|
|
[NSNumber numberWithDouble:totalFrames], @"totalFrames",
|
|
[NSNumber numberWithInt:bitrate], @"bitrate",
|
|
[NSNumber numberWithBool:([source seekable] && seekable)], @"seekable",
|
|
nil];
|
|
}
|
|
|
|
|
|
+ (NSArray *)fileTypes
|
|
{
|
|
return [NSArray arrayWithObjects:@"wma",nil];
|
|
}
|
|
|
|
+ (NSArray *)mimeTypes
|
|
{
|
|
return [NSArray arrayWithObjects:@"application/wma", @"application/x-wma", @"audio/x-wma", nil];
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
@end
|