Remove ellipses touching the edges.

[master-thesis.git] / Parasitemia / Parasitemia / Classifier.fs

diff --git a/Parasitemia/Parasitemia/Classifier.fs b/Parasitemia/Parasitemia/Classifier.fs
index 6441635..24ea4a7 100644 (file)
--- a/Parasitemia/Parasitemia/Classifier.fs
+++ b/Parasitemia/Parasitemia/Classifier.fs
@@ -14,5 +14,18 @@ type Cell = {
     center: Point
     elements: Matrix<byte> }
 
+type KdEllipse (e: Types.Ellipse) =
+    inherit Types.Ellipse (e.Cx, e.Cy, e.A, e.B, e.Alpha)
+    
+    interface KdTree.I2DCoords with    
+        member this.X = this.Cx
+        member this.Y = this.Cy
+        
+    
 let findCells (ellipses: Types.Ellipse list) (parasites: ParasitesMarker.Result) (fg: Image<Gray, byte>) : Cell list =
-    []
\ No newline at end of file
+    if ellipses.IsEmpty
+    then
+        []
+    else
+        let tree = KdTree.Tree.buildTree (ellipses |> List.map KdEllipse)
+        []
\ No newline at end of file