Demo: input points

Code/Seg/IPOLdemo/Input/points.txt

+121 149 140 114
+335 107 307 83
+-1

Code/Seg/IPOLdemo/mainIPOL.cpp

@@ -27,15 +27,25 @@ int main (int argc, char *argv[])
 
   
   // getting parameters:
-  // - input file
+  // - input image file
   string input_filename = argv[1];
+  // - input points file
+  string inpoints_filename = "";
+  ifstream fin;  
+  bool with_points = false;
+  if (argc == 4)
+  {
+    inpoints_filename = argv[3];
+    fin.open (inpoints_filename.c_str(), std::fstream::in);
+    with_points = true;
+  }
   // - output file
   string output_filename = argv[2];
   ofstream fout;  
-  fout.open(output_filename.c_str(), std::fstream::out);
+  fout.open (output_filename.c_str(), std::fstream::out);
 
   
-// Calcul de la carte de gradient (utilise qt)
+  // Gradient map extraction (uses qt)
   QImage image;
   
   image.load (QString (input_filename.c_str()));
@@ -53,16 +63,65 @@ int main (int argc, char *argv[])
   }
   VMap *gMap = new VMap (width, height, tabImage, VMap::TYPE_SOBEL_5X5);
 
-  // Detection
+  // Input points reading (uses qt)
+  vector<Pt2i> pts;
+  if (with_points)
+  {
+    int val[4], i = 0;
+    bool reading = true;
+    if (fin)
+    {
+      while (reading)
+      {
+        fin >> val[i++];
+        if (val[i-1] == -1) reading = false;
+        // if (fin.eof ()) reading = false;
+        if (reading && i == 4)
+        {
+          pts.push_back (Pt2i (val[0], val[1]));
+          pts.push_back (Pt2i (val[2], val[3]));
+          i = 0;
+        }
+      }
+    }
+    fin.close();
+  }
+
+  // Blurred segment detection
+  vector<BlurredSegment *> bss;
   BSDetector detector;
   detector.setGradientMap (gMap);
-  detector.setMaxTrials (-1);
-  detector.detectAll ();
+  if (with_points)
+  {
+    vector<Pt2i>::iterator pit = pts.begin ();
+    int nbseg = 0;
+    while (pit != pts.end ())
+    {
+      Pt2i pt1 = *pit++;
+      Pt2i pt2 = *pit++;
+      std::cout << nbseg << " detect (" << pt1.x () << ", " << pt1.y ()
+                << ") (" << pt2.x () << ", " << pt2.y () << ")" << std::endl;
+      detector.detect (pt1, pt2);
+      BlurredSegment *mybs = detector.getBlurredSegment ();
+      if (mybs != NULL)
+      {
+        detector.preserveFormerBlurredSegment ();
+        bss.push_back (mybs);
+        nbseg ++;
+      }
+    }
+    std::cout << nbseg << " detections" << std::endl;
+  }
+  else
+  {
+    detector.setMaxTrials (-1);
+    detector.detectAll ();
+    bss = detector.getBlurredSegments ();
+  }
 
   fout << "# Line detection generated from " << argv[0] << "with format : X1 Y1 X2 Y2 on each line" << std::endl;
   
-  // Affichage
-  vector<BlurredSegment *> bss = detector.getBlurredSegments ();
+  // Display
   if (! bss.empty ())
   {
     vector<BlurredSegment *>::const_iterator it = bss.begin ();