--- trunk/OOPSE/libmdtools/SimSetup.cpp	2003/03/28 19:33:37	435
+++ trunk/OOPSE/libmdtools/SimSetup.cpp	2003/04/07 14:30:36	466
@@ -66,6 +66,7 @@ void SimSetup::createSim( void ){
 
   MakeStamps *the_stamps;
   Globals* the_globals;
+  ExtendedSystem* the_extendedsystem;
   int i, j;
 
   // get the stamps and globals;
@@ -80,16 +81,33 @@ void SimSetup::createSim( void ){
   // get the ones we know are there, yet still may need some work.
   n_components = the_globals->getNComponents();
   strcpy( force_field, the_globals->getForceField() );
+
+  // get the ensemble and set up an extended system if we need it:
   strcpy( ensemble, the_globals->getEnsemble() );
+  if( !strcasecmp( ensemble, "NPT" ) ) {
+    the_extendedsystem = new ExtendedSystem( simnfo );
+    the_extendedsystem->setTargetTemp(the_globals->getTargetTemp());
+    the_extendedsystem->setTargetPressure(the_globals->getTargetPressure());
+  } else if ( !strcasecmp( ensemble, "NVT") ) {
+    the_extendedsystem = new ExtendedSystem( simnfo );
+    the_extendedsystem->setTargetTemp(the_globals->getTargetTemp());
+  } else if ( !strcasecmp( ensemble, "NVE") ) {
+  } else {
+    sprintf( painCave.errMsg,
+	     "SimSetup Warning. Unrecognized Ensemble -> %s, "
+             "reverting to NVE for this simulation.\n",
+	     ensemble );
+    painCave.isFatal = 0;
+    simError();
+    strcpy( ensemble, "NVE" );
+  }  
   strcpy( simnfo->ensemble, ensemble );
 
   strcpy( simnfo->mixingRule, the_globals->getMixingRule() );
   simnfo->usePBC = the_globals->getPBC();
 	  
-
-
-  if( !strcmp( force_field, "TraPPE_Ex" ) ) the_ff = new TraPPE_ExFF();
-  else if( !strcmp( force_field, "LJ" ) ) the_ff = new LJ_FF();
+  if( !strcmp( force_field, "TraPPE_Ex" ) ) the_ff = new TraPPE_ExFF();
+  else if( !strcasecmp( force_field, "LJ" ) ) the_ff = new LJ_FF();
   else{
     sprintf( painCave.errMsg,
 	     "SimSetup Error. Unrecognized force field -> %s\n",
@@ -314,6 +332,7 @@ void SimSetup::createSim( void ){
     
     if(mol2proc[i] == worldRank ){
       the_molecules[molIndex].setStampID( molCompType[i] );
+      the_molecules[molIndex].setMyIndex( molIndex );
       molIndex++;
     }
   }
@@ -324,6 +343,7 @@ void SimSetup::createSim( void ){
   for(i=0; i<n_components; i++){
     for(j=0; j<components_nmol[i]; j++ ){
       the_molecules[molIndex].setStampID( i );
+      the_molecules[molIndex].setMyIndex( molIndex );
       molIndex++;
     }
   }
@@ -334,13 +354,11 @@ void SimSetup::createSim( void ){
 
   if( simnfo->n_SRI ){
     
-    std::cerr << "n_SRI = " << simnfo->n_SRI << "\n";
-    
     Exclude::createArray(simnfo->n_SRI);
     the_excludes = new Exclude*[simnfo->n_SRI];
     for( int ex=0; ex<simnfo->n_SRI; ex++) the_excludes[ex] = new Exclude(ex);
     simnfo->globalExcludes = new int;
-    simnfo->n_exclude = tot_SRI;
+    simnfo->n_exclude = simnfo->n_SRI;
   }
   else{
     
@@ -655,8 +673,12 @@ void SimSetup::createSim( void ){
 
 //   new AllLong( simnfo );
 
-  if( !strcmp( force_field, "TraPPE_Ex" ) ) new Symplectic( simnfo, the_ff );
-  if( !strcmp( force_field, "LJ" ) ) new Verlet( *simnfo, the_ff );
+  if( !strcmp( force_field, "TraPPE_Ex" ) ) new Symplectic(simnfo, 
+                                                           the_ff, 
+                                                           the_extendedsystem);
+  if( !strcmp( force_field, "LJ" ) ) new Verlet( *simnfo, 
+                                                 the_ff, 
+                                                 the_extendedsystem );
 
 #ifdef IS_MPI
   mpiSim->mpiRefresh();
@@ -932,6 +954,8 @@ void SimSetup::makeMolecules( void ){
 
 
     the_molecules[i].initialize( info );
+
+
     atomOffset += info.nAtoms;
     delete[] theBonds;
     delete[] theBends;