Thierry Leconte
21 years ago
2 changed files with 170 additions and 56 deletions
Split View
Diff Options
+ 107
- 43
image.c
View File
| @@ -24,84 +24,148 @@ | |||
| #include <string.h> | |||
| #include <sndfile.h> | |||
| typedef struct { | |||
| double slope; | |||
| double offset; | |||
| } rgparam; | |||
| static void rgcomp(double *x, rgparam *rgpr) | |||
| { | |||
| /*const double y[9] = { 0.106,0.215,0.324,0.434,0.542,0.652,0.78,0.87 ,0.0}; */ | |||
| const double y[9] = { 0.11872,0.2408,0.36288,0.48608,0.60704,0.73024,0.8736,0.9744 , 0.0 }; | |||
| const double yavg=0.48819556; | |||
| double xavg; | |||
| double sxx,sxy; | |||
| int i; | |||
| for(i=0,xavg=0.0;i<9;i++) | |||
| xavg+=x[i]; | |||
| xavg/=9; | |||
| for(i=0,sxx=0.0;i<9;i++) { | |||
| float t=x[i]-xavg; | |||
| sxx+=t*t; | |||
| } | |||
| for(i=0,sxy=0.0;i<9;i++) { | |||
| sxy+=(x[i]-xavg)*(y[i]-yavg); | |||
| } | |||
| rgpr->slope=sxy/sxx; | |||
| rgpr->offset=yavg-rgpr->slope*xavg; | |||
| } | |||
| static double rgcal(float x,rgparam *rgpr) | |||
| { | |||
| return(rgpr->slope*x+rgpr->offset); | |||
| } | |||
| int Calibrate(float **prow,int nrow,int offset) | |||
| { | |||
| /* const double cal[8] = { 0.106,0.215,0.324,0.434,0.542,0.652,0.78,0.87 }; */ | |||
| const double cal[8] = { 0.1155 ,0.234,0.353,0.473,0.59,0.71,0.85,0.95 }; | |||
| float tele[2000]; | |||
| float frm[16]; | |||
| float slv[8]; | |||
| int n,i; | |||
| int shift,channel; | |||
| double tele[3000]; | |||
| double frm[9]; | |||
| rgparam regr[30]; | |||
| int n; | |||
| int k,mk; | |||
| int shift; | |||
| int mshift,channel; | |||
| float max; | |||
| printf("Calibration "); | |||
| fflush(stdout); | |||
| if(nrow<128) { | |||
| fprintf(stderr," impossible, not enought row\n"); | |||
| return (0); | |||
| } | |||
| /* build telemetry values */ | |||
| /* build telemetry values lines */ | |||
| for(n=0;n<nrow;n++) { | |||
| int i; | |||
| tele[n]=0.0; | |||
| for(i=3;i<43;i++) | |||
| tele[n]+=prow[n][i+offset+909]; | |||
| tele[n]/=40; | |||
| } | |||
| max=0.0;shift=0; | |||
| for(n=60;n<nrow-64;n++) { | |||
| if(nrow<128) { | |||
| fprintf(stderr," impossible, not enought row\n"); | |||
| return (0); | |||
| } | |||
| /* find telemetry start */ | |||
| max=-1000.0;mshift=0; | |||
| for(n=60;n<nrow-3;n++) { | |||
| float df; | |||
| df=(tele[n-4]+tele[n-3]+tele[n-2]+tele[n-1])-(tele[n]+tele[n+1]+tele[n+2]+tele[n+3]); | |||
| if(df> max){ shift=n; max=df; } | |||
| if(df> max){ mshift=n; max=df; } | |||
| } | |||
| for(n=0;n<16;n++) { | |||
| frm[n]=0.0; | |||
| for(i=1;i<7;i++) | |||
| frm[n]+=tele[shift-64+n*8+i]; | |||
| frm[n]/=6; | |||
| mshift-=64; | |||
| shift=mshift%128; | |||
| if(nrow<shift+8*9) { | |||
| fprintf(stderr," impossible, not enought row\n"); | |||
| return (0); | |||
| } | |||
| /* channel I.D */ | |||
| max=100000.0; | |||
| for(n=0;n<6;n++) { | |||
| float df; | |||
| df=fabs(frm[n]-frm[15]); | |||
| if(df<max) { | |||
| max=df; | |||
| channel=n; | |||
| /* compute regression params */ | |||
| for(n=shift,k=0;n<nrow-8*9;n+=128,k++) { | |||
| int j; | |||
| for(j=0;j<9;j++) { | |||
| int i; | |||
| frm[j]=0.0; | |||
| for(i=1;i<7;i++) | |||
| frm[j]+=tele[n+j*8+i]; | |||
| frm[j]/=6; | |||
| } | |||
| } | |||
| fflush(stdout); | |||
| rgcomp(frm,&(regr[k])); | |||
| /* compute slopes */ | |||
| slv[0]=cal[0]/(frm[0]-frm[8]); | |||
| for(n=1;n<8;n++) { | |||
| slv[n]=(cal[n]-cal[n-1])/(frm[n]-frm[n-1]); | |||
| if(n==mshift) { | |||
| /* channel ID */ | |||
| int i; | |||
| float cv,df; | |||
| for(i=1,cv=0;i<7;i++) | |||
| cv+=tele[n+15*8+i]; | |||
| cv/=6; | |||
| for(i=0,max=10000.0,channel=-1;i<6;i++) { | |||
| df=cv-frm[i]; df=df*df; | |||
| if (df<max) { channel=i; max=df; } | |||
| } | |||
| } | |||
| } | |||
| mk=k; | |||
| for(n=0;n<nrow;n++) { | |||
| float *pixelv; | |||
| int i,c; | |||
| pixelv=prow[n]; | |||
| for(i=0;i<909;i++) { | |||
| for(i=0;i<954;i++) { | |||
| float pv; | |||
| int k,kof; | |||
| pv=pixelv[i+offset]; | |||
| k=(n-shift)/128; | |||
| kof=(n-shift)%128; | |||
| if(kof<64) { | |||
| if(k<1) | |||
| pv=rgcal(pv,&(regr[k])); | |||
| else | |||
| pv=rgcal(pv,&(regr[k]))*(64+kof)/128.0+ | |||
| rgcal(pv,&(regr[k-1]))*(64-kof)/128.0; | |||
| } else { | |||
| if((k+1)>=mk) | |||
| pv=rgcal(pv,&(regr[k])); | |||
| else | |||
| pv=rgcal(pv,&(regr[k]))*(192-kof)/128.0+ | |||
| rgcal(pv,&(regr[k+1]))*(kof-64)/128.0; | |||
| } | |||
| if(pv<frm[8]) {pv=0.0; goto conv; } | |||
| if(pv<frm[0]) {pv=(pv-frm[8])*slv[0]; goto conv; } | |||
| for(c=1;c<8;c++) | |||
| if(pv<frm[c]) {pv=(pv-frm[c-1])*slv[c]+cal[c-1]; goto conv; } | |||
| pv=(pv-frm[7])*slv[7]+cal[7]; | |||
| if(pv>1.0) pv=1.0; | |||
| conv: | |||
| if(pv<0.0) pv=0.0; | |||
| pixelv[i+offset]=pv; | |||
| } | |||
| } | |||
+ 63
- 13
main.c
View File
| @@ -163,9 +163,9 @@ for(n=0;n<nrow;n++) { | |||
| for(i=0;i<909;i++) { | |||
| int x,y; | |||
| x=(int)(pixelv[i+85]*255.0); | |||
| y=(int)(pixelv[i+1125]*255.0); | |||
| pixel[i]=cmap[x][y]; | |||
| y=(int)(pixelv[i+85]*255.0); | |||
| x=(int)(pixelv[i+1125]*255.0); | |||
| pixel[i]=cmap[y][x]; | |||
| } | |||
| png_write_row(png_ptr,(png_bytep)pixel); | |||
| } | |||
| @@ -232,6 +232,47 @@ png_destroy_read_struct(&png_ptr,&info_ptr,NULL); | |||
| return(0); | |||
| } | |||
| #ifdef DEBUG | |||
| unsigned int distrib[256][256]; | |||
| int Distrib(char *filename,float **prow,int nrow) | |||
| { | |||
| int n; | |||
| int x,y; | |||
| int max=0; | |||
| FILE *df; | |||
| for(y=0;y<256;y++) | |||
| for(x=0;x<256;x++) | |||
| distrib[y][x]=0; | |||
| for(n=0;n<nrow;n++) { | |||
| float *pixelv; | |||
| png_color pixel[909]; | |||
| int i; | |||
| pixelv=prow[n]; | |||
| for(i=0;i<909;i++) { | |||
| y=(int)(pixelv[i+85]*255.0); | |||
| x=(int)(pixelv[i+1125]*255.0); | |||
| if(x>255) x=255; | |||
| if(x<0) x=0; | |||
| if(y>255) y=255; | |||
| if(y<0) y=0; | |||
| distrib[y][x]+=1; | |||
| if(distrib[y][x]> max) max=distrib[y][x]; | |||
| } | |||
| } | |||
| df=fopen(filename,"w"); | |||
| fprintf(df,"P2\n#max %d\n",max); | |||
| fprintf(df,"256 256\n255\n"); | |||
| for(y=0;y<256;y++) | |||
| for(x=0;x<256;x++) | |||
| fprintf(df,"%d\n",(int)((255.0*(double)(distrib[y][x]))/(double)max)); | |||
| fclose(df); | |||
| } | |||
| #endif | |||
| extern int Calibrate(float **prow,int nrow,int offset); | |||
| extern int optind,opterr; | |||
| extern char *optarg; | |||
| @@ -299,25 +340,34 @@ ImageOut(pngfilename,prow,nrow,2080,0); | |||
| } | |||
| /* Channel A */ | |||
| ch=Calibrate(prow,nrow,85); | |||
| if(ch>0) { | |||
| a=1; | |||
| if(((strchr(imgopt,(int)'a')!=NULL) || (strchr(imgopt,ch)!=NULL))) { | |||
| if(((strchr(imgopt,(int)'a')!=NULL) || (strchr(imgopt,(int)'c')!=NULL))) { | |||
| ch=Calibrate(prow,nrow,85); | |||
| if(ch>0) { | |||
| a=1; | |||
| if(strchr(imgopt,(int)'a')!=NULL) { | |||
| sprintf(pngfilename,"%s/%s-%s.png",pngdirname,name,chid[ch]); | |||
| ImageOut(pngfilename,prow,nrow,909,85); | |||
| ImageOut(pngfilename,prow,nrow,954,85); | |||
| } | |||
| } | |||
| } | |||
| /* Channel B */ | |||
| ch=Calibrate(prow,nrow,1125); | |||
| if(ch>0) { | |||
| b=1; | |||
| if(((strchr(imgopt,(int)'b')!=NULL) || (strchr(imgopt,ch)!=NULL))) { | |||
| if(((strchr(imgopt,(int)'b')!=NULL) || (strchr(imgopt,(int)'c')!=NULL))) { | |||
| ch=Calibrate(prow,nrow,1125); | |||
| if(ch>0) { | |||
| b=1; | |||
| if(strchr(imgopt,(int)'b')!=NULL) { | |||
| sprintf(pngfilename,"%s/%s-%s.png",pngdirname,name,chid[ch]); | |||
| ImageOut(pngfilename,prow,nrow,909,1125); | |||
| ImageOut(pngfilename,prow,nrow,954,1125); | |||
| } | |||
| } | |||
| } | |||
| #ifdef DEBUG | |||
| sprintf(pngfilename,"%s/%s-d.pnm",pngdirname,name); | |||
| Distrib(pngfilename,prow,nrow); | |||
| #endif | |||
| /* color image */ | |||
| if(a && b && strchr(imgopt,(int)'c')!=NULL){ | |||
| sprintf(pngfilename,"%s/%s-c.png",pngdirname,name); | |||