root/ggpa/ActQNetwork.h

/*
 * ActQNetwork.h
 */

#ifndef ACTQNETWORK_H
#define ACTQNETWORK_H

#include <string>
#include <map>


#include "State.h"
#include "Action.h"
#include "QFunction.h"
#include "QNeuralNetwork.h"

#include "fann/fann.h"

class ActQNetwork : public QFunction {

private:

    map<Action, QNeuralNetwork> theMap;

public:
    ActQNetwork();
    ~ActQNetwork();

    
     /*
     * ActQNetwork
     * Parameters:   actionLength - max length on action.toInputs()
     *               stateLength - max length on state.toInputs()
     *               legalMoves - all possible actions
     *
     * Constructor. Creates a neural network.
     */
    ActQNetwork(int actionLength, int stateLength, vector<Action*> legalMoves);

    /*
     * update
     * Parameters:   state - a State object that represents a state in the game
     *               action - an Action object that represents an action in the
     *                        game
     *               utility - a double that represents the utility to insert
     *                         for this State/Action pair
     * No return value
     *
     * Updates the entry for the given State/Action pair.
     */
    void update(State &state, Action &action, double utility);

    /*
     * getValue
     * Parameters:   state - a State object that represents a state in the game
     *               action - an Action object that represents an action in the
     *                        game
     * Return value: a double representing the utility of the given State/
     *   Action pair.
     * This function will return the current utility of the given State/Action 
     *   pair.  In other words, it will return a value representing how good 
     *   the action is from the state.
     */
    double getValue(State &state, Action &action);

    /*
     * writeToFile
     *
     * Parameters:   file - String containing the name of the file to write to.
     * Return value: none
     *
     * Writes the QNeuralNetwork to the given file.  If the file does not exist, it
     *   will be created.  If it does exist, then it will be written to the 
     *   end of the file
     */
    void writeToFile(string filename);

    /* 
     * readFromFile
     *
     * Parameters:   file - string containing the name of the file to read from
     * Return value: none
     * 
     * Reads in a QNeuralNetwork from a file.  The current QNeuralNetwork will be set to 
     *    be this saved QNeuralNetwork
     */
    void readFromFile(string filename);
};

#endif