int Masek::findline(Masek::IMAGE *image, double **lines)
{
filter *I2, *orND, *I3, *I4, *edgeimage;
double theta[180];
double *R, *xp;
double maxval;
int size, linecount;
double cx, cy;
int i, j, index, x, y;
double r, t;
int tmpcount;
FILE *fid;
fid = fopen("image.txt", "w");
for (i = 0; i<image->hsize[0]; i++)
{
for (j = 0; j<image->hsize[1]; j++)
fprintf(fid, "%d %d %d\n", i+1, j+1, image->data[i*image->hsize[1]+j]);
}
fclose(fid);
I2 = (filter*)malloc(sizeof(filter));
orND = (filter*)malloc(sizeof(filter));
image = canny(image, 2, 1, 0.00, 1.00, I2, orND);
//imwrite("C:\\testImage090716.bmp", image);
/*fid = fopen("I2.txt", "w");
for (i = 0; i<I2->hsize[0]; i++)
{
for (j = 0; j<I2->hsize[1]; j++)
fprintf(fid, "%d %d %f\n", i+1, j+1, I2->data[i*I2->hsize[1]+j]);
}
fclose(fid);
fid = fopen("or.txt", "w");
for (i = 0; i<or->hsize[0]; i++)
{
for (j = 0; j<or->hsize[1]; j++)
fprintf(fid, "%d %d %f\n", i+1, j+1, or->data[i*or->hsize[1]+j]);
}
fclose(fid);*/
I3 = adjgamma(I2, 1.9);
/*fid = fopen("I3.txt", "w");
for (i = 0; i<I3->hsize[0]; i++)
{
for (j = 0; j<I3->hsize[1]; j++)
fprintf(fid, "%d %d %f\n", i+1, j+1, I3->data[i*I3->hsize[1]+j]);
}
fclose(fid);*/
I4 = nonmaxsup(I3, orND, 1.5);
/*fid = fopen("I4.txt", "w");
for (i = 0; i<I4->hsize[0]; i++)
{
for (j = 0; j<I4->hsize[1]; j++)
fprintf(fid, "%d %d %f\n", i+1, j+1, I4->data[i*I4->hsize[1]+j]);
}
fclose(fid);*/
edgeimage = hysthresh(I4, 0.20, 0.15);
//edgeimage = hysthresh(I4, 0.25, 0.10);//LEE:
for (i = 0; i<180; i++)
theta[i] = i;
/*fid = fopen("edge.txt", "w");
for (i = 0; i<edgeimage->hsize[0]; i++)
{
for (j = 0; j<edgeimage->hsize[1]; j++)
fprintf(fid, "%d %d %f\n", i+1, j+1, edgeimage->data[i*edgeimage->hsize[1]+j]);
}
fclose(fid);*/
size = radonc(edgeimage->data, theta, edgeimage->hsize[0], edgeimage->hsize[1], 180, &R, &xp);
/*fid = fopen("r.txt", "w");
for (i = 0; i<size; i++)
{
for (j = 0; j<180; j++)
fprintf(fid, "%d %d %f\n", i+1, j+1, R[i*180+j]);
}
fclose(fid);
fid = fopen("xp.txt", "w");
for (i = 0; i<size; i++)
{
fprintf(fid, "%f\n", xp[i]);
}
fclose(fid);
*/
maxval = -1;
index = -1;
linecount=0;
for (i = 0; i<size*180; i++)
{
if (R[i]>maxval)
{
maxval = R[i];
index = i;
linecount=1;
}
else if (R[i]==maxval)
{
linecount++;
}
}
if (maxval<=25)
return 0;
*lines = (double*) malloc(sizeof(double)*linecount*3);
cx = image->hsize[1]/2.0-1;
cy = image->hsize[0]/2.0-1;
tmpcount=0;
for (i = index; i<size*180; i++)
{
if (R[i]==maxval)
{
y = i/180;
x = i%180;
t = -theta[x]*PI/180;
r = xp[y];
(*lines)[tmpcount*3] = cos(t);
(*lines)[tmpcount*3+1] = sin(t);
(*lines)[tmpcount*3+2] = -r;
(*lines)[tmpcount*3+2] = (*lines)[tmpcount*3+2] - (*lines)[tmpcount*3]*cx - (*lines)[tmpcount*3+1]*cy;
tmpcount++;
}
}
free(R);
free(xp);
free(I2->data);
free(I2);
free(orND->data);
free(orND);
return linecount;
}
//------------------------------------------------------------------------------
// Color and Gradient based edge detector
//------------------------------------------------------------------------------
void pbBG(IplImage *im, IplImage* grad, IplImage* ori){
double BETA[] = {-3.6945, 2.7430};
//1. Calculate color and texture gradient gradient
int norient = 8;
double *gtheta =(double *) malloc (sizeof(double)*norient);
fprintf(stdout,"\nCalculating Brightness gradient"); fflush(stdout);
CvMat** bg = detBG(im, norient, gtheta);
//2. Compute oriented Pb
CvMat* b;
CvMat** orientedPb = new CvMat* [norient];
for(int i=0; i< norient; i++){
orientedPb[i] = cvCreateMat(im->height, im->width, CV_32FC1);
b = bg[i];
for(int h=0; h < im->height; h++){
for(int w=0; w < im->width; w++){
double b_grad = cvGetReal2D(b,h,w);
double tmpSum = BETA[0] + b_grad*BETA[1];
cvSetReal2D(orientedPb[i], h, w, 1/(1+exp(-tmpSum)) );
}
}
//Release Matrices
cvReleaseMat(&b);
}
//3. nonmax suppression and max over orientations
CvMat* maxOri = cvCreateMat(im->height,im->width,CV_32FC1);
for(int h=0; h < im->height; h++){
for(int w=0; w < im->width; w++){
int maxOriInd=0;
for(int oriInd=0; oriInd < norient; oriInd++){
if ( cvGetReal2D(orientedPb[maxOriInd],h,w) < cvGetReal2D(orientedPb[oriInd],h,w) )
maxOriInd = oriInd;
}
cvSetReal2D(maxOri,h,w,(float)maxOriInd);
}
}
CvMat* pb = cvCreateMat(im->height, im->width, CV_32FC1); cvSetZero(pb);
CvMat* pbi = cvCreateMat(im->height, im->width, CV_32FC1); cvSetZero(pb);
CvMat* theta = cvCreateMat(im->height, im->width, CV_32FC1); cvSetZero(theta);
double r =2.5;
CvMat* tmp; // = cvCreateMat(im->height, im->width, CV_32FC1);
for(int i=0; i < norient; i++){
tmp = fitparab(*(orientedPb[i]),r,r,gtheta[i]);
pbi = nonmaxsup(tmp,gtheta[i]);
for(int h=0; h < pb->rows; h++){
for(int w=0; w < pb->cols; w++){
if( cvGet2D(maxOri,h,w).val[0] == i && cvGet2D(pb,h,w).val[0] < cvGet2D(pbi,h,w).val[0]){
cvSetReal2D(pb,h,w,cvGetReal2D(pbi,h,w));
}
if (cvGet2D(maxOri,h,w).val[0] == i){
cvSetReal2D(theta,h,w,gtheta[i]);
}
}
}
}
// "pb" should be between 0 and 1
for(int h=0; h < pb->rows; h++){
for( int w=0; w< pb->cols; w++){
if(cvGet2D(pb,h,w).val[0] < 0)
cvSet2D(pb,h,w,cvScalar(0.0));
if(cvGet2D(pb,h,w).val[0] > 1)
cvSet2D(pb,h,w,cvScalar(1.0));
}
}
// Copy cvMat into IplImages
for(int i=0; i<im->height; i++){
for(int j=0; j<im->width; j++){
cvSetReal2D(grad,i,j,cvGetReal2D(pb,i,j));
cvSetReal2D(ori,i,j,cvGetReal2D(theta,i,j));
}
}
//release matrices!!
for(int i=0; i<norient; i++)
cvReleaseMat(&orientedPb[i]);
cvReleaseMat(&pb);
cvReleaseMat(&pbi);
cvReleaseMat(&theta);
cvReleaseMat(&maxOri);
free(gtheta);
return;
}
