module Config

open System

open Const
open UnitsOfMeasure

type Debug =
    | DebugOff
    | DebugOn of string // Output directory.

type Parameters = {
    rbcDiameter: float<μm>
    resolution: float<ppi>

    ratioAreaPaleCenter: float32 // The area of the second opening is 'ratioSecondAreaOpen' * mean RBC area. It's applied only if greater than 'initialAreaOpen'.

    granulometryRange: float32 // The radius will be seeked from radius - granulometryRange * radius to radius + granulometryRange * radius.

    minRbcRadius: float32 // Factor of the mean RBC radius.
    maxRbcRadius: float32 // Factor of the mean RBC radius.

    LPFStandardDeviation: float<μm> // Sigma parameter of the gaussian to remove the high frequency noise.

    // Ellipse.
    factorNbPick: float // The number of computed ellipse per edge pixel.

    // Parasites detection.
    darkStainLevel: float // Lower -> more sensitive. Careful about illumination on the borders.
    maxDarkStainRatio: float // When a cell must own less than this ratio to be a RBC.

    stainArea: float32 // Factor of a RBC area. 0.5 means the half of RBC area.
    stainSensitivity: float // between 0 (the least sensitive) and 1 (the most sensitive).
    maxStainRatio: float // A cell must own less than this ratio to be a RBC.

    infectionArea: float32 // Factor of a RBC area. 0.5 means the half of RBC area.
    infectionSensitivity: float // between 0 (the least sensitive) and 1 (the most sensitive).

    standardDeviationMaxRatio: float // The standard deviation of the pixel values of a cell can't be greater than standardDeviationMaxRatio * global standard deviation
    minimumCellAreaFactor: float32 // Factor of the mean RBC area. A cell with an area below this will be rejected.
}

let defaultParameters = {
    rbcDiameter = 8.<μm>
    resolution = 200.e3<ppi> // Correspond to 50X.

    ratioAreaPaleCenter = 1.f / 3.f // The ratio between an RBC area and the area of the its pale center.

    granulometryRange = 0.5f

    minRbcRadius = -0.3f
    maxRbcRadius = 0.3f

    LPFStandardDeviation = 0.2<μm> // 8.5e-6<inch>.

    factorNbPick = 1.0

    darkStainLevel = 0.25 // 0.3
    maxDarkStainRatio = 0.1 // 10 %

    infectionArea = 0.012f // 1.2 %
    infectionSensitivity = 0.9

    stainArea = 0.08f // 8 %
    stainSensitivity = 0.9
    maxStainRatio = 0.12 // 12 %

    standardDeviationMaxRatio = 0.5 // 0.5
    minimumCellAreaFactor = 0.4f }

type Config (param: Parameters) =
    let mutable parameters: Parameters = param

    let initialRBCRadius : float32 =
        let rbcRadiusInch: float<inch> = (μmToInch parameters.rbcDiameter) / 2.
        let rbcRadiusPx: float<px> = parameters.resolution * rbcRadiusInch
        float32 rbcRadiusPx

    new () = Config(defaultParameters)

    member this.Parameters with get() = parameters and set(param) = parameters <- param
    member val Debug = DebugOff with get, set

    member this.LPFStandardDeviation =
        let stdDeviation: float<px> = (μmToInch parameters.LPFStandardDeviation) * param.resolution
        float stdDeviation

    // Mean RBC radius.
    member val RBCRadius : float32 = initialRBCRadius with get, set

    member this.RBCRadiusμm : float<μm> =
        1.<px> * (float this.RBCRadius) / parameters.resolution |> inchToμm

    member this.RBCMinRadius = this.RBCRadius + parameters.minRbcRadius * this.RBCRadius
    member this.RBCMaxRadius = this.RBCRadius + parameters.maxRbcRadius * this.RBCRadius

    member this.RBCArea = PI * this.RBCRadius ** 2.f
    member this.RBCMinArea = parameters.minimumCellAreaFactor * this.RBCArea

    member this.InfectionArea = parameters.infectionArea * this.RBCArea
    member this.StainArea = parameters.stainArea * this.RBCArea

    member this.FormattedRadius =
        sprintf "%d px (%.1f μm)" (Utils.roundInt <| 2.f * this.RBCRadius) (2. * this.RBCRadiusμm)

    member this.Copy () =
        this.MemberwiseClone() :?> Config