Add files via upload

1 files changed, 195 insertions, 0 deletions

diff --git a/sourcecodes/k-best/src/Arguments.h b/sourcecodes/k-best/src/Arguments.h
new file mode 100644
index 00000000..574b2bf7
--- /dev/null
+++ b/sourcecodes/k-best/src/Arguments.h
@@ -0,0 +1,195 @@
+#ifndef ARGUMENTS_H
+#define ARGUMENTS_H
+
+#include<stdio.h>
+#include<stdlib.h>
+
+using namespace std;
+
+class Arguments {
+public:
+	static char* datafile;
+	static char* layeringfile;
+	static char* maxindegree;
+	static char* model;
+	static char* task;
+	static char* maxnumrecords;
+	
+	//HR: Add for in_out features
+	static char* inFeasFileName;
+	static char* outFeasFileName;
+	//
+	
+	//HR: Add for in_out features
+	static int option;
+	//
+	
+	//HR: Add for ADtree option
+	static int ADtree;
+	//
+	
+	//HR: Add for Topk dir option
+	static char* directoryName;
+	//
+
+
+
+	static void init(int argc, char **args){
+		for(int i = 1; i < argc; i ++){
+			
+			if(args[i][0]=='-' && args[i][1]=='d' && args[i][2]=='\0'){
+				int j = i + 1;
+				while (j < argc && args[j][0]!='-') {
+					Arguments::datafile = args[j];
+					j ++;
+				} 
+			}
+			else if(args[i][0]=='-' && args[i][1]=='l' && args[i][2]=='\0'){
+				int j = i + 1;
+				while (j < argc && args[j][0]!='-') {
+					Arguments::layeringfile = args[j]; 
+					j ++;
+				}
+			}
+			else if(args[i][0]=='-' && args[i][1]=='m' && args[i][2]=='\0'){
+				int j = i + 1;
+				while (j < argc && args[j][0]!='-') {
+					Arguments::maxindegree = args[j]; 
+					j ++;
+				}
+			}
+			else if(args[i][0]=='-' && args[i][1]=='u' && args[i][2]=='\0'){
+				int j = i + 1;
+				while (j < argc && args[j][0]!='-') {
+					Arguments::maxnumrecords = args[j]; 
+					j ++;
+				}
+			}
+			else if(args[i][0]=='-' && args[i][1]=='M' && args[i][2]=='\0'){
+				int j = i + 1;
+				while (j < argc && args[j][0]!='-') {
+					Arguments::model = args[j]; 
+					j ++;
+				}
+			}
+			else if(args[i][0]=='-' && args[i][1]=='T' && args[i][2]=='\0'){
+				int j = i + 1;
+				while (j < argc && args[j][0]!='-') {
+					Arguments::task = args[j]; 
+					j ++;
+				}
+			}
+			//HR: Add for in_out feature
+			else if(args[i][0]=='-' && args[i][1]=='i' && args[i][2]=='n' && args[i][3]=='\0'){
+				int j = i + 1;
+				while (j < argc && args[j][0]!='-') {
+					Arguments::inFeasFileName = args[j]; 
+					j ++;
+				}
+			}
+			else if(args[i][0]=='-' && args[i][1]=='o' && args[i][2]=='u' && args[i][3]=='t' && args[i][4]=='\0'){
+				int j = i + 1;
+				while (j < argc && args[j][0]!='-') {
+					Arguments::outFeasFileName = args[j]; 
+					j ++;
+				}
+			}
+			//
+			//HR: Add for in_out feature
+			else if(args[i][0]=='-' && args[i][1]=='o' && args[i][2]=='p' && args[i][3]=='t' && args[i][4]=='\0'){
+				int j = i + 1;
+				while (j < argc && args[j][0]!='-') {
+					Arguments::option = atoi(args[j]); 
+					j ++;
+				}
+			}
+			//
+			//HR: Add for ADtree feature
+			else if(args[i][0]=='-' && args[i][1]=='a' && args[i][2]=='d' && args[i][3]=='\0'){
+				int j = i + 1;
+				while (j < argc && args[j][0]!='-') {
+					Arguments::ADtree = atoi(args[j]); 
+					j ++;
+				}
+			}
+			//
+			//HR: Add for Topk dir
+			else if(args[i][0]=='-' && args[i][1]=='d' && args[i][2]=='i' && args[i][3]=='r' && args[i][4]=='\0'){
+				int j = i + 1;
+				while (j < argc && args[j][0]!='-') {
+					Arguments::directoryName = args[j];
+					j++;
+				}
+			}
+			//
+		}
+	
+		print_arguments(stderr);
+		
+	}
+	static void print_arguments(FILE *f){
+		fprintf(f, " -d Data file:\n");
+		fprintf(f, "   %62s\n", Arguments::datafile);	
+		//fprintf(f, " -l Layering file:\n");
+		//fprintf(f, "   %62s\n", Arguments::layeringfile);
+		fprintf(f, " -m Maximum indegree:\n");
+		fprintf(f, "   %62s\n", Arguments::maxindegree);
+		fprintf(f, " -u Maximum number of data records read:\n");
+		fprintf(f, "   %62s\n", Arguments::maxnumrecords);
+		//fprintf(f, " -M Model:\n");
+		//fprintf(f, "   %62s\n", Arguments::model);
+		//fprintf(f, " -T Task (Infer=I, Generate=G):\n");
+		//fprintf(f, "   %62s\n", Arguments::task);
+		
+		//HR: Add for in_out
+		fprintf(f, " -opt Option:\n");
+		fprintf(f, "   %62d\n", Arguments::option);
+		if(Arguments::option == 5){
+			fprintf(f, " -in In_Feature File:\n");
+			fprintf(f, "   %62s\n", Arguments::inFeasFileName);
+			fprintf(f, " -out Out_Feature File:\n");
+			fprintf(f, "   %62s\n", Arguments::outFeasFileName);
+		}
+		//
+		//HR: Add for ADtree
+		fprintf(f, " -ad ADtree:\n");
+		fprintf(f, "   %62d\n", Arguments::ADtree);
+		//
+		//HR: Add for Topk dir
+		fprintf(f, " -dir directory name:\n");
+		fprintf(f, "   %62s\n", Arguments::directoryName);
+		//
+	}
+};
+
+char* Arguments::datafile = "testdata.dat";
+char* Arguments::layeringfile = "%";
+char* Arguments::maxindegree = "3";
+char* Arguments::model = "M";
+char* Arguments::task = "I";
+char* Arguments::maxnumrecords = "999999";
+
+
+//HR: For test
+//char* Arguments::datafile = "cases/iris.idt";
+//char* Arguments::layeringfile = "%";
+//char* Arguments::maxindegree = "4";
+//char* Arguments::model = "M";
+//char* Arguments::task = "I";
+//char* Arguments::maxnumrecords = "150";
+
+
+//HR: Add for in_out
+char* Arguments::inFeasFileName = "inFeature.txt";
+char* Arguments::outFeasFileName = "outFeature.txt";
+int Arguments::option = 5;
+
+//HR: Add for ADtree
+int Arguments::ADtree = 0;
+
+
+//HR: Add for Topk dir option
+char* Arguments::directoryName = "./";
+//
+
+#endif