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

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

#define MODULO 5

#define SWAP_LONG( A )  ((A << 24) | ((A & 0xff00L) << 8 ) | ((A & 0xff0000L) >> 8 ) | ((unsigned long)A >> 24))

struct imageHeader {
        int magic; // 4 bytes
        int ndim; // 4 bytes, little endian
        int dim[4];
};

struct infoHeader {
        int magic; // 4 bytes
        int ndim; // 4 bytes, little endian
        int dim[3];
};

struct categoryHeader {
        int magic; // 4 bytes
        int ndim; // 4 bytes, little endian
        int dim[3];
};


class NORB_Dataset : public Dataset{
protected:
        long imageBaseOffset;
        long imageSize;
        FILE* imageFile;
        OutputImage* imageL;
        OutputImage* imageR;

        long categoryBaseOffset;
        long categorySize;
        FILE* categoryFile;
        
        long infoBaseOffset;
        long infoSize;
        FILE* infoFile;
        int info[10];

        void init(char* basefilename);
        void readImageFile(char* filename);
        void readCategoryFile(char* filename);
        void readInfoFile(char* filename);
        int IMAGE_SIZE, INPUT_IMAGE_SIZE;

public:
        NORB_Dataset(char* basefilename);
        ~NORB_Dataset();
        OutputImage* getImage(int index, OutputImage** list, int LorR);
        Matrix* getDesiredOutcome(int index);
        int* getInfo(int index);
};
#endif/*NORBDATASET_H_INCLUDED*/