cog/Plugins/vgmstream/vgmstream/VGMDecoder.m

356 lines
11 KiB
Objective-C

//
// VGMDecoder.m
// vgmstream
//
// Created by Christopher Snowhill on 02/25/14.
// Copyright 2014 __NoWork, Inc__. All rights reserved.
//
#import "VGMDecoder.h"
#import "VGMInterface.h"
#import "PlaylistController.h"
@implementation VGMInfoCache
+(id)sharedCache {
static VGMInfoCache *sharedMyCache = nil;
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
sharedMyCache = [[self alloc] init];
});
return sharedMyCache;
}
-(id)init {
if (self = [super init]) {
storage = [[NSMutableDictionary alloc] init];
}
return self;
}
-(void)stuffURL:(NSURL *)url stream:(VGMSTREAM *)stream {
int track_num = [[url fragment] intValue];
int sampleRate = stream->sample_rate;
int channels = stream->channels;
long totalFrames = get_vgmstream_play_samples( 2.0, 10.0, 10.0, stream );
int bitrate = get_vgmstream_average_bitrate(stream);
NSURL *folder = [url URLByDeletingLastPathComponent];
NSURL *tagurl = [folder URLByAppendingPathComponent:@"!tags.m3u" isDirectory:NO];
NSString *filename = [url lastPathComponent];
NSString *album = @"";
NSString *artist = @"";
NSNumber *year = [NSNumber numberWithInt:0];
NSNumber *track = [NSNumber numberWithInt:0];
NSString *title = @"";
NSNumber *rgTrackGain = [NSNumber numberWithInt:0];
NSNumber *rgTrackPeak = [NSNumber numberWithInt:0];
NSNumber *rgAlbumGain = [NSNumber numberWithInt:0];
NSNumber *rgAlbumPeak = [NSNumber numberWithInt:0];
STREAMFILE *tagFile = cogsf_create_from_url(tagurl);
if (tagFile) {
VGMSTREAM_TAGS *tags;
const char *tag_key, *tag_val;
tags = vgmstream_tags_init(&tag_key, &tag_val);
vgmstream_tags_reset(tags, [filename UTF8String]);
while (vgmstream_tags_next_tag(tags, tagFile)) {
NSString *value = [NSString stringWithUTF8String:tag_val];
if (!strncasecmp(tag_key, "REPLAYGAIN_", strlen("REPLAYGAIN_"))) {
if (!strncasecmp(tag_key+strlen("REPLAYGAIN_"), "TRACK_", strlen("TRACK_"))) {
if (!strcasecmp(tag_key+strlen("REPLAYGAIN_TRACK_"), "GAIN")) {
rgTrackGain = [NSNumber numberWithFloat:[value floatValue]];
}
else if (!strcasecmp(tag_key+strlen("REPLAYGAIN_TRACK_"), "PEAK")) {
rgTrackPeak = [NSNumber numberWithFloat:[value floatValue]];
}
}
else if (!strncasecmp(tag_key+strlen("REPLAYGAIN_"), "ALBUM_", strlen("ALBUM_"))) {
if (!strcasecmp(tag_key+strlen("REPLAYGAIN_ALBUM_"), "GAIN")) {
rgAlbumGain = [NSNumber numberWithFloat:[value floatValue]];
}
else if (!strcasecmp(tag_key+strlen("REPLAYGAIN_ALBUM_"), "PEAK")) {
rgAlbumPeak = [NSNumber numberWithFloat:[value floatValue]];
}
}
}
else if (!strcasecmp(tag_key, "ALBUM")) {
album = value;
}
else if (!strcasecmp(tag_key, "ARTIST")) {
artist = value;
}
else if (!strcasecmp(tag_key, "DATE")) {
year = [NSNumber numberWithInt:[value intValue]];
}
else if (!strcasecmp(tag_key, "TRACK") ||
!strcasecmp(tag_key, "TRACKNUMBER")) {
track = [NSNumber numberWithInt:[value intValue]];
}
else if (!strcasecmp(tag_key, "TITLE")) {
title = value;
}
}
vgmstream_tags_close(tags);
close_streamfile(tagFile);
}
NSDictionary * properties =
[NSDictionary dictionaryWithObjectsAndKeys:
[NSNumber numberWithInt:bitrate / 1000], @"bitrate",
[NSNumber numberWithInt:sampleRate], @"sampleRate",
[NSNumber numberWithDouble:totalFrames], @"totalFrames",
[NSNumber numberWithInt:16], @"bitsPerSample",
[NSNumber numberWithBool:NO], @"floatingPoint",
[NSNumber numberWithInt:channels], @"channels",
[NSNumber numberWithBool:YES], @"seekable",
@"host", @"endian",
nil];
if ( [title isEqualToString:@""] ) {
if ( stream->num_streams > 1 ) {
title = [NSString stringWithFormat:@"%@ - %s", [[url URLByDeletingPathExtension] lastPathComponent], stream->stream_name];
} else {
title = [[url URLByDeletingPathExtension] lastPathComponent];
}
}
if ( [track isEqualToNumber:[NSNumber numberWithInt:0]] )
track = [NSNumber numberWithInt:track_num];
NSMutableDictionary * mutableMetadata =
[NSMutableDictionary dictionaryWithObjectsAndKeys:
title, @"title",
track, @"track",
nil];
if ( ![album isEqualToString:@""] )
[mutableMetadata setValue:album forKey:@"album"];
if ( ![artist isEqualToString:@""] )
[mutableMetadata setValue:artist forKey:@"artist"];
if ( ![year isEqualToNumber:[NSNumber numberWithInt:0]] )
[mutableMetadata setValue:year forKey:@"year"];
NSDictionary * metadata = mutableMetadata;
NSDictionary * package =
[NSDictionary dictionaryWithObjectsAndKeys:
properties, @"properties",
metadata, @"metadata",
nil];
@synchronized (self) {
[storage setValue:package forKey:[url absoluteString]];
}
}
-(NSDictionary*)getPropertiesForURL:(NSURL *)url {
NSDictionary *properties = nil;
@synchronized (self) {
NSDictionary * package = [storage objectForKey:[url absoluteString]];
if (package) {
properties = [package objectForKey:@"properties"];
}
}
return properties;
}
-(NSDictionary*)getMetadataForURL:(NSURL *)url {
NSDictionary *metadata = nil;
@synchronized (self) {
NSDictionary * package = [storage objectForKey:[url absoluteString]];
if (package) {
metadata = [package objectForKey:@"metadata"];
}
}
return metadata;
}
@end
@implementation VGMDecoder
- (BOOL)open:(id<CogSource>)s
{
int track_num = [[[s url] fragment] intValue];
NSString * path = [[s url] absoluteString];
NSRange fragmentRange = [path rangeOfString:@"#" options:NSBackwardsSearch];
if (fragmentRange.location != NSNotFound) {
path = [path substringToIndex:fragmentRange.location];
}
NSLog(@"Opening %@ subsong %d", path, track_num);
stream = init_vgmstream_from_cogfile([[path stringByRemovingPercentEncoding] UTF8String], track_num);
if ( !stream )
return NO;
vgmstream_mixing_autodownmix(stream, 6);
playForever = IsRepeatOneSet();
sampleRate = stream->sample_rate;
channels = stream->channels;
totalFrames = get_vgmstream_play_samples( 2.0, 10.0, 10.0, stream );
framesFade = stream->loop_flag ? sampleRate * 10 : 0;
framesLength = totalFrames - framesFade;
vgmstream_cfg_t vcfg = {0};
vcfg.allow_play_forever = 1;
vcfg.play_forever = playForever;
vcfg.loop_count = 2;
vcfg.fade_time = 10;
vcfg.fade_delay = 0;
vcfg.ignore_loop = 0;
vgmstream_apply_config(stream, &vcfg);
framesRead = 0;
bitrate = get_vgmstream_average_bitrate(stream);
[self willChangeValueForKey:@"properties"];
[self didChangeValueForKey:@"properties"];
return YES;
}
- (NSDictionary *)properties
{
return [NSDictionary dictionaryWithObjectsAndKeys:
[NSNumber numberWithInt:bitrate / 1000], @"bitrate",
[NSNumber numberWithInt:sampleRate], @"sampleRate",
[NSNumber numberWithDouble:totalFrames], @"totalFrames",
[NSNumber numberWithInt:16], @"bitsPerSample",
[NSNumber numberWithBool:NO], @"floatingPoint",
[NSNumber numberWithInt:channels], @"channels",
[NSNumber numberWithBool:YES], @"seekable",
@"host", @"endian",
nil];
}
- (int)readAudio:(void *)buf frames:(UInt32)frames
{
BOOL repeatone = IsRepeatOneSet();
if (repeatone != playForever) {
playForever = repeatone;
vgmstream_set_play_forever(stream, repeatone);
}
BOOL loopokay = repeatone && stream->loop_flag;
if (!loopokay) {
if (framesRead >= totalFrames) return 0;
else if (framesRead + frames > totalFrames)
frames = (UInt32)(totalFrames - framesRead);
}
sample * sbuf = (sample *) buf;
render_vgmstream( sbuf, frames, stream );
if ( !repeatone && framesFade && framesRead + frames > framesLength ) {
long fadeStart = (framesLength > framesRead) ? framesLength : framesRead;
long fadeEnd = (framesRead + frames) > totalFrames ? totalFrames : (framesRead + frames);
long fadePos;
long i;
int64_t fadeScale = (int64_t)(totalFrames - fadeStart) * INT_MAX / framesFade;
int64_t fadeStep = INT_MAX / framesFade;
sbuf += (fadeStart - framesRead) * channels;
for (fadePos = fadeStart; fadePos < fadeEnd; ++fadePos) {
for (i = 0; i < channels; ++i) {
sbuf[ i ] = (int16_t)((int64_t)(sbuf[ i ]) * fadeScale / INT_MAX);
}
sbuf += channels;
fadeScale -= fadeStep;
if (fadeScale <= 0) break;
}
frames = (UInt32)(fadePos - framesRead);
}
framesRead += frames;
return frames;
}
- (long)seek:(long)frame
{
// Constrain the seek offset to within the loop, if any
if(stream->loop_flag && (stream->loop_end_sample - stream->loop_start_sample) && frame >= stream->loop_end_sample) {
frame -= stream->loop_start_sample;
frame %= (stream->loop_end_sample - stream->loop_start_sample);
frame += stream->loop_start_sample;
}
if (frame < framesRead) {
reset_vgmstream( stream );
framesRead = 0;
}
while (framesRead < frame) {
sample buffer[1024];
long max_sample_count = 1024 / channels;
long samples_to_skip = frame - framesRead;
if ( samples_to_skip > max_sample_count )
samples_to_skip = max_sample_count;
render_vgmstream( buffer, (int)samples_to_skip, stream );
framesRead += samples_to_skip;
}
return framesRead;
}
- (void)close
{
close_vgmstream( stream );
stream = NULL;
}
- (void)dealloc
{
[self close];
}
+ (NSArray *)fileTypes
{
NSMutableArray *array = [[NSMutableArray alloc] init];
size_t count;
const char ** formats = vgmstream_get_formats(&count);
for (size_t i = 0; i < count; ++i)
{
[array addObject:[NSString stringWithUTF8String:formats[i]]];
}
return array;
}
+ (NSArray *)mimeTypes
{
return nil;
}
+ (float)priority
{
return 0.0;
}
@end