/*M///////////////////////////////////////////////////////////////////////////////////////
//
//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
//
//  By downloading, copying, installing or using the software you agree to this license.
//  If you do not agree to this license, do not download, install,
//  copy or use the software.
//
//
//                           License Agreement
//                For Open Source Computer Vision Library
//
// Copyright (C) 2014, Itseez Inc, all rights reserved.
// Third party copyrights are property of their respective owners.
//
// Redistribution and use in source and binary forms, with or without modification,
// are permitted provided that the following conditions are met:
//
//   * Redistribution's of source code must retain the above copyright notice,
//     this list of conditions and the following disclaimer.
//
//   * Redistribution's in binary form must reproduce the above copyright notice,
//     this list of conditions and the following disclaimer in the documentation
//     and/or other materials provided with the distribution.
//
//   * The name of the copyright holders may not be used to endorse or promote products
//     derived from this software without specific prior written permission.
//
// This software is provided by the copyright holders and contributors "as is" and
// any express or implied warranties, including, but not limited to, the implied
// warranties of merchantability and fitness for a particular purpose are disclaimed.
// In no event shall the Itseez Inc or contributors be liable for any direct,
// indirect, incidental, special, exemplary, or consequential damages
// (including, but not limited to, procurement of substitute goods or services;
// loss of use, data, or profits; or business interruption) however caused
// and on any theory of liability, whether in contract, strict liability,
// or tort (including negligence or otherwise) arising in any way out of
// the use of this software, even if advised of the possibility of such damage.
//
//M*/

#include "opencv2/datasets/pd_caltech.hpp"
#include "opencv2/datasets/util.hpp"

namespace cv
{
namespace datasets
{

using namespace std;

class PD_caltechImp : public PD_caltech
{
public:
    PD_caltechImp() {}
    //PD_caltechImp(const string &path);
    virtual ~PD_caltechImp() {}

    virtual void load(const string &path);

private:
    void loadDataset(const string &path);
};

/*PD_caltechImp::PD_caltechImp(const string &path)
{
    loadDataset(path);
}*/

void PD_caltechImp::load(const string &path)
{
    loadDataset(path);
}

void PD_caltechImp::loadDataset(const string &path)
{
    train.push_back(vector< Ptr<Object> >());
    test.push_back(vector< Ptr<Object> >());
    validation.push_back(vector< Ptr<Object> >());

    createDirectory((path + "../images/"));

    vector<string> objectNames;
    getDirList(path, objectNames);
    for (vector<string>::iterator it=objectNames.begin(); it!=objectNames.end(); ++it)
    {
        Ptr<PD_caltechObj> curr(new PD_caltechObj);
        curr->name = *it;

        string objectPath(path + "../images/" + curr->name + "/");
        createDirectory(objectPath);

        string seqImagesPath(path + curr->name + "/");
        vector<string> seqNames;
        getDirList(seqImagesPath, seqNames);
        for (vector<string>::iterator itSeq=seqNames.begin(); itSeq!=seqNames.end(); ++itSeq)
        {
            string &seqName = *itSeq;

            createDirectory((objectPath + seqName));

            FILE *f = fopen((seqImagesPath + seqName).c_str(), "rb");

            #define SKIP 28+8+512
            fseek(f, SKIP, SEEK_CUR);

            unsigned int header[9];
            size_t res = fread(header, 9, 4, f);
            double fps;
            res = fread(&fps, 1, 8, f);
            fseek(f, 432, SEEK_CUR);

            /*printf("width %u\n", header[0]);
            printf("height %u\n", header[1]);
            printf("imageBitDepth %u\n", header[2]);
            printf("imageBitDepthReal %u\n", header[3]);
            printf("imageSizeBytes %u\n", header[4]);
            printf("imageFormat %u\n", header[5]);
            printf("numFrames %u\n", numFrames);
            printf("fps %f\n", fps);
            printf("trueImageSize %u\n", header[8]);*/
            unsigned int numFrames = header[6];

            string ext;
            switch (header[5])
            {
            case 100:
            case 200:
                ext = "raw";
                break;
            case 101:
                ext = "brgb8";
                break;
            case 102:
            case 201:
                ext = "jpg";
                break;
            case 103:
                ext = "jbrgb";
                break;
            case 001:
            case 002:
                ext = "png";
                break;
            }

            for (unsigned int i=0; i<numFrames; ++i)
            {
                unsigned int size;
                res = fread(&size, 1, 4, f);

                char imgName[20];
                sprintf(imgName, "/%u.%s", i, ext.c_str());
                curr->imageNames.push_back(imgName);

                // comment fseek and uncomment next block to unpack all frames
                fseek(f, size, SEEK_CUR);
                /*char *img = new char[size];
                fread(img, size, 1, f);
                string imgPath(objectPath + seqName + imgName);
                FILE *fImg = fopen(imgPath.c_str(), "wb");
                fwrite(img, size, 1, fImg);
                fclose(fImg);
                delete[] img;*/

                fseek(f, 12, SEEK_CUR);
            }

            if (0 != res) // should fix unused variable warning
            {
                res = 0;
            }

            fclose(f);
        }

        train.back().push_back(curr);
    }
}

Ptr<PD_caltech> PD_caltech::create()
{
    return Ptr<PD_caltechImp>(new PD_caltechImp);
}

}
}
