C:/Users/albertlr/projects/TarzaNN/TarzaNN/Dataset.h

Go to the documentation of this file.

/****************************************************************************
 **
 ** Copyright C 2002-2012 Laboratory for Active and Attentive Vision (LAAV), Department of Computer Science and Engineering, York University, Toronto, ON, Canada.
 ** All rights reserved.
 **
 ** This file is part of the TarzaNN Neural Network Simulator.
 **
 ** This file may be distributed and/or modified under the terms of the
 ** GNU General Public License version 2 as published by the Free Software
 ** Foundation and appearing in the file LICENSE.GPL included in the
 ** packaging of this file.
 **
 ** See http://www.tarzann.org/gpl/ for GPL licensing information.
 **
 ** Contact info@tarzann.org if any conditions of this licensing are
 ** not clear to you.
 **
 ** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 ** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 **
 ****************************************************************************/

#ifndef DATASET_H_INCLUDED
#define DATASET_H_INCLUDED
#include "OutputImage.h"

class Dataset {
protected:
        int count;

    int maxTestSteps;
    int maxSteps;
    int currentStep;
    bool sameAsLast;
    int invarianceLevel;

public:
        Dataset(char* basefilename);
        virtual ~Dataset();
        virtual OutputImage* getImage(int index, OutputImage** list, int LorR=0)=0;
        virtual Matrix* getDesiredOutcome(int index)=0;
        virtual int getRandomIndex();

    void setMaxTestSteps(int mTS){
        maxTestSteps=mTS;
    }
    int getMaxTestSteps(){
        return maxTestSteps;
    }
    void setMaxSteps(int mS){
        maxSteps=mS;
    }
    void setInvarianceLevel(int levels){
        invarianceLevel=levels;
    }
    int getInvarianceLevel(){
        return invarianceLevel;
    }

    int getMaxSteps(){
        return maxSteps;
    }

    bool isSameAsLast(){
        return sameAsLast;
    }

};
#endif/*DATASET_H_INCLUDED*/