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){ |