| 57 |
|
#include "selection/SelectionEvaluator.hpp" |
| 58 |
|
#include "selection/SelectionManager.hpp" |
| 59 |
|
#include "visitors/LipidTransVisitor.hpp" |
| 60 |
+ |
#include "visitors/AtomNameVisitor.hpp" |
| 61 |
|
|
| 62 |
|
using namespace oopse; |
| 63 |
|
|
| 115 |
|
|
| 116 |
|
LinearAtomVisitor* linearVisitor = new LinearAtomVisitor(info); |
| 117 |
|
compositeVisitor->addVisitor(linearVisitor, 750); |
| 118 |
+ |
if (args_info.gb_given) { |
| 119 |
+ |
linearVisitor->addGayBerneAtomType(args_info.gb_arg); |
| 120 |
+ |
} |
| 121 |
|
|
| 122 |
+ |
GBLipidAtomVisitor* gbLipidVisitor = new GBLipidAtomVisitor(info); |
| 123 |
+ |
compositeVisitor->addVisitor(gbLipidVisitor, 740); |
| 124 |
+ |
|
| 125 |
|
//create default atom visitor |
| 126 |
|
DefaultAtomVisitor* defaultAtomVisitor = new DefaultAtomVisitor(info); |
| 127 |
|
compositeVisitor->addVisitor(defaultAtomVisitor, 700); |
| 131 |
|
WaterTypeVisitor* waterTypeVisitor = new WaterTypeVisitor; |
| 132 |
|
compositeVisitor->addVisitor(waterTypeVisitor, 600); |
| 133 |
|
} |
| 134 |
+ |
|
| 135 |
+ |
if (args_info.basetype_flag) { |
| 136 |
+ |
AtomNameVisitor* atomNameVisitor = new AtomNameVisitor(info); |
| 137 |
+ |
compositeVisitor->addVisitor(atomNameVisitor, 550); |
| 138 |
+ |
|
| 139 |
+ |
} |
| 140 |
|
|
| 141 |
|
//create ZconsVisitor |
| 142 |
|
if(args_info.zconstraint_flag){ |