ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/OpenMD/trunk/src/applications/nanoparticleBuilder/nanoparticleBuilder.cpp
(Generate patch)

Comparing trunk/src/applications/nanoparticleBuilder/nanoparticleBuilder.cpp (file contents):
Revision 1032 by gezelter, Fri Sep 1 19:16:02 2006 UTC vs.
Revision 1793 by gezelter, Fri Aug 31 21:16:10 2012 UTC

# Line 6 | Line 6
6   * redistribute this software in source and binary code form, provided
7   * that the following conditions are met:
8   *
9 < * 1. Acknowledgement of the program authors must be made in any
10 < *    publication of scientific results based in part on use of the
11 < *    program.  An acceptable form of acknowledgement is citation of
12 < *    the article in which the program was described (Matthew
13 < *    A. Meineke, Charles F. Vardeman II, Teng Lin, Christopher
14 < *    J. Fennell and J. Daniel Gezelter, "OOPSE: An Object-Oriented
15 < *    Parallel Simulation Engine for Molecular Dynamics,"
16 < *    J. Comput. Chem. 26, pp. 252-271 (2005))
17 < *
18 < * 2. Redistributions of source code must retain the above copyright
9 > * 1. Redistributions of source code must retain the above copyright
10   *    notice, this list of conditions and the following disclaimer.
11   *
12 < * 3. Redistributions in binary form must reproduce the above copyright
12 > * 2. Redistributions in binary form must reproduce the above copyright
13   *    notice, this list of conditions and the following disclaimer in the
14   *    documentation and/or other materials provided with the
15   *    distribution.
# Line 37 | Line 28
28   * arising out of the use of or inability to use software, even if the
29   * University of Notre Dame has been advised of the possibility of
30   * such damages.
31 + *
32 + * SUPPORT OPEN SCIENCE!  If you use OpenMD or its source code in your
33 + * research, please cite the appropriate papers when you publish your
34 + * work.  Good starting points are:
35 + *                                                                      
36 + * [1]  Meineke, et al., J. Comp. Chem. 26, 252-271 (2005).            
37 + * [2]  Fennell & Gezelter, J. Chem. Phys. 124, 234104 (2006).          
38 + * [3]  Sun, Lin & Gezelter, J. Chem. Phys. 128, 24107 (2008).          
39 + * [4]  Kuang & Gezelter,  J. Chem. Phys. 133, 164101 (2010).
40 + * [5]  Vardeman, Stocker & Gezelter, J. Chem. Theory Comput. 7, 834 (2011).
41   */
42  
43   #include <cstdlib>
# Line 64 | Line 65 | using namespace std;
65   #include "utils/StringUtils.hpp"
66  
67   using namespace std;
68 < using namespace oopse;
68 > using namespace OpenMD;
69   void createMdFile(const std::string&oldMdFileName,
70                    const std::string&newMdFileName,
71 <                  int components,int* numMol);
71 >                  std::vector<int> numMol);
72  
73   int main(int argc, char *argv []) {
74    
74  //register force fields
75  registerForceFields();
75    registerLattice();
76    
77    gengetopt_args_info args_info;
78    std::string latticeType;
79    std::string inputFileName;
80 <  std::string outPrefix;
82 <  std::string outMdFileName;
83 <  std::string outInitFileName;
80 >  std::string outputFileName;
81  
85  
86  
87  Lattice *simpleLat;
82    MoLocator* locator;
89  int* numMol;
83    int nComponents;
84    double latticeConstant;
85    std::vector<double> lc;
93  double mass;                                                                                      
94  const double rhoConvertConst = 1.661;
95  double density;
96  double particleRadius;
97  
98  
86  
87 +  RealType particleRadius;
88 +
89    Mat3x3d hmat;
90    std::vector<Vector3d> latticePos;
91    std::vector<Vector3d> latticeOrt;
92 <  int numMolPerCell;
104 <  int nShells; /* Number of shells in nanoparticle*/
105 <  int numSites;
106 <  
92 >
93    DumpWriter *writer;
94    
95    // Parse Command Line Arguments
96    if (cmdline_parser(argc, argv, &args_info) != 0)
97      exit(1);
98 <  
113 <        
114 <        
98 >        
99    /* get lattice type */
100 <  latticeType = UpperCase(args_info.latticetype_arg);
101 <    
100 >  latticeType = "FCC";
101 >
102    /* get input file name */
103    if (args_info.inputs_num)
104      inputFileName = args_info.inputs[0];
105    else {
106 <    std::cerr << "You must specify a input file name.\n" << std::endl;
106 >    sprintf(painCave.errMsg, "No input .md file name was specified "
107 >            "on the command line");
108 >    painCave.isFatal = 1;
109      cmdline_parser_print_help();
110 <    exit(1);
110 >    simError();
111    }
112    
113    /* parse md file and set up the system */
114    SimCreator oldCreator;
115    SimInfo* oldInfo = oldCreator.createSim(inputFileName, false);
116    
117 <  
132 <  /*calculate lattice constant (in Angstrom)
133 <    latticeConstant = pow(rhoConvertConst * numMolPerCell * mass / density,
134 <    1.0 / 3.0);*/
135 <  
136 <  latticeConstant = args_info.latticeCnst_arg;
117 >  latticeConstant = args_info.latticeConstant_arg;
118    particleRadius = args_info.radius_arg;
119    Globals* simParams = oldInfo->getSimParams();
120    
140  /* Find out how many different components in this simualtion */
141  nComponents =simParams->getNComponents();
142  
143  /*determine the output file names*/
144  if (args_info.output_given){
145    outInitFileName = args_info.output_arg;
146  }else{
147    outInitFileName = getPrefix(inputFileName.c_str()) + ".in";
148  }
149  
150  std::cout <<"Before build shaped lattice. "<<std::endl;
151  
152  /* create Molocators */
153  locator = new MoLocator(oldInfo->getMoleculeStamp(0), oldInfo->getForceField());
154  
121    /* Create nanoparticle */
122 <  shapedLatticeSpherical nanoParticle(latticeConstant,latticeType,particleRadius);
122 >  shapedLatticeSpherical nanoParticle(latticeConstant, latticeType,
123 >                                      particleRadius);
124    
158  std::cout <<"Before build getPoints. "<<std::endl;
125    /* Build a lattice and get lattice points for this lattice constant */
126 <  vector<Vector3d> nanoParticleSites = nanoParticle.getPoints();
126 >  vector<Vector3d> sites = nanoParticle.getSites();
127 >  vector<Vector3d> orientations = nanoParticle.getOrientations();
128 >
129 >
130 >  std::vector<int> vacancyTargets;
131 >  vector<bool> isVacancy;
132    
133 <  /* Get number of lattice sites */
134 <  numSites = nanoParticleSites.size();
133 >  Vector3d myLoc;
134 >  RealType myR;
135  
136 <  //std::cout <<"numSites are %d "<<numSites<<std::endl;
137 <  // std::cout <<"nComponents are %d "<<nComponents<<std::endl;
138 <  numMol = new int[nComponents];
139 <
136 >  for (int i = 0; i < sites.size(); i++)
137 >    isVacancy.push_back(false);
138 >
139 >  if (args_info.vacancyPercent_given) {
140 >    if (args_info.vacancyPercent_arg < 0.0 || args_info.vacancyPercent_arg > 100.0) {
141 >      sprintf(painCave.errMsg, "vacancyPercent was set to a non-sensical value.");
142 >      painCave.isFatal = 1;
143 >      simError();
144 >    } else {
145 >      RealType vF = args_info.vacancyPercent_arg / 100.0;
146 >      RealType vIR;
147 >      RealType vOR;
148 >      if (args_info.vacancyInnerRadius_given) {
149 >        vIR = args_info.vacancyInnerRadius_arg;
150 >      } else {
151 >        vIR = 0.0;
152 >      }
153 >      if (args_info.vacancyOuterRadius_given) {
154 >        vOR = args_info.vacancyOuterRadius_arg;
155 >      } else {
156 >        vOR = particleRadius;
157 >      }
158 >      if (vIR >= 0.0 && vOR <= particleRadius && vOR >= vIR) {
159 >        
160 >        for (int i = 0; i < sites.size(); i++) {
161 >          myLoc = sites[i];
162 >          myR = myLoc.length();
163 >          if (myR >= vIR && myR <= vOR) {
164 >            vacancyTargets.push_back(i);
165 >          }          
166 >        }
167 >        std::random_shuffle(vacancyTargets.begin(), vacancyTargets.end());
168 >        
169 >        int nTargets = vacancyTargets.size();
170 >        vacancyTargets.resize((int)(vF * nTargets));
171 >        
172 >                  
173 >        sprintf(painCave.errMsg, "Removing %d atoms from randomly-selected\n"
174 >                "\tsites between %lf and %lf.", (int) vacancyTargets.size(),
175 >                vIR, vOR);
176 >        painCave.isFatal = 0;
177 >        simError();
178 >
179 >        isVacancy.clear();
180 >        for (int i = 0; i < sites.size(); i++) {
181 >          bool vac = false;
182 >          for (int j = 0; j < vacancyTargets.size(); j++) {
183 >            if (i == vacancyTargets[j]) vac = true;
184 >          }
185 >          isVacancy.push_back(vac);
186 >        }
187 >              
188 >      } else {
189 >        sprintf(painCave.errMsg, "Something is strange about the vacancy\n"
190 >                "\tinner or outer radii.  Check their values.");
191 >        painCave.isFatal = 1;
192 >        simError();
193 >      }
194 >    }
195 >  }
196 >
197 >  /* Get number of lattice sites */
198 >  int nSites = sites.size() - vacancyTargets.size();
199 >
200 >  std::vector<Component*> components = simParams->getComponents();
201 >  std::vector<RealType> molFractions;
202 >  std::vector<RealType> shellRadii;
203 >  std::vector<RealType> molecularMasses;
204 >  std::vector<int> nMol;
205 >  std::map<int, int> componentFromSite;
206 >  nComponents = components.size();
207 >
208 >  if (args_info.molFraction_given && args_info.shellRadius_given) {
209 >    sprintf(painCave.errMsg, "Specify either molFraction or shellRadius "
210 >            "arguments, but not both!");
211 >    painCave.isFatal = 1;
212 >    simError();
213 >  }
214    
215 <  /* Random particle is the default case*/
216 <  if (!args_info.ShellRadius_given){
217 <    std::cout << "Creating a random nanoparticle" << std::endl;
218 <    /* Check to see if we have enough components */
219 <    if (nComponents != args_info.molFraction_given && nComponents != 1){
220 <      std::cerr << "Number of components does not equal molFraction occurances." << std::endl;
221 <      exit(1);
215 >  if (nComponents == 1) {
216 >    molFractions.push_back(1.0);    
217 >    shellRadii.push_back(particleRadius);
218 >  } else if (args_info.molFraction_given) {
219 >    if ((int)args_info.molFraction_given == nComponents) {
220 >      for (int i = 0; i < nComponents; i++) {
221 >        molFractions.push_back(args_info.molFraction_arg[i]);
222 >      }
223 >    } else if ((int)args_info.molFraction_given == nComponents-1) {
224 >      RealType remainingFraction = 1.0;
225 >      for (int i = 0; i < nComponents-1; i++) {
226 >        molFractions.push_back(args_info.molFraction_arg[i]);
227 >        remainingFraction -= molFractions[i];
228 >      }
229 >      molFractions.push_back(remainingFraction);
230 >    } else {    
231 >      sprintf(painCave.errMsg, "nanoparticleBuilder can't figure out molFractions "
232 >              "for all of the components in the <MetaData> block.");
233 >      painCave.isFatal = 1;
234 >      simError();
235      }
236 <    /* Build the mole fractions and number of molecules of each type */  
237 <    int totComponents = 0;
238 <    for (int i = 0;i<nComponents-1;i++){ /* Figure out Percent for each component */
239 <      numMol[i] = int((double)numSites * args_info.molFraction_arg[i]);
240 <      std::cout<<numMol[i]<<std::endl;
241 <      totComponents += numMol[i];
242 <    }
243 <    numMol[nComponents-1] = numSites - totComponents;
244 <  
245 <    /* do the iPod thing, Shuffle da vector */
246 <    std::random_shuffle(nanoParticleSites.begin(), nanoParticleSites.end());
247 <  } else{ /*Handle core-shell with multiple components.*/
248 <    std::cout << "Creating a core-shell nanoparticle." << std::endl;
249 <    if (nComponents != args_info.ShellRadius_given + 1){
250 <      std::cerr << "Number of components does not equal ShellRadius occurances." << std::endl;
251 <      exit(1);
252 <    }  
236 >  } else if ((int)args_info.shellRadius_given) {
237 >    if ((int)args_info.shellRadius_given == nComponents) {
238 >      for (int i = 0; i < nComponents; i++) {
239 >        shellRadii.push_back(args_info.shellRadius_arg[i]);
240 >      }
241 >    } else if ((int)args_info.shellRadius_given == nComponents-1) {
242 >      for (int i = 0; i < nComponents-1; i++) {
243 >        shellRadii.push_back(args_info.shellRadius_arg[i]);
244 >      }
245 >      shellRadii.push_back(particleRadius);
246 >    } else {    
247 >      sprintf(painCave.errMsg, "nanoparticleBuilder can't figure out the\n"
248 >              "\tshell radii for all of the components in the <MetaData> block.");
249 >      painCave.isFatal = 1;
250 >      simError();
251 >    }
252 >  } else {
253 >    sprintf(painCave.errMsg, "You have a multi-component <MetaData> block,\n"
254 >            "\tbut have not specified either molFraction or shellRadius arguments.");
255 >    painCave.isFatal = 1;
256 >    simError();
257 >  }
258      
259 +  if (args_info.molFraction_given) {
260 +    RealType totalFraction = 0.0;
261 +    
262 +    /* Do some simple sanity checking*/
263 +    
264 +    for (int i = 0; i < nComponents; i++) {
265 +      if (molFractions.at(i) < 0.0) {
266 +        sprintf(painCave.errMsg, "One of the requested molFractions was"
267 +                " less than zero!");
268 +        painCave.isFatal = 1;
269 +        simError();
270 +      }
271 +      if (molFractions.at(i) > 1.0) {
272 +        sprintf(painCave.errMsg, "One of the requested molFractions was"
273 +                " greater than one!");
274 +        painCave.isFatal = 1;
275 +        simError();
276 +      }
277 +      totalFraction += molFractions.at(i);
278 +    }
279 +    if (abs(totalFraction - 1.0) > 1e-6) {
280 +      sprintf(painCave.errMsg, "The sum of molFractions was not close enough to 1.0");
281 +      painCave.isFatal = 1;
282 +      simError();
283 +    }
284 +    
285 +    int remaining = nSites;
286 +    for (int i=0; i < nComponents-1; i++) {    
287 +      nMol.push_back(int((RealType)nSites * molFractions.at(i)));
288 +      remaining -= nMol.at(i);
289 +    }
290 +    nMol.push_back(remaining);
291 +    
292 +    // recompute actual mol fractions and perform final sanity check:
293 +    
294 +    int totalMolecules = 0;
295 +    for (int i=0; i < nComponents; i++) {
296 +      molFractions[i] = (RealType)(nMol.at(i))/(RealType)nSites;
297 +      totalMolecules += nMol.at(i);
298 +    }
299 +    
300 +    if (totalMolecules != nSites) {
301 +      sprintf(painCave.errMsg, "Computed total number of molecules is not equal "
302 +              "to the number of lattice sites!");
303 +      painCave.isFatal = 1;
304 +      simError();
305 +    }
306 +  } else {
307 +
308 +    for (int i = 0; i < shellRadii.size(); i++) {
309 +      if (shellRadii.at(i) > particleRadius + 1e-6 ) {
310 +        sprintf(painCave.errMsg, "One of the shellRadius values exceeds the particle Radius.");
311 +        painCave.isFatal = 1;
312 +        simError();
313 +      }
314 +      if (shellRadii.at(i) <= 0.0 ) {
315 +        sprintf(painCave.errMsg, "One of the shellRadius values is smaller than zero!");
316 +        painCave.isFatal = 1;
317 +        simError();
318 +      }
319 +    }
320    }
321  
322 +  vector<int> ids;          
323 +  if ((int)args_info.molFraction_given){
324 +    sprintf(painCave.errMsg, "Creating a randomized spherical nanoparticle.");
325 +    painCave.isFatal = 0;
326 +    simError();
327 +    /* Random particle is the default case*/
328 +
329 +    for (int i = 0; i < sites.size(); i++)
330 +      if (!isVacancy[i]) ids.push_back(i);
331 +    
332 +    std::random_shuffle(ids.begin(), ids.end());
333 +    
334 +  } else{
335 +    sprintf(painCave.errMsg, "Creating a core-shell spherical nanoparticle.");
336 +    painCave.isFatal = 0;
337 +    simError();
338 +
339 +    RealType smallestSoFar;
340 +    int myComponent = -1;
341 +    nMol.clear();
342 +    nMol.resize(nComponents);
343 +
344 +    for (int i = 0; i < sites.size(); i++) {
345 +      myLoc = sites[i];
346 +      myR = myLoc.length();
347 +      smallestSoFar = particleRadius;      
348 +      if (!isVacancy[i]) {
349 +        for (int j = 0; j < nComponents; j++) {
350 +          if (myR <= shellRadii[j]) {
351 +            if (shellRadii[j] <= smallestSoFar) {
352 +              smallestSoFar = shellRadii[j];
353 +              myComponent = j;
354 +            }
355 +          }
356 +        }
357 +        componentFromSite[i] = myComponent;
358 +        nMol[myComponent]++;
359 +      }
360 +    }      
361 +  }
362    
363 <  //get the orientation of the cell sites
364 <  //for the same type of molecule in same lattice, it will not change
201 <  latticeOrt = nanoParticle.getPointsOrt();
202 <  std::cout<<"Orientational vector Size: "<< std::endl;
203 <  std::cout<<latticeOrt.size()<< std::endl;
204 <  
205 <  
206 <  
207 <  // needed for writing out new md file.
208 <  
209 <  outPrefix = getPrefix(inputFileName.c_str()) + "_" + latticeType;
210 <  outMdFileName = outPrefix + ".md";
211 <  
363 >  outputFileName = args_info.output_arg;
364 >  
365    //creat new .md file on fly which corrects the number of molecule    
366 <  createMdFile(inputFileName, outMdFileName, nComponents,numMol);
366 >  createMdFile(inputFileName, outputFileName, nMol);
367    
368    if (oldInfo != NULL)
369      delete oldInfo;
370    
218  
219  // We need to read in new siminfo object.    
220  //parse md file and set up the system
221  //SimCreator NewCreator;
371    SimCreator newCreator;
372 <  SimInfo* NewInfo = newCreator.createSim(outMdFileName, false);
373 <  
225 <  
372 >  SimInfo* NewInfo = newCreator.createSim(outputFileName, false);
373 >    
374    // Place molecules
375    Molecule* mol;
376    SimInfo::MoleculeIterator mi;
377    mol = NewInfo->beginMolecule(mi);
378 +
379    int l = 0;
380 <  for (mol = NewInfo->beginMolecule(mi); mol != NULL; mol = NewInfo->nextMolecule(mi)) {
381 <    locator->placeMol(nanoParticleSites[l], latticeOrt[l], mol);
382 <    l++;
380 >
381 >  for (int i = 0; i < nComponents; i++){
382 >    locator = new MoLocator(NewInfo->getMoleculeStamp(i),
383 >                            NewInfo->getForceField());
384 >    
385 >    if (!args_info.molFraction_given) {
386 >      for (int n = 0; n < sites.size(); n++) {
387 >        if (!isVacancy[n]) {
388 >          if (componentFromSite[n] == i) {
389 >            mol = NewInfo->getMoleculeByGlobalIndex(l);
390 >            locator->placeMol(sites[n], orientations[n], mol);
391 >            l++;
392 >          }
393 >        }
394 >      }
395 >    } else {
396 >      for (int n = 0; n < nMol.at(i); n++) {
397 >        mol = NewInfo->getMoleculeByGlobalIndex(l);
398 >        locator->placeMol(sites[ids[l]], orientations[ids[l]], mol);
399 >        l++;
400 >      }
401 >    }
402    }
235
403    
404    //fill Hmat
405 <  hmat(0, 0)=  latticeConstant;
405 >  hmat(0, 0)=  10.0*particleRadius;
406    hmat(0, 1) = 0.0;
407    hmat(0, 2) = 0.0;
408    
409    hmat(1, 0) = 0.0;
410 <  hmat(1, 1) =  latticeConstant;
410 >  hmat(1, 1) =  10.0*particleRadius;
411    hmat(1, 2) = 0.0;
412    
413    hmat(2, 0) = 0.0;
414    hmat(2, 1) = 0.0;
415 <  hmat(2, 2) =  latticeConstant;
415 >  hmat(2, 2) =  10.0*particleRadius;
416    
417    //set Hmat
418    NewInfo->getSnapshotManager()->getCurrentSnapshot()->setHmat(hmat);
419    
420    
421    //create dumpwriter and write out the coordinates
422 <  NewInfo->setFinalConfigFileName(outInitFileName);
256 <  writer = new DumpWriter(NewInfo);
422 >  writer = new DumpWriter(NewInfo, outputFileName);
423    
424    if (writer == NULL) {
425 <    std::cerr << "error in creating DumpWriter" << std::endl;
426 <    exit(1);
425 >    sprintf(painCave.errMsg, "Error in creating dumpwriter object ");
426 >    painCave.isFatal = 1;
427 >    simError();
428    }
429    
430    writer->writeDump();
431 <  std::cout << "new initial configuration file: " << outInitFileName
432 <            << " is generated." << std::endl;
433 <  
434 <  //delete objects
435 <  
436 <  //delete oldInfo and oldSimSetup
437 <  
438 <  if (NewInfo != NULL)
439 <    delete NewInfo;
440 <  
274 <  if (writer != NULL)
275 <    delete writer;      
276 <  cmdline_parser_free(&args_info);
431 >
432 >  // deleting the writer will put the closing at the end of the dump file
433 >
434 >  delete writer;
435 >
436 >  // cleanup a by calling sim error.....
437 >  sprintf(painCave.errMsg, "A new OpenMD file called \"%s\" has been "
438 >          "generated.\n", outputFileName.c_str());
439 >  painCave.isFatal = 0;
440 >  simError();
441    return 0;
442   }
443  
444 < void createMdFile(const std::string&oldMdFileName, const std::string&newMdFileName,
445 <                  int components,int* numMol) {
444 > void createMdFile(const std::string&oldMdFileName,
445 >                  const std::string&newMdFileName,
446 >                  std::vector<int> nMol) {
447    ifstream oldMdFile;
448    ofstream newMdFile;
449    const int MAXLEN = 65535;
# Line 287 | Line 452 | void createMdFile(const std::string&oldMdFileName, con
452    //create new .md file based on old .md file
453    oldMdFile.open(oldMdFileName.c_str());
454    newMdFile.open(newMdFileName.c_str());
290  
455    oldMdFile.getline(buffer, MAXLEN);
456 <
456 >
457    int i = 0;
458    while (!oldMdFile.eof()) {
459 <    
459 >
460      //correct molecule number
461      if (strstr(buffer, "nMol") != NULL) {
462 <      if(i<components){
463 <        sprintf(buffer, "\tnMol = %i;", numMol[i]);                            
462 >      if(i<nMol.size()){
463 >        sprintf(buffer, "\tnMol = %i;", nMol.at(i));
464          newMdFile << buffer << std::endl;
465          i++;
466        }
# Line 308 | Line 472 | void createMdFile(const std::string&oldMdFileName, con
472    
473    oldMdFile.close();
474    newMdFile.close();
475 +
476 +  if (i != nMol.size()) {
477 +    sprintf(painCave.errMsg, "Couldn't replace the correct number of nMol\n"
478 +            "\tstatements in component blocks.  Make sure that all\n"
479 +            "\tcomponents in the template file have nMol=1");
480 +    painCave.isFatal = 1;
481 +    simError();
482 +  }
483 +    
484   }
485  

Comparing trunk/src/applications/nanoparticleBuilder/nanoparticleBuilder.cpp (property svn:keywords):
Revision 1032 by gezelter, Fri Sep 1 19:16:02 2006 UTC vs.
Revision 1793 by gezelter, Fri Aug 31 21:16:10 2012 UTC

# Line 0 | Line 1
1 + Author Id Revision Date

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines