X-Git-Url: http://git.euphorik.ch/?p=master-thesis.git;a=blobdiff_plain;f=Parasitemia%2FParasitemia%2FGUI%2FState.fs;h=45272cab7de11b7bb1525ee28959e49cfc1986a1;hp=a1ae0521c01ea831a35876f05053eef6d4052366;hb=05be8164d308447b916544ae3ce4211500dfd8da;hpb=999a48b8eb25c92e4403c9280fd4fe68f9bc4b7b

diff --git a/Parasitemia/Parasitemia/GUI/State.fs b/Parasitemia/Parasitemia/GUI/State.fs
index a1ae052..45272ca 100644
--- a/Parasitemia/Parasitemia/GUI/State.fs
+++ b/Parasitemia/Parasitemia/GUI/State.fs
@@ -1,5 +1,6 @@
 module Parasitemia.GUI.State
 
+open System
 open System.Collections.Generic
 open System.Windows
 
@@ -10,41 +11,116 @@ open Types
 
 type State () =
     let sourceImages = List<SourceImage>()
+    let mutable alteredSinceLastSave = false
+    let mutable patientID = ""
 
+    member this.AlteredSinceLastSave = alteredSinceLastSave
     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.PatientID
+        with get () : string = patientID
+        and set id =
+            if id <> patientID
+            then
+                alteredSinceLastSave <- true
+                patientID <- id
+
+    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.SetAsInfected (rbc: RBC) (infected: bool) =
+        if infected <> rbc.infected
+        then
+            alteredSinceLastSave <- true
+            rbc.infected <- infected
+            rbc.setManually <- not rbc.setManually
 
     member this.Save () =
-        let data = { PiaZ.sources = List.ofSeq sourceImages; PiaZ.patientID = this.PatientID }
+        let data = { PiaZ.FileData.patientID = this.PatientID; PiaZ.FileData.images = List.ofSeq sourceImages }
         PiaZ.save this.FilePath data
+        alteredSinceLastSave <- false
 
     member this.Load () =
         let data = PiaZ.load this.FilePath
         this.PatientID <- data.patientID
         sourceImages.Clear()
-        sourceImages.InsertRange(0, data.sources)
+        sourceImages.InsertRange(0, data.images)
+        if sourceImages.Count > 0
+        then this.CurrentImage <- Some sourceImages.[0]
+        alteredSinceLastSave <- false
 
-    member this.AddSourceImage (filePath: string) : SourceImage =
-        let srcImg = { num = sourceImages.Count + 1; rbcs = []; img = new Image<Bgr, byte>(filePath) }
+    member this.AddSourceImage (filePath: string) (defaultConfig: Config.Config) : SourceImage =
+        let srcImg = { num = sourceImages.Count + 1; config = defaultConfig.Copy(); dateLastAnalysis = DateTime(0L); rbcs = []; img = new Image<Bgr, byte>(filePath) }
         sourceImages.Add(srcImg)
+        if sourceImages.Count = 1
+        then this.CurrentImage <- Some sourceImages.[0]
+        alteredSinceLastSave <- true
         srcImg
 
-    member this.SetResult (num: int) (cells: Cell list) =
-        let sourceImage = sourceImages.Find(fun srcImg -> srcImg.num = num)
+    member this.RemoveSourceImage (srcImg: SourceImage) =
+        let isCurrent =
+            match this.CurrentImage with
+            | Some srcImg' -> srcImg = srcImg'
+            | _ -> false
+
+        if sourceImages.Remove(srcImg)
+        then
+            alteredSinceLastSave <- true
+            if isCurrent
+            then
+                this.CurrentImage <- if sourceImages.Count > 0 then Some sourceImages.[0] else None
+        // Re-numbered the images.
+        sourceImages |> Seq.iteri (fun i srcImg -> srcImg.num <- i + 1)
+
+    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.dateLastAnalysis <- DateTime.UtcNow
+
+        // To match with previously manually altered RBC.
+        let manuallyAlteredPreviousRBCS = sourceImage.rbcs |> List.filter (fun rbc -> rbc.setManually)
+        let tolerance = (float sourceImage.config.RBCRadius.Pixel) * 0.5 // +/-.
+        let getPreviousRBC (center: Point) : RBC option =
+            manuallyAlteredPreviousRBCS |> List.tryFind (fun rbc -> rbc.center.X > center.X - tolerance && rbc.center.X < center.X + tolerance &&
+                                                                    rbc.center.Y > center.Y - tolerance && rbc.center.Y < center.Y + tolerance)
+
         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
-                  infected = cell.cellClass = InfectedRBC
-                  setManually = false
-                  center = Point(float cell.center.X, float cell.center.Y)
+                let center = Point(float cell.center.X, float cell.center.Y)
+                let infected, setManually =
+                    match getPreviousRBC center with
+                    | Some rbc -> rbc.infected, true
+                    | _ -> cell.cellClass = InfectedRBC, false
+
+                { num = i + 1
+                  infected = infected
+                  setManually = setManually
+                  center = center
                   size = Size(float cell.elements.Width, float cell.elements.Height)
-                  stainArea = cell.stainArea })
+                  infectedArea = cell.infectedArea })
+
+        alteredSinceLastSave <- true
 
     member this.SourceImages : SourceImage seq =
         sourceImages :> SourceImage seq
 
     member this.Reset () =
         this.PatientID <- ""
-        sourceImages.Clear()
\ No newline at end of file
+        this.FilePath <- ""
+        this.CurrentImage <- None
+        sourceImages.Clear()
+        alteredSinceLastSave <- false
\ No newline at end of file