Changeset c5324f for src/Actions

Timestamp:

Nov 8, 2012, 1:13:19 PM (12 years ago)

Author:

Frederik Heber <heber@…>

Branches:

Action_Thermostats, Add_AtomRandomPerturbation, Add_FitFragmentPartialChargesAction, Add_RotateAroundBondAction, Add_SelectAtomByNameAction, Added_ParseSaveFragmentResults, AddingActions_SaveParseParticleParameters, Adding_Graph_to_ChangeBondActions, Adding_MD_integration_tests, Adding_ParticleName_to_Atom, Adding_StructOpt_integration_tests, AtomFragments, Automaking_mpqc_open, AutomationFragmentation_failures, Candidate_v1.5.4, Candidate_v1.6.0, Candidate_v1.6.1, ChangeBugEmailaddress, ChangingTestPorts, ChemicalSpaceEvaluator, CombiningParticlePotentialParsing, Combining_Subpackages, Debian_Package_split, Debian_package_split_molecuildergui_only, Disabling_MemDebug, Docu_Python_wait, EmpiricalPotential_contain_HomologyGraph, EmpiricalPotential_contain_HomologyGraph_documentation, Enable_parallel_make_install, Enhance_userguide, Enhanced_StructuralOptimization, Enhanced_StructuralOptimization_continued, Example_ManyWaysToTranslateAtom, Exclude_Hydrogens_annealWithBondGraph, FitPartialCharges_GlobalError, Fix_BoundInBox_CenterInBox_MoleculeActions, Fix_ChargeSampling_PBC, Fix_ChronosMutex, Fix_FitPartialCharges, Fix_FitPotential_needs_atomicnumbers, Fix_ForceAnnealing, Fix_IndependentFragmentGrids, Fix_ParseParticles, Fix_ParseParticles_split_forward_backward_Actions, Fix_PopActions, Fix_QtFragmentList_sorted_selection, Fix_Restrictedkeyset_FragmentMolecule, Fix_StatusMsg, Fix_StepWorldTime_single_argument, Fix_Verbose_Codepatterns, Fix_fitting_potentials, Fixes, ForceAnnealing_goodresults, ForceAnnealing_oldresults, ForceAnnealing_tocheck, ForceAnnealing_with_BondGraph, ForceAnnealing_with_BondGraph_continued, ForceAnnealing_with_BondGraph_continued_betteresults, ForceAnnealing_with_BondGraph_contraction-expansion, FragmentAction_writes_AtomFragments, FragmentMolecule_checks_bonddegrees, GeometryObjects, Gui_Fixes, Gui_displays_atomic_force_velocity, ImplicitCharges, IndependentFragmentGrids, IndependentFragmentGrids_IndividualZeroInstances, IndependentFragmentGrids_IntegrationTest, IndependentFragmentGrids_Sole_NN_Calculation, JobMarket_RobustOnKillsSegFaults, JobMarket_StableWorkerPool, JobMarket_unresolvable_hostname_fix, MoreRobust_FragmentAutomation, ODR_violation_mpqc_open, PartialCharges_OrthogonalSummation, PdbParser_setsAtomName, PythonUI_with_named_parameters, QtGui_reactivate_TimeChanged_changes, Recreated_GuiChecks, Rewrite_FitPartialCharges, RotateToPrincipalAxisSystem_UndoRedo, SaturateAtoms_findBestMatching, SaturateAtoms_singleDegree, StoppableMakroAction, Subpackage_CodePatterns, Subpackage_JobMarket, Subpackage_LinearAlgebra, Subpackage_levmar, Subpackage_mpqc_open, Subpackage_vmg, Switchable_LogView, ThirdParty_MPQC_rebuilt_buildsystem, TrajectoryDependenant_MaxOrder, TremoloParser_IncreasedPrecision, TremoloParser_MultipleTimesteps, TremoloParser_setsAtomName, Ubuntu_1604_changes, stable

Children:

188639

Parents:

1e7dd4

git-author:

Frederik Heber <heber@…> (07/15/12 11:26:51)

git-committer:

Frederik Heber <heber@…> (11/08/12 13:13:19)

Message:

Refactored FragmentationAutomationAction, pulling out bracketed function parts into distinct functions.

File:

• src/Actions/FragmentationAction/FragmentationAutomationAction.cpp (modified) (2 diffs)

src/Actions/FragmentationAction/FragmentationAutomationAction.cpp

@@ -276 +276 @@
 }
 
-Action::state_ptr FragmentationFragmentationAutomationAction::performCall() {
-  boost::asio::io_service io_service;
-  FragmentController controller(io_service);
-
-  // TODO: Have io_service run in second thread and merge with current again eventually
-
-  // Phase One: obtain ids
-  std::vector< boost::filesystem::path > jobfiles = params.jobfiles.get();
-  controller.requestIds(params.host.get(), params.port.get(), jobfiles.size());
-  {
-    io_service.reset();
-    Info info("io_service: Phase One");
-    io_service.run();
-  }
-  // Phase Two: create and add jobs
-  {
-    std::vector<FragmentJob::ptr> jobs;
-    for (std::vector< boost::filesystem::path >::const_iterator iter = jobfiles .begin();
-        iter != jobfiles .end(); ++iter) {
-      const std::string &filename = (*iter).string();
-      if (boost::filesystem::exists(filename)) {
-        const JobId_t next_id = controller.getAvailableId();
-        LOG(1, "INFO: Creating MPQCCommandJob with filename'"
-            +filename+"', and id "+toString(next_id)+".");
-        parsejob(jobs, params.executable.get().string(), filename, next_id);
-      } else {
-        ELOG(1, "Fragment job "+filename+" does not exist.");
-        return Action::failure;
-      }
+
+void RunService(
+    boost::asio::io_service &io_service,
+    std::string message)
+{
+  message = std::string("io_service: ") + message;
+  io_service.reset();
+  Info info(message.c_str());
+  io_service.run();
+}
+
+void requestIds(
+    FragmentController &controller,
+    const FragmentationFragmentationAutomationAction::FragmentationFragmentationAutomationParameters &params,
+    const size_t numberjobs)
+{
+  controller.requestIds(params.host.get(), params.port.get(), numberjobs);
+}
+
+bool createJobsFromFiles(
+    FragmentController &controller,
+    const FragmentationFragmentationAutomationAction::FragmentationFragmentationAutomationParameters &params,
+    const std::vector< boost::filesystem::path > &jobfiles)
+{
+  std::vector<FragmentJob::ptr> jobs;
+  for (std::vector< boost::filesystem::path >::const_iterator iter = jobfiles.begin();
+      iter != jobfiles .end(); ++iter) {
+    const std::string &filename = (*iter).string();
+    if (boost::filesystem::exists(filename)) {
+      const JobId_t next_id = controller.getAvailableId();
+      LOG(1, "INFO: Creating MPQCCommandJob with filename'"
+          +filename+"', and id "+toString(next_id)+".");
+      parsejob(jobs, params.executable.get().string(), filename, next_id);
+    } else {
+      ELOG(1, "Fragment job "+filename+" does not exist.");
+      return false;
     }
-    controller.addJobs(jobs);
-    controller.sendJobs(params.host.get(), params.port.get());
-  }
-  {
-    io_service.reset();
-    Info info("io_service: Phase Two");
-    io_service.run();
-  }
-  // Phase Three: calculate result
+  }
+  controller.addJobs(jobs);
+  controller.sendJobs(params.host.get(), params.port.get());
+  return true;
+}
+
+void WaitforResults(
+    boost::asio::io_service &io_service,
+    FragmentController &controller,
+    const FragmentationFragmentationAutomationAction::FragmentationFragmentationAutomationParameters &params,
+    const size_t NoExpectedResults
+    )
+{
   size_t NoCalculatedResults = 0;
-  while (NoCalculatedResults != jobfiles.size()) {
+  while (NoCalculatedResults != NoExpectedResults) {
     // wait a bit
     boost::asio::deadline_timer timer(io_service);
@@ -323 +334 @@
     // then request status
     controller.checkResults(params.host.get(), params.port.get());
-    {
-      io_service.reset();
-      Info info("io_service: Phase Three");
-      io_service.run();
-    }
+    RunService(io_service, "Checking on results");
+
     const std::pair<size_t, size_t> JobStatus = controller.getJobStatus();
     LOG(1, "INFO: #" << JobStatus.first << " are waiting in the queue and #" << JobStatus.second << " jobs are calculated so far.");
     NoCalculatedResults = JobStatus.second;
   }
+}
+
+
+Action::state_ptr FragmentationFragmentationAutomationAction::performCall() {
+  boost::asio::io_service io_service;
+  FragmentController controller(io_service);
+
+  // TODO: Have io_service run in second thread and merge with current again eventually
+
+  // Phase One: obtain ids
+  std::vector< boost::filesystem::path > jobfiles = params.jobfiles.get();
+  requestIds(controller, params, jobfiles.size());
+  RunService(io_service, "Requesting ids");
+
+  // Phase Two: create and add jobs
+  if (!createJobsFromFiles(controller, params, jobfiles))
+    return Action::failure;
+  RunService(io_service, "Adding jobs");
+
+  // Phase Three: calculate result
+  WaitforResults(io_service, controller, params, jobfiles.size());
+
   // Phase Three: get result
   controller.receiveResults(params.host.get(), params.port.get());
-  {
-    io_service.reset();
-    Info info("io_service: Phase Four");
-    io_service.run();
-  }
+  RunService(io_service, "Phase Four");
+
   // Final phase: print result
   {