Skip to content
Snippets Groups Projects
Commit 9af17c2e authored by Wuttke, Joachim's avatar Wuttke, Joachim
Browse files

expand run_and_plot|save to 3 lines

parent 0d99a11d
No related branches found
No related tags found
No related merge requests found
...@@ -480,7 +480,9 @@ std::string SimulationToPython::simulationPlotCode(const ISimulation& simulation ...@@ -480,7 +480,9 @@ std::string SimulationToPython::simulationPlotCode(const ISimulation& simulation
return simulationCode(simulation) return simulationCode(simulation)
+ "if __name__ == '__main__':\n" + "if __name__ == '__main__':\n"
" import ba_plot\n" " import ba_plot\n"
" ba_plot.run_and_plot(get_simulation(get_sample()))\n"; " sample = get_sample()\n"
" simulation = get_simulation(sample)\n"
" ba_plot.run_and_plot(simulation)\n";
} }
std::string SimulationToPython::simulationSaveCode(const ISimulation& simulation, std::string SimulationToPython::simulationSaveCode(const ISimulation& simulation,
...@@ -489,6 +491,7 @@ std::string SimulationToPython::simulationSaveCode(const ISimulation& simulation ...@@ -489,6 +491,7 @@ std::string SimulationToPython::simulationSaveCode(const ISimulation& simulation
return simulationCode(simulation) return simulationCode(simulation)
+ "if __name__ == '__main__':\n" + "if __name__ == '__main__':\n"
" import ba_plot\n" " import ba_plot\n"
" ba_plot.run_and_save(get_simulation(get_sample()), \"" " sample = get_sample()\n"
+ fname + "\")\n"; " simulation = get_simulation(sample)\n"
" ba_plot.run_and_save(simulation, \"" + fname + "\")\n";
} }
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment