Histogram.cpp

Aller à la documentation de ce fichier.

/// \file 
/// Définition de la classe Histogram

#include "Histogram.hpp"
///////////////////////////////////////////////////////////////////////////////
/// HistogramClass
///////////////////////////////////////////////////////////////////////////////
HistogramClass::HistogramClass(Color _color):
        color(_color), 
        total(0) {
}
string HistogramClass::toString()const{
        ostringstream o;
        o << color.toString() << " #" << data.size()  << " #TOTAL " << total;
        return o.str();
}
void HistogramClass::addPixel(Pixel pixel){
        data.push_back(pixel);
}
int HistogramClass::getSize()const{
        return data.size();
}
void HistogramClass::setTotal(int _total){
        total = _total;
}
int HistogramClass::getTotal()const{
        return total;
}
Pixel HistogramClass::getElementAt(const int i)const{
        return data[i];
}
///////////////////////////////////////////////////////////////////////////////
/// Histogram
///
/// Met en correspondance une couleur et sa classe
///////////////////////////////////////////////////////////////////////////////
void Histogram::addPixel(Pixel pixel){
        // MAJ nombre de pixel
        pixelNumber++;
        Color color = pixel.getColor();
        // Cherche si la classe de cette couleur existe, sinon on la crée
        map_iter it = data.find(color);
        if(it != data.end()){
                it->second.addPixel(pixel);
        } else {
                HistogramClass cl(color);
                cl.addPixel(pixel);
                data.insert(make_pair(color, cl));
        }
}
string Histogram::toString()const{
        const_map_iter b = data.begin();
        const_map_iter e = data.end();
        ostringstream o;
        while(b != e){
                o << "Class " << b->first.toString() << " " << b->second.toString() << endl;
                b++;
        }
        return o.str();
}
///     .
/// Calcul de l'histogramme cumulé. On parcourt chaque classe en mettant
/// à jour le total de la suivante
void Histogram::makeCumul(){
        map_iter b = data.begin();
        map_iter e = data.end();
        int total = 0;
        while(b != e){
                total += b->second.getSize();
                b->second.setTotal(total);
                b++;
        }
}
///     .
/// Sépare l'ensemble des couleurs en un nombre classe. Le but de cette
/// méthode est de fournir de bonnes valeurs de départ à l'algorithme des
/// KMean
void Histogram::makeClass(int classNumber){
        // Calcul de la "largeur" d'une classe
        float delta = (float)pixelNumber / (float)classNumber;
        for(int i = 0; i < classNumber; i++){
                float j = i;
                // Indice de début et fin
                int start = (int)(j * delta);
                int end = (int)(start + delta - 1);
                // On prent la première couleur de la classe
                Color c1 = getElementAt(start).getColor();
                Color c2 = getElementAt(end).getColor();
                Color res;
                /*
                Variant faire la moyenne des couleurs extremes
    if(COLOR_GRAY == c1.getType()){
                        float gray = (c1.getGray() + c2.getGray()) / 2;
                        res = Color(gray);
                }else if(COLOR_COLOR == c1.getType()){
                        float h = (c1.getHue() + c2.getHue()) / 2;
                        float s = (c1.getSaturation() + c2.getSaturation()) / 2;
                        float v = (c1.getValue() + c2.getValue()) / 2;
                        res = Color(h, s, v, false);
                }
                Autre variante faire la moyenne de toutes les couleurs
                for(int j = start + 1; j < end; j++){
                        Color c1 = getElementAt(j).getColor();
                        c = c + c1;
                }
                c = c / delta;
                */
                res = c1;
                classColor.push_back(res);
        }
}
///     .
/// Parcourt chaque classe jusqu'à arriver au rang demandé
Pixel Histogram::getElementAt(const int i)const{
        const_map_iter b = data.begin();
        const_map_iter e = data.end();

        int total = 0;

        while(b != e){
                total += b->second.getSize();
                if(total > i){
                        total -= b->second.getSize();
                        break;
                }
                b++;
        }
        int ndx = i - total;
        Pixel ret = b->second.getElementAt(ndx);
        return ret;
}
Color Histogram::getColor(int i)const{
        return classColor[i];
}

---