Only show models with positive score, and convert the enum to strings

[germs.git] / src / modelidentifier.cpp

diff --git a/src/modelidentifier.cpp b/src/modelidentifier.cpp
index 98fae9bbc48c2261dec4648efd6edb3b4733a435..56b6935c48a4050bda6be087f9899f8f2af45b40 100644 (file)
--- a/src/modelidentifier.cpp
+++ b/src/modelidentifier.cpp
@@ -86,3 +86,14 @@ std::map<ModelIdentifier::Model,double> ModelIdentifier::identify(const GeneOrde
        scores[Cloud] = output[4];
        return scores;
 }
+
+string ModelIdentifier::modelName(Model m){
+       switch (m){
+               case Whirl : return "Whirl";
+               case X : return "X";
+               case FatX : return "FatX";
+               case Zipper : return "Zipper";
+               case Cloud : return "Cloud";
+       }
+       return "Unknown model";
+}