X-Git-Url: https://ruin.nu/git/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmain.cpp;h=abd520f43a29f7a3ec9b4cb7cb8f893895f16cbc;hb=69cfdf79b7597ee6133dc3add1a759bc94861fa6;hp=fe581385516cc1133f80a49a9e338211a7a69279;hpb=0b90077b65cab03d00ef8f8c761afc3593f1b386;p=germs.git

diff --git a/src/main.cpp b/src/main.cpp
index fe58138..abd520f 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -21,10 +21,11 @@ int main(int argc, char** argv){
 	Model model(0);
 	bool detectModel = true;
 	bool onlyIdentify = false;
+	bool printPerm = false;
 
 	//Parse command line arguments
 	int opt;
-	while ((opt = getopt(argc, argv, "im:n:h")) != -1) {
+	while ((opt = getopt(argc, argv, "im:n:hp")) != -1) {
 		switch (opt) {
 			case 'm':
 				model = Model::modelFactory(optarg);
@@ -36,10 +37,15 @@ int main(int argc, char** argv){
 			case 'i':
 				onlyIdentify = true;
 				break;
+			case 'p':
+				printPerm = true;
+				break;
 			case 'h':
 				cout << "Usage: " << argv[0] << " [OPTION] [FILE]" << endl
 					<< endl << "  -m <model> Specifies which model to use for sorting: Whirl, X, Zipper, FatX or Cloud "
 					<< endl << "  -n <ann>   Specifies which artificial neural network to use for identification. '" << ann << "' is used by default"
+					<< endl << "  -i         Only identify the model, don't try to sort it"
+					<< endl << "  -p         Print the resulting permutation at each sorting step"
 					<< endl << "  -h         Prints this help message"
 					<< endl << endl
 					<< "With no FILE, or if FILE is '-', stdin will be used"
@@ -97,9 +103,17 @@ int main(int argc, char** argv){
 
 	//Print the result
 	double score = 0;
+
+	GeneOrder temp(go);
 	for (GeneSorter::ActionList::iterator sa = al.begin(); sa != al.end(); ++sa){
-		cout << "Action: " << sa->toString() << " model score: " << model.score(*sa,go) << endl;
-		score += model.score(*sa,go);
+		cout << "Action: " << sa->toString() << " model score: " << model.score(*sa,temp) << endl;
+		(*sa)(temp);
+		score += model.score(*sa,temp);
+
+		if (printPerm){
+			copy(temp.begin(), temp.end(), ostream_iterator<int>(cout, " "));
+			cout << endl;
+		}
 	}
 	cout << "Avg score: " << score / al.size() << endl;