AIFSM.h

Go to the documentation of this file.

//    Copyright (C) 2004-2008 Dylan Blair
//
//    email: dblair@alumni.cs.utexas.edu
//
//    This library is free software; you can redistribute it and/or
//    modify it under the terms of the GNU Lesser General Public
//    License as published by the Free Software Foundation; either
//    version 2.1 of the License, or (at your option) any later version.
//
//    This library 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
//    Lesser General Public License for more details.
//
//    You should have received a copy of the GNU Lesser General Public
//    License along with this library; if not, write to the Free Software
//    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

#ifndef AIFSM_H
#define AIFSM_H

#include "AIBase.h"

namespace OpenSkyNet {
    namespace AI {
        class State {
        public:
            int                 _stateId;
            std::vector<Action> _actions;
            std::vector<int>    _inputs;
            std::vector<State*> _next;

            State(int stateId_=-1) : _stateId(stateId_) {}
            State(int stateId_, const std::vector<Action>& actions_) : _stateId(stateId_), _actions(actions_) {}

            State* getNextState(int input_);

            int serialize(std::ofstream& fout_) const;
            int assemble(std::ifstream& fin_, std::vector<int>& nextStateIds_);
        };

        extern std::vector<State> g_states;

        class FSM {
            static bool _areGlobalStatesSerialized;
            static bool _areGlobalStatesAssembled;
            FSM_IDS     _fSMId;
            bool        _isInitialStateSet;
            int         _statesIndex;
            State*      _states[FSM_STATE_MAX];
        public:
            FSM(FSM_IDS fSMId_);
            FSM(FSM_IDS fSMId_, State& initialState_);

            inline FSM_IDS getFSMId() const { return _fSMId; }

            bool setInitialState(State& initialState_);

            inline State*& getInitialState() { return _states[0]; }

            bool addState(State& currState_, int input_, State& nextState_);

            State* getStateById(int Id_) const;

            static void initAIFSMFileIO() {
                _areGlobalStatesSerialized = false;
                _areGlobalStatesAssembled = false;
            }

            int serialize(std::ofstream& fout_) const;

            int assemble(std::ifstream& fin_);
        };

        namespace FileIO {
            template <>
            int serialize(std::ofstream& fout_, const std::vector<State>& stateVector_);

            template <>
            int assemble(std::ifstream& fin_, std::vector<State>& stateVector_);
        }
    }
}

#endif //AIFSM_H