X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=Parasitemia%2FParasitemia%2FGUI%2FState.fs;h=44de2e335f3f2245f077dbda2e3d3cdcd0995bb9;hb=8013d6a586604e443332e6e9a715c86df444a875;hp=b5136c2269d93d3e74c16f0056b48d00e4d41882;hpb=da6b1ee15d0368ef9f2d9734b0d72996cbc227d8;p=master-thesis.git

diff --git a/Parasitemia/Parasitemia/GUI/State.fs b/Parasitemia/Parasitemia/GUI/State.fs
index b5136c2..44de2e3 100644
--- a/Parasitemia/Parasitemia/GUI/State.fs
+++ b/Parasitemia/Parasitemia/GUI/State.fs
@@ -1,34 +1,67 @@
-module State
+module Parasitemia.GUI.State
 
 open System.Collections.Generic
+open System.Windows
 
 open Emgu.CV
 open Emgu.CV.Structure
 
-type ImageSource = {
-    cells: List<Types.Cell>
-    img: Image<Bgr, byte> }
+open Types
 
 type State () =
-    let imagesSource = List<ImageSource>()
+    let sourceImages = List<SourceImage>()
 
+    member val CurrentImage: SourceImage option = None with get, set
     member val FilePath: string = "" with get, set
     member val PatientID: string = "" with get, set
 
+    member this.ImageParasitemia (srcImg: SourceImage) : int * int =
+        List.length srcImg.rbcs,
+        srcImg.rbcs |> List.fold (fun nbInfected rbc -> if rbc.infected then nbInfected + 1 else nbInfected) 0
+
+    member this.GlobalParasitemia : int * int =
+        sourceImages
+        |> Seq.fold (fun (nbTotal, nbTotalInfected) srcImg ->
+                        let nb, nbInfected = this.ImageParasitemia srcImg
+                        nbTotal + nb, nbTotalInfected + nbInfected) (0, 0)
+
     member this.Save () =
-        let data = { Pia.sources = []; Pia.patientID = this.PatientID }
-        Pia.save this.FilePath data
+        let data = { PiaZ.sources = List.ofSeq sourceImages; PiaZ.patientID = this.PatientID }
+        PiaZ.save this.FilePath data
 
     member this.Load () =
-        let data = Pia.load this.FilePath
+        let data = PiaZ.load this.FilePath
         this.PatientID <- data.patientID
+        sourceImages.Clear()
+        sourceImages.InsertRange(0, data.sources)
+        if sourceImages.Count > 0
+        then this.CurrentImage <- Some sourceImages.[0]
+
+    member this.AddSourceImage (filePath: string) : SourceImage =
+        let srcImg = { num = sourceImages.Count + 1; rbcs = []; img = new Image<Bgr, byte>(filePath) }
+        sourceImages.Add(srcImg)
+        if sourceImages.Count = 1
+        then this.CurrentImage <- Some sourceImages.[0]
+        srcImg
 
-    member this.AddSourceImage (filePath: string) =
-        imagesSource.Add({ cells = List<Types.Cell>(); img = new Image<Bgr, byte>(filePath) })
+    member this.SetResult (imgNum: int) (cells: Cell list) =
+        let sourceImage = sourceImages.Find(fun srcImg -> srcImg.num = imgNum)
+        let w = sourceImage.img.Width
+        let h = sourceImage.img.Height
+        sourceImage.rbcs <- cells
+            |> List.filter (fun cell -> match cell.cellClass with HealthyRBC | InfectedRBC -> true | _ -> false )
+            |> List.sortByDescending (fun cell -> cell.infectedArea, (w - cell.center.X) + (h - cell.center.Y))
+            |> List.mapi (fun i cell ->
+                { num = i + 1
+                  infected = cell.cellClass = InfectedRBC
+                  setManually = false
+                  center = Point(float cell.center.X, float cell.center.Y)
+                  size = Size(float cell.elements.Width, float cell.elements.Height)
+                  infectedArea = cell.infectedArea })
 
-    member x.ImagesSource : ImageSource seq =
-        imagesSource :> ImageSource seq
+    member this.SourceImages : SourceImage seq =
+        sourceImages :> SourceImage seq
 
     member this.Reset () =
         this.PatientID <- ""
-        imagesSource.Clear()
\ No newline at end of file
+        sourceImages.Clear()
\ No newline at end of file