Changeset 9b0dcd for src

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.

Location:

src/Actions/PotentialAction

Files:

• GeneratePotentialsAction.cpp (modified) (4 diffs)
• GeneratePotentialsAction.def (modified) (1 diff)

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();

src/Actions/PotentialAction/GeneratePotentialsAction.def

@@ -9 +9 @@
 #include "Parameters/Validators/STLVectorValidator.hpp"
 #include "Parameters/Validators/Specific/ElementValidator.hpp"
+#include "Parameters/Validators/Specific/PotentialTypeValidator.hpp"
+#include <string>
+#include <vector>
 
 // i.e. there is an integer with variable name Z that can be found in
 // ValueStorage by the token "Z" -> first column: int, Z, "Z"
 // "undefine" if no parameters are required, use (NOPARAM_DEFAULT) for each (undefined) default value
-#define paramtypes (std::vector<const element *>)
-#define paramtokens ("fragment-charges")
-#define paramdescriptions ("charges specifying the fragment")
-#define paramdefaults (NOPARAM_DEFAULT)
-#define paramreferences (fragment)
+#define paramtypes (std::vector<const element *>)(std::vector<std::string>)
+#define paramtokens ("fragment-charges")("potential-list")
+#define paramdescriptions ("charges specifying the fragment")("list of potentials to generate or empty for all")
+#define paramdefaults (NOPARAM_DEFAULT)(NOPARAM_DEFAULT)
+#define paramreferences (fragment)(potential_list)
 #define paramvalids \
-(STLVectorValidator< std::vector<const element *> >(0,99, ElementValidator()))
+(STLVectorValidator< std::vector<const element *> >(0,99, ElementValidator())) \
+(STLVectorValidator< std::vector<std::string> >(0,99, PotentialTypeValidator()))
 
 #undef statetypes