UniversalDetector.m

#import "UniversalDetector.h"
#import "WrappedUniversalDetector.h"

@implementation UniversalDetector

- (id)init {
	self = [super init];
	
	if (self) {
		detectorPtr = AllocUniversalDetector();
		charsetName = nil;
		confidence  = 0;
	}
	return self;
}

- (void)dealloc {
	FreeUniversalDetector(detectorPtr);
	[charsetName release];
	[super dealloc];
}

- (void)analyzeContentsOfFile:(NSString *)path {
    NSData *data = [[NSData alloc] initWithContentsOfFile:path options:NSDataReadingMappedIfSafe error:NULL];

	if (data) {
		[self analyzeBytes:(const char *)[data bytes] length:(int)[data length]];
	}
	[data release];
}

- (void)analyzeData:(NSData *)data {
	[self analyzeBytes:(const char *)[data bytes] length:(int)[data length]];
}

- (void)analyzeBytes:(const char *)data length:(int)len {
	UniversalDetectorHandleData(detectorPtr, data, len);
	[charsetName release];
	charsetName=nil;
}

- (void)reset {
	UniversalDetectorReset(detectorPtr);
}

- (BOOL)done {
	return UniversalDetectorDone(detectorPtr);
}

- (NSString *)MIMECharset {
	if(!charsetName)
	{
		const char *cstr=UniversalDetectorCharset(detectorPtr, &confidence);
		if(!cstr) return nil;
		charsetName=[[NSString alloc] initWithUTF8String:cstr];
	}
	return charsetName;
}

- (NSStringEncoding)encoding {
	NSString *mimecharset=[self MIMECharset];
    if (!mimecharset) { return 0; }

	CFStringEncoding cfenc=CFStringConvertIANACharSetNameToEncoding((CFStringRef)mimecharset);
    if (cfenc==kCFStringEncodingInvalidId) { return 0; }

	// UniversalDetector detects CP949 but returns "EUC-KR" because CP949 lacks an IANA name.
	// Kludge to make strings decode properly anyway.
    if (cfenc==kCFStringEncodingEUC_KR) { cfenc=kCFStringEncodingDOSKorean; }

	return CFStringConvertEncodingToNSStringEncoding(cfenc);
}

- (float)confidence {
	if(!charsetName) [self MIMECharset];
	return confidence;
}

@end