Changeset 7790 for trunk/MagicSoft/Cosy/videodev/FilterLed.cc

Timestamp:

07/14/06 13:19:54 (18 years ago)

Author:

tbretz

Message:

*** empty log message ***

File:

• trunk/MagicSoft/Cosy/videodev/FilterLed.cc (modified) (12 diffs)

trunk/MagicSoft/Cosy/videodev/FilterLed.cc

@@ -45 +45 @@
 }
 
+void FilterLed::DrawHexagon(float cx, float cy, float r, byte col) const
+{
+    MGMap::DrawHexagon(fImg, 768, 576, cx, cy, r, col);
+}
+
 void FilterLed::DrawCircle(const Ring &l, byte col) const
 {
@@ -53 +58 @@
 {
     DrawCircle(l.GetX(), l.GetY(), r, col);
+}
+
+void FilterLed::DrawHexagon(const Ring &l, double r, byte col) const
+{
+    DrawHexagon(l.GetX(), l.GetY(), r, col);
 }
 
@@ -121 +131 @@
 }
 
-int FilterLed::GetMeanPositionCircle(const int x, const int y,
-                                     const int box, float &mx, 
-                                     float &my, unsigned int &sum) const
+int FilterLed::GetMeanPositionBox(const int x, const int y,
+                                  const int box, float &mx,
+                                  float &my, unsigned int &sum) const
 {
     //-------------------------------
@@ -188 +198 @@
 }
 
-int FilterLed::GetMeanPositionCircle(const int x, const int y, 
-                                     const int box) const
+int FilterLed::GetMeanPositionBox(const int x, const int y,
+                                  const int box) const
 {
     float mx, my;
     unsigned int sum;
-    return GetMeanPositionCircle(x, y, box, mx, my, sum);
+    return GetMeanPositionBox(x, y, box, mx, my, sum);
 }
 
@@ -334 +344 @@
 }
 
+class ClusterFinder
+{
+private:
+    byte *fImg;
+
+    UInt_t fW;
+    UInt_t fH;
+
+    Int_t fX0;
+    Int_t fX1;
+
+    Int_t fY0;
+    Int_t fY1;
+
+    UInt_t fCount;
+    Float_t fSumX;
+    Float_t fSumY;
+
+    Float_t FindCluster(Int_t x, Int_t y)
+    {
+        // if edge is touched stop finding cluster
+        if (x<fX0 || x>=fX1 || y<fY0 || y>=fY1)
+            return -1;
+
+        if (fCount>999)
+            return -1;
+
+        // get the value
+        Float_t val = fImg[y*fW+x];
+
+        // if its empty we have found the border of the cluster
+        if (val==0)
+            return 0;
+
+        // mark the point as processed
+        fImg[y*fW+x] = 0;
+
+        fSumX += x*val; // sumx
+        fSumY += y*val; // sumy
+        fCount++;
+
+        Float_t rc[4];
+        rc[0] = FindCluster(x+1, y  );
+        rc[1] = FindCluster(x,   y+1);
+        rc[2] = FindCluster(x-1, y  );
+        rc[3] = FindCluster(x,   y-1);
+
+        for (int i=0; i<4; i++)
+        {
+            if (rc[i]<0) // check if edge is touched
+                return -1;
+
+            val += rc[i];
+        }
+
+        return val;
+    }
+public:
+    ClusterFinder(byte *img, UInt_t w, UInt_t h)
+    {
+        fW = w;
+        fH = h;
+
+        fImg = new byte[fW*fH];
+
+        memcpy(fImg, img, fW*fH);
+    }
+
+    ~ClusterFinder()
+    {
+        delete fImg;
+    }
+    void FindCluster(Leds &leds, Int_t x0=0, Int_t y0=0, Int_t x1=0, Int_t y1=0)
+    {
+        fX0 = x0;
+        fY0 = y0;
+        fX1 = x1==0?fW:x1;
+        fY1 = y1==0?fH:y1;
+
+        for (Int_t x=fX0; x<fX1; x++)
+            for (Int_t y=fY0; y<fY1; y++)
+            {
+                const byte &b = fImg[y*fW+x];
+                if (b==0)
+                    continue;
+
+                fCount = 0;
+                fSumX  = 0;
+                fSumY  = 0;
+
+                const Float_t mag = FindCluster(x, y);
+                if (fCount>999)
+                {
+                    cout << "ERROR - Spot with Size>999 detected..." << endl;
+                    return;
+                }
+
+                if (mag>0 && fCount>6)
+                {
+                    Float_t M = mag/0xff;
+                    if (M>0xf0)
+                        M=0xf0;
+
+                    leds.Add(fSumX/mag, fSumY/mag, 0, 0, 0xf0);
+                }
+            }
+        leds.Compress();
+    }
+};
 
 void FilterLed::Execute(Leds &leds, int xc, int yc, double &bright) const
@@ -353 +472 @@
             byte &b = fImg[y*fW+x];
 
+            // Skip saturating pixels
+            if (b>0xf0)
+                continue;
+
+            sum += b;
+            sq  += b*b;
+        }
+
+    sum /= wx*hy;
+    sq  /= wx*hy;
+
+    bright=sum;
+
+    
+    // 254 because b<=max and not b<max
+    const double sdev = TMath::Sqrt(sq-sum*sum);
+    const byte   max  = sum+fCut*sdev>254 ? 254 : (byte)(sum+fCut*sdev);
+
+    //
+    // clean image from noise
+    // (FIXME: A lookup table could accelerate things...
+    //
+    for (int x=x0; x<x1; x++)
+        for (int y=y0; y<y1; y++)
+        {
+            byte &b = fImg[y*fW+x];
+            if (b<=max)
+                b = 0;
+        }
+
+    ClusterFinder find(fImg, fW, fH);
+    find.FindCluster(leds, x0, y0, x1, y1);
+}
+
+/*
+void FilterLed::Execute(Leds &leds, int xc, int yc, double &bright) const
+{
+    const int x0 = TMath::Max(xc-fBox, 0);
+    const int y0 = TMath::Max(yc-fBox, 0);
+    const int x1 = TMath::Min(xc+fBox, fW);
+    const int y1 = TMath::Min(yc+fBox, fH);
+
+    const int wx = x1-x0;
+    const int hy = y1-y0;
+
+    double sum = 0;
+    double sq  = 0;
+
+    for (int x=x0; x<x1; x++)
+        for (int y=y0; y<y1; y++)
+        {
+            byte &b = fImg[y*fW+x];
+
             sum += b;
             sq  += b*b;
@@ -387 +559 @@
     byte mag[maxpnt]; // (Use 'new' instead for big numbers!)
 
+    const int r = 5;
+
     int cnt = 0;
-    for (int x=x0; x<x1; x++)
-        for (int y=y0; y<y1; y++)
+    for (int x=x0+r; x<x1-r; x++)
+        for (int y=y0+r; y<y1-r; y++)
         {
             byte &b = fImg[y*fW+x];
@@ -395 +569 @@
                 continue;
 
-            const int ipos = GetMeanPosition(x, y, 5);
+            const int ipos = GetMeanPosition(x, y, r);
 
             int j;
@@ -402 +576 @@
                 if (pos[j]==ipos)
                 {
-                    if (mag[j] < 0xf0)
-                        mag[j] += 0x10;
+                    if (mag[j]<0xff)
+                        mag[j]++;  // how often found (area)
                     break;
                 }
@@ -411 +585 @@
 
             pos[cnt] = ipos;
-            mag[cnt] = 0x10;
+            mag[cnt] = 1;
 
             cnt++;
@@ -428 +602 @@
     for (int i=0; i<cnt; i++)
     {
-        if (mag[i]<=0x80) // 0xa0
+        if (mag[i]<=7)
             continue;
 
         Float_t mx, my;
         unsigned int sum;
-        GetMeanPosition(pos[i]%fW, pos[i]/fW, 5, mx, my, sum);
-
-        leds.Add(mx, my, 0, 0, mag[i]);
-    }
-
-    RemoveTwinsInterpol(leds, first, 5);
-}
-
-void FilterLed::FindStar(Leds &leds, int xc, int yc, bool circle) const
+        GetMeanPosition(pos[i]%fW, pos[i]/fW, r, mx, my, sum);
+
+        leds.Add(mx, my, 0, 0, 0);//mag[i]*10);
+    }
+
+    RemoveTwinsInterpol(leds, first, r);
+}
+*/
+void FilterLed::FindStar(Leds &leds, int xc, int yc, bool box) const
 {
     // fBox: radius of the inner (signal) box
@@ -535 +709 @@
     float mx, my;
     unsigned int mag;
-    int pos = circle ? GetMeanPositionCircle(xc, yc, fBox-1, mx, my, mag) : GetMeanPosition(xc, yc, fBox-1, mx, my, mag);
+    int pos = box ? GetMeanPositionBox(xc, yc, fBox-1, mx, my, mag) : GetMeanPosition(xc, yc, fBox-1, mx, my, mag);
 
     if (pos<0 || pos>=fW*fH || fImg[pos]<sum+fCut*sdev)