AI.m

Go to the documentation of this file.

/*
 
AI.m
 
Oolite
Copyright (C) 2004-2013 Giles C Williams and contributors
 
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., 51 Franklin Street, Fifth Floor, Boston,
MA 02110-1301, USA.
 
*/
 
#import "AI.h"
#import "ResourceManager.h"
#import "OOStringParsing.h"
#import "OOWeakReference.h"
#import "OOCacheManager.h"
#import "OOCollectionExtractors.h"
#import "OOPListParsing.h"
 
#import "ShipEntity.h"
#import "ShipEntityAI.h"
 
 
enum
{
    kRecursionLimiter       = 32,   // reactToMethod: recursion
    kStackLimiter           = 32    // setAITo: stack overflow
};
 
 
typedef struct
{
    AI              *ai;
    SEL             selector;
    id              parameter;
} OOAIDeferredCallTrampolineInfo;
 
 
static AI *sCurrentlyRunningAI = nil;
 
 
@interface AI (OOPrivate)
 
// Wrapper for performSelector:withObject:afterDelay: to catch/fix bugs.
- (void) performDeferredCall:(SEL)selector withObject:(id)object afterDelay:(NSTimeInterval)delay;
+ (void) deferredCallTrampolineWithInfo:(NSValue *)info;
 
- (void) refreshOwnerDesc;
 
// Set state machine and state without side effects.
- (void) directSetStateMachine:(NSDictionary *)newSM name:(NSString *)name;
- (void) directSetState:(NSString *)state;
 
// Loading/whitelisting
- (NSDictionary *) loadStateMachine:(NSString *)smName jsName:(NSString *)script;
- (NSDictionary *) cleanHandlers:(NSDictionary *)handlers forState:(NSString *)stateKey stateMachine:(NSString *)smName;
- (NSArray *) cleanActions:(NSArray *)actions forHandler:(NSString *)handlerKey state:(NSString *)stateKey stateMachine:(NSString *)smName;
 
@end
 
 
#if DEBUG_GRAPHVIZ
extern void GenerateGraphVizForAIStateMachine(NSDictionary *stateMachine, NSString *name);
#endif
 
 
@interface OOPreservedAIStateMachine: NSObject
{
@private
    NSDictionary        *_stateMachine;
    NSString            *_name;
    NSString            *_state;
    NSMutableSet        *_pendingMessages;
    NSString      *_jsScript;
}
 
- (id) initWithStateMachine:(NSDictionary *)stateMachine
                       name:(NSString *)name
                      state:(NSString *)state
            pendingMessages:(NSSet *)pendingMessages
                                     jsScript:(NSString *)script;
 
- (NSDictionary *) stateMachine;
- (NSString *) name;
- (NSString *) state;
- (NSSet *) pendingMessages;
- (NSString *) jsScript;
 
@end
 
 
@implementation AI
 
+ (AI *) currentlyRunningAI
{
    return sCurrentlyRunningAI;
}
 
 
+ (NSString *) currentlyRunningAIDescription
{
    if (sCurrentlyRunningAI != nil)
    {
        return [NSString stringWithFormat:@"%@ in state %@", [sCurrentlyRunningAI name], [sCurrentlyRunningAI state]];
    }
    else
    {
        return @"<no AI running>";
    }
}
 
 
- (id) init
{
    if ((self = [super init]))
    {
        nextThinkTime = INFINITY;   // don't think for a while
        thinkTimeInterval = AI_THINK_INTERVAL;
        
        stateMachineName = @"<no AI>";  // no initial brain
    }
    
    return self;
}
 
 
- (id) initWithStateMachine:(NSString *)smName andState:(NSString *)stateName
{
    if ((self = [self init]))
    {
        if (smName != nil)  [self setStateMachine:smName withJSScript:@"oolite-nullAI.js"];
        if (stateName != nil)  currentState = [stateName retain];
    }
    
    return self;
}
 
 
- (void) dealloc
{
    if (sCurrentlyRunningAI == self)
    {
        sCurrentlyRunningAI = nil;
    }
    
    DESTROY(_owner);
    DESTROY(ownerDesc);
    DESTROY(aiStack);
    DESTROY(stateMachine);
    DESTROY(stateMachineName);
    DESTROY(currentState);
    DESTROY(pendingMessages);
    DESTROY(jsScript);
    
    [super dealloc];
}
 
 
- (NSString *) descriptionComponents
{
    return [NSString stringWithFormat:@"\"%@\" in state: \"%@\" for %@", stateMachineName, currentState, ownerDesc];
}
 
 
- (NSString *) shortDescriptionComponents
{
    return [NSString stringWithFormat:@"%@:%@ / %@", stateMachineName, currentState, [stateMachine objectForKey:@"jsScript"]];
}
 
 
- (ShipEntity *)owner
{
    ShipEntity      *owner = [_owner weakRefUnderlyingObject];
    if (owner == nil)
    {
        [_owner release];
        _owner = nil;
    }
    
    return owner;
}
 
 
- (void) setOwner:(ShipEntity *)ship
{
    [_owner release];
    _owner = [ship weakRetain];
    [self refreshOwnerDesc];
}
 
 
- (void) reportStackOverflow
{
    if (OOLogWillDisplayMessagesInClass(@"ai.error.stackOverflow"))
    {
        BOOL stackDump = OOLogWillDisplayMessagesInClass(@"ai.error.stackOverflow.dump");
        
        NSString *trailer = stackDump ? @" -- stack:" : @".";
        OOLogERR(@"ai.error.stackOverflow", @"AI stack overflow for %@ in %@: %@%@\n", [_owner shortDescription], stateMachineName, currentState, trailer);
        
        if (stackDump)
        {
            OOLogIndent();
            
            NSUInteger count = [aiStack count];
            while (count--)
            {
                OOPreservedAIStateMachine *preservedMachine = [aiStack objectAtIndex:count];
                OOLog(@"ai.error.stackOverflow.dump", @"%3llu: %@: %@", count, [preservedMachine name], [preservedMachine state]);
            }
            
            OOLogOutdent();
        }
    }
}
 
 
- (void) preserveCurrentStateMachine
{
    if (stateMachine == nil)  return;
    
    if (aiStack == nil)
    {
        aiStack = [[NSMutableArray alloc] init];
    }
    
    if ([aiStack count] >= kStackLimiter)
    {
        [self reportStackOverflow];
        
        [NSException raise:@"OoliteException"
                    format:@"AI stack overflow for %@", _owner];
    }
    
    OOPreservedAIStateMachine *preservedMachine = [[OOPreservedAIStateMachine alloc]
                                                   initWithStateMachine:stateMachine
                                                                   name:stateMachineName
                                                                  state:currentState
                                                        pendingMessages:pendingMessages
                                                                                                    jsScript:[stateMachine objectForKey:@"jsScript"]];
    
#ifndef NDEBUG
    if ([[self owner] reportAIMessages])  OOLog(@"ai.stack.push", @"Pushing state machine for %@", self);
#endif
    
    [aiStack addObject:preservedMachine];  // PUSH
    
    [preservedMachine release];
}
 
 
- (void) restorePreviousStateMachine
{
    if ([aiStack count] == 0)  return;
    
    OOPreservedAIStateMachine *preservedMachine = [aiStack lastObject];
    
#ifndef NDEBUG
    if ([[self owner] reportAIMessages])  OOLog(@"ai.stack.pop", @"Popping previous state machine for %@", self);
#endif
    
    [self directSetStateMachine:[preservedMachine stateMachine]
                           name:[preservedMachine name]];
    
    [self directSetState:[preservedMachine state]];
    
    // restore JS script
    [[self owner] setAIScript:[preservedMachine jsScript]];
 
    [pendingMessages release];
    pendingMessages = [[preservedMachine pendingMessages] mutableCopy];  // restore a MUTABLE set
    
    [aiStack removeLastObject];  //  POP
}
 
 
- (BOOL) hasSuspendedStateMachines
{
    return [aiStack count] != 0;
}
 
 
- (void) exitStateMachineWithMessage:(NSString *)message
{
    if ([aiStack count] != 0)
    {
        [self restorePreviousStateMachine];
        if (message == nil)  message = @"RESTARTED";
        [self reactToMessage:message context:@"suspended AI restart"];
    }
}
 
 
- (void) setStateMachine:(NSString *)smName withJSScript:(NSString *)script
{
    NSDictionary *newSM = [self loadStateMachine:smName jsName:script];
 
    if (newSM)
    {
        [self preserveCurrentStateMachine];
        [self directSetStateMachine:newSM name:smName];
        [self directSetState:@"GLOBAL"];
        
        nextThinkTime = 0.0;    // think at next tick
 
        /*  CRASH in objc_msgSend, apparently on [self reactToMessage:@"ENTER"] (1.69, OS X/x86).
            Analysis: self corrupted. We're being called by __NSFireDelayedPerform, which doesn't go
            through -[NSObject performSelector:withObject:], suggesting it's using IMP caching. An
            invalid self is therefore possible.
            Attempted fix: new delayed dispatch with trampoline, see -[AI setStateMachine:afterDelay:].
             -- Ahruman, 20070706
        */
        [self reactToMessage:@"ENTER" context:@"changing AI"];
        
        // refresh name
        [self refreshOwnerDesc];
    }
}
 
 
- (void) setState:(NSString *) stateName
{
    if ([stateMachine objectForKey:stateName])
    {
        /*  CRASH in objc_msgSend, apparently on [self reactToMessage:@"EXIT"] (1.69, OS X/x86).
            Analysis: self corrupted. We're being called by __NSFireDelayedPerform, which doesn't go
            through -[NSObject performSelector:withObject:], suggesting it's using IMP caching. An
            invalid self is therefore possible.
            Attempted fix: new delayed dispatch with trampoline, see -[AI setState:afterDelay:].
             -- Ahruman, 20070706
        */
        [self reactToMessage:@"EXIT" context:@"changing state"];
        [self directSetState:stateName];
        [self reactToMessage:@"ENTER" context:@"changing state"];
    }
}
 
 
- (void) setStateMachine:(NSString *)smName afterDelay:(NSTimeInterval)delay
{
    [self performDeferredCall:@selector(setStateMachine:) withObject:smName afterDelay:delay];
}
 
 
- (void) setState:(NSString *)stateName afterDelay:(NSTimeInterval)delay
{
    [self performDeferredCall:@selector(setState:) withObject:stateName afterDelay:delay];
}
 
 
- (NSString *) name
{
    return [[stateMachineName retain] autorelease];
}
 
 
- (NSString *) associatedJS
{
    return [stateMachine objectForKey:@"jsScript"];
}
 
 
- (NSString *) state
{
    return [[currentState retain] autorelease];
}
 
 
- (NSUInteger) stackDepth
{
    return [aiStack count];
}
 
 
#ifndef NDEBUG
typedef struct AIStackElement AIStackElement;
struct AIStackElement
{
    AIStackElement          *back;
    ShipEntity              *owner;
    NSString                *aiName;
    NSString                *state;
    NSString                *message;
    NSString                *context;
};
 
static AIStackElement *sStack = NULL;
#endif
 
 
- (void) reactToMessage:(NSString *) message context:(NSString *)debugContext
{
    unsigned        i;
    NSArray         *actions = nil;
    NSDictionary    *messagesForState = nil;
    ShipEntity      *owner = [self owner];
    static unsigned recursionLimiter = 0;
    AI              *previousRunning = sCurrentlyRunningAI;
    
    /*  CRASH in _freedHandler when called via -setState: __NSFireDelayedPerform (1.69, OS X/x86).
        Analysis: owner invalid.
        Fix: make owner an OOWeakReference.
         -- Ahruman, 20070706
    */
    if (message == nil || owner == nil || [owner universalID] == NO_TARGET)  return;
 
#ifndef NDEBUG
    // Push debug stack frame.
    if (debugContext == nil)  debugContext = @"unspecified";
    AIStackElement stackElement =
    {
        .back = sStack,
        .owner = owner,
        .aiName = [[stateMachineName retain] autorelease],
        .state = [[currentState retain] autorelease],
        .message = message,
        .context = debugContext
    };
    sStack = &stackElement;
#endif
    
    /*  CRASH when calling reactToMessage: FOO in state FOO causes infinite
        recursion. (NB: there are other ways of triggering this.)
        FIX: recursion limiter. Alternative is to explicitly catch this case
        in takeAction:, but that could potentially miss indirect recursion via
        scripts.
    */
    if (recursionLimiter > kRecursionLimiter)
    {
        OOLogERR(@"ai.error.recursion", @"AI dispatch: hit stack depth limit in AI %@, state %@ handling message %@ in context \"%@\", aborting.", stateMachineName, currentState, message, debugContext);
        
#ifndef NDEBUG
        AIStackElement *stack = sStack;
        unsigned depth = 0;
        while (stack != NULL)
        {
            OOLog(@"ai.error.recursion.stackTrace", @"%4u  %@ - %@:%@.%@ (%@)", depth++, [stack->owner shortDescription], stack->aiName, stack->state, stack->message, stack->context);
            stack = stack->back;
        }
        
        // unwind.
        if (sStack != NULL)  sStack = sStack->back;
#endif
        
        return;
    }
    
    messagesForState = [stateMachine objectForKey:currentState];
    if (messagesForState == nil)  return;
    
#ifndef NDEBUG
    if (currentState != nil && ![message isEqual:@"UPDATE"] && [owner reportAIMessages])
    {
        OOLog(@"ai.message.receive", @"AI %@ for %@ in state '%@' receives message '%@'. Context: %@, stack depth: %u", stateMachineName, ownerDesc, currentState, message, debugContext, recursionLimiter);
    }
#endif
    
    actions = [[[messagesForState objectForKey:message] copy] autorelease];
    
    sCurrentlyRunningAI = self;
    if ([actions count] > 0)
    {
        ++recursionLimiter;
        @try
        {
            for (i = 0; i < [actions count]; i++)
            {
                [self takeAction:[actions objectAtIndex:i]];
            }
        }
        @catch (NSException *exception)
        {
            OOLog(kOOLogException, @"Squashing exception %@:%@ in AI handler %@:%@.%@", [exception name], [exception reason], stateMachineName, currentState, message);
        }
        
        --recursionLimiter;
    }
    else
    {
        if (currentState != nil)
        {
            if ([owner respondsToSelector:@selector(interpretAIMessage:)])
            {
                [owner performSelector:@selector(interpretAIMessage:) withObject:message];
            }
        }
    }
    
    sCurrentlyRunningAI = previousRunning;
#ifndef NDEBUG
    // Unwind stack.
    if (sStack != NULL)  sStack = sStack->back;
#endif
}
 
 
- (void) takeAction:(NSString *)action
{
    ShipEntity *owner = [self owner];
    
#ifndef NDEBUG
    BOOL report = [owner reportAIMessages];
    if (report)
    {
        OOLog(@"ai.takeAction", @"%@ to take action %@", ownerDesc, action);
        OOLogIndent();
    }
#endif
    
    NSArray *tokens = ScanTokensFromString(action);
    NSUInteger tokenCount = [tokens count];
    
    if (tokenCount != 0)
    {
        NSString *selectorStr = [tokens objectAtIndex:0];
        
        if (owner != nil)
        {
            NSString *dataString = nil;
            
            if (tokenCount == 2)
            {
                dataString = [tokens objectAtIndex:1];
            }
            else if ([tokens count] > 1)
            {
                dataString = [[tokens subarrayWithRange:NSMakeRange(1, tokenCount - 1)] componentsJoinedByString:@" "];
            }
            
            SEL selector = NSSelectorFromString(selectorStr);
            if ([owner respondsToSelector:selector])
            {
                if (dataString != nil)  [owner performSelector:selector withObject:dataString];
                else  [owner performSelector:selector];
            }
            else
            {
                OOLogERR(@"ai.takeAction.badSelector", @"in AI %@ in state %@: %@ does not respond to %@", stateMachineName, currentState, ownerDesc, selectorStr);
            }
        }
        else
        {
            OOLog(@"ai.takeAction.orphaned", @"***** AI %@, trying to perform %@, is orphaned (no owner)", stateMachineName, selectorStr);
        }
    }
    else
    {
#ifndef NDEBUG
        if (report)  OOLog(@"ai.takeAction.noAction", @"DEBUG: - no action '%@'", action);
#endif
    }
    
#ifndef NDEBUG
    if (report)
    {
        OOLogOutdent();
    }
#endif
}
 
 
- (void) think
{
    NSArray         *ms_list = nil;
    unsigned        i;
    
    if ([[self owner] universalID] == NO_TARGET || stateMachine == nil)  return;  // don't think until launched
    
    [self reactToMessage:@"UPDATE" context:@"periodic update"];
 
    if ([pendingMessages count] > 0)
    {
        ms_list = [pendingMessages allObjects];
        [pendingMessages removeAllObjects];
    }
    
    if (ms_list != nil)
    {
        for (i = 0; i < [ms_list count]; i++)
        {
            [self reactToMessage:[ms_list objectAtIndex:i] context:@"handling deferred message"];
        }
    }
}
 
 
- (void) message:(NSString *)ms
{
    if ([[self owner] universalID] == NO_TARGET)  return;  // don't think until launched
 
    if (EXPECT_NOT([pendingMessages count] > 32))
    {
        // Generate the error, but don't crash Oolite! Fixes bug #18055 - Pending message overflow for thargoids, -> crash !
        OOLogERR(@"ai.message.failed.overflow", @"AI message \"%@\" received by '%@' AI while pending messages stack full; message discarded. Pending messages:\n%@", ms, ownerDesc, pendingMessages);
    }
    else
    {
        if (pendingMessages == nil)
        {
            pendingMessages = [[NSMutableSet alloc] init];
        }
        [pendingMessages addObject:ms];
    }
}
 
 
- (void) dropMessage:(NSString *)ms
{
    [pendingMessages removeObject:ms];
}
    
 
- (NSSet *) pendingMessages
{
    if (pendingMessages != nil)
    {
        return [[pendingMessages copy] autorelease];
    }
    else
    {
        return [NSSet set];
    }
}
 
 
- (void) debugDumpPendingMessages
{
    NSArray             *sortedMessages = nil;
    NSString            *displayMessages = nil;
    
    if ([pendingMessages count] > 0)
    {
        sortedMessages = [[pendingMessages allObjects] sortedArrayUsingSelector:@selector(caseInsensitiveCompare:)];
        displayMessages = [sortedMessages componentsJoinedByString:@", "];
    }
    else
    {
        displayMessages = @"none";
    }
    
    OOLog(@"ai.debug.pendingMessages", @"Pending messages for AI %@: %@", [self descriptionComponents], displayMessages);
}
 
 
- (void) setNextThinkTime:(OOTimeAbsolute) ntt
{
    nextThinkTime = ntt;
}
 
 
- (OOTimeAbsolute) nextThinkTime
{
    if (!stateMachine)
        return INFINITY;
 
    return nextThinkTime;
}
 
 
- (void) setThinkTimeInterval:(OOTimeDelta) tti
{
    thinkTimeInterval = tti;
}
 
 
- (OOTimeDelta) thinkTimeInterval
{
    return thinkTimeInterval;
}
 
 
- (void) clearStack
{
    [aiStack removeAllObjects];
}
 
 
- (void) clearAllData
{
    [aiStack removeAllObjects];
    [pendingMessages removeAllObjects];
    
    nextThinkTime += 36000.0;   // should dealloc in under ten hours!
}
 
 
- (void)dumpState
{
    OOLog(@"dumpState.ai", @"State machine name: %@", stateMachineName);
    OOLog(@"dumpState.ai", @"Current state: %@", currentState);
    OOLog(@"dumpState.ai", @"Next think time: %g", nextThinkTime);
    OOLog(@"dumpState.ai", @"Next think interval: %g", thinkTimeInterval);
}
 
@end
 
 
/*  This is an attempt to fix the bugs referred to above regarding calls from
    __NSFireDelayedPerform with a corrupt self. I'm not certain whether this
    will fix the issue or merely cause a less weird crash in
    +deferredCallTrampolineWithInfo:.
    -- Ahruman 20070706
*/
@implementation AI (OOPrivate)
 
- (void)performDeferredCall:(SEL)selector withObject:(id)object afterDelay:(NSTimeInterval)delay
{
    OOAIDeferredCallTrampolineInfo  infoStruct;
    NSValue                         *info = nil;
    
    if (selector != NULL)
    {
        infoStruct.ai = [self retain];
        infoStruct.selector = selector;
        infoStruct.parameter = object;
        
        info = [[NSValue alloc] initWithBytes:&infoStruct objCType:@encode(OOAIDeferredCallTrampolineInfo)];
        
        [[AI class] performSelector:@selector(deferredCallTrampolineWithInfo:)
                         withObject:info
                         afterDelay:delay];
        [info release];
    }
}
 
 
+ (void)deferredCallTrampolineWithInfo:(NSValue *)info
{
    OOAIDeferredCallTrampolineInfo  infoStruct;
    
    if (info != nil)
    {
        assert(strcmp([info objCType], @encode(OOAIDeferredCallTrampolineInfo)) == 0);
        [info getValue:&infoStruct];
        
        [infoStruct.ai performSelector:infoStruct.selector withObject:infoStruct.parameter];
        
        [infoStruct.ai release];
        [infoStruct.parameter release];
    }
}
 
 
- (void)refreshOwnerDesc
{
    ShipEntity *owner = [self owner];
    [ownerDesc release];
    if ([owner isPlayer])
    {
        ownerDesc = @"player autopilot";
    }
    else if (owner != nil)
    {
        ownerDesc = [[NSString alloc] initWithFormat:@"%@ %d", [owner name], [owner universalID]];
    }
    else
    {
        ownerDesc = @"no owner";
    }
}
 
 
- (void) directSetStateMachine:(NSDictionary *)newSM name:(NSString *)name
{
    if (stateMachine != newSM)
    {
        [stateMachine release];
        stateMachine = [newSM copy];
    }
    if (stateMachineName != name)
    {
        [stateMachineName release];
        stateMachineName = [name copy];
    }
}
 
 
- (void) directSetState:(NSString *)state
{
    if (currentState != state)
    {
        [currentState release];
        currentState = [state copy];
    }
}
 
 
- (NSDictionary *) loadStateMachine:(NSString *)smName jsName:(NSString *)script
{
    NSDictionary            *newSM = nil;
    NSMutableDictionary     *cleanSM = nil;
    OOCacheManager          *cacheMgr = [OOCacheManager sharedCache];
    NSString                *stateKey = nil;
    NSDictionary            *stateHandlers = nil;
    NSAutoreleasePool       *pool = nil;
    
    if (![smName isEqualToString:@"nullAI.plist"])
    {
        // don't cache nullAI since they're different depending on associated JS AI
        newSM = [cacheMgr objectForKey:smName inCache:@"AIs"];
        if (newSM != nil && ![newSM isKindOfClass:[NSDictionary class]])  return nil;   // catches use of @"nil" to indicate no AI found.
    }
    
    if (newSM == nil)
    {
        pool = [[NSAutoreleasePool alloc] init];
        OOLog(@"ai.load", @"Loading and sanitizing AI \"%@\"", smName);
        OOLogPushIndent();
        OOLogIndentIf(@"ai.load");
        
        @try
        {
            // Load state machine and validate against whitelist.
            NSString *aiPath = [ResourceManager pathForFileNamed:smName inFolder:@"AIs"];
            if (aiPath != nil)
            {
                newSM = OODictionaryFromFile(aiPath);
            }
            if (newSM == nil)
            {
                [cacheMgr setObject:@"nil" forKey:smName inCache:@"AIs"];
                NSString *fromString = @"";
                if ([self state] != nil)
                {
                    fromString = [NSString stringWithFormat:@" from %@:%@", [self name], [self state]];
                }
                OOLog(@"ai.load.failed.unknownAI", @"Can't switch AI for %@%@ to \"%@\" - could not load file.", [[self owner] shortDescription], fromString, smName);
                return nil;
            }
            
            cleanSM = [NSMutableDictionary dictionaryWithCapacity:[newSM count]];
            
            foreachkey (stateKey, newSM)
            {
                stateHandlers = [newSM objectForKey:stateKey];
                if (![stateHandlers isKindOfClass:[NSDictionary class]])
                {
                    OOLogWARN(@"ai.invalidFormat.state", @"State \"%@\" in AI \"%@\" is not a dictionary, ignoring.", stateKey, smName);
                    continue;
                }
                
                stateHandlers = [self cleanHandlers:stateHandlers forState:stateKey stateMachine:smName];
                [cleanSM setObject:stateHandlers forKey:stateKey];
            }
            [cleanSM setObject:script forKey:@"jsScript"];
 
            // Make immutable.
            newSM = [[cleanSM copy] autorelease];
            
#if DEBUG_GRAPHVIZ
            if ([[NSUserDefaults standardUserDefaults] boolForKey:@"generate-ai-graphviz"])
            {
                GenerateGraphVizForAIStateMachine(newSM, smName);
            }
#endif
            
            // Cache.
            [cacheMgr setObject:newSM forKey:smName inCache:@"AIs"];
        }
        @finally
        {
            OOLogPopIndent();
        }
        
        [newSM retain];
        [pool release];
        [newSM autorelease];
    }
    
    return newSM;
}
 
 
- (NSDictionary *) cleanHandlers:(NSDictionary *)handlers forState:(NSString *)stateKey stateMachine:(NSString *)smName
{
    NSString                *handlerKey = nil;
    NSArray                 *handlerActions = nil;
    NSMutableDictionary     *result = nil;
    
    result = [NSMutableDictionary dictionaryWithCapacity:[handlers count]];
    foreachkey (handlerKey, handlers)
    {
        handlerActions = [handlers objectForKey:handlerKey];
        if (![handlerActions isKindOfClass:[NSArray class]])
        {
            OOLogWARN(@"ai.invalidFormat.handler", @"Handler \"%@\" for state \"%@\" in AI \"%@\" is not an array, ignoring.", handlerKey, stateKey, smName);
            continue;
        }
        
        handlerActions = [self cleanActions:handlerActions forHandler:handlerKey state:stateKey stateMachine:smName];
        [result setObject:handlerActions forKey:handlerKey];
    }
    
    // Return immutable copy.
    return [[result copy] autorelease];
}
 
 
- (NSArray *) cleanActions:(NSArray *)actions forHandler:(NSString *)handlerKey state:(NSString *)stateKey stateMachine:(NSString *)smName
{
    NSString                *action = nil;
    NSRange                 spaceRange;
    NSString                *selector = nil;
    id                      aliasedSelector = nil;
    NSMutableArray          *result = nil;
    static NSSet            *whitelist = nil;
    static NSDictionary     *aliases = nil;
    NSArray                 *whitelistArray1 = nil;
    NSArray                 *whitelistArray2 = nil;
    
    if (whitelist == nil)
    {
        whitelistArray1 = [[ResourceManager whitelistDictionary] oo_arrayForKey:@"ai_methods"];
        if (whitelistArray1 == nil)  whitelistArray1 = [NSArray array];
        whitelistArray2 = [[ResourceManager whitelistDictionary] oo_arrayForKey:@"ai_and_action_methods"];
        if (whitelistArray2 != nil)  whitelistArray1 = [whitelistArray1 arrayByAddingObjectsFromArray:whitelistArray2];
        
        whitelist = [[NSSet alloc] initWithArray:whitelistArray1];
        aliases = [[[ResourceManager whitelistDictionary] oo_dictionaryForKey:@"ai_method_aliases"] retain];
    }
    
    result = [NSMutableArray arrayWithCapacity:[actions count]];
    foreach (action, actions)
    {
        if (![action isKindOfClass:[NSString class]])
        {
            OOLogWARN(@"ai.invalidFormat.action", @"An action in handler \"%@\" for state \"%@\" in AI \"%@\" is not a string, ignoring.", handlerKey, stateKey, smName);
            continue;
        }
        
        // Trim spaces from beginning and end.
        action = [action stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceCharacterSet]];
        
        // Cut off parameters.
        spaceRange = [action rangeOfString:@" "];
        if (spaceRange.location == NSNotFound)  selector = action;
        else  selector = [action substringToIndex:spaceRange.location];
        
        // Look in alias table.
        aliasedSelector = [aliases objectForKey:selector];
        if (aliasedSelector != nil)
        {
            if ([aliasedSelector isKindOfClass:[NSString class]])
            {
                // Change selector and action to use real method name.
                selector = aliasedSelector;
                if (spaceRange.location == NSNotFound)  action = aliasedSelector;
                else action = [aliasedSelector stringByAppendingString:[action substringFromIndex:spaceRange.location]];
            }
            else if ([aliasedSelector isKindOfClass:[NSArray class]] && [aliasedSelector count] != 0)
            {
                // Alias is complete expression, pretokenized in anticipation of a tokenized future.
                action = [aliasedSelector componentsJoinedByString:@" "];
                selector = [[aliasedSelector objectAtIndex:0] description];
            }
        }
        
        // Check for selector in whitelist.
        if (![whitelist containsObject:selector])
        {
            OOLog(@"ai.unpermittedMethod", @"Handler \"%@\" for state \"%@\" in AI \"%@\" uses \"%@\", which is not a permitted AI method.", handlerKey, stateKey, smName, selector);
            continue;
        }
        
        [result addObject:action];
    }
    
    // Return immutable copy.
    return [[result copy] autorelease];
}
 
@end
 
 
@implementation OOPreservedAIStateMachine
 
- (id) initWithStateMachine:(NSDictionary *)stateMachine
                       name:(NSString *)name
                      state:(NSString *)state
            pendingMessages:(NSSet *)pendingMessages
                                     jsScript:(NSString *)script
{
    if ((self = [super init]))
    {
        _stateMachine = [stateMachine copy];
        _name = [name copy];
        _state = [state copy];
        _pendingMessages = [pendingMessages copy];
        _jsScript = [script copy];
    }
    
    return self;
}
 
 
- (void) dealloc
{
    [_stateMachine autorelease];
    [_name autorelease];
    [_state autorelease];
    [_pendingMessages autorelease];
    [_jsScript autorelease];
    
    [super dealloc];
}
 
 
- (NSDictionary *) stateMachine
{
    return _stateMachine;
}
 
 
- (NSString *) name
{
    return _name;
}
 
 
- (NSString *) state
{
    return _state;
}
 
 
- (NSSet *) pendingMessages
{
    return _pendingMessages;
}
 
- (NSString *) jsScript
{
    return _jsScript;
}
 
@end