BlurEdgeDetection/smoothfunctioncreator.cpp at master · iguha94/BlurEdgeDetection · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include <iostream>
#include "CImg.h"
#include "Commons.h"
#include "Global.h"
#include <fstream>
#include <string>
#include <utility>
#include <list>
#include <map>
using namespace std;
using namespace cimg_library;

class  CPoint{
public:
    int x;
    int y;
    int value;
    CPoint(int x, int y, int value){
        this->x=x;
        this->y=y;
        this->value=value;
    }
};

int main(int argc, char*argv[]){
    string imagename;
    basepath=string(argv[1]);
    int interval;
    cout<<"Enter ImageName: ";
    cin>>imagename;
   // cout<<"Enter sampling interval (square)";
    //cin>>interval;
    int range;
    cout<<"Maximum scale";
    cin>>range;
    string imagefile=basepath+imagename;
    string blurimagefile=basepath+"blur_"+imagename;
    fstream fin;
    string blurpointsfile=basepath+"BlurPoints.txt";
    fin.open(blurpointsfile.c_str());

    CImg<unsigned short> image(imagefile.c_str());
    CImg<unsigned short> blurimage(image.width(), image.height(), 1, 1, 0);;
    allocateMemory(image.width(), image.height());
    list<CPoint > controlpoints;
    map<long long int, CPoint*> pointsmap;
    map<long long int, CPoint*>::iterator it;

    int x,y,value;

//    for(int i=0;i<image.width();i+=interval){
//        for(int j=0;j<image.height();j+=interval){
//            long long int index=calculateIndex(i,j,image.width(),image.height());
//            int value=rand()%range;
//            CPoint* cp=new CPoint(i,j,value);
//            CPoint cp1(x,y,value);
//            controlpoints.push_back(cp1);
//            pointsmap.insert(pair<long long int,CPoint*>(index, cp));
//        }
//    }

        while(fin>>x>>y>>value){
            //if(value>=15 ) value+=10;
            //else if(value>=10 ) value+=30;
            cout<<"x: "<<x<<" y: "<<y<<" value: "<<value<<"\n";
            CPoint* cp=new CPoint(x,y,value);
            CPoint cp1(x,y,value);
            controlpoints.push_back(cp1);
            pointsmap.insert(pair<long long int,CPoint*>(calculateIndex(x,y,image.width(),image.height()), cp));
            blurimage(x,y)=value;
        }

    /*for(int i=0;i<image.width();i+=interval){
        for(int j=0;j<image.height();j+=interval){
            long long int index=calculateIndex(i,j,image.width(),image.height());
            if(pointsmap.find(index)!=pointsmap.end()){
                blurimage(i,j)=pointsmap[index]->value;
                //cout<<"Here\n";
            }
            else{
                double totalsum=0;
                double totalweight=0;
                double minimum=INT_MAX;
                double smin=INT_MAX;
                double sminval,minval;
                for (list<CPoint>::iterator it=controlpoints.begin(); it != controlpoints.end(); ++it){
                    int x=(*it).x;
                    int y=(*it).y;
                    int val=(*it).value;
                    double distance=CartesianDistance2D(i,j,x,y);
                    if(distance<minimum) {smin=minimum;minimum=distance;sminval=minval;minval=val;}
                    else if(distance<smin) {smin=distance;sminval=val;}
                }
                double sminw=1/smin;
                double minw=1/minimum;
                totalweight=(sminval*sminw+minval*minw);
                totalsum=sminw+minw;
               blurimage(i,j)=totalweight/totalsum;
               //blurimage(i,j)=maximum;
            }
        }
    }*/

    blurimage.display();
    blurimage.save_tiff(blurimagefile.c_str());
}