Changeset 9b0dcd for src/Actions/PotentialAction/GeneratePotentialsAction.cpp

Timestamp:

May 19, 2021, 7:06:29 PM (4 years ago)

Author:

Frederik Heber <frederik.heber@…>

Branches:

Candidate_v1.7.0, stable

Children:

20fc6f

Parents:

c98620

git-author:

Frederik Heber <frederik.heber@…> (05/19/21 19:03:24)

git-committer:

Frederik Heber <frederik.heber@…> (05/19/21 19:06:29)

Message:

Added potential-list argument to GeneratePotentialsAction.

• DOCU: Extended documentation on new option.
• TEST: Added a test case on this.

File:

• src/Actions/PotentialAction/GeneratePotentialsAction.cpp (modified) (4 diffs)

src/Actions/PotentialAction/GeneratePotentialsAction.cpp

@@ -43 +43 @@
 #include <string>
 
+#include <boost/foreach.hpp>
+
 #include "Actions/PotentialAction/GeneratePotentialsAction.hpp"
 
@@ -80 +82 @@
   }
 
+  // gather list of potential candidates
+  std::vector<std::string> potentials;
+  if (!params.potential_list.isSet()) {
+    for (unsigned int i=0; i<PotentialTypesMax; ++i)
+      potentials.push_back(PotentialFactory::getNameForType((enum PotentialTypes)i));
+  } else
+    potentials = params.potential_list.get();
+
   // go through all potential potentials :)4
   const PotentialFactory& factory = PotentialFactory::getConstInstance();
@@ -86 +96 @@
   typedef std::set<BindingModel> unique_models_t;
   unique_models_t unique_models;
-  for (unsigned int i=0; i<PotentialTypesMax; ++i) {
-    const std::string potential_name = PotentialFactory::getNameForType((enum PotentialTypes)i);
+  BOOST_FOREACH(std::string &potential_name, potentials) {
     unique_models.clear();
 
@@ -102 +111 @@
 
     // first need to construct potential, then may access it
-    EmpiricalPotential const * const defaultPotential = factory.getDefaultPotential((enum PotentialTypes)i);
+    const enum PotentialTypes potential_type = factory.getTypeForName(potential_name);
+    EmpiricalPotential const * const defaultPotential = factory.getDefaultPotential(potential_type);
     /// 1. get its number of particles
     const unsigned int num_particles = defaultPotential->getParticleTypeNumber();