ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/group/trunk/OOPSE_old/src/mdtools/libmdCode/LJ_FF.cpp
(Generate patch)

Comparing trunk/OOPSE_old/src/mdtools/libmdCode/LJ_FF.cpp (file contents):
Revision 275 by mmeineke, Tue Feb 18 21:06:36 2003 UTC vs.
Revision 296 by mmeineke, Thu Mar 6 20:05:39 2003 UTC

# Line 14 | Line 14 | using namespace std;
14   #include "SRI.hpp"
15   #include "simError.h"
16  
17 + #include <fortranWrappers.hpp>
18  
18
19 // Declare the structures that will be passed by the parser and  MPI
20
21 typedef struct{
22  char name[15];
23  double mass;
24  double epslon;
25  double sigma;
26  int ident;
27  int last;      //  0  -> default
28                 //  1  -> in MPI: tells nodes to stop listening
29 } atomStruct;
30
31 int parseAtomLJ( char *lineBuffer, int lineNum, atomStruct &info );
32
19   #ifdef IS_MPI
20   #include "mpiForceField.h"
21 + #endif // is_mpi
22  
36 MPI_Datatype mpiAtomStructType;
23  
38 #endif
24  
25 + namespace {
26  
27 < // declaration of functions needed to wrap the fortran module
27 >  // Declare the structures that will be passed by the parser and  MPI
28 >  
29 >  typedef struct{
30 >    char name[15];
31 >    double mass;
32 >    double epslon;
33 >    double sigma;
34 >    int ident;
35 >    int last;      //  0  -> default
36 >                   //  1  -> in MPI: tells nodes to stop listening
37 >  } atomStruct;
38  
39 < extern "C" {
40 <  void forcefactory_( char* forceName,
41 <                      int* status,
42 <                      void (*wrapFunction)( void (*p1)( int* ident,
43 <                                                        double* mass,
44 <                                                        double* epslon,
45 <                                                        double* sigma,
50 <                                                        int* status ),
51 <                                            void (*p2)( int *nLocal,
52 <                                                        int *identArray,
53 <                                                        int *isError ),
54 <                                            void (*p3)( double* positionArray,
55 <                                                        double* forceArray,
56 <                                                        double* potentialEnergy,
57 <                                                        short int* doPotentialCalc )),
58 <                      int forceNameLength );
39 >  int parseAtom( char *lineBuffer, int lineNum, atomStruct &info );
40 >  
41 > #ifdef IS_MPI
42 >  
43 >  MPI_Datatype mpiAtomStructType;
44 >  
45 > #endif
46   }
47  
48  
62 void LJfunctionWrapper( void (*p1)( int* ident, double* mass, double* epslon,
63                                   double* sigma, int* status ),
64                        void (*p2)( int *nLocal, int *identArray, int *isError ),
65                        void (*p3)( double* positionArray,double* forceArray,
66                                    double* potentialEnergy,
67                                    short int* doPotentialCalc ) );
68
69 void (*newLJtype)( int* ident, double* mass, double* epslon, double* sigma,
70                   int* status );
71
72 void (*initLJfortran) ( int *nLocal, int *identArray, int *isError );
73
74 LJ_FF* currentLJwrap;
75
76
49   //****************************************************************
50   // begins the actual forcefield stuff.  
51   //****************************************************************
# Line 88 | Line 60 | LJ_FF::LJ_FF(){
60    char errMsg[1000];
61  
62    // do the funtion wrapping
63 <  currentLJwrap = this;
92 <  wrapMe();
63 >  wrapMeFF( this );
64  
65   #ifdef IS_MPI
66    int i;
# Line 188 | Line 159 | void LJ_FF::wrapMe( void ){
159   }
160  
161  
191 void LJ_FF::wrapMe( void ){
192  
193  char* currentFF = "LJ";
194  int isError = 0;
195  
196  forcefactory_( currentFF, &isError, LJfunctionWrapper, strlen(currentFF) );
162  
198  if( isError ){
199    
200    sprintf( painCave.errMsg,
201             "LJ_FF error: an error was returned from fortran when the "
202             "the functions were being wrapped.\n" );
203    painCave.isFatal = 1;
204    simError();
205  }
206
207 #ifdef IS_MPI
208  sprintf( checkPointMsg, "LJ_FF functions succesfully wrapped." );
209  MPIcheckPoint();
210 #endif // is_mpi
211 }
212  
213
214 void LJfunctionWrapper( void (*p1)( int* ident, double* mass, double* epslon,
215                                   double* sigma, int* status ),
216                        void (*p2)( int*, int*, int* ),
217                        void (*p3)( double* positionArray,double* forceArray,
218                                    double* potentialEnergy,
219                                    short int* doPotentialCalc ) ){
220  
221  
222  newLJtype = p1;
223  initLJfortran = p2;
224  currentLJwrap->setLJfortran( p3 );
225 }
226
227
228
163   void LJ_FF::initializeAtoms( void ){
164    
165    class LinkedType {
# Line 402 | Line 336 | void LJ_FF::initializeAtoms( void ){
336    // call new A_types in fortran
337    
338    int isError;
339 +
340 +  // dummy variables
341 +  int isLJ = 1;
342 +  int isDipole = 0;
343 +  int isSSD = 0;
344 +  int isGB = 0;
345 +  double w0 = 0.0;
346 +  double v0 = 0.0;
347 +  double dipole = 0.0;
348 +  
349 +  
350    currentAtomType = headAtomType;
351    while( currentAtomType != NULL ){
352      
353      if( currentAtomType->name[0] != '\0' ){
354        isError = 0;
355 <          newLJtype( &(currentAtomType->ident),
356 <                 &(currentAtomType->mass),
357 <                 &(currentAtomType->epslon),
358 <                 &(currentAtomType->sigma),
359 <                 &isError );
355 >      newAtype( &(currentAtomType->ident),
356 >                &(currentAtomType->mass),
357 >                &(currentAtomType->epslon),
358 >                &(currentAtomType->sigma),
359 >                &isLJ, &isSSD, &isDipole, &isGB, &w0, &v0, &dipole,
360 >                &isError );
361        if( isError ){
362          sprintf( painCave.errMsg,
363                   "Error initializing the \"%s\" atom type in fortran\n",
# Line 482 | Line 428 | void LJ_FF::initializeAtoms( void ){
428  
429    initFortran();
430    entry_plug->refreshSim();
431 <
431 >  
432   }
433  
434   void LJ_FF::initializeBonds( bond_pair* the_bonds ){
# Line 527 | Line 473 | void LJ_FF::initializeTorsions( torsion_set* the_torsi
473  
474   }
475  
476 + void LJ_FF::doForces( int calcPot ){
477  
478 +  int i, isError;
479 +  double* frc;
480 +  double* pos;
481 +  double* trq;
482 +  double* tau;
483 +  short int passedCalcPot = (short int)calcPot;
484 +
485 +  // forces are zeroed here, before any are acumulated.
486 +  // NOTE: do not rezero the forces in Fortran.
487 +
488 +  for(i=0; i<entry_plug->n_atoms; i++){
489 +    entry_plug->atoms[i]->zeroForces();
490 +  }
491 +
492 +  frc = Atom::getFrcArray();
493 +  pos = Atom::getPosArray();
494 +  trq = Atom::getTrqArray();
495 +  tau = entry_plug->tau;
496 +
497 +  // dummy variables
498 +  // NOTE: THIS IS A MAJOR POTENTIAL BUG!!!!!! be careful.
499 +  // that said, here we go anyway.
500 +  
501 +  double* A     = NULL;
502 +  double* mu    = NULL;
503 +  double* u_l   = NULL;
504 +    
505 +  isError = 0;
506 +  fortranForceLoop( pos, A, mu, u_l, frc, trq, tau, &(entry_plug->lrPot),
507 +                    &passedCalcPot, &isError );
508 +
509 +
510 +  if( isError ){
511 +    sprintf( painCave.errMsg,
512 +             "Error returned from the fortran force calculation.\n" );
513 +    painCave.isFatal = 1;
514 +    simError();
515 +  }
516 +
517 + #ifdef IS_MPI
518 +  sprintf( checkPointMsg,
519 +           "returned from the force calculation.\n" );
520 +  MPIcheckPoint();
521 + #endif // is_mpi
522 +
523 + }
524 +  
525 + void LJ_FF::initFortran( void ){
526 +  
527 +  int nLocal = entry_plug->n_atoms;
528 +  int *ident;
529 +  int isError;
530 +  int i;
531 +
532 +  ident = new int[nLocal];
533 +
534 +  for(i=0; i<nLocal; i++){
535 +    ident[i] = entry_plug->atoms[i]->getIdent();
536 +  }
537 +
538 +  isError = 0;
539 +  initfortran( &nLocal, ident, &isError );
540 +  
541 +  if(isError){
542 +    sprintf( painCave.errMsg,
543 +             "LJ_FF error: There was an error initializing the component list in fortran.\n" );
544 +    painCave.isFatal = 1;
545 +    simError();
546 +  }
547 +
548 +  
549 + #ifdef IS_MPI
550 +  sprintf( checkPointMsg, "LJ_FF successfully initialized the fortran component list.\n" );
551 +  MPIcheckPoint();
552 + #endif // is_mpi
553 +  
554 +  delete[] ident;
555 +
556 + }
557 +
558 +  
559   void LJ_FF::fastForward( char* stopText, char* searchOwner ){
560  
561    int foundText = 0;
# Line 584 | Line 612 | int parseAtomLJ( char *lineBuffer, int lineNum,  atomS
612  
613  
614  
615 < int parseAtomLJ( char *lineBuffer, int lineNum,  atomStruct &info ){
615 > int parseAtom( char *lineBuffer, int lineNum,  atomStruct &info ){
616  
617    char* the_token;
618    
# Line 624 | Line 652 | int parseAtomLJ( char *lineBuffer, int lineNum,  atomS
652    }
653    else return 0;
654   }
627
628
629 void LJ_FF::doForces( int calcPot ){
630
631  int i;
632  double* frc;
633  double* pos;
634  short int passedCalcPot = (short int)calcPot;
635
636  // forces are zeroed here, before any are acumulated.
637  // NOTE: do not rezero the forces in Fortran.
638
639  for(i=0; i<entry_plug->n_atoms; i++){
640    entry_plug->atoms[i]->zeroForces();
641  }
642
643  frc = Atom::getFrcArray();
644  pos = Atom::getPosArray();
645
646 //   entry_plug->lrPot = -1;
647  doLJfortran( pos, frc, &(entry_plug->lrPot), &passedCalcPot );
648
649
650  //   fprintf( stderr,
651  //   "lrPot =  %lf\n", entry_plug->lrPot );
652  
653 }
654  
655 void LJ_FF::initFortran( void ){
656  
657  int nLocal = entry_plug->n_atoms;
658  int *ident;
659  int isError;
660  int i;
661
662  ident = new int[nLocal];
663
664  for(i=0; i<nLocal; i++){
665    ident[i] = entry_plug->atoms[i]->getIdent();
666  }
667
668  isError = 0;
669  initLJfortran( &nLocal, ident, &isError );
670  
671  if(isError){
672    sprintf( painCave.errMsg,
673             "LJ_FF error: There was an error initializing the component list in fortran.\n" );
674    painCave.isFatal = 1;
675    simError();
676  }
677
678  
679 #ifdef IS_MPI
680  sprintf( checkPointMsg, "LJ_FF successfully initialized the fortran component list.\n" );
681  MPIcheckPoint();
682 #endif // is_mpi
683  
684  delete[] ident;
685
686 }
687  

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines