ARQUIVO: main.

cpp
#include "menus.h" #include "realce.h" #include <stdlib.h> using namespace cv; int main (){ int op_menu, op_img; string img[9] = {"airplane.pgm", "barbara.pgm", "boat.pgm", "frog.pgm", "goldhill.pgm", "gull.pgm", "lena.pgm", "mandrill.pgm", "peppers.pgm"}; string nome, nome2; char aux[50], n1[100]; int histo[256]; apresentacao(); //chama funo para ttulo do sistema op_img = selec_img(img); //chama funo para menu de selao de imagens && L a escolha do usurio apresentacao(); //chama funo para ttulo do sistema op_menu = menu_principal(); //chama funo para menu de selao do tipo de realce && L a escolha do usurio //Monta matriz com a imagem selecionada: Mat image = imread("img/"+img[op_img-1], 0); //Exibe a imagem selacionada: namedWindow("Imagem selecionada"); imshow("Imagem selecionada",image); Mat imageT; switch (op_menu){ case 1: nome = "T linear reta a= "; for (int i=0; i<5; i++){ imageT = tranf_linear_reta(image, (i+1), 0); itoa((i+1), aux, 10); namedWindow(nome+aux); imshow(nome+aux, imageT); } break; case 2: nome = "T linear reta b= "; for (int i=0; i<161; i+=40){ imageT = tranf_linear_reta(image, 1, (i-80) ); itoa((i-80), aux, 10); namedWindow(nome+aux); imshow(nome+aux, imageT); } break; case 3: nome = "T linear reta a= "; strcpy(n1, "Histograma a= "); for (int i=0; i<5; i++){ imageT = tranf_linear_reta(image, (i+1), 0); itoa((i+1), aux, 10); namedWindow(nome+aux); imshow(nome+aux, imageT); histograma(imageT, histo); strcat(n1, aux); plota_hist(histo, n1); strcpy(n1, "Histograma a= "); } break; case 4: nome = "T linear reta b= "; strcpy(n1, "Histograma b= "); for (int i=0; i<161; i+=40){ imageT = tranf_linear_reta(image, 1, (i-80) ); itoa((i-80), aux, 10); namedWindow(nome+aux);

imshow(nome+aux, imageT); histograma(imageT, histo); strcat(n1, aux); plota_hist(histo, n1); strcpy(n1, "Histograma b= "); } break; case 5: imageT = equal_histog(image); namedWindow("Imagem Equalizada"); imshow("Imagem Equalizada", imageT); int hist[256], hist2[256]; histograma(image, hist); plota_hist(hist, "Histograma original"); histograma(imageT, hist2); plota_hist(hist2, "Histograma equalizado"); break; } waitKey(); return 0; }

ARQUIVO: realce.cpp
#include "realce.h" using std::cout; using std::cin; using std::endl; void histograma(Mat image, int H[]){ for(int i = 0; i < 256; i++){ H[i] = 0; } for(int i = 0; i < image.rows; i++) for(int j = 0; j < image.cols; j++) H[(int)image.at<uchar>(i,j)]++; } void plota_hist(int histogram[], const char* name) { int hist[256]; for(int i = 0; i < 256; i++) { hist[i]=histogram[i]; } // Desenha o histograma int hist_w = 512; int hist_h = 400; int bin_w = cvRound((double) hist_w/256); Mat histImage(hist_h, hist_w, CV_8UC1, Scalar(255, 255, 255)); // histImage e o nome de uma variavel. // CV_8UC1 - Tipo de elementos da matriz no formato CV_<bit depth><S|U|F>C<numero de canais> , onde S=signed, U=unsigned, F=float. Por exemplo, CV _ 8UC1 significa que os elementos so 8-bit unsigned e existe 1 canal, and CV _ 32SC2 elementos de 32-bit signed com 2 canais. // Acha o elemento de maior intensidade no histograma int max = hist[0]; //Comeca pelo 0 for(int i = 1; i < 256; i++){ if(max < hist[i]){ max = hist[i]; } } // normalize the histogram between 0 and histImage.rows for(int i = 0; i < 256; i++){ hist[i] = ((double)hist[i]/max)*histImage.rows; }

// draw the intensity line for histogram for(int i = 0; i < 256; i++) { line(histImage, Point(bin_w*(i), hist_h), //line desenha uma linha: line(Mat& img, Point pt1, Point pt2, const Scalar& color, int thickness=1, int lineType=8, int shift=0) Point(bin_w*(i), hist_h - hist[i]), Scalar(0,0,0), 1, 8, 0); } // display histogram namedWindow(name, CV_WINDOW_AUTOSIZE); imshow(name, histImage); } uchar calc_partes(uchar x, int f1, int g1, int f2, int g2) { float result; if (0 <= x && x <= f1){ result = g1/f1 * x; } else if (f1 < x && x <= f2){ result = ((g2 - g1)/(f2 - f1)) * (x - f1) + g1; } else if (f2 < x && x <= 255){ result = ((255 - g2)/(255 - f2)) * (x - f2) + g2; } return (uchar)result; } Mat tranf_linear_partes(Mat& image){ uchar output; Mat new_image = image.clone(); int f1, g1, f2, g2; cout<<"Entre cout<<"Entre com cout<<"Entre com cout<<"Entre com com um valor para um valor para g1: um valor para f2: um valor para g2: f1: "<<endl; cin>>f1; "<<endl; cin>>g1; "<<endl; cin>>f2; "<<endl; cin>>g2;

void

for(int y = 0; y < image.rows; y++){ for(int x = 0; x < image.cols; x++){ output = calc_partes(image.at<uchar>(y,x), f1, g1, f2, g2); new_image.at<uchar>(y,x) = saturate_cast<uchar>(output); } } return (new_image); } Mat tranf_linear_reta(Mat& image, float a, int b){ //S = a*f + b Mat new_image = image.clone(); uchar output; /*int a, b; cout<<"Entre com um valor para a: "<<endl; cin>>a; cout<<"Entre com um valor para b: "<<endl; cin>>b; */ for(int y = 0; y < image.rows; y++){ for(int x = 0; x < image.cols; x++){ if (a* image.at<uchar>(y,x) +b >= 255){ output = 255; } else if (a* image.at<uchar>(y,x) +b <= 0){

output =0; } else { output = a* image.at<uchar>(y,x) +b; } new_image.at<uchar>(y,x) = saturate_cast<uchar>(output); } } return(new_image); } Mat tranf_nlinear_log(Mat& image){ // s = c log(1 + f) Mat new_image = image.clone(); uchar output; int c; double L; cout<<"Entre com um valor para 'c': "<<endl; cin>>c; for(int y = 0; y < image.rows; y++){ for(int x = 0; x < image.cols; x++){ output = image.at<uchar>(y,x); L = c* log ( (double)output + 1); output = (uchar) L; new_image.at<uchar>(y,x) = saturate_cast<uchar>(output); } } return(new_image); } Mat tranf_negativo (Mat& image){ // s = 255-f Mat new_image = image.clone(); uchar output; for(int y = 0; y < image.rows; y++){ for(int x = 0; x < image.cols; x++){ output = (uchar) 255 - image.at<uchar>(y,x); new_image.at<uchar>(y,x) = saturate_cast<uchar>(output); } } return(new_image); } Mat equal_histog (Mat& image) { int histogram[256]; histograma(image, histogram); // Calcula o tamanho da imagem: int size = image.rows * image.cols; // Fator de ormalizao o histograma com 0 <= alfa <= 255 float alpha = (float) 255.0/size; // Calcula probabilidade float Proba[256]; for(int i = 0; i < 256; i++) { Proba[i] = (float) histogram[i] / size; } // Calcula funo acumulativa de probabilidade int acumuH[256]; acumuH[0] = histogram[0]; for(int i = 1; i < 256; i++) { acumuH[i] = acumuH[i-1] + histogram[i]; }

// Escala do histograma int Sk[256]; for(int i = 0; i < 256; i++) { Sk[i] = cvRound((double) acumuH[i] * alpha); mais prximo } // Gera um histograma equalizado float PsSk[256]; for(int i = 0; i < 256; i++) { PsSk[i] = 0; } for(int i = 0; i < 256; i++) { PsSk[Sk[i]] += Proba[i]; } int final[256]; for(int i = 0; i < 256; i++) final[i] = cvRound(PsSk[i]*255); // Gera imagem equalizada Mat new_image = image.clone();

// cvRound arredonda um numero de ponto flutuante para um inteiro

for(int y = 0; y < image.rows; y++) for(int x = 0; x < image.cols; x++) new_image.at<uchar>(y,x) = saturate_cast<uchar>(Sk[image.at<uchar>(y,x)]); return(new_image); }