//
//  MPT_Emotions1.m
//  GenderController
//
//  Created by Andrew Salamon on 9/22/08.
//  Copyright 2008 Machine Perception Laboratory, University of California San Diego. All rights reserved.
//

#import "MPT_Emotions1.h"

#import <DemoPlugin/NSImageToRimage.h>

@interface MPT_Emotions1(PrivateMethods)
-(void)makeNewEmotions;
@end

@implementation MPT_Emotions1
- (id)initWithBundle:(NSBundle *)_bundle
{
	if( self = [super initWithBundle:_bundle] )
	{
		[self makeNewEmotions];
	}
	
	return self;
}

- (void)dealloc
{
	delete emotions;
	[super dealloc];
}

- (id)copyWithZone:(NSZone *)zone
{
	id cp = [super copyWithZone:zone];
	return cp;
}

- (void)makeThreadsafe
{
	[super makeThreadsafe];
	[self makeNewEmotions];
}

- (NSString *)localizedName
{
	return NSLocalizedStringFromTableInBundle(@"Emotion Detector", nil, [self bundle], @"");
}

- (NSString *)internalName
{
	return @"emotion1";
}

- (NSString *)step
{
	return MPT_PluginStep_Results; 
}

-(void)makeNewEmotions
{
	emotions =  new mpt::Emotions1;
	NSString *weightsPath = [[self bundle] pathForResource:@"emotions1" ofType:@"txt"];
	// need to load the emotion weights
	if( [weightsPath length] > 0 )
		emotions->readFromFile( [weightsPath UTF8String] );
	else
		NSLog( @"Unable to read emotion weights file in Emotions1 plugin." );
}

- (NSMutableDictionary *)processResults:(NSDictionary *)results usingController:(id<MPT_PluginControllerProtocol>)controller;
{
	NSMutableDictionary *res = [NSMutableDictionary dictionary];
	// edu.ucsd.mplab.plugins.CERT4_4_Weights: 1, 2, 4, 5, 9, 10, 12, 14, 15, 17, 20
	// edu.ucsd.mplab.plugins.AdditonalAUs: 25, 17, 7, 6, 26, 15, 24, 23, 27, 18, 9, 28
	
	NSDictionary *aus = [res objectForKey:@"edu.ucsd.mplab.plugins.CERT4_4_Weights"];
	NSDictionary *addAUs = [res objectForKey:@"edu.ucsd.mplab.plugins.AdditonalAUs"];
	std::vector<float> inputs;
	
	// 
	// Find the inputs we need and add them to a vector
	// need the following aus:      1     2     4     5     6     7     9    10    12    14    15    17    18    20    23    24    25    26    28
	/* Main AUs
	 <string>(AU 1) Inner Brow Raise</string>
	 <string>(AU 2) Outer Brow Raise</string>
	 <string>(AU 4) Brow Lower</string>
	 <string>(AU 5) Eye Widen</string>
	 <string>(AU 9) Nose Wrinkle</string>
	 <string>(AU 10) Lip Raise</string>
	 <string>(AU 12) Lip Corner Pull</string>
	 <string>(AU 14) Dimpler</string>
	 <string>(AU 15) Lip Corner Depressor</string>
	 <string>(AU 17) Chin Raise</string>
	 <string>(AU 20) Lip stretch</string>
*/	 
/* Additional AUs
 <string>AU 25</string>
 <string>AU 17</string>
 <string>AU 7</string>
 <string>AU 6</string>
 <string>AU 26</string>
 <string>AU 15</string>
 <string>AU 24</string>
 <string>AU 23</string>
 <string>AU 27</string>
 <string>AU 18</string>
 <string>AU 9</string>
 <string>AU 28</string>
*/
	
	//call
	std::vector<float> ans = emotions->calc( inputs );

	return res;
}

@end