applications/randomBuilder/randomBuilder.cpp

/* Copyright (c) 2006 The University of Notre Dame. All Rights Reserved.
 *
 * The University of Notre Dame grants you ("Licensee") a
 * non-exclusive, royalty free, license to use, modify and
 * redistribute this software in source and binary code form, provided
 * that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the
 *    distribution.
 *
 * This software is provided "AS IS," without a warranty of any
 * kind. All express or implied conditions, representations and
 * warranties, including any implied warranty of merchantability,
 * fitness for a particular purpose or non-infringement, are hereby
 * excluded.  The University of Notre Dame and its licensors shall not
 * be liable for any damages suffered by licensee as a result of
 * using, modifying or distributing the software or its
 * derivatives. In no event will the University of Notre Dame or its
 * licensors be liable for any lost revenue, profit or data, or for
 * direct, indirect, special, consequential, incidental or punitive
 * damages, however caused and regardless of the theory of liability,
 * arising out of the use of or inability to use software, even if the
 * University of Notre Dame has been advised of the possibility of
 * such damages.
 *
 * SUPPORT OPEN SCIENCE!  If you use OpenMD or its source code in your
 * research, please cite the appropriate papers when you publish your
 * work.  Good starting points are:
 *                                                                      
 * [1]  Meineke, et al., J. Comp. Chem. 26, 252-271 (2005).             
 * [2]  Fennell & Gezelter, J. Chem. Phys. 124, 234104 (2006).          
 * [3]  Sun, Lin & Gezelter, J. Chem. Phys. 128, 24107 (2008).          
 * [4]  Vardeman & Gezelter, in progress (2009).                        
 *
 *
 *  randomBuilder.cpp
 *
 *  Created by Charles F. Vardeman II on 10 Apr 2006.
 *  @author  Charles F. Vardeman II
 *  @version $Id: randomBuilder.cpp,v 1.8 2009-11-25 20:01:58 gezelter Exp $
 *
 */


#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <string>
#include <map>
#include <fstream>

#include "applications/randomBuilder/randomBuilderCmd.h"
#include "lattice/LatticeFactory.hpp"
#include "utils/MoLocator.hpp"
#include "lattice/Lattice.hpp"
#include "brains/Register.hpp"
#include "brains/SimInfo.hpp"
#include "brains/SimCreator.hpp"
#include "io/DumpWriter.hpp"
#include "math/Vector3.hpp"
#include "math/SquareMatrix3.hpp"
#include "utils/StringUtils.hpp"

using namespace std;
using namespace OpenMD;

void createMdFile(const std::string&oldMdFileName, 
                  const std::string&newMdFileName,
                  std::vector<int> nMol);

int main(int argc, char *argv []) {

  // register force fields
  registerForceFields();
  registerLattice();
    
  gengetopt_args_info args_info;
  std::string latticeType;
  std::string inputFileName;
  std::string outputFileName;
  Lattice *simpleLat;
  RealType latticeConstant;
  std::vector<RealType> lc;
  const RealType rhoConvertConst = 1.661;
  RealType density;
  int nx, ny, nz;
  Mat3x3d hmat;
  MoLocator *locator;
  std::vector<Vector3d> latticePos;
  std::vector<Vector3d> latticeOrt;
  int nMolPerCell;
  DumpWriter *writer;

  // parse command line arguments
  if (cmdline_parser(argc, argv, &args_info) != 0)
    exit(1);

  density = args_info.density_arg;

  //get lattice type
  latticeType = "FCC";

  simpleLat = LatticeFactory::getInstance()->createLattice(latticeType);
    
  if (simpleLat == NULL) {
    sprintf(painCave.errMsg, "Lattice Factory can not create %s lattice\n",
            latticeType.c_str());
    painCave.isFatal = 1;
    simError();
  }
  nMolPerCell = simpleLat->getNumSitesPerCell();

  //get the number of unit cells in each direction:

  nx = args_info.nx_arg;

  if (nx <= 0) {
    sprintf(painCave.errMsg, "The number of unit cells in the x direction "
            "must be greater than 0.");
    painCave.isFatal = 1;
    simError();
  }

  ny = args_info.ny_arg;

  if (ny <= 0) {
    sprintf(painCave.errMsg, "The number of unit cells in the y direction "
            "must be greater than 0.");
    painCave.isFatal = 1;
    simError();
  }

  nz = args_info.nz_arg;

  if (nz <= 0) {
    sprintf(painCave.errMsg, "The number of unit cells in the z direction "
            "must be greater than 0.");
    painCave.isFatal = 1;
    simError();
  }

  int nSites = nMolPerCell * nx * ny * nz;

  //get input file name
  if (args_info.inputs_num)
    inputFileName = args_info.inputs[0];
  else {
    sprintf(painCave.errMsg, "No input .md file name was specified "
            "on the command line");
    painCave.isFatal = 1;
    simError();
  }

  //parse md file and set up the system

  SimCreator oldCreator;
  SimInfo* oldInfo = oldCreator.createSim(inputFileName, false);
  Globals* simParams = oldInfo->getSimParams();

  // Calculate lattice constant (in Angstroms)

  std::vector<Component*> components = simParams->getComponents();
  std::vector<RealType> molFractions;
  std::vector<RealType> molecularMasses;
  std::vector<int> nMol;
  int nComponents = components.size();

  if (nComponents == 1) {
    molFractions.push_back(1.0);    
  } else {
    if (args_info.molFraction_given == nComponents) {
      for (int i = 0; i < nComponents; i++) {
        molFractions.push_back(args_info.molFraction_arg[i]);
      }
    } else if (args_info.molFraction_given == nComponents-1) {
      RealType remainingFraction = 1.0;
      for (int i = 0; i < nComponents-1; i++) {
        molFractions.push_back(args_info.molFraction_arg[i]);
        remainingFraction -= molFractions[i];
      }
      molFractions.push_back(remainingFraction);
    } else {    
      sprintf(painCave.errMsg, "randomBuilder can't figure out molFractions "
              "for all of the components in the <MetaData> block.");
      painCave.isFatal = 1;
      simError();
    }
  }

  // do some sanity checking:
  
  RealType totalFraction = 0.0;

  for (int i = 0; i < nComponents; i++) {
    if (molFractions.at(i) < 0.0) {
      sprintf(painCave.errMsg, "One of the requested molFractions was"
              " less than zero!");
      painCave.isFatal = 1;
      simError();
    }
    if (molFractions.at(i) > 1.0) {
      sprintf(painCave.errMsg, "One of the requested molFractions was"
              " greater than one!");
      painCave.isFatal = 1;
      simError();
    }
    totalFraction += molFractions.at(i);
  }
  if (abs(totalFraction - 1.0) > 1e-6) {
    sprintf(painCave.errMsg, "The sum of molFractions was not close enough to 1.0");
    painCave.isFatal = 1;
    simError();
  }

  int remaining = nSites;
  for (int i=0; i < nComponents-1; i++) {    
    nMol.push_back(int((RealType)nSites * molFractions.at(i)));
    remaining -= nMol.at(i);
  }
  nMol.push_back(remaining);

  // recompute actual mol fractions and perform final sanity check:

  int totalMolecules = 0;
  RealType totalMass = 0.0;
  for (int i=0; i < nComponents; i++) {
    molFractions[i] = (RealType)(nMol.at(i))/(RealType)nSites;
    totalMolecules += nMol.at(i);
    molecularMasses.push_back(getMolMass(oldInfo->getMoleculeStamp(i),
                                         oldInfo->getForceField()));
    totalMass += (RealType)(nMol.at(i)) * molecularMasses.at(i);
  }
  RealType avgMass = totalMass / (RealType) totalMolecules;

  if (totalMolecules != nSites) {
    sprintf(painCave.errMsg, "Computed total number of molecules is not equal "
            "to the number of lattice sites!");
    painCave.isFatal = 1;
    simError();
  }
     
  latticeConstant = pow(rhoConvertConst * nMolPerCell * avgMass / density,
                        (RealType)(1.0 / 3.0));
  
  // Set the lattice constant
  
  lc.push_back(latticeConstant);
  simpleLat->setLatticeConstant(lc);
  
  // Calculate the lattice sites and fill the lattice vector.

  // Get the standard orientations of the cell sites

  latticeOrt = simpleLat->getLatticePointsOrt();

  vector<Vector3d> sites;
  vector<Vector3d> orientations;
  
  for(int i = 0; i < nx; i++) {
    for(int j = 0; j < ny; j++) {
      for(int k = 0; k < nz; k++) {

        // Get the position of the cell sites

        simpleLat->getLatticePointsPos(latticePos, i, j, k);

        for(int l = 0; l < nMolPerCell; l++) {
          sites.push_back(latticePos[l]);
          orientations.push_back(latticeOrt[l]);
        }
      }
    }
  }
  
  outputFileName = args_info.output_arg;

  // create a new .md file on the fly which corrects the number of molecules

  createMdFile(inputFileName, outputFileName, nMol);

  if (oldInfo != NULL)
    delete oldInfo;

  // We need to read in the new SimInfo object, then Parse the 
  // md file and set up the system

  SimCreator newCreator;
  SimInfo* newInfo = newCreator.createSim(outputFileName, false);

  // fill Hmat

  hmat(0, 0) = nx * latticeConstant;
  hmat(0, 1) = 0.0;
  hmat(0, 2) = 0.0;

  hmat(1, 0) = 0.0;
  hmat(1, 1) = ny * latticeConstant;
  hmat(1, 2) = 0.0;

  hmat(2, 0) = 0.0;
  hmat(2, 1) = 0.0;
  hmat(2, 2) = nz * latticeConstant;

  // Set Hmat

  newInfo->getSnapshotManager()->getCurrentSnapshot()->setHmat(hmat);

  // place the molecules

  // Randomize a vector of ints:

  vector<int> ids;
  for (int i = 0; i < sites.size(); i++) ids.push_back(i);
  std::random_shuffle(ids.begin(), ids.end());

  Molecule* mol;
  int l = 0;
  for (int i = 0; i < nComponents; i++){
    locator = new MoLocator(newInfo->getMoleculeStamp(i), 
                            newInfo->getForceField());
    for (int n = 0; n < nMol.at(i); n++) {
      mol = newInfo->getMoleculeByGlobalIndex(l);
      locator->placeMol(sites[ids[l]], orientations[ids[l]], mol);
      l++;
    }
  }
  
  // Create DumpWriter and write out the coordinates

  writer = new DumpWriter(newInfo, outputFileName);

  if (writer == NULL) {
    sprintf(painCave.errMsg, "error in creating DumpWriter");
    painCave.isFatal = 1;
    simError();
  }

  writer->writeDump();

  // deleting the writer will put the closing at the end of the dump file.

  delete writer;

  sprintf(painCave.errMsg, "A new OpenMD file called \"%s\" has been "
          "generated.\n", outputFileName.c_str());
  painCave.isFatal = 0;
  simError();
  return 0;
}

void createMdFile(const std::string&oldMdFileName, 
                  const std::string&newMdFileName, 
                  std::vector<int> nMol) {
  ifstream oldMdFile;
  ofstream newMdFile;
  const int MAXLEN = 65535;
  char buffer[MAXLEN];
  
  //create new .md file based on old .md file

  oldMdFile.open(oldMdFileName.c_str());
  newMdFile.open(newMdFileName.c_str());
  
  oldMdFile.getline(buffer, MAXLEN);
 
  int i = 0;
  while (!oldMdFile.eof()) {
    
    //correct molecule number
    if (strstr(buffer, "nMol") != NULL) {
      if(i<nMol.size()){
        sprintf(buffer, "\tnMol = %i;", nMol.at(i));
        newMdFile << buffer << std::endl;
        i++;
      }
    } else
      newMdFile << buffer << std::endl;
    
    oldMdFile.getline(buffer, MAXLEN);
  }
  
  oldMdFile.close();
  newMdFile.close();

  if (i != nMol.size()) {
    sprintf(painCave.errMsg, "Couldn't replace the correct number of nMol\n"
            "\tstatements in component blocks.  Make sure that all\n"
            "\tcomponents in the template file have nMol=1");
    painCave.isFatal = 1;
    simError();
  }

}

Revision:	1390
Committed:	Wed Nov 25 20:02:06 2009 UTC (15 years, 11 months ago) by gezelter
File size:	11610 byte(s)
Log Message:	Almost all of the changes necessary to create OpenMD out of our old project (OOPSE-4)
#	User	Rev	Content
1	chuckv	950	/* Copyright (c) 2006 The University of Notre Dame. All Rights Reserved.
2			*
3			* The University of Notre Dame grants you ("Licensee") a
4			* non-exclusive, royalty free, license to use, modify and
5			* redistribute this software in source and binary code form, provided
6			* that the following conditions are met:
7			*
8	gezelter	1390	* 1. Redistributions of source code must retain the above copyright
9	chuckv	950	* notice, this list of conditions and the following disclaimer.
10			*
11	gezelter	1390	* 2. Redistributions in binary form must reproduce the above copyright
12	chuckv	950	* notice, this list of conditions and the following disclaimer in the
13			* documentation and/or other materials provided with the
14			* distribution.
15			*
16			* This software is provided "AS IS," without a warranty of any
17			* kind. All express or implied conditions, representations and
18			* warranties, including any implied warranty of merchantability,
19			* fitness for a particular purpose or non-infringement, are hereby
20			* excluded. The University of Notre Dame and its licensors shall not
21			* be liable for any damages suffered by licensee as a result of
22			* using, modifying or distributing the software or its
23			* derivatives. In no event will the University of Notre Dame or its
24			* licensors be liable for any lost revenue, profit or data, or for
25			* direct, indirect, special, consequential, incidental or punitive
26			* damages, however caused and regardless of the theory of liability,
27			* arising out of the use of or inability to use software, even if the
28			* University of Notre Dame has been advised of the possibility of
29			* such damages.
30			*
31	gezelter	1390	* SUPPORT OPEN SCIENCE! If you use OpenMD or its source code in your
32			* research, please cite the appropriate papers when you publish your
33			* work. Good starting points are:
34			*
35			* [1] Meineke, et al., J. Comp. Chem. 26, 252-271 (2005).
36			* [2] Fennell & Gezelter, J. Chem. Phys. 124, 234104 (2006).
37			* [3] Sun, Lin & Gezelter, J. Chem. Phys. 128, 24107 (2008).
38			* [4] Vardeman & Gezelter, in progress (2009).
39	chuckv	950	*
40	gezelter	1390	*
41	chuckv	950	* randomBuilder.cpp
42			*
43			* Created by Charles F. Vardeman II on 10 Apr 2006.
44			* @author Charles F. Vardeman II
45	gezelter	1390	* @version $Id: randomBuilder.cpp,v 1.8 2009-11-25 20:01:58 gezelter Exp $
46	chuckv	950	*
47			*/
48
49
50			#include <cstdlib>
51			#include <cstdio>
52			#include <cstring>
53			#include <cmath>
54			#include <iostream>
55			#include <string>
56			#include <map>
57			#include <fstream>
58
59			#include "applications/randomBuilder/randomBuilderCmd.h"
60			#include "lattice/LatticeFactory.hpp"
61			#include "utils/MoLocator.hpp"
62			#include "lattice/Lattice.hpp"
63			#include "brains/Register.hpp"
64			#include "brains/SimInfo.hpp"
65			#include "brains/SimCreator.hpp"
66			#include "io/DumpWriter.hpp"
67			#include "math/Vector3.hpp"
68			#include "math/SquareMatrix3.hpp"
69			#include "utils/StringUtils.hpp"
70
71			using namespace std;
72	gezelter	1390	using namespace OpenMD;
73	chuckv	950
74			void createMdFile(const std::string&oldMdFileName,
75			const std::string&newMdFileName,
76	gezelter	1065	std::vector<int> nMol);
77	chuckv	950
78			int main(int argc, char *argv []) {
79
80	gezelter	1062	// register force fields
81	chuckv	950	registerForceFields();
82			registerLattice();
83
84			gengetopt_args_info args_info;
85			std::string latticeType;
86			std::string inputFileName;
87	gezelter	1062	std::string outputFileName;
88	chuckv	950	Lattice *simpleLat;
89	tim	963	RealType latticeConstant;
90			std::vector<RealType> lc;
91			const RealType rhoConvertConst = 1.661;
92			RealType density;
93	gezelter	1062	int nx, ny, nz;
94	chuckv	950	Mat3x3d hmat;
95			MoLocator *locator;
96			std::vector<Vector3d> latticePos;
97			std::vector<Vector3d> latticeOrt;
98	gezelter	1065	int nMolPerCell;
99	chuckv	950	DumpWriter *writer;
100
101			// parse command line arguments
102			if (cmdline_parser(argc, argv, &args_info) != 0)
103			exit(1);
104
105			density = args_info.density_arg;
106
107			//get lattice type
108	gezelter	1067	latticeType = "FCC";
109	chuckv	950
110			simpleLat = LatticeFactory::getInstance()->createLattice(latticeType);
111
112			if (simpleLat == NULL) {
113			sprintf(painCave.errMsg, "Lattice Factory can not create %s lattice\n",
114			latticeType.c_str());
115			painCave.isFatal = 1;
116			simError();
117			}
118	gezelter	1065	nMolPerCell = simpleLat->getNumSitesPerCell();
119	chuckv	950
120	gezelter	1062	//get the number of unit cells in each direction:
121
122	chuckv	950	nx = args_info.nx_arg;
123
124			if (nx <= 0) {
125	gezelter	1062	sprintf(painCave.errMsg, "The number of unit cells in the x direction "
126			"must be greater than 0.");
127			painCave.isFatal = 1;
128			simError();
129	chuckv	950	}
130
131			ny = args_info.ny_arg;
132
133			if (ny <= 0) {
134	gezelter	1062	sprintf(painCave.errMsg, "The number of unit cells in the y direction "
135			"must be greater than 0.");
136			painCave.isFatal = 1;
137			simError();
138	chuckv	950	}
139
140			nz = args_info.nz_arg;
141
142			if (nz <= 0) {
143	gezelter	1062	sprintf(painCave.errMsg, "The number of unit cells in the z direction "
144			"must be greater than 0.");
145			painCave.isFatal = 1;
146			simError();
147	chuckv	950	}
148
149	gezelter	1065	int nSites = nMolPerCell * nx * ny * nz;
150
151	chuckv	950	//get input file name
152			if (args_info.inputs_num)
153			inputFileName = args_info.inputs[0];
154			else {
155	gezelter	1062	sprintf(painCave.errMsg, "No input .md file name was specified "
156			"on the command line");
157			painCave.isFatal = 1;
158			simError();
159	chuckv	950	}
160
161			//parse md file and set up the system
162	gezelter	1062
163	chuckv	950	SimCreator oldCreator;
164			SimInfo* oldInfo = oldCreator.createSim(inputFileName, false);
165			Globals* simParams = oldInfo->getSimParams();
166
167	gezelter	1065	// Calculate lattice constant (in Angstroms)
168
169			std::vector<Component*> components = simParams->getComponents();
170			std::vector<RealType> molFractions;
171			std::vector<RealType> molecularMasses;
172			std::vector<int> nMol;
173			int nComponents = components.size();
174
175			if (nComponents == 1) {
176			molFractions.push_back(1.0);
177			} else {
178			if (args_info.molFraction_given == nComponents) {
179			for (int i = 0; i < nComponents; i++) {
180			molFractions.push_back(args_info.molFraction_arg[i]);
181			}
182			} else if (args_info.molFraction_given == nComponents-1) {
183			RealType remainingFraction = 1.0;
184			for (int i = 0; i < nComponents-1; i++) {
185			molFractions.push_back(args_info.molFraction_arg[i]);
186			remainingFraction -= molFractions[i];
187			}
188			molFractions.push_back(remainingFraction);
189			} else {
190			sprintf(painCave.errMsg, "randomBuilder can't figure out molFractions "
191			"for all of the components in the <MetaData> block.");
192			painCave.isFatal = 1;
193			simError();
194			}
195			}
196
197			// do some sanity checking:
198
199			RealType totalFraction = 0.0;
200
201			for (int i = 0; i < nComponents; i++) {
202			if (molFractions.at(i) < 0.0) {
203			sprintf(painCave.errMsg, "One of the requested molFractions was"
204			" less than zero!");
205			painCave.isFatal = 1;
206			simError();
207			}
208			if (molFractions.at(i) > 1.0) {
209			sprintf(painCave.errMsg, "One of the requested molFractions was"
210			" greater than one!");
211			painCave.isFatal = 1;
212			simError();
213			}
214			totalFraction += molFractions.at(i);
215			}
216			if (abs(totalFraction - 1.0) > 1e-6) {
217			sprintf(painCave.errMsg, "The sum of molFractions was not close enough to 1.0");
218	gezelter	1062	painCave.isFatal = 1;
219			simError();
220	chuckv	950	}
221
222	gezelter	1065	int remaining = nSites;
223			for (int i=0; i < nComponents-1; i++) {
224			nMol.push_back(int((RealType)nSites * molFractions.at(i)));
225			remaining -= nMol.at(i);
226			}
227			nMol.push_back(remaining);
228	chuckv	950
229	gezelter	1065	// recompute actual mol fractions and perform final sanity check:
230	chuckv	950
231	gezelter	1065	int totalMolecules = 0;
232			RealType totalMass = 0.0;
233			for (int i=0; i < nComponents; i++) {
234			molFractions[i] = (RealType)(nMol.at(i))/(RealType)nSites;
235			totalMolecules += nMol.at(i);
236			molecularMasses.push_back(getMolMass(oldInfo->getMoleculeStamp(i),
237			oldInfo->getForceField()));
238			totalMass += (RealType)(nMol.at(i)) * molecularMasses.at(i);
239			}
240			RealType avgMass = totalMass / (RealType) totalMolecules;
241	gezelter	1062
242	gezelter	1065	if (totalMolecules != nSites) {
243			sprintf(painCave.errMsg, "Computed total number of molecules is not equal "
244			"to the number of lattice sites!");
245	gezelter	1062	painCave.isFatal = 1;
246			simError();
247	chuckv	950	}
248	gezelter	1065
249			latticeConstant = pow(rhoConvertConst * nMolPerCell * avgMass / density,
250	tim	963	(RealType)(1.0 / 3.0));
251	gezelter	1065
252	gezelter	1062	// Set the lattice constant
253	gezelter	1065
254	chuckv	950	lc.push_back(latticeConstant);
255			simpleLat->setLatticeConstant(lc);
256	gezelter	1065
257	gezelter	1062	// Calculate the lattice sites and fill the lattice vector.
258
259			// Get the standard orientations of the cell sites
260
261			latticeOrt = simpleLat->getLatticePointsOrt();
262
263			vector<Vector3d> sites;
264			vector<Vector3d> orientations;
265	chuckv	950
266			for(int i = 0; i < nx; i++) {
267			for(int j = 0; j < ny; j++) {
268			for(int k = 0; k < nz; k++) {
269
270	gezelter	1062	// Get the position of the cell sites
271
272	chuckv	950	simpleLat->getLatticePointsPos(latticePos, i, j, k);
273
274	gezelter	1065	for(int l = 0; l < nMolPerCell; l++) {
275	gezelter	1062	sites.push_back(latticePos[l]);
276			orientations.push_back(latticeOrt[l]);
277	chuckv	950	}
278			}
279			}
280			}
281
282	gezelter	1062	outputFileName = args_info.output_arg;
283	chuckv	950
284	gezelter	1062	// create a new .md file on the fly which corrects the number of molecules
285	chuckv	950
286	gezelter	1065	createMdFile(inputFileName, outputFileName, nMol);
287	chuckv	950
288	gezelter	1062	if (oldInfo != NULL)
289			delete oldInfo;
290
291			// We need to read in the new SimInfo object, then Parse the
292			// md file and set up the system
293
294			SimCreator newCreator;
295			SimInfo* newInfo = newCreator.createSim(outputFileName, false);
296
297			// fill Hmat
298
299			hmat(0, 0) = nx * latticeConstant;
300	chuckv	950	hmat(0, 1) = 0.0;
301			hmat(0, 2) = 0.0;
302
303			hmat(1, 0) = 0.0;
304			hmat(1, 1) = ny * latticeConstant;
305			hmat(1, 2) = 0.0;
306
307			hmat(2, 0) = 0.0;
308			hmat(2, 1) = 0.0;
309			hmat(2, 2) = nz * latticeConstant;
310
311	gezelter	1062	// Set Hmat
312	chuckv	950
313	gezelter	1062	newInfo->getSnapshotManager()->getCurrentSnapshot()->setHmat(hmat);
314	chuckv	950
315	gezelter	1062	// place the molecules
316
317			// Randomize a vector of ints:
318	chuckv	950
319	gezelter	1062	vector<int> ids;
320			for (int i = 0; i < sites.size(); i++) ids.push_back(i);
321			std::random_shuffle(ids.begin(), ids.end());
322	chuckv	950
323			Molecule* mol;
324			int l = 0;
325	gezelter	1062	for (int i = 0; i < nComponents; i++){
326			locator = new MoLocator(newInfo->getMoleculeStamp(i),
327			newInfo->getForceField());
328	gezelter	1065	for (int n = 0; n < nMol.at(i); n++) {
329	gezelter	1062	mol = newInfo->getMoleculeByGlobalIndex(l);
330			locator->placeMol(sites[ids[l]], orientations[ids[l]], mol);
331			l++;
332			}
333	chuckv	950	}
334	gezelter	1062
335			// Create DumpWriter and write out the coordinates
336	chuckv	950
337	gezelter	1062	writer = new DumpWriter(newInfo, outputFileName);
338	chuckv	950
339			if (writer == NULL) {
340	gezelter	1062	sprintf(painCave.errMsg, "error in creating DumpWriter");
341			painCave.isFatal = 1;
342			simError();
343	chuckv	950	}
344
345	gezelter	1062	writer->writeDump();
346
347			// deleting the writer will put the closing at the end of the dump file.
348
349			delete writer;
350
351	gezelter	1390	sprintf(painCave.errMsg, "A new OpenMD file called \"%s\" has been "
352	gezelter	1065	"generated.\n", outputFileName.c_str());
353	gezelter	1062	painCave.isFatal = 0;
354			simError();
355	chuckv	950	return 0;
356			}
357
358	gezelter	1062	void createMdFile(const std::string&oldMdFileName,
359			const std::string&newMdFileName,
360	gezelter	1065	std::vector<int> nMol) {
361	chuckv	950	ifstream oldMdFile;
362			ofstream newMdFile;
363			const int MAXLEN = 65535;
364			char buffer[MAXLEN];
365
366			//create new .md file based on old .md file
367	gezelter	1062
368	chuckv	950	oldMdFile.open(oldMdFileName.c_str());
369			newMdFile.open(newMdFileName.c_str());
370
371			oldMdFile.getline(buffer, MAXLEN);
372
373			int i = 0;
374			while (!oldMdFile.eof()) {
375
376			//correct molecule number
377			if (strstr(buffer, "nMol") != NULL) {
378	gezelter	1065	if(i<nMol.size()){
379			sprintf(buffer, "\tnMol = %i;", nMol.at(i));
380	chuckv	950	newMdFile << buffer << std::endl;
381			i++;
382			}
383			} else
384			newMdFile << buffer << std::endl;
385
386			oldMdFile.getline(buffer, MAXLEN);
387			}
388
389			oldMdFile.close();
390			newMdFile.close();
391	gezelter	1077
392			if (i != nMol.size()) {
393			sprintf(painCave.errMsg, "Couldn't replace the correct number of nMol\n"
394			"\tstatements in component blocks. Make sure that all\n"
395			"\tcomponents in the template file have nMol=1");
396			painCave.isFatal = 1;
397			simError();
398			}
399
400	chuckv	950	}
401