#include <Proximity.h>
#include <FileReader.h>
#include <TestImages.h>
#include <Canny.h>
#include <Convolution.h>
using namespace std;
using namespace icl;

int main(int nArgs, char **ppcArg){
   const ImgBase *src;
   ImgBase *dst=0,*dst2=0,*dst3=0,*dst4=0;
   string srcName("src.ppm");
   string dstName("wiener.ppm");
   if (nArgs > 2) dstName = ppcArg[2];
   if (nArgs > 1) {
      // read image from file
      FileReader reader(ppcArg[1]);
      src = reader.grab();
   } else src = TestImages::create("women",Size(63,60),formatGray,depth32f);
   
   
   // test 1
   Convolution* pConv = new Convolution(Convolution::kernelSobelX3x3);
   pConv->setClipToROI (true);
   pConv->apply (src, &dst2);
   
   pConv = new Convolution(Convolution::kernelSobelY3x3);
   pConv->setClipToROI (true);
   pConv->apply (src, &dst3);

   Canny* pCanny = new Canny(src->asImg<icl32f>());
   
   pCanny->apply (dst3, dst2,&dst,20,40);
   // test2
   pCanny->apply (src,&dst4,20,40);
   
   // write and display the image
   src->print("src");
   dst->print("dst");
   dst2->print("dst2");
   dst3->print("dst3");
   dst4->print("dst4");
   
   
   TestImages::xv (src, string("src.pgm"));
   TestImages::xv (dst, string("dst.pgm"));
   TestImages::xv (dst2, string("sobelX.pgm"));
   TestImages::xv (dst3, string("sobelY.pgm"));
   TestImages::xv (dst4, string("dst4.pgm"));

   return 0;
}