implement Marcos

Author: jbytecode

src/main/scala/marcos.scala

@@ -1,6 +1,16 @@
 package org.expr.mcdm
 
-case class MarcosResult() extends MCDMResult 
+case class MarcosResult(
+  normalizedDecmat: Mat,
+  S: Vec,
+  KPlus: Vec,
+  KMinus: Vec,
+  fKPlus: Vec,
+  fKMinus: Vec,
+  scores: Vec,
+  orderings: VecInt,
+  bestIndex: Int
+) extends MCDMResult 
 
 def marcos(
     decmat: Mat,
@@ -9,4 +19,48 @@ def marcos(
     normalization: NormalizationFunction =
       Normalization.MarcosNormalization,
     options: Map[String, Any] = Map.empty 
-): MarcosResult = ???
+): MarcosResult = 
+
+  val (row, col) = Matrix.size(decmat)
+
+  val normalizedDecmat = normalization(decmat, weights, directions)
+
+  val S = normalizedDecmat.map(row => Matrix.sumproduct(weights, row))
+
+  // KPlus = S[1:row] ./ S[row+1]
+  // S.slice(0, 4) takes S(0), S(1), S(2), S(3)
+  val KPlus = S.slice(0, row).map(x => x / S(row))
+
+  // KMinus = S[1:row] ./ S[row+2]
+  val KMinus = S.slice(0, row).map(x => x / S(row + 1))
+
+  // fKPlus = KPlus ./ (KPlus .+ KMinus)
+  val fKPlus = KPlus.zip(KMinus).map((x, y) => x / (x + y))
+
+  // fKMinus = KMinus ./ (KPlus .+ KMinus)
+  val fKMinus = KMinus.zip(KPlus).map((x, y) => x / (x + y))
+
+  var scores = Matrix.zeros(row)
+  for i <- 0 until row do
+    scores(i) =
+      (KPlus(i) + KMinus(i)) /
+        ((1 + (1 - fKPlus(i)) / fKPlus(i)) + ((1 - fKMinus(i)) / fKMinus(i)))
+
+  val orderings = scores.zipWithIndex
+    .sortBy(-_._1)
+    .map(_._2)
+    .toArray
+
+  val bestIndex = orderings.last
+
+  MarcosResult(
+    normalizedDecmat,
+    S,
+    KPlus,
+    KMinus,
+    fKPlus,
+    fKMinus,
+    scores,
+    orderings,
+    bestIndex
+  )

src/test/scala/testmarcos.scala

@@ -18,11 +18,8 @@ class TestMarcos extends munit.FunSuite {
 
     val expected_scores = Array(0.684865943528, 0.672767106696, 0.662596906139, 0.661103207660)
 
-    //val result = marcos(decmat, weights, fns)
+    val result = marcos(decmat, weights, fns)
 
-    println("*****************************")
-    println("Marcos not implemented yet!")
-    println("*****************************")
-    //A.assert(Matrix.elementwise_equal(result, expected_scores, 1e-6), "Marcos scores do not match expected values")
+    A.assert(Matrix.elementwise_equal(result.scores, expected_scores, 1e-6), "Marcos scores do not match expected values")
   }
 }