ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/OpenMD/branches/devel_omp/src/nonbonded/InteractionManager.cpp
(Generate patch)

Comparing:
branches/development/src/nonbonded/InteractionManager.cpp (file contents), Revision 1502 by gezelter, Sat Oct 2 19:53:32 2010 UTC vs.
branches/devel_omp/src/nonbonded/InteractionManager.cpp (file contents), Revision 1614 by mciznick, Tue Aug 23 20:55:51 2011 UTC

# Line 42 | Line 42 | namespace OpenMD {
42   #include "nonbonded/InteractionManager.hpp"
43  
44   namespace OpenMD {
45
46  bool InteractionManager::initialized_ = false;
47  ForceField* InteractionManager::forceField_ = NULL;  
48  InteractionManager* InteractionManager::_instance = NULL;
49  map<int, AtomType*> InteractionManager::typeMap_;
50  map<pair<AtomType*, AtomType*>, set<NonBondedInteraction*> > InteractionManager::interactions_;
51
52  InteractionManager* InteractionManager::Instance() {
53    if (!_instance) {
54      _instance = new InteractionManager();
55    }
56    return _instance;
57  }
45  
46 <  void InteractionManager::initialize() {
47 <    
46 >  InteractionManager::InteractionManager() {
47 >
48 >    initialized_ = false;
49 >        
50      lj_ = new LJ();
51      gb_ = new GB();
52      sticky_ = new Sticky();
53 +    morse_ = new Morse();
54      eam_ = new EAM();
55      sc_ = new SC();
66    morse_ = new Morse();
56      electrostatic_ = new Electrostatic();
57 +    maw_ = new MAW();
58 +  }
59  
60 +  void InteractionManager::initialize() {
61 +    
62 +    ForceField* forceField_ = info_->getForceField();
63 +    
64      lj_->setForceField(forceField_);
65      gb_->setForceField(forceField_);
66      sticky_->setForceField(forceField_);
67      eam_->setForceField(forceField_);
68      sc_->setForceField(forceField_);
69      morse_->setForceField(forceField_);
70 +    electrostatic_->setSimInfo(info_);
71      electrostatic_->setForceField(forceField_);
72 +    maw_->setForceField(forceField_);
73  
74      ForceField::AtomTypeContainer* atomTypes = forceField_->getAtomTypes();
75      ForceField::AtomTypeContainer::MapTypeIterator i1, i2;
# Line 143 | Line 140 | namespace OpenMD {
140          // look for an explicitly-set non-bonded interaction type using the
141          // two atom types.
142          NonBondedInteractionType* nbiType = forceField_->getNonBondedInteractionType(atype1->getName(), atype2->getName());
143 +        
144 +        if (nbiType != NULL) {
145  
146 <        if (nbiType->isLennardJones()) {
147 <          // We found an explicit Lennard-Jones interaction.  
148 <          // override all other vdw entries for this pair of atom types:
149 <          set<NonBondedInteraction*>::iterator it;
150 <          for (it = interactions_[key].begin(); it != interactions_[key].end(); ++it) {
151 <            InteractionFamily ifam = (*it)->getFamily();
152 <            if (ifam == VANDERWAALS_FAMILY) interactions_[key].erase(*it);
146 >          if (nbiType->isLennardJones()) {
147 >            // We found an explicit Lennard-Jones interaction.  
148 >            // override all other vdw entries for this pair of atom types:
149 >            set<NonBondedInteraction*>::iterator it;
150 >            for (it = interactions_[key].begin();
151 >                 it != interactions_[key].end(); ++it) {
152 >              InteractionFamily ifam = (*it)->getFamily();
153 >              if (ifam == VANDERWAALS_FAMILY) interactions_[key].erase(*it);
154 >            }
155 >            interactions_[key].insert(lj_);
156 >            vdwExplicit = true;
157            }
158 <          interactions_[key].insert(lj_);
159 <          vdwExplicit = true;
160 <        }
161 <
162 <        if (nbiType->isMorse()) {
163 <          if (vdwExplicit) {
164 <            sprintf( painCave.errMsg,
165 <                     "InteractionManager::initialize found more than one explicit\n"
166 <                     "\tvan der Waals interaction for atom types %s - %s\n",
167 <                     atype1->getName().c_str(), atype2->getName().c_str());
168 <            painCave.severity = OPENMD_ERROR;
169 <            painCave.isFatal = 1;
170 <            simError();
158 >          
159 >          if (nbiType->isMorse()) {
160 >            if (vdwExplicit) {
161 >              sprintf( painCave.errMsg,
162 >                       "InteractionManager::initialize found more than one "
163 >                       "explicit \n"
164 >                       "\tvan der Waals interaction for atom types %s - %s\n",
165 >                       atype1->getName().c_str(), atype2->getName().c_str());
166 >              painCave.severity = OPENMD_ERROR;
167 >              painCave.isFatal = 1;
168 >              simError();
169 >            }
170 >            // We found an explicit Morse interaction.  
171 >            // override all other vdw entries for this pair of atom types:
172 >            set<NonBondedInteraction*>::iterator it;
173 >            for (it = interactions_[key].begin();
174 >                 it != interactions_[key].end(); ++it) {
175 >              InteractionFamily ifam = (*it)->getFamily();
176 >              if (ifam == VANDERWAALS_FAMILY) interactions_[key].erase(*it);
177 >            }
178 >            interactions_[key].insert(morse_);
179 >            vdwExplicit = true;
180            }
181 <          // We found an explicit Morse interaction.  
182 <          // override all other vdw entries for this pair of atom types:
183 <          set<NonBondedInteraction*>::iterator it;
184 <          for (it = interactions_[key].begin(); it != interactions_[key].end(); ++it) {
185 <            InteractionFamily ifam = (*it)->getFamily();
186 <            if (ifam == VANDERWAALS_FAMILY) interactions_[key].erase(*it);
181 >          
182 >          if (nbiType->isEAM()) {
183 >            // We found an explicit EAM interaction.  
184 >            // override all other metallic entries for this pair of atom types:
185 >            set<NonBondedInteraction*>::iterator it;
186 >            for (it = interactions_[key].begin();
187 >                 it != interactions_[key].end(); ++it) {
188 >              InteractionFamily ifam = (*it)->getFamily();
189 >              if (ifam == METALLIC_FAMILY) interactions_[key].erase(*it);
190 >            }
191 >            interactions_[key].insert(eam_);
192 >            metExplicit = true;
193            }
194 <          interactions_[key].insert(morse_);
195 <          vdwExplicit = true;
196 <        }
197 <
198 <        if (nbiType->isEAM()) {
199 <          // We found an explicit EAM interaction.  
200 <          // override all other metallic entries for this pair of atom types:
201 <          set<NonBondedInteraction*>::iterator it;
202 <          for (it = interactions_[key].begin(); it != interactions_[key].end(); ++it) {
203 <            InteractionFamily ifam = (*it)->getFamily();
204 <            if (ifam == METALLIC_FAMILY) interactions_[key].erase(*it);
205 <          }
206 <          interactions_[key].insert(eam_);
207 <          metExplicit = true;
208 <        }
209 <
210 <        if (nbiType->isSC()) {
211 <          if (metExplicit) {
212 <            sprintf( painCave.errMsg,
213 <                     "InteractionManager::initialize found more than one explicit\n"
214 <                     "\tmetallic interaction for atom types %s - %s\n",
215 <                     atype1->getName().c_str(), atype2->getName().c_str());
198 <            painCave.severity = OPENMD_ERROR;
199 <            painCave.isFatal = 1;
200 <            simError();
201 <          }
202 <          // We found an explicit Sutton-Chen interaction.  
203 <          // override all other metallic entries for this pair of atom types:
204 <          set<NonBondedInteraction*>::iterator it;
205 <          for (it = interactions_[key].begin(); it != interactions_[key].end(); ++it) {
206 <            InteractionFamily ifam = (*it)->getFamily();
207 <            if (ifam == METALLIC_FAMILY) interactions_[key].erase(*it);
194 >          
195 >          if (nbiType->isSC()) {
196 >            if (metExplicit) {
197 >              sprintf( painCave.errMsg,
198 >                       "InteractionManager::initialize found more than one "
199 >                       "explicit\n"
200 >                       "\tmetallic interaction for atom types %s - %s\n",
201 >                       atype1->getName().c_str(), atype2->getName().c_str());
202 >              painCave.severity = OPENMD_ERROR;
203 >              painCave.isFatal = 1;
204 >              simError();
205 >            }
206 >            // We found an explicit Sutton-Chen interaction.  
207 >            // override all other metallic entries for this pair of atom types:
208 >            set<NonBondedInteraction*>::iterator it;
209 >            for (it = interactions_[key].begin();
210 >                 it != interactions_[key].end(); ++it) {
211 >              InteractionFamily ifam = (*it)->getFamily();
212 >              if (ifam == METALLIC_FAMILY) interactions_[key].erase(*it);
213 >            }
214 >            interactions_[key].insert(sc_);
215 >            metExplicit = true;
216            }
217 <          interactions_[key].insert(sc_);
218 <          metExplicit = true;
217 >          
218 >          if (nbiType->isMAW()) {
219 >            if (vdwExplicit) {
220 >              sprintf( painCave.errMsg,
221 >                       "InteractionManager::initialize found more than one "
222 >                       "explicit\n"
223 >                       "\tvan der Waals interaction for atom types %s - %s\n",
224 >                       atype1->getName().c_str(), atype2->getName().c_str());
225 >              painCave.severity = OPENMD_ERROR;
226 >              painCave.isFatal = 1;
227 >              simError();
228 >            }
229 >            // We found an explicit MAW interaction.  
230 >            // override all other vdw entries for this pair of atom types:
231 >            set<NonBondedInteraction*>::iterator it;
232 >            for (it = interactions_[key].begin();
233 >                 it != interactions_[key].end(); ++it) {
234 >              InteractionFamily ifam = (*it)->getFamily();
235 >              if (ifam == VANDERWAALS_FAMILY) interactions_[key].erase(*it);
236 >            }
237 >            interactions_[key].insert(maw_);
238 >            vdwExplicit = true;
239 >          }        
240          }
241        }
242      }
243      
244 <    // make sure every pair of atom types has a non-bonded interaction:
245 <    for (atype1 = atomTypes->beginType(i1); atype1 != NULL;
246 <         atype1 = atomTypes->nextType(i1)) {
247 <      for (atype2 = atomTypes->beginType(i2); atype2 != NULL;
248 <           atype2 = atomTypes->nextType(i2)) {
244 >    
245 >    // Make sure every pair of atom types in this simulation has a
246 >    // non-bonded interaction.  If not, just inform the user.
247 >
248 >    set<AtomType*> simTypes = info_->getSimulatedAtomTypes();
249 >    set<AtomType*>::iterator it, jt;
250 >
251 >    for (it = simTypes.begin(); it != simTypes.end(); ++it) {
252 >      atype1 = (*it);
253 >      for (jt = it; jt != simTypes.end(); ++jt) {
254 >        atype2 = (*jt);
255          key = make_pair(atype1, atype2);
256          
257          if (interactions_[key].size() == 0) {
258            sprintf( painCave.errMsg,
259 <                   "InteractionManager unable to find an appropriate non-bonded\n"
260 <                   "\tinteraction for atom types %s - %s\n",
259 >                   "InteractionManager could not find a matching non-bonded\n"
260 >                   "\tinteraction for atom types %s - %s\n"
261 >                   "\tProceeding without this interaction.\n",
262                     atype1->getName().c_str(), atype2->getName().c_str());
263            painCave.severity = OPENMD_INFO;
264 <          painCave.isFatal = 1;
264 >          painCave.isFatal = 0;
265            simError();
266          }
267        }
268      }
233  }
269  
270 +    initialized_ = true;
271 +  }
272  
273 <  void InteractionManager::doPrePair(AtomType* atype1,
237 <                                     AtomType* atype2,
238 <                                     RealType rij,
239 <                                     RealType &rho_i_at_j,
240 <                                     RealType &rho_j_at_i) {
273 >  void InteractionManager::setCutoffRadius(RealType rcut) {
274      
275 +    electrostatic_->setCutoffRadius(rcut);
276 +    eam_->setCutoffRadius(rcut);
277    }
243  
244  void InteractionManager::doPreForce(AtomType* atype,
245                                      RealType rho,      
246                                      RealType &frho,
247                                      RealType &dfrhodrho) {
248  }
278  
279 <  void InteractionManager::doSkipCorrection(AtomType* atype1,      
280 <                                            AtomType* atype2,
252 <                                            Vector3d d,
253 <                                            RealType rij,
254 <                                            RealType &skippedCharge1,
255 <                                            RealType &skippedCharge2,
256 <                                            RealType sw,
257 <                                            RealType electroMult,
258 <                                            RealType &pot,
259 <                                            RealType &vpair,
260 <                                            Vector3d &f1,
261 <                                            Mat3x3d eFrame1,
262 <                                            Mat3x3d eFrame2,
263 <                                            Vector3d &t1,
264 <                                            Vector3d &t2) {
279 >  void InteractionManager::setSwitchingRadius(RealType rswitch) {
280 >    electrostatic_->setSwitchingRadius(rswitch);
281    }
282    
283 <  void InteractionManager::doSelfCorrection(AtomType* atype,
284 <                                            Mat3x3d eFrame,
285 <                                            RealType skippedCharge,
286 <                                            RealType &pot,
287 <                                            Vector3d &t) {
288 <  }
283 >  void InteractionManager::doPrePair(InteractionData idat){
284 > //      printf("%s\tERROR!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n", __FUNCTION__);
285 >    if (!initialized_) initialize();
286 >        
287 >    // excluded interaction, so just return
288 >    if (idat.excluded) return;
289  
290 <  void InteractionManager::do_prepair(int *atid1, int *atid2, RealType *rij, RealType *rho_i_at_j, RealType *rho_j_at_i){
290 >    set<NonBondedInteraction*>::iterator it;
291 >
292 >    for (it = interactions_[ idat.atypes ].begin();
293 >         it != interactions_[ idat.atypes ].end(); ++it){
294 >      if ((*it)->getFamily() == METALLIC_FAMILY) {
295 >        dynamic_cast<MetallicInteraction*>(*it)->calcDensity(idat);
296 >      }
297 >    }
298      
276    if (!initialized_) initialize();
277    AtomType* atype1 = typeMap_[*atid1];
278    AtomType* atype2 = typeMap_[*atid2];
279    
280    doPrePair(atype1, atype2, *rij, *rho_i_at_j, *rho_j_at_i);
281    
299      return;    
300    }
301 +  
302 +  void InteractionManager::initializeOMP() {
303 +        if (!initialized_) initialize();
304 +  }
305  
306 <  void InteractionManager::do_preforce(int *atid, RealType *rho, RealType *frho, RealType *dfrhodrho){
306 >  void InteractionManager::doPrePairOMP(InteractionDataPrv idatThread){
307 >        InteractionData idat;
308 >    // excluded interaction, so just return
309 >    if (idat.excluded) return;
310  
311 <    if (!initialized_) initialize();    
288 <    AtomType* atype = typeMap_[*atid];
311 >    wrapData(idatThread, idat);
312  
313 <    doPreForce(atype, *rho, *frho, *dfrhodrho);
314 <    
315 <    return;    
313 >    set<NonBondedInteraction*>::iterator it;
314 >
315 >    for (it = interactions_[ idat.atypes ].begin();
316 >         it != interactions_[ idat.atypes ].end(); ++it){
317 >      if ((*it)->getFamily() == METALLIC_FAMILY) {
318 >        dynamic_cast<MetallicInteraction*>(*it)->calcDensity(idat);
319 >      }
320 >    }
321 >
322 >    return;
323    }
324  
325 <  void InteractionManager::do_pair(int *atid1, int *atid2, RealType *d, RealType *r, RealType *r2, RealType *rcut, RealType *sw, RealType *vdwMult,RealType *electroMult, RealType *pot, RealType *vpair, RealType *f1, RealType *eFrame1, RealType *eFrame2, RealType *A1, RealType *A2, RealType *t1, RealType *t2, RealType *rho1, RealType *rho2, RealType *dfrho1, RealType *dfrho2, RealType *fshift1, RealType *fshift2){
325 >  void InteractionManager::doPreForce(SelfData sdat){
326  
327      if (!initialized_) initialize();
328 <          
329 <    InteractionData idat;
328 >    
329 >    pair<AtomType*, AtomType*> key = make_pair(sdat.atype, sdat.atype);
330 >    set<NonBondedInteraction*>::iterator it;
331 >    
332 >    for (it = interactions_[key].begin(); it != interactions_[key].end(); ++it){
333 >      if ((*it)->getFamily() == METALLIC_FAMILY) {
334 >        dynamic_cast<MetallicInteraction*>(*it)->calcFunctional(sdat);
335 >      }
336 >    }
337 >    
338 >    return;    
339 >  }
340  
341 <    idat.atype1 = typeMap_[*atid1];
342 <    idat.atype2 = typeMap_[*atid2];
343 <    idat.d = Vector3d(d);
344 <    idat.rij = *r;
345 <    idat.r2 = *r2;
346 <    idat.rcut = *rcut;
347 <    idat.sw = *sw;
308 <    idat.vdwMult = *vdwMult;
309 <    idat.electroMult = *electroMult;
310 <    idat.pot = *pot;
311 <    idat.vpair = *vpair;
312 <    idat.f1 = Vector3d(f1);
313 <    idat.eFrame1 = Mat3x3d(eFrame1);
314 <    idat.eFrame2 = Mat3x3d(eFrame2);
315 <    idat.A1 = RotMat3x3d(A1);
316 <    idat.A2 = RotMat3x3d(A2);
317 <    idat.t1 = Vector3d(t1);
318 <    idat.t2 = Vector3d(t2);
319 <    idat.rho1 = *rho1;
320 <    idat.rho2 = *rho2;
321 <    idat.dfrho1 = *dfrho1;
322 <    idat.dfrho2 = *dfrho2;
323 <    idat.fshift1 = *fshift1;
324 <    idat.fshift2 = *fshift2;
341 >  void InteractionManager::initNonbondedForces() {
342 >        pair<AtomType*, AtomType*> key;
343 >                set<AtomType*> simTypes = info_->getSimulatedAtomTypes();
344 >                set<AtomType*>::iterator it, jt;
345 >                set<NonBondedInteraction*>::iterator kt;
346 >    AtomType* atype1;
347 >    AtomType* atype2;
348  
349 <    pair<AtomType*, AtomType*> key = make_pair(idat.atype1, idat.atype2);
350 <    set<NonBondedInteraction*>::iterator it;
349 >                for (it = simTypes.begin(); it != simTypes.end(); ++it) {
350 >                        atype1 = (*it);
351 >                        for (jt = it; jt != simTypes.end(); ++jt) {
352 >                                atype2 = (*jt);
353 >                                key = make_pair(atype1, atype2);
354  
355 <    for (it = interactions_[key].begin(); it != interactions_[key].end(); ++it)
356 <      (*it)->calcForce(idat);
331 <    
332 <    f1[0] = idat.f1.x();
333 <    f1[1] = idat.f1.y();
334 <    f1[2] = idat.f1.z();
335 <    
336 <    t1[0] = idat.t1.x();
337 <    t1[1] = idat.t1.y();
338 <    t1[2] = idat.t1.z();
339 <    
340 <    t2[0] = idat.t2.x();
341 <    t2[1] = idat.t2.y();
342 <    t2[2] = idat.t2.z();
355 >                                for (kt = interactions_[ key ].begin();
356 >                                                kt != interactions_[ key ].end(); ++kt) {
357  
358 <    return;    
359 <  }
358 >                                                (*kt)->initForce();
359 >                                }
360 >                        }
361 >                }
362  
363 <  void InteractionManager::do_skip_correction(int *atid1, int *atid2, RealType *d, RealType *r, RealType *skippedCharge1, RealType *skippedCharge2, RealType *sw, RealType *electroMult, RealType *pot, RealType *vpair, RealType *f1, RealType *eFrame1, RealType *eFrame2, RealType *t1, RealType *t2){
363 > /*      lj_->initForce();
364 >        electrostatic_->initForce();
365 >        sticky_->initForce();
366 >        eam_->initForce();
367 >        sc_->initForce();
368 >        gb_->initForce();*/
369  
370 +                return;
371 +  }
372 +
373 +  void InteractionManager::doPair(InteractionData idat){
374 + //      printf("%s\tERROR!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n", __FUNCTION__);
375      if (!initialized_) initialize();
350          
351    AtomType* atype1 = typeMap_[*atid1];
352    AtomType* atype2 = typeMap_[*atid2];
353    Vector3d disp(d);
354    Vector3d frc(f1);
355    Vector3d trq1(t1);
356    Vector3d trq2(t2);
357    Mat3x3d eFi(eFrame1);
358    Mat3x3d eFj(eFrame2);
359      
360    doSkipCorrection(atype1, atype2, disp, *r, *skippedCharge1, *skippedCharge2, *sw,
361                     *electroMult, *pot,  *vpair, frc, eFi, eFj, trq1, trq2);
376  
377 <    f1[0] = frc.x();
378 <    f1[1] = frc.y();
379 <    f1[2] = frc.z();
377 >    set<NonBondedInteraction*>::iterator it;
378 >
379 >    for (it = interactions_[ idat.atypes ].begin();
380 >         it != interactions_[ idat.atypes ].end(); ++it) {
381 >
382 >      // electrostatics still has to worry about indirect
383 >      // contributions from excluded pairs of atoms:
384 >
385 >      if (!idat.excluded || (*it)->getFamily() == ELECTROSTATIC_FAMILY) {
386 >        (*it)->calcForce(idat);
387 >      }
388 >    }
389      
390 <    t1[0] = trq1.x();
391 <    t1[1] = trq1.y();
369 <    t1[2] = trq1.z();
370 <    
371 <    t2[0] = trq2.x();
372 <    t2[1] = trq2.y();
373 <    t2[2] = trq2.z();
390 >    return;    
391 >  }
392  
393 <    return;
393 >  void InteractionManager::wrapData(InteractionDataPrv &src, InteractionData &dst) {
394 >        dst.A1 = src.A1;
395 >        dst.A2 = src.A2;
396 >        dst.atypes = src.atypes;
397 >        dst.d = &src.d;
398 >        dst.dfrho1 = src.dfrho1;
399 >        dst.dfrho2 = src.dfrho2;
400 >        dst.eFrame1 = src.eFrame1;
401 >        dst.eFrame2 = src.eFrame2;
402 >        dst.electroMult = &src.electroMult;
403 >        dst.excluded = src.excluded;
404 >        dst.f1 = &src.f1;
405 >        dst.frho1 = src.frho1;
406 >        dst.frho2 = src.frho2;
407 >        dst.particlePot1 = src.particlePot1;
408 >        dst.particlePot2 = src.particlePot2;
409 >        dst.pot = &src.pot;
410 >        dst.r2 = &src.r2;
411 >        dst.rcut = &src.rcut;
412 >        dst.rho1 = src.rho1;
413 >        dst.rho2 = src.rho2;
414 >        dst.rij = &src.rij;
415 >        dst.shiftedForce = src.shiftedForce;
416 >        dst.shiftedPot = src.shiftedPot;
417 >        dst.skippedCharge1 = src.skippedCharge1;
418 >        dst.skippedCharge2 = src.skippedCharge2;
419 >        dst.sw = &src.sw;
420 >        dst.t1 = src.t1;
421 >        dst.t2 = src.t2;
422 >        dst.topoDist = src.topoDist;
423 >        dst.vdwMult = &src.vdwMult;
424 >        dst.vpair = &src.vpair;
425    }
426  
427 <  void InteractionManager::do_self_correction(int *atid, RealType *eFrame, RealType *skippedCharge, RealType *pot, RealType *t){
427 >  void InteractionManager::doPairOMP(InteractionDataPrv &idatThread){
428 >    set<NonBondedInteraction*>::iterator it;
429 >    InteractionData idat;
430  
431 <    if (!initialized_) initialize();
381 <          
382 <    AtomType* atype = typeMap_[*atid];
383 <    Mat3x3d eFi(eFrame);
384 <    Vector3d trq1(t);
385 <      
386 <    doSelfCorrection(atype, eFi, *skippedCharge, *pot, trq1);
431 >    wrapData(idatThread, idat);
432  
433 <    t[0] = trq1.x();
434 <    t[1] = trq1.y();
390 <    t[2] = trq1.z();
433 >    for (it = interactions_[ idat.atypes ].begin();
434 >         it != interactions_[ idat.atypes ].end(); ++it) {
435  
436 +      // electrostatics still has to worry about indirect
437 +      // contributions from excluded pairs of atoms:
438 +
439 +      if (!idat.excluded || (*it)->getFamily() == ELECTROSTATIC_FAMILY) {
440 +        (*it)->calcForce(idat);
441 +      }
442 +    }
443 +
444      return;
445    }
446  
447 < } //end namespace OpenMD
447 >  void InteractionManager::doSelfCorrection(SelfData sdat){
448  
449 < extern "C" {
450 <  
451 < #define fortranDoPrePair FC_FUNC(do_prepair, DO_PREPAIR)
452 < #define fortranDoPreForce FC_FUNC(do_preforce, DO_PREFORCE)
401 < #define fortranDoPair FC_FUNC(do_pair, DO_PAIR)
402 < #define fortranDoSkipCorrection FC_FUNC(do_skip_correction, DO_SKIP_CORRECTION)
403 < #define fortranDoSelfCorrection FC_FUNC(do_self_correction, DO_SELF_CORRECTION)
404 < #define fortranGetCutoff FC_FUNC(get_cutoff, GET_CUTOFF)
449 >    if (!initialized_) initialize();
450 >    
451 >    pair<AtomType*, AtomType*> key = make_pair(sdat.atype, sdat.atype);
452 >    set<NonBondedInteraction*>::iterator it;
453  
454 <  void fortranDoPrePair(int *atid1, int *atid2, RealType *rij,
455 <                        RealType *rho_i_at_j, RealType *rho_j_at_i) {
456 <            
457 <    return OpenMD::InteractionManager::Instance()->do_prepair(atid1, atid2, rij,
458 <                                                              rho_i_at_j,  
459 <                                                              rho_j_at_i);
454 >    for (it = interactions_[key].begin(); it != interactions_[key].end(); ++it){
455 >      if ((*it)->getFamily() == ELECTROSTATIC_FAMILY) {
456 >        dynamic_cast<ElectrostaticInteraction*>(*it)->calcSelfCorrection(sdat);
457 >      }
458 >    }
459 >      
460 >    return;    
461    }
462 <  void fortranDoPreforce(int *atid, RealType *rho, RealType *frho,
463 <                         RealType *dfrhodrho) {  
462 >
463 >  RealType InteractionManager::getSuggestedCutoffRadius(int *atid) {
464 >    if (!initialized_) initialize();
465      
466 <    return OpenMD::InteractionManager::Instance()->do_preforce(atid, rho, frho,
417 <                                                               dfrhodrho);    
418 <  }
419 <  
420 <  void fortranDoPair(int *atid1, int *atid2, RealType *d, RealType *r,
421 <                     RealType *r2, RealType *rcut, RealType *sw, RealType *vdwMult,
422 <                     RealType *electroMult, RealType *pot, RealType *vpair, RealType *f1,
423 <                     RealType *eFrame1, RealType *eFrame2, RealType *A1, RealType *A2,
424 <                     RealType *t1, RealType *t2,
425 <                     RealType *rho1, RealType *rho2, RealType *dfrho1, RealType *dfrho2,
426 <                     RealType *fshift1, RealType *fshift2){
427 <    
428 <    return OpenMD::InteractionManager::Instance()->do_pair(atid1, atid2, d, r, r2, rcut,
429 <                                                           sw, vdwMult, electroMult, pot,
430 <                                                           vpair, f1, eFrame1, eFrame2,
431 <                                                           A1, A2, t1, t2, rho1, rho2,
432 <                                                           dfrho1, dfrho2, fshift1, fshift2);
433 <  }
466 >    AtomType* atype = typeMap_[*atid];
467  
468 <  void fortranDoSkipCorrection(int *atid1, int *atid2, RealType *d, RealType *r,
469 <                               RealType *skippedCharge1, RealType *skippedCharge2,
470 <                               RealType *sw, RealType *electroMult, RealType *pot,
438 <                               RealType *vpair, RealType *f1,
439 <                               RealType *eFrame1, RealType *eFrame2,
440 <                               RealType *t1, RealType *t2){
468 >    pair<AtomType*, AtomType*> key = make_pair(atype, atype);
469 >    set<NonBondedInteraction*>::iterator it;
470 >    RealType cutoff = 0.0;
471      
472 <    return OpenMD::InteractionManager::Instance()->do_skip_correction(atid1, atid2, d, r,
473 <                                                                      skippedCharge1,
474 <                                                                      skippedCharge2,
445 <                                                                      sw, electroMult, pot,
446 <                                                                      vpair, f1, eFrame1,
447 <                                                                      eFrame2, t1, t2);
472 >    for (it = interactions_[key].begin(); it != interactions_[key].end(); ++it)
473 >      cutoff = max(cutoff, (*it)->getSuggestedCutoffRadius(key));  
474 >    return cutoff;    
475    }
476  
477 <  void fortranDoSelfCorrection(int *atid, RealType *eFrame, RealType *skippedCharge,
478 <                               RealType *pot, RealType *t) {
477 >  RealType InteractionManager::getSuggestedCutoffRadius(AtomType* atype) {
478 >    if (!initialized_) initialize();
479      
480 <    return OpenMD::InteractionManager::Instance()->do_self_correction(atid, eFrame,
481 <                                                                      skippedCharge,
482 <                                                                      pot, t);
480 >    pair<AtomType*, AtomType*> key = make_pair(atype, atype);
481 >    set<NonBondedInteraction*>::iterator it;
482 >    RealType cutoff = 0.0;
483 >    
484 >    for (it = interactions_[key].begin(); it != interactions_[key].end(); ++it)
485 >      cutoff = max(cutoff, (*it)->getSuggestedCutoffRadius(key));  
486 >    return cutoff;    
487    }
488 <  RealType fortranGetCutoff() {
458 <    return OpenMD::InteractionManager::Instance()->getCutoff();
459 <  }
460 < }
488 > } //end namespace OpenMD

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines