X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=Parasitemia%2FParasitemiaCore%2FEllipse.fs;h=e95cf2780c988461d8223e284a53fa7922892a18;hb=85adb74195fe7145535d3d36263aec2f7879cd60;hp=db23ece8976d2e82a53c42533f26d787a74936a5;hpb=04d4504e7b248a82ddfc1a41d325e59d24146590;p=master-thesis.git

diff --git a/Parasitemia/ParasitemiaCore/Ellipse.fs b/Parasitemia/ParasitemiaCore/Ellipse.fs
index db23ece..e95cf27 100644
--- a/Parasitemia/ParasitemiaCore/Ellipse.fs
+++ b/Parasitemia/ParasitemiaCore/Ellipse.fs
@@ -77,19 +77,18 @@ let ellipse (p1x: float) (p1y: float) (m1: float) (p2x: float) (p2y: float) (m2:
 
             Some (Types.Ellipse(float32 cx, float32 cy, float32 majorAxis, float32 minorAxis, float32 phi'))
 
-
-let private vectorRotation (p1x: float32) (p1y: float32) (v1x: float32) (v1y: float32) (px: float32) (py: float32) : float32 =
-    if p1y > py
+let inline private vectorRotation (px: float32) (py: float32) (vx: float32) (vy: float32) (p0x: float32) (p0y: float32) : float32 =
+    if py > p0y
     then
-        if v1x > 0.f then -1.f else 1.f
-    elif p1y < py
+        if vx > 0.f then -1.f else 1.f
+    elif py < p0y
     then
-        if v1x < 0.f then -1.f else 1.f
-    elif p1x > px
+        if vx < 0.f then -1.f else 1.f
+    elif px > p0x
     then
-        if v1y < 0.f then -1.f else 1.f
+        if vy < 0.f then -1.f else 1.f
     else // p1x < px
-        if v1y > 0.f then -1.f else 1.f
+        if vy > 0.f then -1.f else 1.f
 
 let private areVectorsValid (p1x: float32) (p1y: float32) (p2x: float32) (p2y: float32) (v1x: float32) (v1y: float32) (v2x: float32) (v2y: float32) : (float32 * float32) option =
     let m1 = -v1x / v1y
@@ -97,23 +96,25 @@ let private areVectorsValid (p1x: float32) (p1y: float32) (p2x: float32) (p2y: f
 
     let b1 = -m1 * p1x + p1y
     let b2 = -m2 * p2x + p2y
-    let px = -((b1 - b2) / (m1 - m2))
-    let py = -((m2 * b1 - m1 * b2) / (m1 - m2))
+    let p0x = -(b1 - b2) / (m1 - m2)
+    let p0y = -(m2 * b1 - m1 * b2) / (m1 - m2)
 
-    let rot1 = vectorRotation p1x p1y v1x v1y px py
-    let rot2 = vectorRotation p2x p2y v2x v2y px py
+    let rot1 = vectorRotation p1x p1y v1x v1y p0x p0y
+    let rot2 = vectorRotation p2x p2y v2x v2y p0x p0y
 
     if rot1 = rot2
     then
         None
     else
-        let alpha1 = atan2 (p1y - py) (p1x - px)
-        let alpha2 = atan2 (p2y - py) (p2x - px)
+        let alpha1 =
+            let alpha1' = atan2 (p1y - p0y) (p1x - p0x)
+            if alpha1' < 0.f then 2.f * PI + alpha1' else alpha1'
 
-        let alpha1' = if alpha1 < 0.f then 2.f * PI + alpha1 else alpha1
-        let alpha2' = if alpha2 < 0.f then 2.f * PI + alpha2 else alpha2
+        let alpha2 =
+            let alpha2' = atan2 (p2y - p0y) (p2x - p0x)
+            if alpha2' < 0.f then 2.f * PI + alpha2' else alpha2'
 
-        let diff = rot1 * alpha1' + rot2 * alpha2'
+        let diff = rot1 * alpha1 + rot2 * alpha2
 
         if diff > PI || (diff < 0.f && diff > -PI)
         then
@@ -131,7 +132,9 @@ let find (edges: Matrix<byte>)
 
     // We choose a window size for which the biggest ellipse can always be fitted in.
     let windowSize = roundInt (2.f * r2)
-    let factorNbPick = config.Parameters.factorNbPick
+    let factorNbValidPick = config.Parameters.factorNbValidPick
+    let factorNbMaxPick = config.Parameters.factorNbMaxPick
+    let nbPickElementsMin = config.Parameters.nbPickElementsMin
 
     let increment = windowSize / (int incrementWindowDivisor)
 
@@ -178,10 +181,11 @@ let find (edges: Matrix<byte>)
                     if edgesData.[i, j] = 1uy
                     then currentElements.Add(Point(j, i))
 
-            if currentElements.Count >= 10
+            if currentElements.Count >= nbPickElementsMin
             then
-                let mutable nbOfPicks = (float currentElements.Count) * factorNbPick |> int
-                while nbOfPicks > 0 do
+                let mutable nbOfPicks = (float currentElements.Count) * factorNbMaxPick |> int
+                let mutable nbOfValidPicks = (float currentElements.Count) * factorNbValidPick |> int
+                while nbOfPicks > 0 && nbOfValidPicks > 0 do
                     let p1 = currentElements.[rng.Next(currentElements.Count)]
                     let p2 = currentElements.[rng.Next(currentElements.Count)]
                     let p3 = currentElements.[rng.Next(currentElements.Count)]
@@ -198,9 +202,10 @@ let find (edges: Matrix<byte>)
                             match areVectorsValid (float32 p1xf) (float32 p1yf) (float32 p2xf) (float32 p2yf) -xDirData.[p1.Y, p1.X] -yDirData.[p1.Y, p1.X] -xDirData.[p2.Y, p2.X] -yDirData.[p2.Y, p2.X] with
                             | Some (m1, m2) ->
                                 match ellipse p1xf p1yf (float m1) p2xf p2yf (float m2) p3xf p3yf with
-                                | Some e when e.Cx > 0.f && e.Cx < w_f - 1.f && e.Cy > 0.f && e.Cy < h_f - 1.f &&
+                                | Some e when e.Cx > 0.f && e.Cx <  w_f - 1.f && e.Cy > 0.f && e.Cy < h_f - 1.f &&
                                               e.A >= r1 - radiusTolerance && e.A <= r2 + radiusTolerance && e.B >= r1 - radiusTolerance && e.B <= r2 + radiusTolerance ->
-                                     ellipses.Add e
+                                    nbOfValidPicks <- nbOfValidPicks - 1
+                                    ellipses.Add e
                                 | _ -> ()
                             | _ -> ()