X-Git-Url: https://ruin.nu/git/?a=blobdiff_plain;f=src%2Fmodelidentifier.h;h=a42e0efb7097fe05ed34a8175c9a05633baf5d18;hb=67df9f01df85be7e0ddcc0d53c89534aae009ed9;hp=1ed4809ff72319836c798cbdfd14c1c231183c86;hpb=868e1b08fbbd49e489dc1349cb3657521c5c1dd9;p=germs.git

diff --git a/src/modelidentifier.h b/src/modelidentifier.h
index 1ed4809..a42e0ef 100644
--- a/src/modelidentifier.h
+++ b/src/modelidentifier.h
@@ -35,6 +35,8 @@ struct fann;
  */
 class ModelIdentifier{
 	public:
+		//TODO: Make a Model class with functions for scoring SortActions
+		//one subclass for each model. Possibly storing name as string too.
 		enum Model{Whirl,X,FatX,Zipper,Cloud};
 
 		/**
@@ -54,6 +56,8 @@ class ModelIdentifier{
 		 */
 		std::map<Model,double> identify(const GeneOrder& go);
 
+		static std::string modelName(Model m);
+
 		~ModelIdentifier();
 	private:
 		fann* _ann;