Reorganized code for better readability.

This commit is contained in:
Bhargava Shastry
2021-01-26 16:16:03 +01:00
parent f5e0faaa37
commit be733d5f61
2 changed files with 16 additions and 31 deletions
+2 -11
View File
@@ -64,7 +64,6 @@ string TestCaseGenerator::visit()
<< "\n";
m_numSourceUnits++;
os << visitChildren();
generator<SourceUnitGenerator>()->reset();
}
return os.str();
}
@@ -78,15 +77,7 @@ void SourceUnitGenerator::setup()
string SourceUnitGenerator::visit()
{
string sourceUnit = visitChildren();
reset();
return sourceUnit;
}
void SourceUnitGenerator::reset()
{
for (auto& g: generators)
std::visit(ResetVisitor{}, g);
return visitChildren();
}
string PragmaGenerator::visit()
@@ -132,7 +123,7 @@ string SolidityGenerator::generateTestProgram()
createGenerators();
for (auto& g: m_generators)
std::visit(AddDependenciesVisitor{}, g);
string program = generator<TestCaseGenerator>()->visit();
string program = generator<TestCaseGenerator>()->generate();
destroyGenerators();
return program;
}