AIBase.h

Go to the documentation of this file.

//    Copyright (C) 2004-2012 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 AIBASE_H
#define AIBASE_H

#include "AIConsts.h"

#include "../Utils_LIB/UTypes.h"

#include <vector>
#include <fstream>

namespace OpenSkyNet {
    namespace Utils {
        class PrecisionTimer;
    }

    namespace AI {
        extern OpenSkyNet::Utils::PrecisionTimer g_timer;

        class Premise {
        public:
            std::vector<int> _params; 
            float _entryTime;         
            bool _negated;            

            Premise() : _entryTime(-1.0f), _negated(false) {}

            Premise(const std::vector<int>& params_) : _params(params_), _entryTime(-1.0f), _negated(false) {}

            virtual bool operator==(const Premise& rhs_) const;

            virtual bool equals(const Premise& rhs_, unsigned int numParamsToCheck_) const;

            Premise& operator=(int param_) {
                _params.clear();
                _params.push_back(param_);
                _entryTime = -1.0f;
                _negated = false;
                return *this;
            }

            Premise& operator+=(int param_) {
                _params.push_back(param_);
                return *this;
            }

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

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

            virtual ~Premise() {} 
        };

        class Action : public Premise {
        public:
            Action() {}

            Action(const std::vector<int>& params_) : Premise(params_) {}

            Action& operator=(int param_) {
                Premise::operator=(param_);
                return *this;
            }

            Action& operator+=(int param_) {
                Premise::operator+=(param_);
                return *this;
            }

            virtual ~Action() {} 
        };

        class Wrapper;

        class Record : public Premise {
        public:
            OpenSkyNet::AI::Wrapper* _n; 
            float _badGoodScaleValue;    
            int _stateIdAtCreation;      

            Record(OpenSkyNet::AI::Wrapper* n_=0, float badGoodScaleValue_=0.0f, 
                int stateIdAtCreation_=-1) : _n(n_), 
                _badGoodScaleValue(badGoodScaleValue_), 
                _stateIdAtCreation(stateIdAtCreation_) {}

            Record(const std::vector<int>& params_, OpenSkyNet::AI::Wrapper* n_=0, 
                float badGoodScaleValue_=0.0f, int stateIdAtCreation_=-1) : Premise(params_), 
                _n(n_), _badGoodScaleValue(badGoodScaleValue_), 
                _stateIdAtCreation(stateIdAtCreation_) {}

            Record(const Premise& premise_) : Premise(premise_), _n(0), _badGoodScaleValue(0.0f), _stateIdAtCreation(-1) {}

            Record& operator=(int param_) {
                Premise::operator=(param_);
                return *this;
            }

            Record& operator+=(int param_) {
                Premise::operator+=(param_);
                return *this;
            }

            virtual ~Record() {} 
        };

        namespace FileIO {
            template <class T>
            int serialize(std::ofstream& fout_, const std::vector<T>& vector_) {
                static const unsigned int sizeOfType = sizeof(T);
                unsigned int bytesWritten = 0;
                unsigned int vectorSize = static_cast<unsigned int>(vector_.size());

                fout_.write(reinterpret_cast<char*>(&vectorSize),sizeof(vectorSize));
                bytesWritten += sizeof(vectorSize);

                T element;
                for (unsigned int i = 0; i < vectorSize; i++) {
                    element = vector_[i];
                    fout_.write(reinterpret_cast<char*>(&element),sizeOfType);
                }
                bytesWritten += vectorSize*sizeOfType;

                return bytesWritten;
            }

            template <class T>
            int assemble(std::ifstream& fin_, std::vector<T>& vector_) {
                static const unsigned int sizeOfType = sizeof(T);
                unsigned int bytesRead = 0;
                unsigned int vectorSize = 0;
                vector_.clear();

                fin_.read(reinterpret_cast<char*>(&vectorSize),sizeof(vectorSize));
                bytesRead += sizeof(vectorSize);

                vector_.resize(vectorSize);
                T element;
                for (unsigned int i = 0; i < vectorSize; i++) {
                    fin_.read(reinterpret_cast<char*>(&element),sizeOfType);
                    vector_[i] = element;
                }
                bytesRead += vectorSize*sizeOfType;

                return bytesRead;
            }

            template <>
            int serialize(std::ofstream& fout_, const std::vector<Premise>& premVector_);

            template <>
            int assemble(std::ifstream& fin_, std::vector<Premise>& premVector_);

            template <>
            int serialize(std::ofstream& fout_, const std::vector<Action>& actVector_);

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

#endif //AIBASE_H