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

Comparing trunk/OOPSE-4/src/applications/nanoparticleBuilder/nanoparticleBuilder.cpp (file contents):
Revision 2574 by chuckv, Thu Jan 19 22:54:23 2006 UTC vs.
Revision 2677 by chuckv, Wed Mar 29 19:17:20 2006 UTC

# Line 47 | Line 47
47   #include <string>
48   #include <map>
49   #include <fstream>
50 + #include <algorithm>
51  
52   #include "config.h"
53 <
53 > #include "shapedLatticeSpherical.hpp"
54   #include "nanoparticleBuilderCmd.h"
54 #include "sphericalNanoparticle.hpp"
55   #include "lattice/LatticeFactory.hpp"
56   #include "utils/MoLocator.hpp"
57   #include "lattice/Lattice.hpp"
# Line 67 | Line 67 | void createMdFile(const std::string&oldMdFileName,
67   using namespace oopse;
68   void createMdFile(const std::string&oldMdFileName,
69                    const std::string&newMdFileName,
70 <                  int numMol);
70 >                  int components,int* numMol);
71  
72   int main(int argc, char *argv []) {
73    
# Line 85 | Line 85 | int main(int argc, char *argv []) {
85    
86    
87    Lattice *simpleLat;
88 <  int numMol;
88 >  MoLocator* locator;
89 >  int* numMol;
90 >  int nComponents;
91    double latticeConstant;
92    std::vector<double> lc;
93    double mass;
94    const double rhoConvertConst = 1.661;
95    double density;
96 +  double particleRadius;
97    
98    
99  
100    Mat3x3d hmat;
98  MoLocator *locator;
99  sphericalNanoparticle *nanoparticle;
101    std::vector<Vector3d> latticePos;
102    std::vector<Vector3d> latticeOrt;
103    int numMolPerCell;
# Line 127 | Line 128 | int main(int argc, char *argv []) {
128    SimCreator oldCreator;
129    SimInfo* oldInfo = oldCreator.createSim(inputFileName, false);
130    
130  nShells = 0;
131  if (args_info.coreShellRadius_given){
132    nShells = args_info.coreShellRadius_given;
133  }
131    
135  nComponents = oldInfo->getNMoleculeStamp();
136  
137  /* Check to see if we have enough components to build that many shells. */
138  if (nShells){
139    if (oldInfo->getNMoleculeStamp() != nShells) {
140      std::cerr << "Not enough components present in MD file to build specified number of shells"
141      << std::endl;
142      exit(1);
143    }
144  }
145  
146  
147  //creat lattice
148  simpleLat = LatticeFactory::getInstance()->createLattice(latticeType);
149  
150  if (simpleLat == NULL) {
151    std::cerr << "Error in creating lattice" << std::endl;
152    exit(1);
153  }
154  
155  numMolPerCell = simpleLat->getNumSitesPerCell();
156  
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;
137    particleRadius = args_info.radius_arg;
138 <  particleDiameter = 2.0 * particleRadius;
138 >  Globals* simParams = oldInfo->getSimParams();
139    
140 <  /* set lattice constant */
141 <  lc.push_back(latticeConstant);
167 <  simpleLat->setLatticeConstant(lc);
140 >  /* Find out how many different components in this simualtion */
141 >  nComponents =simParams->getNComponents();
142    
169  
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    
177        
178  
179  
180  
152    /* create Molocators */
153    locator = new MoLocator(oldInfo->getMoleculeStamp(0), oldInfo->getForceField());
154    
155 <  /* create a new spherical nanoparticle */
156 <  nanoparticle = new sphericalNanoparticle(particleRadius,latticeConstant);
186 <  /* Build a nanoparticle to see how many sites are there */
187 <  numSites = new int[nComponents]
188 <  int numSites = nanoparticle.getNMol(simpleLat);
155 >  /* Create nanoparticle */
156 >  shapedLatticeSpherical nanoParticle(latticeConstant,latticeType,particleRadius);
157    
158 <
158 >  std::cout <<"Before build getPoints. "<<std::endl;
159 >  /* Build a lattice and get lattice points for this lattice constant */
160 >  vector<Vector3d> nanoParticleSites = nanoParticle.getPoints();
161 >  
162 >  /* Get number of lattice sites */
163 >  numSites = nanoParticleSites.size();
164 >
165 > //std::cout <<"numSites are %d "<<numSites<<std::endl;
166 > // std::cout <<"nComponents are %d "<<nComponents<<std::endl;
167 >  numMol = new int[nComponents];
168 >
169 >  
170    /* Random particle is the default case*/
171    if (!args_info.ShellRadius_given){
172      std::cout << "Creating a random nanoparticle" << std::endl;
173      /* Check to see if we have enough components */
174      if (nComponents != args_info.molFraction_given + 1){
175        std::cerr << "Number of components does not equal molFraction occurances." << std::endl;
176 <      exit 1;
176 >      exit(1);
177      }
178 +    /* Build the mole fractions and number of molecules of each type */  
179      int totComponents = 0;
180      for (int i = 0;i<nComponents-2;i++){ /* Figure out Percent for each component */
181        numMol[i] = int((double)numSites * args_info.molFraction_arg[i]);
182        totComponents += numMol[i];
183      }
184      numMol[nComponents-1] = numSites - totComponents;
185 <
185 >        /* do the iPod thing, Shuffle da vector */
186 >        std::random_shuffle(nanoParticleSites.begin(), nanoParticleSites.end());
187    } else{ /*Handle core-shell with multiple components.*/
188      std::cout << "Creating a core-shell nanoparticle." << std::endl;
189      if (nComponents != args_info.ShellRadius_given + 1){
190        std::cerr << "Number of components does not equal ShellRadius occurances." << std::endl;
191 <      exit 1;
192 <    }
191 >      exit(1);
192 >    }  
193      
213    
214    
194    }
195  
196 +
197     //get the orientation of the cell sites
198    //for the same type of molecule in same lattice, it will not change
199 <   latticeOrt = simpleLat->getLatticePointsOrt();
199 >   latticeOrt = nanoParticle.getPointsOrt();
200 > std::cout<<"Orientational vector Size: "<< std::endl;
201 > std::cout<<latticeOrt.size()<< std::endl;
202 >
203    
204 <  
222 <  
204 >
205    // needed for writing out new md file.
206    
207      outPrefix = getPrefix(inputFileName.c_str()) + "_" + latticeType;
208      outMdFileName = outPrefix + ".md";
209    
210      //creat new .md file on fly which corrects the number of molecule    
211 <    createMdFile(inputFileName, outMdFileName, numcomponents,numMol);
211 >    createMdFile(inputFileName, outMdFileName, nComponents,numMol);
212    
213    if (oldInfo != NULL)
214      delete oldInfo;
# Line 238 | Line 220 | int main(int argc, char *argv []) {
220    
221    SimInfo* NewInfo = oldCreator.createSim(outMdFileName, false);
222    
241  // This was so much fun the first time, lets do it again.
223    
224 +  // Place molecules
225    Molecule* mol;
226    SimInfo::MoleculeIterator mi;
227    mol = NewInfo->beginMolecule(mi);
228 + int l = 0;
229 + for (mol = NewInfo->beginMolecule(mi); mol != NULL; mol = NewInfo->nextMolecule(mi)) {
230 +        locator->placeMol(nanoParticleSites[l], latticeOrt[l], mol);
231 +        l++;
232 +  }
233 +
234  
235  
248  for(int i = -nx; i < nx; i++) {
249     for(int j = -ny; j < ny; j++) {
250        for(int k = -nz; k < nz; k++) {
251          
252           //get the position of the cell sites
253           simpleLat->getLatticePointsPos(latticePos, i, j, k);
254          
255           for(int l = 0; l < numMolPerCell; l++) {
256 #ifdef HAVE_CGAL              
257              if (myGeometry->isInsidePolyhedron(latticePos[l][0],latticePos[l][1],latticePos[l][2])){
258 #endif                              
259                 if (mol != NULL) {
260                    locator->placeMol(latticePos[l], latticeOrt[l], mol);
261                 } else {
262                    std::cerr<<"Error in placing molecule " << std::endl;                    
263                 }
264                 mol = NewInfo->nextMolecule(mi);
265 #ifdef HAVE_CGAL                
266              }
267 #endif              
268           }
269        }
270     }
271  }
236    
273
274  
237    //fill Hmat
238 <  hmat(0, 0)= nx * latticeConstant;
238 >  hmat(0, 0)=  latticeConstant;
239    hmat(0, 1) = 0.0;
240    hmat(0, 2) = 0.0;
241    
242    hmat(1, 0) = 0.0;
243 <  hmat(1, 1) = ny * latticeConstant;
243 >  hmat(1, 1) =  latticeConstant;
244    hmat(1, 2) = 0.0;
245    
246    hmat(2, 0) = 0.0;
247    hmat(2, 1) = 0.0;
248 <  hmat(2, 2) = nz * latticeConstant;
248 >  hmat(2, 2) =  latticeConstant;
249    
250    //set Hmat
251    NewInfo->getSnapshotManager()->getCurrentSnapshot()->setHmat(hmat);
# Line 310 | Line 272 | int main(int argc, char *argv []) {
272      delete NewInfo;
273    
274    if (writer != NULL)
275 <    delete writer;
314 <  delete simpleLat;    
275 >    delete writer;      
276    cmdline_parser_free(&args_info);
277    return 0;
278   }
279  
280   void createMdFile(const std::string&oldMdFileName, const std::string&newMdFileName,
281 <                  int components,int &nummol) {
281 >                  int components,int* numMol) {
282    ifstream oldMdFile;
283    ofstream newMdFile;
284    const int MAXLEN = 65535;
# Line 328 | Line 289 | void createMdFile(const std::string&oldMdFileName, con
289    newMdFile.open(newMdFileName.c_str());
290    
291    oldMdFile.getline(buffer, MAXLEN);
292 <  
292 >
293 >  int i = 0;
294    while (!oldMdFile.eof()) {
295      
296      //correct molecule number
297      if (strstr(buffer, "nMol") != NULL) {
298 <      sprintf(buffer, "\tnMol = %i;", numMol);                          
298 >        if(i<components){
299 >      sprintf(buffer, "\tnMol = %i;", numMol[i]);                              
300        newMdFile << buffer << std::endl;
301 +      i++;
302 +        }
303      } else
304        newMdFile << buffer << std::endl;
305      

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines