home *** CD-ROM | disk | FTP | other *** search
/ Club Amiga de Montreal - CAM / CAM_CD_1.iso / files / 625.lha / STScan_v2.0 / vectorop.c < prev   
C/C++ Source or Header  |  1992-02-22  |  9KB  |  337 lines
  1. /* verctorop.c vectorization routine for stscan.c */
  2. #include <exec/exec.h>
  3. #include <exec/types.h>
  4. #include <intuition/intuition.h>
  5. #include <intuition/intuitionbase.h>
  6. #include <libraries/dos.h>
  7. #include <libraries/dosextens.h>
  8. #include <graphics/rastport.h>
  9. #include <libraries/reqbase.h>
  10. #include <req_pragmas.h>
  11. #include <stdio.h>
  12. #include <string.h>
  13. #include <functions.h>
  14. #include "stscan.h"
  15.  
  16. #define REGION_SIZE 64
  17. #define CHAINLEN 16384
  18.  
  19. static UWORD chpixels[2][CHAINLEN];
  20. static UWORD lnpixels[2][CHAINLEN];
  21. static UWORD regio_x[128][128];
  22. static UWORD regio_y[128][128];
  23. static UBYTE regio_done[128][128];
  24. static UBYTE regio_numx,regio_numy;
  25. static char str_filenam[REQ_FCHARS+REQ_DSIZE]="";
  26. static UWORD maxdiff=4;
  27.  
  28. static UBYTE getpixel(UWORD x,UWORD y)
  29. { 
  30.   return((*(memptr+(ULONG)y*(ULONG)membpl+(ULONG)(x>>3))&bitval[x&7])?0:1);
  31. }
  32.  
  33. static void clrpixel(UWORD x,UWORD y)
  34. { *(memptr+(ULONG)y*(ULONG)membpl+(ULONG)(x>>3))|=bitval[x&7];
  35. }
  36.  
  37. static void setpixel(UWORD x,UWORD y)
  38. { *(memptr+(ULONG)y*(ULONG)membpl+(ULONG)(x>>3))&=invbitval[x&7];
  39. }
  40.  
  41. static UBYTE subiter(UWORD nx,UWORD ny)
  42. { UWORD x,y;
  43.   UBYTE f,vw,vnw,vn,vne,ve,vse,vs,vsw;
  44.  
  45.   f=0;
  46.   for (y=regio_y[nx][ny];y<regio_y[nx][ny+1];y++)
  47.   { for (x=regio_x[nx][ny];x<regio_x[nx+1][ny];x++)
  48.     { if ((getpixel(x,y))&&((x&1)==(y&1)))
  49.       { vw= getpixel(x-1,y);
  50.         vnw=getpixel(x-1,y-1);
  51.         vn= getpixel(x  ,y-1);
  52.         vne=getpixel(x+1,y-1);
  53.         ve= getpixel(x+1,y);
  54.         vse=getpixel(x+1,y+1);
  55.         vs= getpixel(x  ,y+1);
  56.         vsw=getpixel(x-1,y+1);
  57.         if ( (( ((vw^1)&(vnw|vn))
  58.                +((vn^1)&(vne|ve))
  59.                +((ve^1)&(vse|vs))
  60.                +((vs^1)&(vsw|vw)) )==1)
  61.             || ((!(vw))&&(!(vnw))&&(!(vn))&&(!(vne))&&
  62.                 (!(ve))&&(!(vse))&&(!(vs))&&(!(vsw)) ) )
  63.         { clrpixel(x,y);
  64.           if (!(f)) f=((vnw|vse)&(vne|vsw)&(vn|vs)&(vw|ve));
  65.         }
  66.       }
  67.     }
  68.   }
  69.   for (y=regio_y[nx][ny];y<regio_y[nx][ny+1];y++)
  70.   { for (x=regio_x[nx][ny];x<regio_x[nx+1][ny];x++)
  71.     { if ((getpixel(x,y))&&((x&1)!=(y&1)))
  72.       { vw= getpixel(x-1,y);
  73.         vnw=getpixel(x-1,y-1);
  74.         vn= getpixel(x  ,y-1);
  75.         vne=getpixel(x+1,y-1);
  76.         ve= getpixel(x+1,y);
  77.         vse=getpixel(x+1,y+1);
  78.         vs= getpixel(x  ,y+1);
  79.         vsw=getpixel(x-1,y+1);
  80.         if ( (( ((vw^1)&(vnw|vn))
  81.                +((vn^1)&(vne|ve))
  82.                +((ve^1)&(vse|vs))
  83.                +((vs^1)&(vsw|vw)) )==1)
  84.             || ((!(vw))&&(!(vnw))&&(!(vn))&&(!(vne))&&
  85.                 (!(ve))&&(!(vse))&&(!(vs))&&(!(vsw)) ) )
  86.         { clrpixel(x,y);
  87.           if (!(f)) f=((vnw|vse)&(vne|vsw)&(vn|vs)&(vw|ve));
  88.         }
  89.       }
  90.     }
  91.   }
  92.   return(f);
  93. }
  94.  
  95. static UBYTE chkblack(UWORD nx,UWORD ny)
  96. { UWORD x,y;
  97.   UBYTE f=1;
  98.   for (y=regio_y[nx][ny];y<regio_y[nx][ny+1];y++)
  99.   { for (x=regio_x[nx][ny];x<regio_x[nx+1][ny];x++)
  100.     if (!(getpixel(x,y))) 
  101.     { f=0;
  102.       y=regio_y[nx][ny+1];
  103.       x=regio_x[nx+1][ny];
  104.     }
  105.   }
  106.   return(f);
  107. }
  108.  
  109. static void straighten()
  110. { UWORD x,y;
  111.   UBYTE f,vnw,vne,vse,vsw;
  112.  
  113.   for (y=1;y<memheight-1;y++)
  114.   { for (x=1;x<memwidth-1;x++)
  115.     { if ((getpixel(x,y))
  116.           &&(!(getpixel(x-1,y)))&&(!(getpixel(x+1,y)))
  117.           &&(!(getpixel(x,y-1)))&&(!(getpixel(x,y+1))))
  118.       { vnw=getpixel(x-1,y-1);
  119.         vne=getpixel(x+1,y-1);
  120.         vse=getpixel(x+1,y+1);
  121.         vsw=getpixel(x-1,y+1);
  122.         if (((vnw))&&(!(vne))&&(!(vse))&&((vsw))) 
  123.         { clrpixel(x,y);
  124.           setpixel(x-1,y);
  125.         }
  126.         else if ((!(vnw))&&((vne))&&((vse))&&(!(vsw))) 
  127.         { clrpixel(x,y);
  128.           setpixel(x+1,y);
  129.         }
  130.         else if (((vnw))&&((vne))&&(!(vse))&&(!(vsw))) 
  131.         { clrpixel(x,y);
  132.           setpixel(x,y-1);
  133.         }
  134.         else if ((!(vnw))&&(!(vne))&&((vse))&&((vsw))) 
  135.         { clrpixel(x,y);
  136.           setpixel(x,y+1);
  137.         }
  138.       }
  139.     }
  140.   }
  141. }
  142.  
  143. void thin()
  144. { UWORD x,y,i;
  145.   ULONG pgofs1,pgofs2;
  146.   UBYTE p,f,ende;
  147.   char wtitel[80];
  148.   
  149.   membpl=(memwidth>>3);
  150.   pgofs1=(memwidth>>3)*(memheight-1);
  151.   for (i=0;i<(memwidth>>3);i++) 
  152.   { *(memptr+i)=255;
  153.     *(memptr+pgofs1+i)=255;
  154.   }
  155.   pgofs1=0;
  156.   pgofs2=membpl-1;
  157.   for (i=0;i<memheight;i++)
  158.   { *(memptr+pgofs1)|=128;
  159.     *(memptr+pgofs2)|=1;
  160.     pgofs1+=membpl;
  161.     pgofs2+=membpl;
  162.   }
  163.   p=1;
  164.   regio_numx=memwidth/REGION_SIZE;
  165.   regio_numy=memheight/REGION_SIZE;
  166.   regio_x[0][0]=1;
  167.   regio_x[1][0]=REGION_SIZE+(memwidth%REGION_SIZE)/2;
  168.   for (x=2;x<regio_numx;x++) regio_x[x][0]=REGION_SIZE+regio_x[x-1][0];
  169.   regio_x[regio_numx][0]=memwidth-1;
  170.   for (x=0;x<regio_numx;x++) regio_y[x][0]=1;
  171.   for (x=0;x<=regio_numx;x++)
  172.   { regio_y[x][1]=REGION_SIZE+(memheight%REGION_SIZE)/2;
  173.     regio_x[x][1]=regio_x[x][0];
  174.   }
  175.   for (y=2;y<regio_numy;y++) for (x=0;x<=regio_numx;x++)
  176.   { regio_y[x][y]=REGION_SIZE+regio_y[x][y-1];
  177.     regio_x[x][y]=regio_x[x][0];
  178.   }
  179.   for (x=0;x<regio_numx;x++) regio_y[x][regio_numy]=memheight-1;
  180.   for (y=0;y<regio_numy;y++) for (x=0;x<regio_numx;x++) regio_done[x][y]=1;
  181.   do
  182.   { ende=0;
  183.     for (y=0;y<regio_numy;y++) for (x=0;x<regio_numx;x++)
  184.     { if (regio_done[x][y])
  185.       { if (subiter(x,y)) f=regio_done[x][y]=1;
  186.         else if (chkblack(x,y)) f=regio_done[x][y]=1;
  187.         else f=regio_done[x][y]=0;
  188.         ende|=f;
  189.         sprintf(&wtitel,"Region (%02d;%02d) Pass %02d",x,y,p);
  190.         SetWindowTitles(win,&wtitel[0],(UBYTE *)-1);
  191.       }
  192.     }
  193.     p++;
  194.   } while (ende);
  195.   SetWindowTitles(win,"Cleanup",(UBYTE *)-1);
  196.   straighten(); 
  197.   SetWindowTitles(win," ",(UBYTE *)-1);
  198. }
  199.  
  200. static UWORD getchain(UWORD x, UWORD y)
  201. { UWORD lx,ly,pos;
  202.   UBYTE f=1;
  203.  
  204.   lx=x;
  205.   ly=y;
  206.   pos=0;
  207.   while ((f)&&(pos<CHAINLEN))
  208.   { chpixels[0][pos]=lx;
  209.     chpixels[1][pos]=ly;
  210.     clrpixel(lx,ly);
  211.     pos++;
  212.          if (getpixel(lx-1,ly  )) lx--;
  213.     else if (getpixel(lx-1,ly-1)) {lx--; ly--;}
  214.     else if (getpixel(lx  ,ly-1)) ly--;
  215.     else if (getpixel(lx+1,ly-1)) {lx++; ly--;}
  216.     else if (getpixel(lx+1,ly  )) lx++;
  217.     else if (getpixel(lx+1,ly+1)) {lx++; ly++;}
  218.     else if (getpixel(lx  ,ly+1)) ly++;
  219.     else if (getpixel(lx-1,ly+1)) {lx--; ly++;}
  220.     else f=0;
  221.   }
  222.   return(pos);
  223. }
  224.  
  225. static UWORD getline(UWORD x0, UWORD y0, UWORD x1, UWORD y1)
  226. { WORD xdiff,ydiff,xstep,ystep, sum;
  227.   UWORD i;
  228.   WORD step[2]={-1,1};
  229.  
  230.   xstep=step[(x0<x1)]; ystep=step[(y0<y1)];
  231.   xdiff=abs(x0-x1); ydiff=abs(y0-y1);
  232.  
  233.   lnpixels[0][0]=x0;
  234.   lnpixels[1][0]=y0;
  235.   i=1;
  236.   if (xdiff>ydiff)
  237.   { sum=xdiff/2;
  238.     while (x0!=x1)
  239.     { x0+=xstep;
  240.       sum-=ydiff;
  241.       if (sum<0)
  242.       { y0+=ystep;
  243.         sum+=xdiff;
  244.       }
  245.       lnpixels[0][i]=x0;
  246.       lnpixels[1][i]=y0;
  247.       i++;
  248.     }
  249.   }
  250.   else
  251.   { sum=ydiff/2;
  252.     while (y0!=y1)
  253.     { y0+=ystep;
  254.       sum-=xdiff;
  255.       if (sum<0)
  256.       { x0+=xstep;
  257.         sum+=ydiff;
  258.       }
  259.       lnpixels[0][i]=x0;
  260.       lnpixels[1][i]=y0;
  261.       i++;
  262.     }
  263.   }
  264.   return(i);
  265. }
  266.  
  267. static void polygon(UWORD startpos,UWORD len,FILE *vecfile)
  268. { UWORD sx0,sy0,sx1,sy1,m,n,s,lnlen,cbdist,mcbdist,mdpos;
  269.   
  270.   sx0=chpixels[0][startpos];
  271.   sy0=chpixels[1][startpos];
  272.   sx1=chpixels[0][startpos+len-1];
  273.   sy1=chpixels[1][startpos+len-1];
  274.   lnlen=getline(sx0,sy0,sx1,sy1);
  275.   mdpos=0;
  276.   mcbdist=0;
  277.   s=len/lnlen;
  278.   if (s<1) s=1;
  279.   for(m=0,n=startpos;m<lnlen;m++,n+=s)
  280.   { cbdist=abs(chpixels[0][n]-lnpixels[0][m])
  281.           +abs(chpixels[1][n]-lnpixels[1][m]);
  282.     if (cbdist>mcbdist)
  283.     { mdpos=m;
  284.       mcbdist=cbdist;
  285.     }
  286.   }
  287.   if (mcbdist>maxdiff)
  288.   { polygon(startpos,mdpos,vecfile);
  289.     polygon(startpos+mdpos,len-mdpos,vecfile);
  290.   }
  291.   else
  292.   { fprintf(vecfile,"%d %d\n",sx0,memheight-sy0);
  293.     fprintf(vecfile,"%d %d\n\n",sx1,memheight-sy1);
  294.   }
  295. }
  296.  
  297. void vectorize()
  298. { UWORD p,x,y,len;
  299.   FILE *vecfile;
  300.   char wtitel[80];
  301.   
  302.   if (filerequest("Save Vector TXT File",str_filenam))
  303.   { p=0;
  304.     membpl=(memwidth>>3);
  305.     vecfile=fopen(str_filenam,"w");
  306.     for (y=1;y<memheight-1;y++)
  307.     { for (x=1;x<memwidth-1;x++)
  308.       { if (getpixel(x,y))
  309.         { len=getchain(x,y);
  310.           polygon(0,len,vecfile);
  311.           sprintf(&wtitel,"Chain %04d",++p);
  312.           SetWindowTitles(win,&wtitel[0],(UBYTE *)-1);
  313.         }
  314.       }
  315.     }
  316.     fclose(vecfile);
  317.     SetWindowTitles(win," ",(UBYTE *)-1);
  318.   }
  319. }
  320.  
  321. void chngmaxdiff()
  322. { struct Process *OurTask;
  323.   struct Window    *old_pr_WindowPtr;
  324.   struct GetLongStruct    gnum;
  325.  
  326.   OurTask = (struct Process *)FindTask((char *)0L);
  327.   old_pr_WindowPtr = OurTask->pr_WindowPtr;
  328.   OurTask->pr_WindowPtr = win;
  329.   memset(&gnum,0,sizeof(gnum));
  330.   gnum.titlebar = "Enter vectorization accuracy.";
  331.   gnum.defaultval = maxdiff;
  332.   gnum.minlimit = 2;
  333.   gnum.maxlimit = 100;
  334.   if (GetLong(&gnum)) maxdiff=gnum.result;
  335.   OurTask->pr_WindowPtr = old_pr_WindowPtr;
  336. }
  337.