From d33ba54a38ebefe8861d947a620188fcb4fc9db4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kamil=20=C5=9Aliwak?= <kamil.sliwak@codepoets.it>
Date: Sat, 29 Feb 2020 00:57:24 +0100
Subject: [PATCH] [yul-phaser] Add print-optimised-programs and
 print-optimised-asts modes

---
 tools/yulPhaser/Phaser.cpp | 41 +++++++++++++++++++++++++++++++++++++-
 tools/yulPhaser/Phaser.h   |  8 ++++++++
 2 files changed, 48 insertions(+), 1 deletion(-)

diff --git a/tools/yulPhaser/Phaser.cpp b/tools/yulPhaser/Phaser.cpp
index 2a4ee4b86..0bfd6a433 100644
--- a/tools/yulPhaser/Phaser.cpp
+++ b/tools/yulPhaser/Phaser.cpp
@@ -49,6 +49,8 @@ namespace
 map<PhaserMode, string> const PhaserModeToStringMap =
 {
 	{PhaserMode::RunAlgorithm, "run-algorithm"},
+	{PhaserMode::PrintOptimisedPrograms, "print-optimised-programs"},
+	{PhaserMode::PrintOptimisedASTs, "print-optimised-asts"},
 };
 map<string, PhaserMode> const StringToPhaserModeMap = invertMap(PhaserModeToStringMap);
 
@@ -642,11 +644,13 @@ void Phaser::runPhaser(po::variables_map const& _arguments)
 	vector<Program> programs = ProgramFactory::build(programOptions);
 	vector<shared_ptr<ProgramCache>> programCaches = ProgramCacheFactory::build(cacheOptions, programs);
 
-	unique_ptr<FitnessMetric> fitnessMetric = FitnessMetricFactory::build(metricOptions, move(programs), programCaches);
+	unique_ptr<FitnessMetric> fitnessMetric = FitnessMetricFactory::build(metricOptions, programs, programCaches);
 	Population population = PopulationFactory::build(populationOptions, move(fitnessMetric));
 
 	if (_arguments["mode"].as<PhaserMode>() == PhaserMode::RunAlgorithm)
 		runAlgorithm(_arguments, move(population), move(programCaches));
+	else
+		printOptimisedProgramsOrASTs(_arguments, population, move(programs), _arguments["mode"].as<PhaserMode>());
 }
 
 void Phaser::runAlgorithm(
@@ -665,3 +669,38 @@ void Phaser::runAlgorithm(
 	AlgorithmRunner algorithmRunner(move(_population), move(_programCaches), buildAlgorithmRunnerOptions(_arguments), cout);
 	algorithmRunner.run(*geneticAlgorithm);
 }
+
+void Phaser::printOptimisedProgramsOrASTs(
+	po::variables_map const& _arguments,
+	Population const& _population,
+	vector<Program> _programs,
+	PhaserMode phaserMode
+)
+{
+	assert(phaserMode == PhaserMode::PrintOptimisedPrograms || phaserMode == PhaserMode::PrintOptimisedASTs);
+	assert(_programs.size() == _arguments["input-files"].as<vector<string>>().size());
+
+	if (_population.individuals().size() == 0)
+	{
+		cout << "<EMPTY POPULATION>" << endl;
+		return;
+	}
+
+	vector<string> const& paths = _arguments["input-files"].as<vector<string>>();
+	for (auto& individual: _population.individuals())
+	{
+		cout << "Chromosome: " << individual.chromosome << endl;
+
+		for (size_t i = 0; i < _programs.size(); ++i)
+		{
+			for (size_t j = 0; j < _arguments["chromosome-repetitions"].as<size_t>(); ++j)
+				_programs[i].optimise(individual.chromosome.optimisationSteps());
+
+			cout << "Program: " << paths[i] << endl;
+			if (phaserMode == PhaserMode::PrintOptimisedPrograms)
+				cout << _programs[i] << endl;
+			else
+				cout << _programs[i].toJson() << endl;
+		}
+	}
+}
diff --git a/tools/yulPhaser/Phaser.h b/tools/yulPhaser/Phaser.h
index ef7c00f41..77e9e48c8 100644
--- a/tools/yulPhaser/Phaser.h
+++ b/tools/yulPhaser/Phaser.h
@@ -50,6 +50,8 @@ class ProgramCache;
 enum class PhaserMode
 {
 	RunAlgorithm,
+	PrintOptimisedPrograms,
+	PrintOptimisedASTs,
 };
 
 enum class Algorithm
@@ -236,6 +238,12 @@ private:
 		Population _population,
 		std::vector<std::shared_ptr<ProgramCache>> _programCaches
 	);
+	static void printOptimisedProgramsOrASTs(
+		boost::program_options::variables_map const& _arguments,
+		Population const& _population,
+		std::vector<Program> _programs,
+		PhaserMode phaserMode
+	);
 };
 
 }