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

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

#define STEPS_STAGE_1 10000
#define STEPS_STAGE_2 10000
#define STEPS_STAGE_3 10000

#define IMAGE_SIZE_X 54
#define IMAGE_SIZE_Y 54

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

class CurrentImageF;

class Caucasian_Dataset : public Dataset {
        
private:
        long imageBaseOffset;
        long imageSize;
        FILE* imageFile;
        char* currentfileName;
        QFileInfoList fileList;
        Matrix* result;
        
    CurrentImageF* currentImage;
    
    static const char* DataFolder;
    
    QImage* baseQImage;
    
    void goToNextStep();

    QMutex dataPrep;
    
public:
        Caucasian_Dataset(char* basefilename);
        Caucasian_Dataset(char* basefilename, int steps);
        ~Caucasian_Dataset();
        OutputImage* getImage(int index, OutputImage** list, int LorR=0);
        Matrix* getDesiredOutcome(int index);
        int getRandomIndex();

};
#endif/*CaucasianDataset_H_INCLUDED*/