source: molecuilder/src/builder.cpp@ 3f0c46

Last change on this file since 3f0c46 was 3f0c46, checked in by Frederik Heber <heber@…>, 17 years ago

Final touches for Create Clusters in water

repetition in CreateClustersinWater(), some stuff thrown out in VolumeOfConvexEnvelope (loop to throw out some more points, which does not happen if we take any appearing in at least one BoundaryPoints), some variable declarations shifted to the start of a function, BoundaryPoints are given to functions and if NULL is created by calling GetBoundaryPoints() and free'd subsequently.

  • Property mode set to 100644
File size: 54.1 KB
RevLine 
[a0bcf1]1/** \file builder.cpp
2 *
3 * By stating absolute positions or binding angles and distances atomic positions of a molecule can be constructed.
4 * The output is the complete configuration file for PCP for direct use.
5 * Features:
6 * -# Atomic data is retrieved from a file, if not found requested and stored there for later re-use
7 * -# step-by-step construction of the molecule beginning either at a centre of with a certain atom
8 *
9 */
10
11/*! \mainpage Molecuilder - a molecular set builder
12 *
13 * This introductory shall briefly make aquainted with the program, helping in installing and a first run.
14 *
15 * \section about About the Program
16 *
17 * Molecuilder is a short program, written in C++, that enables the construction of a coordinate set for the
18 * atoms making up an molecule by the successive statement of binding angles and distances and referencing to
19 * already constructed atoms.
20 *
21 * A configuration file may be written that is compatible to the format used by PCP - a parallel Car-Parrinello
22 * molecular dynamics implementation.
23 *
24 * \section install Installation
25 *
26 * Installation should without problems succeed as follows:
27 * -# ./configure (or: mkdir build;mkdir run;cd build; ../configure --bindir=../run)
28 * -# make
29 * -# make install
30 *
31 * Further useful commands are
32 * -# make clean uninstall: deletes .o-files and removes executable from the given binary directory\n
33 * -# make doxygen-doc: Creates these html pages out of the documented source
34 *
35 * \section run Running
36 *
37 * The program can be executed by running: ./molecuilder
38 *
39 * Note, that it uses a database, called "elements.db", in the executable's directory. If the file is not found,
40 * it is created and any given data on elements of the periodic table will be stored therein and re-used on
41 * later re-execution.
42 *
43 * \section ref References
44 *
45 * For the special configuration file format, see the documentation of pcp.
46 *
47 */
48
49
50using namespace std;
51
52#include "helpers.hpp"
53#include "molecules.hpp"
[e292c10]54#include "boundary.hpp"
[a0bcf1]55
56/********************************************** Submenu routine **************************************/
57
58/** Submenu for adding atoms to the molecule.
59 * \param *periode periodentafel
60 * \param *mol the molecule to add to
61 */
[f75030]62static void AddAtoms(periodentafel *periode, molecule *mol)
[a0bcf1]63{
64 atom *first, *second, *third, *fourth;
65 vector **atoms;
66 vector x,y,z,n; // coordinates for absolute point in cell volume
67 double a,b,c;
68 char choice; // menu choice char
69 bool valid;
70
71 cout << Verbose(0) << "===========ADD ATOM============================" << endl;
72 cout << Verbose(0) << " a - state absolute coordinates of atom" << endl;
73 cout << Verbose(0) << " b - state relative coordinates of atom wrt to reference point" << endl;
74 cout << Verbose(0) << " c - state relative coordinates of atom wrt to already placed atom" << endl;
75 cout << Verbose(0) << " d - state two atoms, two angles and a distance" << endl;
76 cout << Verbose(0) << " e - least square distance position to a set of atoms" << endl;
77 cout << Verbose(0) << "all else - go back" << endl;
78 cout << Verbose(0) << "===============================================" << endl;
79 cout << Verbose(0) << "Note: Specifiy angles in degrees not multiples of Pi!" << endl;
80 cout << Verbose(0) << "INPUT: ";
81 cin >> choice;
82
83 switch (choice) {
84 case 'a': // absolute coordinates of atom
85 cout << Verbose(0) << "Enter absolute coordinates." << endl;
86 first = new atom;
87 first->x.AskPosition(mol->cell_size, false);
88 first->type = periode->AskElement(); // give type
89 mol->AddAtom(first); // add to molecule
90 break;
91
92 case 'b': // relative coordinates of atom wrt to reference point
93 first = new atom;
94 valid = true;
95 do {
96 if (!valid) cout << Verbose(0) << "Resulting position out of cell." << endl;
97 cout << Verbose(0) << "Enter reference coordinates." << endl;
98 x.AskPosition(mol->cell_size, true);
99 cout << Verbose(0) << "Enter relative coordinates." << endl;
100 first->x.AskPosition(mol->cell_size, false);
101 first->x.AddVector((const vector *)&x);
102 cout << Verbose(0) << "\n";
103 } while (!(valid = mol->CheckBounds((const vector *)&first->x)));
104 first->type = periode->AskElement(); // give type
105 mol->AddAtom(first); // add to molecule
106 break;
107
108 case 'c': // relative coordinates of atom wrt to already placed atom
109 first = new atom;
110 valid = true;
111 do {
112 if (!valid) cout << Verbose(0) << "Resulting position out of cell." << endl;
113 second = mol->AskAtom("Enter atom number: ");
114 cout << Verbose(0) << "Enter relative coordinates." << endl;
115 first->x.AskPosition(mol->cell_size, false);
[f75030]116 for (int i=NDIM;i--;) {
[a0bcf1]117 first->x.x[i] += second->x.x[i];
118 }
119 } while (!(valid = mol->CheckBounds((const vector *)&first->x)));
120 first->type = periode->AskElement(); // give type
121 mol->AddAtom(first); // add to molecule
122 break;
123
124 case 'd': // two atoms, two angles and a distance
125 first = new atom;
126 valid = true;
127 do {
128 if (!valid) {
129 cout << Verbose(0) << "Resulting coordinates out of cell - ";
130 first->x.Output((ofstream *)&cout);
131 cout << Verbose(0) << endl;
132 }
133 cout << Verbose(0) << "First, we need two atoms, the first atom is the central, while the second is the outer one." << endl;
134 second = mol->AskAtom("Enter central atom: ");
135 third = mol->AskAtom("Enter second atom (specifying the axis for first angle): ");
136 fourth = mol->AskAtom("Enter third atom (specifying a plane for second angle): ");
137 a = ask_value("Enter distance between central (first) and new atom: ");
138 b = ask_value("Enter angle between new, first and second atom (degrees): ");
139 b *= M_PI/180.;
140 bound(&b, 0., 2.*M_PI);
141 c = ask_value("Enter second angle between new and normal vector of plane defined by first, second and third atom (degrees): ");
142 c *= M_PI/180.;
143 bound(&c, -M_PI, M_PI);
144 cout << Verbose(0) << "radius: " << a << "\t phi: " << b*180./M_PI << "\t theta: " << c*180./M_PI << endl;
145/*
146 second->Output(1,1,(ofstream *)&cout);
147 third->Output(1,2,(ofstream *)&cout);
148 fourth->Output(1,3,(ofstream *)&cout);
149 n.MakeNormalVector((const vector *)&second->x, (const vector *)&third->x, (const vector *)&fourth->x);
150 x.CopyVector(&second->x);
151 x.SubtractVector(&third->x);
152 x.CopyVector(&fourth->x);
153 x.SubtractVector(&third->x);
154
155 if (!z.SolveSystem(&x,&y,&n, b, c, a)) {
156 cout << Verbose(0) << "Failure solving self-dependent linear system!" << endl;
157 continue;
158 }
159 cout << Verbose(0) << "resulting relative coordinates: ";
160 z.Output((ofstream *)&cout);
161 cout << Verbose(0) << endl;
162 */
163 // calc axis vector
164 x.CopyVector(&second->x);
165 x.SubtractVector(&third->x);
166 x.Normalize();
167 cout << "x: ",
168 x.Output((ofstream *)&cout);
169 cout << endl;
170 z.MakeNormalVector(&second->x,&third->x,&fourth->x);
171 cout << "z: ",
172 z.Output((ofstream *)&cout);
173 cout << endl;
174 y.MakeNormalVector(&x,&z);
175 cout << "y: ",
176 y.Output((ofstream *)&cout);
177 cout << endl;
178
179 // rotate vector around first angle
180 first->x.CopyVector(&x);
181 first->x.RotateVector(&z,b - M_PI);
182 cout << "Rotated vector: ",
183 first->x.Output((ofstream *)&cout);
184 cout << endl;
185 // remove the projection onto the rotation plane of the second angle
186 n.CopyVector(&y);
187 n.Scale(first->x.Projection(&y));
188 cout << "N1: ",
189 n.Output((ofstream *)&cout);
190 cout << endl;
191 first->x.SubtractVector(&n);
192 cout << "Subtracted vector: ",
193 first->x.Output((ofstream *)&cout);
194 cout << endl;
195 n.CopyVector(&z);
196 n.Scale(first->x.Projection(&z));
197 cout << "N2: ",
198 n.Output((ofstream *)&cout);
199 cout << endl;
200 first->x.SubtractVector(&n);
201 cout << "2nd subtracted vector: ",
202 first->x.Output((ofstream *)&cout);
203 cout << endl;
204
205 // rotate another vector around second angle
206 n.CopyVector(&y);
207 n.RotateVector(&x,c - M_PI);
208 cout << "2nd Rotated vector: ",
209 n.Output((ofstream *)&cout);
210 cout << endl;
211
212 // add the two linear independent vectors
213 first->x.AddVector(&n);
214 first->x.Normalize();
215 first->x.Scale(a);
216 first->x.AddVector(&second->x);
217
218 cout << Verbose(0) << "resulting coordinates: ";
219 first->x.Output((ofstream *)&cout);
220 cout << Verbose(0) << endl;
221 } while (!(valid = mol->CheckBounds((const vector *)&first->x)));
222 first->type = periode->AskElement(); // give type
223 mol->AddAtom(first); // add to molecule
224 break;
225
226 case 'e': // least square distance position to a set of atoms
227 first = new atom;
228 atoms = new (vector*[128]);
229 valid = true;
[f75030]230 for(int i=128;i--;)
[a0bcf1]231 atoms[i] = NULL;
232 int i=0, j=0;
233 cout << Verbose(0) << "Now we need at least three molecules.\n";
234 do {
235 cout << Verbose(0) << "Enter " << i+1 << "th atom: ";
236 cin >> j;
237 if (j != -1) {
238 second = mol->FindAtom(j);
239 atoms[i++] = &(second->x);
240 }
241 } while ((j != -1) && (i<128));
242 if (i >= 2) {
243 first->x.LSQdistance(atoms, i);
244
245 first->x.Output((ofstream *)&cout);
246 first->type = periode->AskElement(); // give type
247 mol->AddAtom(first); // add to molecule
248 } else {
249 delete first;
250 cout << Verbose(0) << "Please enter at least two vectors!\n";
251 }
252 break;
253 };
254};
255
256/** Submenu for centering the atoms in the molecule.
257 * \param *mol the molecule with all the atoms
258 */
[f75030]259static void CenterAtoms(molecule *mol)
[a0bcf1]260{
[3ce105]261 vector x, y;
[a0bcf1]262 char choice; // menu choice char
263
264 cout << Verbose(0) << "===========CENTER ATOMS=========================" << endl;
265 cout << Verbose(0) << " a - on origin" << endl;
266 cout << Verbose(0) << " b - on center of gravity" << endl;
267 cout << Verbose(0) << " c - within box with additional boundary" << endl;
[3ce105]268 cout << Verbose(0) << " d - within given simulation box" << endl;
[a0bcf1]269 cout << Verbose(0) << "all else - go back" << endl;
270 cout << Verbose(0) << "===============================================" << endl;
271 cout << Verbose(0) << "INPUT: ";
272 cin >> choice;
273
274 switch (choice) {
275 default:
276 cout << Verbose(0) << "Not a valid choice." << endl;
277 break;
278 case 'a':
279 cout << Verbose(0) << "Centering atoms in config file on origin." << endl;
280 mol->CenterOrigin((ofstream *)&cout, &x);
281 break;
282 case 'b':
283 cout << Verbose(0) << "Centering atoms in config file on center of gravity." << endl;
284 mol->CenterGravity((ofstream *)&cout, &x);
285 break;
286 case 'c':
287 cout << Verbose(0) << "Centering atoms in config file within given additional boundary." << endl;
[f75030]288 for (int i=0;i<NDIM;i++) {
[a0bcf1]289 cout << Verbose(0) << "Enter axis " << i << " boundary: ";
[3ce105]290 cin >> y.x[i];
[a0bcf1]291 }
292 mol->CenterEdge((ofstream *)&cout, &x); // make every coordinate positive
[3ce105]293 mol->Translate(&y); // translate by boundary
294 mol->SetBoxDimension(&(x+y*2)); // update Box of atoms by boundary
295 break;
296 case 'd':
297 cout << Verbose(1) << "Centering atoms in config file within given simulation box." << endl;
[f75030]298 for (int i=0;i<NDIM;i++) {
[3ce105]299 cout << Verbose(0) << "Enter axis " << i << " boundary: ";
300 cin >> x.x[i];
[a0bcf1]301 }
[3ce105]302 // center
303 mol->CenterInBox((ofstream *)&cout, &x);
304 // update Box of atoms by boundary
305 mol->SetBoxDimension(&x);
[a0bcf1]306 break;
307 }
308};
309
310/** Submenu for aligning the atoms in the molecule.
311 * \param *periode periodentafel
312 * \param *mol the molecule with all the atoms
313 */
[f75030]314static void AlignAtoms(periodentafel *periode, molecule *mol)
[a0bcf1]315{
316 atom *first, *second, *third;
317 vector x,n;
318 char choice; // menu choice char
319
320 cout << Verbose(0) << "===========ALIGN ATOMS=========================" << endl;
321 cout << Verbose(0) << " a - state three atoms defining align plane" << endl;
322 cout << Verbose(0) << " b - state alignment vector" << endl;
323 cout << Verbose(0) << " c - state two atoms in alignment direction" << endl;
324 cout << Verbose(0) << " d - align automatically by least square fit" << endl;
325 cout << Verbose(0) << "all else - go back" << endl;
326 cout << Verbose(0) << "===============================================" << endl;
327 cout << Verbose(0) << "INPUT: ";
328 cin >> choice;
329
330 switch (choice) {
331 default:
332 case 'a': // three atoms defining mirror plane
333 first = mol->AskAtom("Enter first atom: ");
334 second = mol->AskAtom("Enter second atom: ");
335 third = mol->AskAtom("Enter third atom: ");
336
337 n.MakeNormalVector((const vector *)&first->x,(const vector *)&second->x,(const vector *)&third->x);
338 break;
339 case 'b': // normal vector of mirror plane
340 cout << Verbose(0) << "Enter normal vector of mirror plane." << endl;
341 n.AskPosition(mol->cell_size,0);
342 n.Normalize();
343 break;
344 case 'c': // three atoms defining mirror plane
345 first = mol->AskAtom("Enter first atom: ");
346 second = mol->AskAtom("Enter second atom: ");
347
348 n.CopyVector((const vector *)&first->x);
349 n.SubtractVector((const vector *)&second->x);
350 n.Normalize();
351 break;
352 case 'd':
353 char shorthand[4];
354 vector a;
355 struct lsq_params param;
356 do {
357 fprintf(stdout, "Enter the element of atoms to be chosen: ");
358 fscanf(stdin, "%3s", shorthand);
359 } while ((param.type = periode->FindElement(shorthand)) == NULL);
360 cout << Verbose(0) << "Element is " << param.type->name << endl;
361 mol->GetAlignVector(&param);
[f75030]362 for (int i=NDIM;i--;) {
[a0bcf1]363 x.x[i] = gsl_vector_get(param.x,i);
[f75030]364 n.x[i] = gsl_vector_get(param.x,i+NDIM);
[a0bcf1]365 }
366 gsl_vector_free(param.x);
367 cout << Verbose(0) << "Offset vector: ";
368 x.Output((ofstream *)&cout);
369 cout << Verbose(0) << endl;
370 n.Normalize();
371 break;
372 };
373 cout << Verbose(0) << "Alignment vector: ";
374 n.Output((ofstream *)&cout);
375 cout << Verbose(0) << endl;
376 mol->Align(&n);
377};
378
379/** Submenu for mirroring the atoms in the molecule.
380 * \param *mol the molecule with all the atoms
381 */
[f75030]382static void MirrorAtoms(molecule *mol)
[a0bcf1]383{
384 atom *first, *second, *third;
385 vector n;
386 char choice; // menu choice char
387
388 cout << Verbose(0) << "===========MIRROR ATOMS=========================" << endl;
389 cout << Verbose(0) << " a - state three atoms defining mirror plane" << endl;
390 cout << Verbose(0) << " b - state normal vector of mirror plane" << endl;
391 cout << Verbose(0) << " c - state two atoms in normal direction" << endl;
392 cout << Verbose(0) << "all else - go back" << endl;
393 cout << Verbose(0) << "===============================================" << endl;
394 cout << Verbose(0) << "INPUT: ";
395 cin >> choice;
396
397 switch (choice) {
398 default:
399 case 'a': // three atoms defining mirror plane
400 first = mol->AskAtom("Enter first atom: ");
401 second = mol->AskAtom("Enter second atom: ");
402 third = mol->AskAtom("Enter third atom: ");
403
404 n.MakeNormalVector((const vector *)&first->x,(const vector *)&second->x,(const vector *)&third->x);
405 break;
406 case 'b': // normal vector of mirror plane
407 cout << Verbose(0) << "Enter normal vector of mirror plane." << endl;
408 n.AskPosition(mol->cell_size,0);
409 n.Normalize();
410 break;
411 case 'c': // three atoms defining mirror plane
412 first = mol->AskAtom("Enter first atom: ");
413 second = mol->AskAtom("Enter second atom: ");
414
415 n.CopyVector((const vector *)&first->x);
416 n.SubtractVector((const vector *)&second->x);
417 n.Normalize();
418 break;
419 };
420 cout << Verbose(0) << "Normal vector: ";
421 n.Output((ofstream *)&cout);
422 cout << Verbose(0) << endl;
423 mol->Mirror((const vector *)&n);
424};
425
426/** Submenu for removing the atoms from the molecule.
427 * \param *mol the molecule with all the atoms
428 */
[f75030]429static void RemoveAtoms(molecule *mol)
[a0bcf1]430{
431 atom *first, *second;
432 int axis;
433 double tmp1, tmp2;
434 char choice; // menu choice char
435
436 cout << Verbose(0) << "===========REMOVE ATOMS=========================" << endl;
437 cout << Verbose(0) << " a - state atom for removal by number" << endl;
438 cout << Verbose(0) << " b - keep only in radius around atom" << endl;
439 cout << Verbose(0) << " c - remove this with one axis greater value" << endl;
440 cout << Verbose(0) << "all else - go back" << endl;
441 cout << Verbose(0) << "===============================================" << endl;
442 cout << Verbose(0) << "INPUT: ";
443 cin >> choice;
444
445 switch (choice) {
446 default:
447 case 'a':
448 if (mol->RemoveAtom(mol->AskAtom("Enter number of atom within molecule: ")))
449 cout << Verbose(1) << "Atom removed." << endl;
450 else
451 cout << Verbose(1) << "Atom not found." << endl;
452 break;
453 case 'b':
454 second = mol->AskAtom("Enter number of atom as reference point: ");
455 cout << Verbose(0) << "Enter radius: ";
456 cin >> tmp1;
457 first = mol->start;
458 while(first->next != mol->end) {
459 first = first->next;
460 if (first->x.Distance((const vector *)&second->x) > tmp1*tmp1) // distance to first above radius ...
461 mol->RemoveAtom(first);
462 }
463 break;
464 case 'c':
465 cout << Verbose(0) << "Which axis is it: ";
466 cin >> axis;
467 cout << Verbose(0) << "Left inward boundary: ";
468 cin >> tmp1;
469 cout << Verbose(0) << "Right inward boundary: ";
470 cin >> tmp2;
471 first = mol->start;
472 while(first->next != mol->end) {
473 first = first->next;
474 if ((first->x.x[axis] > tmp2) || (first->x.x[axis] < tmp1)) // out of boundary ...
475 mol->RemoveAtom(first);
476 }
477 break;
478 };
479 //mol->Output((ofstream *)&cout);
480 choice = 'r';
481};
482
483/** Submenu for measuring out the atoms in the molecule.
484 * \param *periode periodentafel
485 * \param *mol the molecule with all the atoms
486 */
[32b6dc]487static void MeasureAtoms(periodentafel *periode, molecule *mol, config *configuration)
[a0bcf1]488{
489 atom *first, *second, *third;
490 vector x,y;
491 double min[256], tmp1, tmp2, tmp3;
492 int Z;
493 char choice; // menu choice char
494
495 cout << Verbose(0) << "===========MEASURE ATOMS=========================" << endl;
496 cout << Verbose(0) << " a - calculate bond length between one atom and all others" << endl;
497 cout << Verbose(0) << " b - calculate bond length between two atoms" << endl;
498 cout << Verbose(0) << " c - calculate bond angle" << endl;
[32b6dc]499 cout << Verbose(0) << " d - calculate principal axis of the system" << endl;
500 cout << Verbose(0) << " e - calculate volume of the convex envelope" << endl;
[a0bcf1]501 cout << Verbose(0) << "all else - go back" << endl;
502 cout << Verbose(0) << "===============================================" << endl;
503 cout << Verbose(0) << "INPUT: ";
504 cin >> choice;
505
506 switch(choice) {
507 default:
508 cout << Verbose(1) << "Not a valid choice." << endl;
509 break;
510 case 'a':
511 first = mol->AskAtom("Enter first atom: ");
[f75030]512 for (int i=MAX_ELEMENTS;i--;)
[a0bcf1]513 min[i] = 0.;
514
515 second = mol->start;
516 while ((second->next != mol->end)) {
517 second = second->next; // advance
518 Z = second->type->Z;
519 tmp1 = 0.;
520 if (first != second) {
521 x.CopyVector((const vector *)&first->x);
522 x.SubtractVector((const vector *)&second->x);
523 tmp1 = x.Norm();
524 }
525 if ((tmp1 != 0.) && ((min[Z] == 0.) || (tmp1 < min[Z]))) min[Z] = tmp1;
526 //cout << Verbose(0) << "Bond length between Atom " << first->nr << " and " << second->nr << ": " << tmp1 << " a.u." << endl;
527 }
[f75030]528 for (int i=MAX_ELEMENTS;i--;)
[a0bcf1]529 if (min[i] != 0.) cout << Verbose(0) << "Minimum Bond length between " << first->type->name << " Atom " << first->nr << " and next Ion of type " << (periode->FindElement(i))->name << ": " << min[i] << " a.u." << endl;
530 break;
531
532 case 'b':
533 first = mol->AskAtom("Enter first atom: ");
534 second = mol->AskAtom("Enter second atom: ");
[f75030]535 for (int i=NDIM;i--;)
[a0bcf1]536 min[i] = 0.;
537 x.CopyVector((const vector *)&first->x);
538 x.SubtractVector((const vector *)&second->x);
539 tmp1 = x.Norm();
540 cout << Verbose(1) << "Distance vector is ";
541 x.Output((ofstream *)&cout);
542 cout << "." << endl << "Norm of distance is " << tmp1 << "." << endl;
543 break;
544
545 case 'c':
546 cout << Verbose(0) << "Evaluating bond angle between three - first, central, last - atoms." << endl;
547 first = mol->AskAtom("Enter first atom: ");
548 second = mol->AskAtom("Enter central atom: ");
549 third = mol->AskAtom("Enter last atom: ");
550 tmp1 = tmp2 = tmp3 = 0.;
551 x.CopyVector((const vector *)&first->x);
552 x.SubtractVector((const vector *)&second->x);
553 y.CopyVector((const vector *)&third->x);
554 y.SubtractVector((const vector *)&second->x);
555 cout << Verbose(0) << "Bond angle between first atom Nr." << first->nr << ", central atom Nr." << second->nr << " and last atom Nr." << third->nr << ": ";
556 cout << Verbose(0) << (acos(x.ScalarProduct((const vector *)&y)/(y.Norm()*x.Norm()))/M_PI*180.) << " degrees" << endl;
557 break;
[32b6dc]558 case 'd':
559 cout << Verbose(0) << "Evaluating prinicipal axis." << endl;
560 cout << Verbose(0) << "Shall we rotate? [0/1]: ";
561 cin >> Z;
562 if ((Z >=0) && (Z <=1))
563 mol->PrincipalAxisSystem((ofstream *)&cout, (bool)Z);
564 else
565 mol->PrincipalAxisSystem((ofstream *)&cout, false);
566 break;
567 case 'e':
568 cout << Verbose(0) << "Evaluating volume of the convex envelope.";
[7fcea6]569 VolumeOfConvexEnvelope((ofstream *)&cout, configuration, NULL, mol);
[32b6dc]570 break;
[a0bcf1]571 }
572};
573
574/** Submenu for measuring out the atoms in the molecule.
575 * \param *mol the molecule with all the atoms
576 * \param *configuration configuration structure for the to be written config files of all fragments
577 */
[f75030]578static void FragmentAtoms(molecule *mol, config *configuration)
[a0bcf1]579{
[c75363]580 int Order1;
[a0bcf1]581 clock_t start, end;
582
583 cout << Verbose(0) << "Fragmenting molecule with current connection matrix ..." << endl;
584 cout << Verbose(0) << "What's the desired bond order: ";
585 cin >> Order1;
586 if (mol->first->next != mol->last) { // there are bonds
587 start = clock();
[c75363]588 mol->FragmentMolecule((ofstream *)&cout, Order1, configuration);
[a0bcf1]589 end = clock();
590 cout << Verbose(0) << "Clocks for this operation: " << (end-start) << ", time: " << ((double)(end-start)/CLOCKS_PER_SEC) << "s." << endl;
591 } else
592 cout << Verbose(0) << "Connection matrix has not yet been generated!" << endl;
593};
594
595/********************************************** Test routine **************************************/
596
597/** Is called always as option 'T' in the menu.
598 */
[f75030]599static void testroutine(molecule *mol)
[a0bcf1]600{
601 // the current test routine checks the functionality of the KeySet&Graph concept:
602 // We want to have a multiindex (the KeySet) describing a unique subgraph
603 atom *Walker = mol->start;
604 int i, comp, counter=0;
605
606 // generate some KeySets
607 cout << "Generating KeySets." << endl;
608 KeySet TestSets[mol->AtomCount+1];
609 i=1;
610 while (Walker->next != mol->end) {
611 Walker = Walker->next;
612 for (int j=0;j<i;j++) {
613 TestSets[j].insert(Walker->nr);
614 }
615 i++;
616 }
617 cout << "Testing insertion of already present item in KeySets." << endl;
618 KeySetTestPair test;
619 test = TestSets[mol->AtomCount-1].insert(Walker->nr);
620 if (test.second) {
621 cout << Verbose(1) << "Insertion worked?!" << endl;
622 } else {
623 cout << Verbose(1) << "Insertion rejected: Present object is " << (*test.first) << "." << endl;
624 }
625 TestSets[mol->AtomCount].insert(mol->end->previous->nr);
626 TestSets[mol->AtomCount].insert(mol->end->previous->previous->previous->nr);
627
628 // constructing Graph structure
629 cout << "Generating Subgraph class." << endl;
630 Graph Subgraphs;
631
632 // insert KeySets into Subgraphs
633 cout << "Inserting KeySets into Subgraph class." << endl;
634 for (int j=0;j<mol->AtomCount;j++) {
635 Subgraphs.insert(GraphPair (TestSets[j],pair<int, double>(counter++, 1.)));
636 }
637 cout << "Testing insertion of already present item in Subgraph." << endl;
638 GraphTestPair test2;
639 test2 = Subgraphs.insert(GraphPair (TestSets[mol->AtomCount],pair<int, double>(counter++, 1.)));
640 if (test2.second) {
641 cout << Verbose(1) << "Insertion worked?!" << endl;
642 } else {
643 cout << Verbose(1) << "Insertion rejected: Present object is " << (*(test2.first)).second.first << "." << endl;
644 }
645
646 // show graphs
647 cout << "Showing Subgraph's contents, checking that it's sorted." << endl;
648 Graph::iterator A = Subgraphs.begin();
649 while (A != Subgraphs.end()) {
650 cout << (*A).second.first << ": ";
651 KeySet::iterator key = (*A).first.begin();
652 comp = -1;
653 while (key != (*A).first.end()) {
654 if ((*key) > comp)
655 cout << (*key) << " ";
656 else
657 cout << (*key) << "! ";
658 comp = (*key);
659 key++;
660 }
661 cout << endl;
662 A++;
663 }
664};
665
[8129de]666/** Tries given filename or standard on saving the config file.
667 * \param *ConfigFileName name of file
668 * \param *configuration pointer to configuration structure with all the values
669 * \param *periode pointer to periodentafel structure with all the elements
670 * \param *mol pointer to molecule structure with all the atoms and coordinates
671 */
[f75030]672static void SaveConfig(char *ConfigFileName, config *configuration, periodentafel *periode, molecule *mol)
[8129de]673{
[c510a7]674 char filename[MAXSTRINGSIZE];
[8129de]675 ofstream output;
676
[a829d1]677 cout << Verbose(0) << "Storing configuration ... " << endl;
[8129de]678 // get correct valence orbitals
679 mol->CalculateOrbitals(*configuration);
680 configuration->InitMaxMinStopStep = configuration->MaxMinStopStep = configuration->MaxPsiDouble;
[584691]681 if (ConfigFileName != NULL) {
[8129de]682 output.open(ConfigFileName, ios::trunc);
[9bd230]683 } else if (strlen(configuration->configname) != 0) {
[584691]684 output.open(configuration->configname, ios::trunc);
[9bd230]685 } else {
686 output.open(DEFAULTCONFIG, ios::trunc);
687 }
[8129de]688 if (configuration->Save(&output, periode, mol))
689 cout << Verbose(0) << "Saving of config file successful." << endl;
690 else
691 cout << Verbose(0) << "Saving of config file failed." << endl;
692 output.close();
693 output.clear();
694 // and save to xyz file
695 if (ConfigFileName != NULL) {
696 strcpy(filename, ConfigFileName);
697 strcat(filename, ".xyz");
698 output.open(filename, ios::trunc);
699 }
700 if (output == NULL) {
701 strcpy(filename,"main_pcp_linux");
702 strcat(filename, ".xyz");
703 output.open(filename, ios::trunc);
704 }
705 if (mol->OutputXYZ(&output))
706 cout << Verbose(0) << "Saving of XYZ file successful." << endl;
707 else
708 cout << Verbose(0) << "Saving of XYZ file failed." << endl;
709 output.close();
710 output.clear();
[584691]711
[36b128]712 if (!strcmp(configuration->configpath, configuration->GetDefaultPath())) {
713 cerr << "WARNING: config is found under different path then stated in config file::defaultpath!" << endl;
714 }
[8129de]715};
716
[3ce105]717/** Parses the command line options.
718 * \param argc argument count
719 * \param **argv arguments array
720 * \param *mol molecule structure
721 * \param *periode elements structure
722 * \param configuration config file structure
723 * \param *ConfigFileName pointer to config file name in **argv
[8b486e]724 * \param *PathToDatabases pointer to db's path in **argv
[3ce105]725 * \return exit code (0 - successful, all else - something's wrong)
726 */
[8b486e]727static int ParseCommandLineOptions(int argc, char **argv, molecule *&mol, periodentafel *&periode, config& configuration, char *&ConfigFileName, char *&PathToDatabases)
[a0bcf1]728{
729 element *finder;
730 vector x,y,z,n; // coordinates for absolute point in cell volume
[3ce105]731 double *factor; // unit factor if desired
[a0bcf1]732 ifstream test;
733 ofstream output;
734 string line;
[3ce105]735 atom *first;
[0e348e]736 int ExitFlag = 0;
[3ce105]737 int j;
[8129de]738 enum ConfigStatus config_present = absent;
[a0bcf1]739 clock_t start,end;
[a829d1]740 int argptr;
[8b486e]741 PathToDatabases = LocalPath;
[3ce105]742
[584691]743 if (argc > 1) { // config file specified as option
[a829d1]744 // 1. : Parse options that just set variables or print help
[3ce105]745 argptr = 1;
746 do {
747 if (argv[argptr][0] == '-') {
[a829d1]748 cout << Verbose(0) << "Recognized command line argument: " << argv[argptr][1] << ".\n";
749 argptr++;
750 switch(argv[argptr-1][1]) {
751 case 'h':
752 case 'H':
753 case '?':
754 cout << "MoleCuilder suite" << endl << "==================" << endl << endl;
[c75363]755 cout << "Usage: " << argv[0] << "[config file] [-{acefpsthH?vfrp}] [further arguments]" << endl;
[a829d1]756 cout << "or simply " << argv[0] << " without arguments for interactive session." << endl;
757 cout << "\t-a Z x1 x2 x3\tAdd new atom of element Z at coordinates (x1,x2,x3)." << endl;
[3ce105]758 cout << "\t-b x1 x2 x3\tCenter atoms in domain with given edge lengths of (x1,x2,x3)." << endl;
[a829d1]759 cout << "\t-c x1 x2 x3\tCenter atoms in domain with a minimum distance to boundary of (x1,x2,x3)." << endl;
[3f0c46]760 cout << "\t-d x1 x2 x3\tDuplicate cell along each axis by given factor." << endl;
[8b486e]761 cout << "\t-e <file>\tSets the databases path to be parsed (default: ./)." << endl;
[0e348e]762 cout << "\t-f <dist> <order>\tFragments the molecule in BOSSANOVA manner and stores config files in same dir as config." << endl;
[a829d1]763 cout << "\t-h/-H/-?\tGive this help screen." << endl;
[32b6dc]764 cout << "\t-m\tAlign in PAS with greatest EV along z axis." << endl;
[ffcf49]765 cout << "\t-n\tFast parsing (i.e. no trajectories are looked for)." << endl;
[e292c10]766 cout << "\t-o\tGet volume of the convex envelope (and store to tecplot file)." << endl;
[a829d1]767 cout << "\t-p <file>\tParse given xyz file and create raw config file from it." << endl;
768 cout << "\t-r\t\tConvert file from an old pcp syntax." << endl;
769 cout << "\t-t x1 x2 x3\tTranslate all atoms by this vector (x1,x2,x3)." << endl;
770 cout << "\t-s x1 x2 x3\tScale all atom coordinates by this vector (x1,x2,x3)." << endl;
[3f0c46]771 cout << "\t-u rho\tsuspend in water solution and output necessary cell lengths, average density rho and repetition." << endl;
[a829d1]772 cout << "\t-v/-V\t\tGives version information." << endl;
[f5306f]773 cout << "Note: config files must not begin with '-' !" << endl;
[36b128]774 delete(mol);
775 delete(periode);
[3ce105]776 return (1);
[a829d1]777 break;
778 case 'v':
779 case 'V':
780 cout << argv[0] << " " << VERSIONSTRING << endl;
781 cout << "Build your own molecule position set." << endl;
[36b128]782 delete(mol);
783 delete(periode);
[3ce105]784 return (1);
[a829d1]785 break;
786 case 'e':
787 cout << "Using " << argv[argptr] << " as elements database." << endl;
[8b486e]788 PathToDatabases = argv[argptr];
[a829d1]789 argptr+=1;
790 break;
[ffcf49]791 case 'n':
792 cout << "I won't parse trajectories." << endl;
793 configuration.FastParsing = true;
[a829d1]794 default: // no match? Step on
795 argptr++;
796 break;
797 }
[3ce105]798 } else
799 argptr++;
[9bd230]800 } while (argptr < argc);
[3ce105]801
802 // 2. Parse the element database
[8b486e]803 if (periode->LoadPeriodentafel(PathToDatabases)) {
[a829d1]804 cout << Verbose(0) << "Element list loaded successfully." << endl;
[8b486e]805 periode->Output((ofstream *)&cout);
806 } else
[a829d1]807 cout << Verbose(0) << "Element list loading failed." << endl;
[9bd230]808
[3ce105]809 // 3. Find config file name and parse if possible
[9bd230]810 if (argv[1][0] != '-') {
[8129de]811 cout << Verbose(0) << "Config file given." << endl;
[9bd230]812 test.open(argv[1], ios::in);
[8129de]813 if (test == NULL) {
814 //return (1);
[9bd230]815 output.open(argv[1], ios::out);
[8129de]816 if (output == NULL) {
[9bd230]817 cout << Verbose(1) << "Specified config file " << argv[1] << " not found." << endl;
[8129de]818 config_present = absent;
819 } else {
820 cout << "Empty configuration file." << endl;
[9bd230]821 ConfigFileName = argv[1];
[8129de]822 config_present = empty;
823 output.close();
824 }
825 } else {
[f5306f]826 test.close();
[9bd230]827 ConfigFileName = argv[1];
[9e831e]828 cout << Verbose(1) << "Specified config file found, parsing ... ";
[f5306f]829 switch (configuration.TestSyntax(ConfigFileName, periode, mol)) {
[8129de]830 case 1:
831 cout << "new syntax." << endl;
[f5306f]832 configuration.Load(ConfigFileName, periode, mol);
[8129de]833 config_present = present;
834 break;
835 case 0:
836 cout << "old syntax." << endl;
[f5306f]837 configuration.LoadOld(ConfigFileName, periode, mol);
[8129de]838 config_present = present;
839 break;
840 default:
841 cout << "Unknown syntax or empty, yet present file." << endl;
842 config_present = empty;
843 }
[a0bcf1]844 }
[a829d1]845 } else
846 config_present = absent;
[9bd230]847
[a829d1]848 // 4. parse again through options, now for those depending on elements db and config presence
849 argptr = 1;
850 do {
[3ce105]851 cout << "Current Command line argument: " << argv[argptr] << "." << endl;
[a829d1]852 if (argv[argptr][0] == '-') {
[8129de]853 argptr++;
854 if ((config_present == present) || (config_present == empty)) {
855 switch(argv[argptr-1][1]) {
856 case 'p':
[0e348e]857 ExitFlag = 1;
[8129de]858 cout << Verbose(1) << "Parsing xyz file for new atoms." << endl;
[3ce105]859 if (!mol->AddXYZFile(argv[argptr]))
[8129de]860 cout << Verbose(2) << "File not found." << endl;
861 else
862 cout << Verbose(2) << "File found and parsed." << endl;
[a829d1]863 config_present = present;
[a0bcf1]864 break;
[a829d1]865 default: // no match? Don't step on (this is done in next switch's default)
[a0bcf1]866 break;
[8129de]867 }
868 }
[3ce105]869 if (config_present == present) {
870 switch(argv[argptr-1][1]) {
871 case 't':
872 ExitFlag = 1;
873 cout << Verbose(1) << "Translating all ions to new origin." << endl;
[f75030]874 for (int i=NDIM;i--;)
[3ce105]875 x.x[i] = atof(argv[argptr+i]);
876 mol->Translate((const vector *)&x);
877 argptr+=3;
878 break;
879 case 'a':
880 ExitFlag = 1;
881 cout << Verbose(1) << "Adding new atom." << endl;
882 first = new atom;
[f75030]883 for (int i=NDIM;i--;)
[3ce105]884 first->x.x[i] = atof(argv[argptr+1+i]);
885 finder = periode->start;
886 while (finder != periode->end) {
887 finder = finder->next;
888 if (strncmp(finder->symbol,argv[argptr+1],3) == 0) {
889 first->type = finder;
890 break;
[a0bcf1]891 }
[3ce105]892 }
893 mol->AddAtom(first); // add to molecule
894 argptr+=4;
895 break;
896 case 's':
897 ExitFlag = 1;
898 j = -1;
899 cout << Verbose(1) << "Scaling all ion positions by factor." << endl;
[f75030]900 factor = new double[NDIM];
[3ce105]901 factor[0] = atof(argv[argptr]);
902 if (argc > argptr+1)
903 argptr++;
904 factor[1] = atof(argv[argptr]);
905 if (argc > argptr+1)
906 argptr++;
907 factor[2] = atof(argv[argptr]);
908 mol->Scale(&factor);
[f75030]909 for (int i=0;i<NDIM;i++) {
[3ce105]910 j += i+1;
[f75030]911 x.x[i] = atof(argv[NDIM+i]);
[3ce105]912 mol->cell_size[j]*=factor[i];
913 }
[f75030]914 delete[](factor);
[3ce105]915 argptr+=1;
916 break;
917 case 'b':
918 ExitFlag = 1;
919 j = -1;
920 cout << Verbose(1) << "Centering atoms in config file within given simulation box." << endl;
921 j=-1;
[f75030]922 for (int i=0;i<NDIM;i++) {
[3ce105]923 j += i+1;
924 x.x[i] = atof(argv[argptr++]);
925 mol->cell_size[j] += x.x[i]*2.;
926 }
927 // center
928 mol->CenterInBox((ofstream *)&cout, &x);
929 // update Box of atoms by boundary
930 mol->SetBoxDimension(&x);
931 break;
932 case 'c':
933 ExitFlag = 1;
934 j = -1;
935 cout << Verbose(1) << "Centering atoms in config file within given additional boundary." << endl;
936 // make every coordinate positive
937 mol->CenterEdge((ofstream *)&cout, &x);
938 // update Box of atoms by boundary
939 mol->SetBoxDimension(&x);
940 // translate each coordinate by boundary
941 j=-1;
[f75030]942 for (int i=0;i<NDIM;i++) {
[3ce105]943 j += i+1;
944 x.x[i] = atof(argv[argptr++]);
945 mol->cell_size[j] += x.x[i]*2.;
946 }
947 mol->Translate((const vector *)&x);
948 break;
949 case 'r':
950 ExitFlag = 1;
951 cout << Verbose(1) << "Converting config file from supposed old to new syntax." << endl;
952 break;
953 case 'f':
954 if (ExitFlag ==0) ExitFlag = 2; // only set if not already by other command line switch
955 cout << "Fragmenting molecule with bond distance " << argv[argptr] << " angstroem, order of " << argv[argptr+1] << "." << endl;
956 if (argc >= argptr+2) {
957 cout << Verbose(0) << "Creating connection matrix..." << endl;
958 start = clock();
[5eb05a]959 mol->CreateAdjacencyList((ofstream *)&cout, atof(argv[argptr++]), configuration.GetIsAngstroem());
[3ce105]960 cout << Verbose(0) << "Fragmenting molecule with current connection matrix ..." << endl;
961 if (mol->first->next != mol->last) {
962 mol->FragmentMolecule((ofstream *)&cout, atoi(argv[argptr]), &configuration);
[8129de]963 }
[3ce105]964 end = clock();
965 cout << Verbose(0) << "Clocks for this operation: " << (end-start) << ", time: " << ((double)(end-start)/CLOCKS_PER_SEC) << "s." << endl;
[8129de]966 argptr+=1;
[3ce105]967 } else {
968 cerr << "Not enough arguments for fragmentation: -f <max. bond distance> <bond order>" << endl;
969 }
970 break;
[32b6dc]971 case 'm':
972 ExitFlag = 1;
[e292c10]973 j = atoi(argv[argptr++]);
974 if ((j<0) || (j>1)) {
975 cerr << Verbose(1) << "ERROR: Argument of '-m' should be either 0 for no-rotate or 1 for rotate." << endl;
976 j = 0;
977 }
978 if (j)
979 cout << Verbose(0) << "Converting to prinicipal axis system." << endl;
980 else
981 cout << Verbose(0) << "Evaluating prinicipal axis." << endl;
982 mol->PrincipalAxisSystem((ofstream *)&cout, (bool)j);
983 break;
984 case 'o':
985 ExitFlag = 1;
986 cout << Verbose(0) << "Evaluating volume of the convex envelope.";
[7fcea6]987 VolumeOfConvexEnvelope((ofstream *)&cout, &configuration, NULL, mol);
[32b6dc]988 break;
[3f0c46]989 case 'u':
990 {
991 double tmp;
992 ExitFlag = 1;
993 cout << Verbose(0) << "Evaluating necessary cell volume for a cluster suspended in water.";
994 tmp = atof(argv[argptr++]);
995 if (tmp < 1.0) {
996 cerr << Verbose(0) << "Density must be greater than 1.0g/cm^3 !" << endl;
997 tmp = 1.3;
998 }
999// for(int i=0;i<NDIM;i++) {
1000// repetition[i] = atoi(argv[argptr++]);
1001// if (repetition[i] < 1)
1002// cerr << Verbose(0) << "ERROR: repetition value must be greater 1!" << endl;
1003// repetition[i] = 1;
1004// }
1005 PrepareClustersinWater((ofstream *)&cout, &configuration, mol, tmp);
1006 }
1007 break;
1008 case 'd':
1009 for (int axis = 1; axis <= NDIM; axis++) {
1010 int faktor = atoi(argv[argptr++]);
1011 int count;
1012 element ** Elements;
1013 vector ** Vectors;
1014 if (faktor < 1) {
1015 cerr << Verbose(0) << "ERROR: Repetition faktor mus be greater than 1!" << endl;
1016 faktor = 1;
1017 }
1018 mol->CountAtoms((ofstream *)&cout); // recount atoms
1019 if (mol->AtomCount != 0) { // if there is more than none
1020 count = mol->AtomCount; // is changed becausing of adding, thus has to be stored away beforehand
1021 Elements = new element *[count];
1022 Vectors = new vector *[count];
1023 j = 0;
1024 first = mol->start;
1025 while (first->next != mol->end) { // make a list of all atoms with coordinates and element
1026 first = first->next;
1027 Elements[j] = first->type;
1028 Vectors[j] = &first->x;
1029 j++;
1030 }
1031 if (count != j)
1032 cout << Verbose(0) << "ERROR: AtomCount " << count << " is not equal to number of atoms in molecule " << j << "!" << endl;
1033 x.Zero();
1034 y.Zero();
1035 y.x[abs(axis)-1] = mol->cell_size[(abs(axis) == 2) ? 2 : ((abs(axis) == 3) ? 5 : 0)] * abs(axis)/axis; // last term is for sign, first is for magnitude
1036 for (int i=1;i<faktor;i++) { // then add this list with respective translation factor times
1037 x.AddVector(&y); // per factor one cell width further
1038 for (int k=count;k--;) { // go through every atom of the original cell
1039 first = new atom(); // create a new body
1040 first->x.CopyVector(Vectors[k]); // use coordinate of original atom
1041 first->x.AddVector(&x); // translate the coordinates
1042 first->type = Elements[k]; // insert original element
1043 mol->AddAtom(first); // and add to the molecule (which increments ElementsInMolecule, AtomCount, ...)
1044 }
1045 }
1046 // free memory
1047 delete[](Elements);
1048 delete[](Vectors);
1049 // correct cell size
1050 if (axis < 0) { // if sign was negative, we have to translate everything
1051 x.Zero();
1052 x.AddVector(&y);
1053 x.Scale(-(faktor-1));
1054 mol->Translate(&x);
1055 }
1056 mol->cell_size[(abs(axis) == 2) ? 2 : ((abs(axis) == 3) ? 5 : 0)] *= faktor;
1057 }
1058 }
1059 break;
[3ce105]1060 default: // no match? Step on
[e292c10]1061 if (argv[argptr][0] != '-') // if it started with a '-' we've already made a step!
1062 argptr++;
[3ce105]1063 break;
[a0bcf1]1064 }
1065 }
[3ce105]1066 } else argptr++;
1067 } while (argptr < argc);
[0e348e]1068 if (ExitFlag == 1) // 1 means save and exit
[8129de]1069 SaveConfig(ConfigFileName, &configuration, periode, mol);
[0e348e]1070 if (ExitFlag >= 1) { // 2 means just exit
[f5306f]1071 delete(mol);
1072 delete(periode);
[8b486e]1073 return (1);
[f5306f]1074 }
[584691]1075 } else { // no arguments, hence scan the elements db
[8b486e]1076 if (periode->LoadPeriodentafel(PathToDatabases))
[584691]1077 cout << Verbose(0) << "Element list loaded successfully." << endl;
1078 else
1079 cout << Verbose(0) << "Element list loading failed." << endl;
1080 configuration.RetrieveConfigPathAndName("main_pcp_linux");
[a0bcf1]1081 }
[3ce105]1082 return(0);
1083};
1084
1085/********************************************** Main routine **************************************/
[a0bcf1]1086
[3ce105]1087int main(int argc, char **argv)
1088{
1089 periodentafel *periode = new periodentafel; // and a period table of all elements
1090 molecule *mol = new molecule(periode); // first we need an empty molecule
1091 config configuration;
1092 double tmp1;
1093 double bond, min_bond;
1094 atom *first, *second;
1095 char choice; // menu choice char
1096 vector x,y,z,n; // coordinates for absolute point in cell volume
1097 double *factor; // unit factor if desired
1098 bool valid; // flag if input was valid or not
1099 ifstream test;
1100 ofstream output;
1101 string line;
1102 char filename[MAXSTRINGSIZE];
1103 char *ConfigFileName = NULL;
1104 char *ElementsFileName = NULL;
1105 int Z;
1106 int j, axis, count, faktor;
1107 int *MinimumRingSize = NULL;
1108 MoleculeLeafClass *Subgraphs = NULL;
1109 clock_t start,end;
1110 element **Elements;
1111 vector **Vectors;
1112
1113 // =========================== PARSE COMMAND LINE OPTIONS ====================================
1114 j = ParseCommandLineOptions(argc, argv, mol, periode, configuration, ConfigFileName, ElementsFileName);
[8b486e]1115 if (j == 1) return 0; // just for -v and -h options
[3ce105]1116 if (j) return j; // something went wrong
[a0bcf1]1117
1118 // General stuff
1119 if (mol->cell_size[0] == 0.) {
1120 cout << Verbose(0) << "enter lower triadiagonal form of basis matrix" << endl << endl;
1121 for (int i=0;i<6;i++) {
1122 cout << Verbose(1) << "Cell size" << i << ": ";
1123 cin >> mol->cell_size[i];
1124 }
1125 }
1126
[a829d1]1127 // =========================== START INTERACTIVE SESSION ====================================
1128
[a0bcf1]1129 // now the main construction loop
1130 cout << Verbose(0) << endl << "Now comes the real construction..." << endl;
1131 do {
1132 cout << Verbose(0) << endl << endl;
1133 cout << Verbose(0) << "============Element list=======================" << endl;
1134 mol->Checkout((ofstream *)&cout);
1135 cout << Verbose(0) << "============Atom list==========================" << endl;
1136 mol->Output((ofstream *)&cout);
1137 cout << Verbose(0) << "============Menu===============================" << endl;
1138 cout << Verbose(0) << "a - add an atom" << endl;
1139 cout << Verbose(0) << "r - remove an atom" << endl;
1140 cout << Verbose(0) << "b - scale a bond between atoms" << endl;
1141 cout << Verbose(0) << "u - change an atoms element" << endl;
1142 cout << Verbose(0) << "l - measure lengths, angles, ... for an atom" << endl;
1143 cout << Verbose(0) << "-----------------------------------------------" << endl;
1144 cout << Verbose(0) << "p - Parse xyz file" << endl;
1145 cout << Verbose(0) << "e - edit the current configuration" << endl;
1146 cout << Verbose(0) << "o - create connection matrix" << endl;
1147 cout << Verbose(0) << "f - fragment molecule many-body bond order style" << endl;
1148 cout << Verbose(0) << "-----------------------------------------------" << endl;
1149 cout << Verbose(0) << "d - duplicate molecule/periodic cell" << endl;
1150 cout << Verbose(0) << "i - realign molecule" << endl;
1151 cout << Verbose(0) << "m - mirror all molecules" << endl;
1152 cout << Verbose(0) << "t - translate molecule by vector" << endl;
1153 cout << Verbose(0) << "c - scale by unit transformation" << endl;
1154 cout << Verbose(0) << "g - center atoms in box" << endl;
1155 cout << Verbose(0) << "-----------------------------------------------" << endl;
1156 cout << Verbose(0) << "s - save current setup to config file" << endl;
1157 cout << Verbose(0) << "T - call the current test routine" << endl;
1158 cout << Verbose(0) << "q - quit" << endl;
1159 cout << Verbose(0) << "===============================================" << endl;
1160 cout << Verbose(0) << "Input: ";
1161 cin >> choice;
1162
1163 switch (choice) {
1164 default:
1165 case 'a': // add atom
1166 AddAtoms(periode, mol);
1167 choice = 'a';
1168 break;
1169
[3ce105]1170 case 'b': // scale a bond
1171 cout << Verbose(0) << "Scaling bond length between two atoms." << endl;
1172 first = mol->AskAtom("Enter first (fixed) atom: ");
1173 second = mol->AskAtom("Enter second (shifting) atom: ");
1174 min_bond = 0.;
[f75030]1175 for (int i=NDIM;i--;)
[3ce105]1176 min_bond += (first->x.x[i]-second->x.x[i])*(first->x.x[i] - second->x.x[i]);
1177 min_bond = sqrt(min_bond);
1178 cout << Verbose(0) << "Current Bond length between " << first->type->name << " Atom " << first->nr << " and " << second->type->name << " Atom " << second->nr << ": " << min_bond << " a.u." << endl;
1179 cout << Verbose(0) << "Enter new bond length [a.u.]: ";
1180 cin >> bond;
[f75030]1181 for (int i=NDIM;i--;) {
[3ce105]1182 second->x.x[i] -= (second->x.x[i]-first->x.x[i])/min_bond*(min_bond-bond);
1183 }
1184 //cout << Verbose(0) << "New coordinates of Atom " << second->nr << " are: ";
1185 //second->Output(second->type->No, 1, (ofstream *)&cout);
1186 break;
1187
1188 case 'c': // unit scaling of the metric
1189 cout << Verbose(0) << "Angstroem -> Bohrradius: 1.8897261\t\tBohrradius -> Angstroem: 0.52917721" << endl;
1190 cout << Verbose(0) << "Enter three factors: ";
1191 factor = new double[NDIM];
1192 cin >> factor[0];
1193 cin >> factor[1];
1194 cin >> factor[2];
1195 valid = true;
1196 mol->Scale(&factor);
1197 delete[](factor);
1198 break;
1199
[a0bcf1]1200 case 'd': // duplicate the periodic cell along a given axis, given times
1201 cout << Verbose(0) << "State the axis [(+-)123]: ";
1202 cin >> axis;
1203 cout << Verbose(0) << "State the factor: ";
1204 cin >> faktor;
1205
1206 mol->CountAtoms((ofstream *)&cout); // recount atoms
1207 if (mol->AtomCount != 0) { // if there is more than none
1208 count = mol->AtomCount; // is changed becausing of adding, thus has to be stored away beforehand
[3ce105]1209 Elements = new element *[count];
1210 Vectors = new vector *[count];
[a0bcf1]1211 j = 0;
1212 first = mol->start;
1213 while (first->next != mol->end) { // make a list of all atoms with coordinates and element
1214 first = first->next;
1215 Elements[j] = first->type;
1216 Vectors[j] = &first->x;
1217 j++;
1218 }
1219 if (count != j)
1220 cout << Verbose(0) << "ERROR: AtomCount " << count << " is not equal to number of atoms in molecule " << j << "!" << endl;
1221 x.Zero();
1222 y.Zero();
1223 y.x[abs(axis)-1] = mol->cell_size[(abs(axis) == 2) ? 2 : ((abs(axis) == 3) ? 5 : 0)] * abs(axis)/axis; // last term is for sign, first is for magnitude
1224 for (int i=1;i<faktor;i++) { // then add this list with respective translation factor times
1225 x.AddVector(&y); // per factor one cell width further
[f75030]1226 for (int k=count;k--;) { // go through every atom of the original cell
[a0bcf1]1227 first = new atom(); // create a new body
1228 first->x.CopyVector(Vectors[k]); // use coordinate of original atom
1229 first->x.AddVector(&x); // translate the coordinates
1230 first->type = Elements[k]; // insert original element
1231 mol->AddAtom(first); // and add to the molecule (which increments ElementsInMolecule, AtomCount, ...)
1232 }
1233 }
1234 if (mol->first->next != mol->last) // if connect matrix is present already, redo it
[5eb05a]1235 mol->CreateAdjacencyList((ofstream *)&cout, mol->BondDistance, configuration.GetIsAngstroem());
[a0bcf1]1236 // free memory
[3ce105]1237 delete[](Elements);
1238 delete[](Vectors);
[a0bcf1]1239 // correct cell size
1240 if (axis < 0) { // if sign was negative, we have to translate everything
1241 x.Zero();
1242 x.AddVector(&y);
1243 x.Scale(-(faktor-1));
1244 mol->Translate(&x);
1245 }
1246 mol->cell_size[(abs(axis) == 2) ? 2 : ((abs(axis) == 3) ? 5 : 0)] *= faktor;
1247 }
1248 break;
1249
[3ce105]1250 case 'e': // edit each field of the configuration
1251 configuration.Edit(mol);
1252 break;
1253
1254 case 'f':
1255 FragmentAtoms(mol, &configuration);
1256 break;
1257
[a0bcf1]1258 case 'g': // center the atoms
1259 CenterAtoms(mol);
1260 break;
1261
1262 case 'i': // align all atoms
1263 AlignAtoms(periode, mol);
1264 break;
1265
1266 case 'l': // measure distances or angles
[32b6dc]1267 MeasureAtoms(periode, mol, &configuration);
[a0bcf1]1268 break;
1269
[3ce105]1270 case 'm': // mirror atoms along a given axis
1271 MirrorAtoms(mol);
[a0bcf1]1272 break;
[3ce105]1273
[a0bcf1]1274 case 'o': // create the connection matrix
1275 cout << Verbose(0) << "What's the maximum bond distance: ";
1276 cin >> tmp1;
1277 start = clock();
[5eb05a]1278 mol->CreateAdjacencyList((ofstream *)&cout, tmp1, configuration.GetIsAngstroem());
[a0bcf1]1279 //mol->CreateListOfBondsPerAtom((ofstream *)&cout);
[32b6dc]1280 Subgraphs = mol->DepthFirstSearchAnalysis((ofstream *)&cout, MinimumRingSize);
[a0bcf1]1281 while (Subgraphs->next != NULL) {
1282 Subgraphs = Subgraphs->next;
1283 delete(Subgraphs->previous);
1284 }
1285 delete(Subgraphs); // we don't need the list here, so free everything
[0716ff4]1286 delete[](MinimumRingSize);
[a0bcf1]1287 Subgraphs = NULL;
1288 end = clock();
1289 cout << Verbose(0) << "Clocks for this operation: " << (end-start) << ", time: " << ((double)(end-start)/CLOCKS_PER_SEC) << "s." << endl;
1290 break;
1291
[3ce105]1292 case 'p': // parse and XYZ file
1293 cout << Verbose(0) << "Format should be XYZ with: ShorthandOfElement\tX\tY\tZ" << endl;
1294 do {
1295 cout << Verbose(0) << "Enter file name: ";
1296 cin >> filename;
1297 } while (!mol->AddXYZFile(filename));
1298 break;
1299
1300 case 'q': // quit
1301 break;
[a0bcf1]1302
[3ce105]1303 case 'r': // remove atom
1304 RemoveAtoms(mol);
1305 break;
1306
1307 case 's': // save to config file
1308 SaveConfig(ConfigFileName, &configuration, periode, mol);
1309 break;
1310
1311 case 't': // translate all atoms
1312 cout << Verbose(0) << "Enter translation vector." << endl;
1313 x.AskPosition(mol->cell_size,0);
1314 mol->Translate((const vector *)&x);
1315 break;
1316
1317 case 'T':
1318 testroutine(mol);
1319 break;
1320
[a0bcf1]1321 case 'u': // change an atom's element
1322 first = NULL;
1323 do {
1324 cout << Verbose(0) << "Change the element of which atom: ";
1325 cin >> Z;
1326 } while ((first = mol->FindAtom(Z)) == NULL);
1327 cout << Verbose(0) << "New element by atomic number Z: ";
1328 cin >> Z;
1329 first->type = periode->FindElement(Z);
1330 cout << Verbose(0) << "Atom " << first->nr << "'s element is " << first->type->name << "." << endl;
1331 break;
1332 };
1333 } while (choice != 'q');
1334
1335 // save element data base
[e292c10]1336 if (periode->StorePeriodentafel(ElementsFileName)) //ElementsFileName
[a0bcf1]1337 cout << Verbose(0) << "Saving of elements.db successful." << endl;
1338 else
1339 cout << Verbose(0) << "Saving of elements.db failed." << endl;
1340
1341 // Free all
1342 if (Subgraphs != NULL) { // free disconnected subgraph list of DFS analysis was performed
1343 while (Subgraphs->next != NULL) {
1344 Subgraphs = Subgraphs->next;
1345 delete(Subgraphs->previous);
1346 }
1347 delete(Subgraphs);
1348 }
1349 delete(mol);
1350 delete(periode);
1351 return (0);
1352}
1353
1354/********************************************** E N D **************************************************/
Note: See TracBrowser for help on using the repository browser.