/*
----
This file is part of SECONDO.

Copyright (C) 2004, University in Hagen, Department of Computer Science,
Database Systems for New Applications.

SECONDO is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

SECONDO is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with SECONDO; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
----

//paragraph [1] title: [{\Large \bf ]   [}]

January 2017 Michael Loris


[1] Declarations for the JPEGImage class

*/
 

#ifndef IMAGE_H_
#define IMAGE_H_

#include<vector>
#include<string>

/* 
Color model conversion functions were taken 
from the GeneralTree Algebra .
 
*/
namespace conversion {
/********************************************************************
1.1 Struct Lab

This struct models a lab-color value, which will be computed in the constructor from a rgb-color value.

********************************************************************/

struct Lab
{
    double L, a, b;
    Lab (unsigned char r_, unsigned char g_, unsigned char b_);

}; // struct Lab

/********************************************************************
1.1 Struct HSV

This struct models a hsv-color value, which will be computed in the constructor from a rgb-color value.

********************************************************************/

struct HSV
{
    int h, s, v;
    HSV (unsigned char r, unsigned char g, unsigned char b);

}; // struct HSV


} //end namespace

struct Feature 
{
    int x;
    int y;
    double colorValue1;
    double colorValue2;
    double colorValue3;
    double coarseness;
    double contrast;
};



struct FeatureSignatureTuple
{
    FeatureSignatureTuple() {}

    FeatureSignatureTuple(double _weight, int _x, int _y, 
                double _colorValue1, double _colorValue2, 
                double _colorValue3, 
                double _coa, double _con)
    {
        weight = _weight;
        centroid.x = _x;
        centroid.y = _y;
        centroid.colorValue1 = _colorValue1;
        centroid.colorValue2 = _colorValue2;
        centroid.colorValue3 = _colorValue3;
        centroid.coarseness = _coa;
        centroid.contrast = _con;
    }    
    FeatureSignatureTuple(double _weight, Feature _centroid):
    weight(_weight), centroid(_centroid) {}
    FeatureSignatureTuple(const FeatureSignatureTuple& fst) : 
    weight(fst.weight), centroid(fst.centroid) {}    
    FeatureSignatureTuple& operator=(const FeatureSignatureTuple& fst)
    {
        weight = fst.weight;        
        centroid = fst.centroid;
        return *this;
    }

  ~FeatureSignatureTuple(){}

  double weight;
  Feature centroid;
};




// this class has more than one purpose, 
// but also JPEG handling, Tamura Features & clustering 
// All has been put here for the sake of simplicity

class JPEGImage
{
public:    
    void importJPEGFile(const std::string _fileName, 
                        const int colorSpace, 
                        const int coaRange,
                        const int conRange,
                        const int patchSize,
                        const int percentSamples,
                        const int noClusters);
    
    void computeCoarsenessValues(const int range);
    void computeContrastValues(const int range);
    void getRandomRepresentants(const unsigned int r);
    void clusterFeatures(const unsigned int k, 
            unsigned int dimensions, 
            unsigned int noDataPoints);
    
    void writeColorImage(const char* fileName);
    void writeGrayscaleImage(const char* fileName);
    void writeCoarsenessImage(const char* fileName, double normalization);
    void writeContrastImage(const char* fileName, double normalization);
    void writeClusterImage(const char* fileName, double normalization);
    
    int width; // image's height
    int height; // image's width

    int* centersX; // position as array
    int* centersY; // position as array
    double* colorValues1; // color value as array
    double* colorValues2; // color value as array
    double* colorValues3; // color value as array
    double* coa; // coarseness as array
    double* con; // constrast as array
    
    double* weights; // weights of centroids/representants    
    
    std::vector<FeatureSignatureTuple> signature;
    int getNoDataPoints() { return this->noDataPoints; };
        
    ~JPEGImage(); 
    
private:
    bool isGrayscale;
    unsigned char* pixels; // store pixles as array during import
    unsigned char*** pixMat4;  // write clustered circle image
    double*** pixMat5;             // stores features
    std::string fileName;

    double ak( int x, int y, unsigned int k);
    double ekh(int x, int y, unsigned int k);
    double ekv(int x, int y, unsigned int k);
    double localCoarseness(int x, int y, const int range);
    double my(int x, int y, const int range);
    double sigma(int x, int y, const int range);
    double eta(int x, int y, const int range);
    double localContrast(int x, int y, const int range);

    unsigned int* randomRepresentantsX;
    unsigned int* randomRepresentantsY;
    unsigned short** assignments; // to which cluster is each centroid assigned
    std::vector<std::vector<Feature> >* clusters; // output of k-means
    void drawCircle(int x, int y, int r);
    int* samplesX;
    int* samplesY;
    int noSamples;
    double* coarsenesses;
    double* contrasts;
    int colorSpace;
    int patchSize; // size of sub images to be extracted
    int noDataPoints; 
    
    void scalePCTDimensions();
};



#endif /* IMAGE_H_ */