C:/Users/albertlr/projects/TarzaNN/TarzaNN/SOMGeomDataset.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 SOMGeomDataset_H_INCLUDED
#define SOMGeomDataset_H_INCLUDED

#include <vector>
using namespace std;
#include "Dataset.h"
#include <QDir>

#define FULL_IMAGE_SIZE_X 166
#define FULL_IMAGE_SIZE_Y 166

#define COIL_IMAGE_SIZE_X 128
#define COIL_IMAGE_SIZE_Y 128

#define MAX_OBJECTS 100
#define MAX_ANGLES 71
#define ANGLE_INCREMENT 5

class CurrentImage;

//Dataset for SOM learning.
class SOM_GeomDataset : public Dataset {
        
private:
        long imageBaseOffset;
        long imageSize;
        FILE* imageFile;
        char* currentfileName;
        QFileInfoList fileList;
        Matrix* result;

    float RandomFloat(float min, float max);

    QMutex dataPrep;
    CurrentImage* currentImage;
    void goToNextStep();

public:
        SOM_GeomDataset(char* basefilename);
        ~SOM_GeomDataset();
        OutputImage* getImage(int index, OutputImage** list, int LorR=0);
        Matrix* getDesiredOutcome(int index);
        int getRandomIndex();
};
#endif/*SOMGeomDataset_H_INCLUDED*/