~njh-aelius/maxosx/submit-discid

/*

 *  $Id$

 *

 *  Copyright (C) 2005 - 2007 Stephen F. Booth <me@sbooth.org>

 *

 *  This program is free software; you can redistribute it and/or modify

 *  it under the terms of the GNU General Public License as published by

 *  the Free Software Foundation; either version 2 of the License, or

 *  (at your option) any later version.

 *

 *  This program is distributed in the hope that it will be useful,

 *  but WITHOUT ANY WARRANTY; without even the implied warranty of

 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 *  GNU General Public License for more details.

 *

 *  You should have received a copy of the GNU General Public License

 *  along with this program; if not, write to the Free Software

 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 */

#import "CompactDisc.h"

#import "Drive.h"

#import "LogController.h"

#include <discid/discid.h>

#include <IOKit/storage/IOCDTypes.h>

@implementation CompactDisc

- (id) initWithDeviceName:(NSString *)deviceName;

{

	if((self = [super init])) {

		unsigned				i;

		unsigned				session			= 1;

		unsigned long			discLength		= 150;

		Drive					*drive			= nil;

		TrackDescriptor			*track			= nil;

		NSMutableDictionary		*trackInfo		= nil;

		NSString				*ISRC			= nil;

		_deviceName		= [deviceName retain];

		// To avoid keeping an open file descriptor, read the disc's properties from the drive

		// and store them in our ivars

		drive = [[Drive alloc] initWithDeviceName:[self deviceName]];

		// Is this is a multisession disc?

		if([drive lastSession] - [drive firstSession] > 0)

			[LogController logMessage:NSLocalizedStringFromTable(@"Multisession disc detected", @"Log", @"")];

		// Use first session for now

		session			= 1;

		// Disc information

		_firstSector	= [drive firstSectorForSession:session];

		_lastSector		= [drive lastSectorForSession:session];

		_leadOut		= [drive leadOutForSession:session];

		_MCN			= [[drive readMCN] retain];

		// Iterate through the tracks and get their information

		_tracks			= [[NSMutableArray alloc] init];

		for(i = [drive firstTrackForSession:session]; i <= [drive lastTrackForSession:session]; ++i) {

			track = [drive trackNumber:i];

			trackInfo = [NSMutableDictionary dictionaryWithCapacity:6];

			[trackInfo setObject:[NSNumber numberWithUnsignedInt:i] forKey:@"number"];

			[trackInfo setObject:[NSNumber numberWithUnsignedInt:[drive firstSectorForTrack:i]] forKey:@"firstSector"];

			[trackInfo setObject:[NSNumber numberWithUnsignedInt:[drive lastSectorForTrack:i]] forKey:@"lastSector"];

			[trackInfo setObject:[NSNumber numberWithUnsignedInt:[track channels]] forKey:@"channels"];

			[trackInfo setObject:[NSNumber numberWithBool:[track preEmphasis]] forKey:@"preEmphasis"];

			[trackInfo setObject:[NSNumber numberWithBool:[track copyPermitted]] forKey:@"allowsDigitalCopy"];

			[trackInfo setObject:[NSNumber numberWithBool:[track dataTrack]] forKey:@"dataTrack"];

			ISRC = [drive readISRC:i];

			if(nil != ISRC)

				[trackInfo setObject:ISRC forKey:@"ISRC"];

			[_tracks addObject:trackInfo];

		}

		for(i = 0; i < [self countOfTracks]; ++i)

			discLength += [self lastSectorForTrack:i] - [self firstSectorForTrack:i] + 1;

		_length = (unsigned) (60 * (discLength / (60 * 75))) + (unsigned)((discLength / 75) % 60);

		[drive release];

		return self;

	}

	return nil;

}

- (void) dealloc

{

	[_deviceName release];		_deviceName = nil;

	[_tracks release];			_tracks = nil;

	[_MCN release];				_MCN = nil;

	[super dealloc];

}

- (NSString *)		deviceName								{ return [[_deviceName retain] autorelease]; }

- (NSString *)		MCN										{ return [[_MCN retain] autorelease]; }

- (unsigned)		firstSector								{ return _firstSector; }

- (unsigned)		lastSector								{ return _lastSector; }

- (unsigned)		leadOut									{ return _leadOut; }

- (unsigned)		firstSectorForTrack:(unsigned)track		{ return [[[self objectInTracksAtIndex:track] objectForKey:@"firstSector"] unsignedIntValue]; }

- (unsigned)		lastSectorForTrack:(unsigned)track		{ return [[[self objectInTracksAtIndex:track] objectForKey:@"lastSector"] unsignedIntValue]; }

- (unsigned)		channelsForTrack:(unsigned)track		{ return [[[self objectInTracksAtIndex:track] objectForKey:@"channels"] unsignedIntValue]; }

- (BOOL)			trackHasPreEmphasis:(unsigned)track		{ return [[[self objectInTracksAtIndex:track] objectForKey:@"hasPreEmphasis"] boolValue]; }

- (BOOL)			trackAllowsDigitalCopy:(unsigned)track	{ return [[[self objectInTracksAtIndex:track] objectForKey:@"allowsDigitalCopy"] boolValue]; }

- (BOOL)			trackContainsData:(unsigned)track		{ return [[[self objectInTracksAtIndex:track] objectForKey:@"dataTrack"] boolValue]; }

- (NSString *)		ISRCForTrack:(unsigned)track			{ return [[self objectInTracksAtIndex:track] objectForKey:@"ISRC"]; }

- (DiscId *) discIdObj

{

	DiscId *discID = discid_new();

	if(NULL == discID)

		return NULL;

	// zero is lead out

	int offsets[100];

	offsets[0] = [self leadOut] + 150;

	unsigned i;

	for(i = 0; i < [self countOfTracks]; ++i)

		offsets[1 + i] = [self firstSectorForTrack:i] + 150;

	int result = discid_put(discID, 1, [self countOfTracks], offsets);

	if(result) {

		return discID;

	} else {

		discid_free(discID);

		return NULL;

	}

}

- (NSString *) discID

{

	NSString *musicBrainzDiscID = nil;

	DiscId *discID = [self discIdObj];

	if(NULL == discID)

		return nil;

	musicBrainzDiscID = [NSString stringWithCString:discid_get_id(discID) encoding:NSASCIIStringEncoding];

	discid_free(discID);

	return [[musicBrainzDiscID retain] autorelease];

}

- (NSString *) freeDBDiscID

{

	NSString *freeDBDiscID = nil;

	DiscId *discID = [self discIdObj];

	if(NULL == discID)

		return nil;

	freeDBDiscID = [NSString stringWithCString:discid_get_freedb_id(discID) encoding:NSASCIIStringEncoding];

	discid_free(discID);

	return [[freeDBDiscID retain] autorelease];

}

- (NSURL *) submissionURL

{

	NSURL *submissionURL = nil;

	DiscId *discID = [self discIdObj];

	if(NULL == discID)

		return nil;

	submissionURL = [NSURL URLWithString:[NSString stringWithCString:discid_get_submission_url(discID) encoding:NSASCIIStringEncoding]];

	discid_free(discID);

	return [[submissionURL retain] autorelease];

}

- (unsigned)		length									{ return _length; }

// KVC

- (unsigned)		countOfTracks							{ return [_tracks count]; }

- (NSDictionary *)	objectInTracksAtIndex:(unsigned)index	{ return [_tracks objectAtIndex:index]; }

@end