Filter.cpp

Aller à la documentation de ce fichier.

/// \file 
///     Implémentation des classes filtres
#include "Filter.hpp"
///////////////////////////////////////////////////////////////////////////////
/// Classe : Filter_Base
///////////////////////////////////////////////////////////////////////////////
Filter_Base::Filter_Base(int _size):size(_size){}
float Filter_Base::getMask(int col, int row){
        return mask[row * size + col];
}
void Filter_Base::setMask(int col, int row, float value){
        mask[row * size + col] = value;
}
///     .
/// Sommation des valeurs du filtres et division de chaque coefficients
/// par cette somme.
void Filter_Base::normalise() {
        float sum = 0.0;
        for (int j = 0; j < size; j++){
                for (int i = 0; i < size; i++) {
                        sum += getMask(i, j);
                }
        }
        if (sum != 0) {
                for (int j = 0; j < size; j++){
                        for (int i = 0; i < size; i++) {
                                setMask(i, j, getMask(i, j) / sum);
                        }
                }
        }
}
///     .
/// Applique le filtre en chaque pointe de l'image source. 
PNG Filter_Base::apply(PNG src){
        Trace trace("apply", channelFilter);
        PNG ret(src.getWidth(), src.getHeight(), src.getType(), "");
        // Dimension de l'image
        int h = src.getHeight();
        int w = src.getWidth();
        int offset = size / 2;
        float r, g, b, gray;
        // Parcourt de l'image
        for(int vy = 0; vy < h; vy++){
                for(int vx = 0; vx < w; vx++){
                        r = g = b = gray = 0;
                        // Parcourt du filtre
                        for(int i = 0; i < size; i++){
                for(int j = 0; j < size; j++){
                Color c = src.getColor(vx + j - offset, vy + i - offset);
                if(COLOR_GRAY == c.getType()){
                                                gray  += c.getGray() * getMask(j, i);
                } else if(COLOR_COLOR == c.getType()){
                        float f = getMask(j, i);
                                r += c.getRed() * f;
                        g += c.getGreen() * f;
                        b += c.getBlue() * f;
                }
           }
                        }
            // Résultat dans l'image destination
          if(PNG_GRAY == src.getType()){
                                ret.setColor(vx, vy, gray);
            } else if(PNG_COLOR == src.getType()){
                                Color result = Color(r, g, b, true);
                                ret.setColor(vx, vy, result);
                        }
                }
  }
        return ret;
}
///////////////////////////////////////////////////////////////////////////////
/// Classe : Filter_Base
///////////////////////////////////////////////////////////////////////////////

///     .
/// Initialise les données membres et calcule du filtre
Filter_Gaussian::Filter_Gaussian(int size, float _mu, float _sigma):Filter_Base(size){
        sigma = _sigma;
        mu = _mu;
        mask = new float[size * size];
        computeMask();
        normalise();
}
///     .
/// Calcul du filtre gaussien
void Filter_Gaussian::computeMask(){
        Trace trace("_computeMask", channelFilter);     
        float xCenter = (size - 1) / 2.0;
        float yCenter = (size - 1) / 2.0;

        for(int j = 0; j < size; j++){
                for(int i = 0; i < size; i++){
                        float value = exp(-(((double) i - xCenter - mu) * ((double) i - xCenter - mu) + ((double) j - yCenter - mu) * ((double) j - yCenter - mu)) / (2 * sigma * sigma));
        setMask(i, j, value);
                }
        }
}
string Filter_Gaussian::toString(){
        Trace trace("toString", channelFilter);         
        std::ostringstream o;
        o << "mu " << mu << " sigma " << sigma << "\n";
        for(int i = 0; i < size; i++){
                for(int j = 0; j < size; j++){
                        o << "[" << i << ":" << j << "]" << getMask(i, j) << "\t";
                }
                o << "\n";
        }
        o << "\n\n";
        return o.str();
}

---