X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=src%2FTuner%2Fsearchtuner.cc;h=9ab6d45d8821318d7726a371f5ebb4acf85f7f6c;hb=b0f6ab9a232beae1ddad1ec0a38e040e2d6187cb;hp=a0fc1bca4a90d5b0d03bff4e541910b767b0d517;hpb=ab3148a0a14a748cb589829029ed0bd40788f972;p=satune.git

diff --git a/src/Tuner/searchtuner.cc b/src/Tuner/searchtuner.cc
index a0fc1bc..9ab6d45 100644
--- a/src/Tuner/searchtuner.cc
+++ b/src/Tuner/searchtuner.cc
@@ -1,4 +1,7 @@
 #include "searchtuner.h"
+#include <iostream>
+#include <fstream>
+using namespace std;
 
 TunableSetting::TunableSetting(VarType _type, TunableParam _param) :
 	hasVar(true),
@@ -41,27 +44,135 @@ void TunableSetting::setDecision(int _low, int _high, int _default, int _selecti
 }
 
 void TunableSetting::print() {
+	model_print("Param %s = %u \t range=[%u,%u]", tunableParameterToString( (Tunables)param), selectedValue, lowValue, highValue);
 	if (hasVar) {
-		model_print("VarType1 %" PRIu64 ", ", type1);
+		model_print("\tVarType1 %" PRIu64 ", ", type1);
 		model_print("VarType2 %" PRIu64 ", ", type2);
 	}
-	model_print("Param %s = %u \t range=[%u,%u]\n", tunableParameterToString( (Tunables)param), selectedValue, lowValue, highValue);
+	model_print("\n");
 }
 
-unsigned int tunableSettingHash(TunableSetting *setting) {
-	return setting->hasVar ^ setting->type1 ^ setting->type2 ^ setting->param;
-}
 
-bool tunableSettingEquals(TunableSetting *setting1, TunableSetting *setting2) {
-	return setting1->hasVar == setting2->hasVar &&
-				 setting1->type1 == setting2->type1 &&
-				 setting1->type2 == setting2->type2 &&
-				 setting1->param == setting2->param;
+ostream &operator<<(ostream &os, const TunableSetting &ts)
+{
+	os << ts.hasVar << " " << ts.type1 << " " << ts.type2 << " " << ts.param << " " << ts.lowValue << " "
+		 << ts.highValue << " " << ts.defaultValue << " " << ts.selectedValue;
+	return os;
 }
 
+
 SearchTuner::SearchTuner() {
 }
 
+SearchTuner::SearchTuner(const char *filename, bool addused) {
+	ifstream myfile;
+	myfile.open (filename, ios::in);
+	if (myfile.is_open()) {
+		bool hasVar;
+		VarType type1;
+		VarType type2;
+		TunableParam param;
+		int lowValue;
+		int highValue;
+		int defaultValue;
+		int selectedValue;
+		while (myfile >> hasVar >> type1 >> type2 >> param >> lowValue >> highValue >> defaultValue >> selectedValue) {
+			TunableSetting *setting;
+
+			if (hasVar) {
+				setting = new TunableSetting(type1, type2, param);
+			} else {
+				setting = new TunableSetting(param);
+			}
+			setting->setDecision(lowValue, highValue, defaultValue, selectedValue);
+			settings.add(setting);
+			if (addused) {
+				usedSettings.add(setting);
+			}
+		}
+		myfile.close();
+	} else {
+		model_print("Warning: Tuner %s couldn't be loaded ... Using default tuner instead ....\n", filename);
+	}
+}
+
+bool SearchTuner::equalUsed(SearchTuner *tuner) {
+	if (tuner->usedSettings.getSize() != usedSettings.getSize()) {
+		return false;
+	}
+	bool result = true;
+	SetIteratorTunableSetting *iterator = usedSettings.iterator();
+	while (iterator->hasNext()) {
+		TunableSetting *setting = iterator->next();
+		if (!tuner->usedSettings.contains(setting)) {
+			result = false;
+			break;
+		} else {
+			TunableSetting *tunerSetting = tuner->usedSettings.get(setting);
+			if (tunerSetting->selectedValue != setting->selectedValue) {
+				result = false;
+				break;
+			}
+		}
+	}
+	delete iterator;
+	return result;
+}
+
+void SearchTuner::addUsed(const char *filename) {
+	ifstream myfile;
+	myfile.open (filename, ios::in);
+	if (myfile.is_open()) {
+		bool hasVar;
+		VarType type1;
+		VarType type2;
+		TunableParam param;
+		int lowValue;
+		int highValue;
+		int defaultValue;
+		int selectedValue;
+		while (myfile >> hasVar >> type1 >> type2 >> param >> lowValue >> highValue >> defaultValue >> selectedValue) {
+			TunableSetting *setting;
+
+			if (hasVar) {
+				setting = new TunableSetting(type1, type2, param);
+			} else {
+				setting = new TunableSetting(param);
+			}
+			setting->setDecision(lowValue, highValue, defaultValue, selectedValue);
+			if (!settings.contains(setting)) {
+				settings.add(setting);
+				usedSettings.add(setting);
+			} else {
+				TunableSetting *tmp = settings.get(setting);
+				settings.remove(tmp);
+				usedSettings.remove(tmp);
+				delete tmp;
+				settings.add(setting);
+				usedSettings.add(setting);
+			}
+		}
+		myfile.close();
+	}
+}
+
+bool SearchTuner::isSubTunerof(SearchTuner *newTuner) {
+	SetIteratorTunableSetting *iterator = usedSettings.iterator();
+	while (iterator->hasNext()) {
+		TunableSetting *setting = iterator->next();
+		if (!newTuner->settings.contains(setting)) {
+			return false;
+		} else {
+			TunableSetting *newSetting = newTuner->settings.get(setting);
+			if (newSetting->selectedValue != setting->selectedValue) {
+				return false;
+			}
+		}
+	}
+	delete iterator;
+	return true;
+}
+
 SearchTuner *SearchTuner::copyUsed() {
 	SearchTuner *tuner = new SearchTuner();
 	SetIteratorTunableSetting *iterator = usedSettings.iterator();
@@ -83,6 +194,24 @@ SearchTuner::~SearchTuner() {
 	delete iterator;
 }
 
+void SearchTuner::setTunable(TunableParam param, TunableDesc *descriptor, uint value) {
+	TunableSetting *result = new TunableSetting(param);
+	result->setDecision(descriptor->lowValue, descriptor->highValue, descriptor->defaultValue, value);
+	settings.add(result);
+	usedSettings.add(result);
+}
+
+void SearchTuner::setVarTunable(VarType vartype, TunableParam param, TunableDesc *descriptor, uint value) {
+	setVarTunable(vartype, 0, param, descriptor, value);
+}
+
+void SearchTuner::setVarTunable(VarType vartype1, VarType vartype2, TunableParam param, TunableDesc *descriptor, uint value) {
+	TunableSetting *result = new TunableSetting(vartype1, vartype2, param);
+	result->setDecision(descriptor->lowValue, descriptor->highValue, descriptor->defaultValue, value);
+	settings.add(result);
+	usedSettings.add(result);
+}
+
 int SearchTuner::getTunable(TunableParam param, TunableDesc *descriptor) {
 	TunableSetting setting(param);
 	TunableSetting *result = usedSettings.get(&setting);
@@ -128,6 +257,9 @@ void SearchTuner::randomMutate() {
 		randomSetting->selectedValue = randomchoice;
 	else
 		randomSetting->selectedValue = randomchoice + 1;
+	model_print("&&&&&&&&Mutating&&&&&&&\n");
+	randomSetting->print();
+	model_print("&&&&&&&&&&&&&&&&&&&&&&&\n");
 }
 
 void SearchTuner::print() {
@@ -140,6 +272,30 @@ void SearchTuner::print() {
 
 }
 
+void SearchTuner::serialize(const char *filename) {
+	ofstream myfile;
+	myfile.open (filename, ios::out | ios::trunc);
+	SetIteratorTunableSetting *iterator = settings.iterator();
+	while (iterator->hasNext()) {
+		TunableSetting *setting = iterator->next();
+		myfile << *setting << endl;
+	}
+	myfile.close();
+	delete iterator;
+}
+
+void SearchTuner::serializeUsed(const char *filename) {
+	ofstream myfile;
+	myfile.open (filename, ios::out | ios::trunc);
+	SetIteratorTunableSetting *iterator = usedSettings.iterator();
+	while (iterator->hasNext()) {
+		TunableSetting *setting = iterator->next();
+		myfile << *setting << endl;
+	}
+	myfile.close();
+	delete iterator;
+}
+
 void SearchTuner::printUsed() {
 	SetIteratorTunableSetting *iterator = usedSettings.iterator();
 	while (iterator->hasNext()) {