Files
EMD/main.cpp

101 lines
3.4 KiB
C++
Raw Normal View History

2011-11-30 12:02:40 +01:00
////////////////////////////////////////////////////////////////////////////////
2011-11-30 12:24:28 +01:00
// Empirical Mode Decomposition //
2011-11-30 12:02:40 +01:00
// BERNARD Guillaume //
// DURAND William //
// ZZ3F2 ISIMA //
////////////////////////////////////////////////////////////////////////////////
#include "CImg.h"
#include <math.h>
using namespace cimg_library;
/*******************************************************************************
2011-11-30 12:24:28 +01:00
Main
*******************************************************************************/
2011-11-30 12:02:40 +01:00
int main()
{
CImg<unsigned char> imgLena("lena.bmp");
CImgDisplay dispBase(imgLena,"Image de base");
2011-11-30 12:24:28 +01:00
///////////////////////////////////////////////////////////////////////////////
// Part 1: Finding minimas and maximas //
///////////////////////////////////////////////////////////////////////////////
2011-12-05 16:10:49 +01:00
CImg<unsigned char> imgMax = imgLena.channel(0);
CImg<unsigned char> imgMin = imgLena.channel(0);
2011-11-30 12:24:28 +01:00
2011-12-05 16:10:49 +01:00
imgMax.print();
2011-11-30 12:02:40 +01:00
for (int i = 0; i<imgLena.width() ; i+=3) {
for (int j = 0; j<imgLena.height() ; j+=3) {
2011-11-30 12:24:28 +01:00
// Save max and min locations
2011-11-30 12:02:40 +01:00
int xmax = i;
int ymax = j;
int xmin = i;
int ymin = j;
2011-11-30 12:24:28 +01:00
// save values
2011-11-30 12:02:40 +01:00
unsigned char max = imgMax(i,j);
unsigned char min = imgMin(i,j);
2011-11-30 12:24:28 +01:00
// 3x3
2011-11-30 12:02:40 +01:00
for (int k = i; k<i+3 ; k++) {
for (int l = j; l<j+3 ; l++) {
2011-11-30 12:24:28 +01:00
2011-12-05 16:10:49 +01:00
// Max
2011-11-30 12:02:40 +01:00
if ((imgMax(k,l) <= max)&&(l!=ymax &&k!=xmax)) {
imgMax(k,l) = 0;
2011-11-30 12:24:28 +01:00
} else {
2011-11-30 12:02:40 +01:00
max = imgMax(k,l);
imgMax(xmax,ymax) = 0;
xmax = k;
ymax = l;
}
2011-11-30 12:24:28 +01:00
2011-12-05 16:10:49 +01:00
// Min
2011-11-30 12:02:40 +01:00
if ((imgMin(k,l) >= min)&&(l!=ymin &&k!=xmin)) {
imgMin(k,l) = 0;
2011-11-30 12:24:28 +01:00
} else {
2011-11-30 12:02:40 +01:00
min = imgMin(k,l);
2011-12-05 16:10:49 +01:00
imgMin(xmin,ymin) = 0;
2011-11-30 12:02:40 +01:00
xmin = k;
ymin = l;
}
}
}
}
}
2011-11-30 12:24:28 +01:00
// Display images for max and min
2011-11-30 12:02:40 +01:00
CImgDisplay dispMax(imgMax,"Image de Max");
CImgDisplay dispMin(imgMin,"Image de Min");
2011-11-30 12:24:28 +01:00
///////////////////////////////////////////////////////////////////////////////
// Part 2: Average //
///////////////////////////////////////////////////////////////////////////////
2011-11-30 12:02:40 +01:00
2011-11-30 23:46:21 +01:00
// Calculate the upper envelope
// ??
// Calculate the lower envelope
// ??
// Calculate the Average
2011-11-30 12:02:40 +01:00
CImg<unsigned char> imgMoyenne = (imgMax+imgMin)/2;
CImgDisplay dispMoyenne(imgMoyenne,"Image Moyenne");
2011-11-30 12:24:28 +01:00
///////////////////////////////////////////////////////////////////////////////
// Partie 3: Deletion //
///////////////////////////////////////////////////////////////////////////////
2011-11-30 12:02:40 +01:00
CImg<unsigned char> imgFin = imgLena - imgMoyenne;
CImgDisplay dispFin(imgFin,"Image Finale");
2011-11-30 12:24:28 +01:00
while (!dispBase.is_closed()) {
2011-11-30 12:02:40 +01:00
dispBase.wait();
}
2011-11-30 12:24:28 +01:00
2011-11-30 12:02:40 +01:00
return 0;
}