source: src/builder.cpp@ a5b2c3a

Action_Thermostats Add_AtomRandomPerturbation Add_FitFragmentPartialChargesAction Add_RotateAroundBondAction Add_SelectAtomByNameAction Added_ParseSaveFragmentResults AddingActions_SaveParseParticleParameters Adding_Graph_to_ChangeBondActions Adding_MD_integration_tests Adding_ParticleName_to_Atom Adding_StructOpt_integration_tests AtomFragments Automaking_mpqc_open AutomationFragmentation_failures Candidate_v1.5.4 Candidate_v1.6.0 Candidate_v1.6.1 ChangeBugEmailaddress ChangingTestPorts ChemicalSpaceEvaluator CombiningParticlePotentialParsing Combining_Subpackages Debian_Package_split Debian_package_split_molecuildergui_only Disabling_MemDebug Docu_Python_wait EmpiricalPotential_contain_HomologyGraph EmpiricalPotential_contain_HomologyGraph_documentation Enable_parallel_make_install Enhance_userguide Enhanced_StructuralOptimization Enhanced_StructuralOptimization_continued Example_ManyWaysToTranslateAtom Exclude_Hydrogens_annealWithBondGraph FitPartialCharges_GlobalError Fix_BoundInBox_CenterInBox_MoleculeActions Fix_ChargeSampling_PBC Fix_ChronosMutex Fix_FitPartialCharges Fix_FitPotential_needs_atomicnumbers Fix_ForceAnnealing Fix_IndependentFragmentGrids Fix_ParseParticles Fix_ParseParticles_split_forward_backward_Actions Fix_PopActions Fix_QtFragmentList_sorted_selection Fix_Restrictedkeyset_FragmentMolecule Fix_StatusMsg Fix_StepWorldTime_single_argument Fix_Verbose_Codepatterns Fix_fitting_potentials Fixes ForceAnnealing_goodresults ForceAnnealing_oldresults ForceAnnealing_tocheck ForceAnnealing_with_BondGraph ForceAnnealing_with_BondGraph_continued ForceAnnealing_with_BondGraph_continued_betteresults ForceAnnealing_with_BondGraph_contraction-expansion FragmentAction_writes_AtomFragments FragmentMolecule_checks_bonddegrees GeometryObjects Gui_Fixes Gui_displays_atomic_force_velocity ImplicitCharges IndependentFragmentGrids IndependentFragmentGrids_IndividualZeroInstances IndependentFragmentGrids_IntegrationTest IndependentFragmentGrids_Sole_NN_Calculation JobMarket_RobustOnKillsSegFaults JobMarket_StableWorkerPool JobMarket_unresolvable_hostname_fix MoreRobust_FragmentAutomation ODR_violation_mpqc_open PartialCharges_OrthogonalSummation PdbParser_setsAtomName PythonUI_with_named_parameters QtGui_reactivate_TimeChanged_changes Recreated_GuiChecks Rewrite_FitPartialCharges RotateToPrincipalAxisSystem_UndoRedo SaturateAtoms_findBestMatching SaturateAtoms_singleDegree StoppableMakroAction Subpackage_CodePatterns Subpackage_JobMarket Subpackage_LinearAlgebra Subpackage_levmar Subpackage_mpqc_open Subpackage_vmg Switchable_LogView ThirdParty_MPQC_rebuilt_buildsystem TrajectoryDependenant_MaxOrder TremoloParser_IncreasedPrecision TremoloParser_MultipleTimesteps TremoloParser_setsAtomName Ubuntu_1604_changes stable
Last change on this file since a5b2c3a was a5b2c3a, checked in by Frederik Heber <heber@…>, 16 years ago

Removing of atoms via command line possible

-R switchs allows for the removal of single atoms
-Left and Right Boundary are now called Upper and Lower Boundary which makes more sense

  • Property mode set to 100755
File size: 75.0 KB
RevLine 
[14de469]1/** \file builder.cpp
[a8bcea6]2 *
[14de469]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
[a8bcea6]8 *
[14de469]9 */
10
11/*! \mainpage Molecuilder - a molecular set builder
[a8bcea6]12 *
[14de469]13 * This introductory shall briefly make aquainted with the program, helping in installing and a first run.
[a8bcea6]14 *
[14de469]15 * \section about About the Program
[a8bcea6]16 *
[6ac7ee]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.
[a8bcea6]20 *
[6ac7ee]21 * A configuration file may be written that is compatible to the format used by PCP - a parallel Car-Parrinello
22 * molecular dynamics implementation.
[a8bcea6]23 *
[14de469]24 * \section install Installation
[a8bcea6]25 *
[6ac7ee]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
[a8bcea6]30 *
[6ac7ee]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
[a8bcea6]34 *
[14de469]35 * \section run Running
[a8bcea6]36 *
[6ac7ee]37 * The program can be executed by running: ./molecuilder
[a8bcea6]38 *
[6ac7ee]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.
[a8bcea6]42 *
[14de469]43 * \section ref References
[a8bcea6]44 *
[6ac7ee]45 * For the special configuration file format, see the documentation of pcp.
[a8bcea6]46 *
[14de469]47 */
48
49
50using namespace std;
51
[6ac7ee]52#include "boundary.hpp"
53#include "ellipsoid.hpp"
[14de469]54#include "helpers.hpp"
55#include "molecules.hpp"
56
[1907a7]57/********************************************* Subsubmenu routine ************************************/
[14de469]58
59/** Submenu for adding atoms to the molecule.
60 * \param *periode periodentafel
[1907a7]61 * \param *molecule molecules with atoms
[14de469]62 */
[7f3b9d]63static void AddAtoms(periodentafel *periode, molecule *mol)
[14de469]64{
[6ac7ee]65 atom *first, *second, *third, *fourth;
66 Vector **atoms;
67 Vector x,y,z,n; // coordinates for absolute point in cell volume
68 double a,b,c;
69 char choice; // menu choice char
70 bool valid;
71
72 cout << Verbose(0) << "===========ADD ATOM============================" << endl;
73 cout << Verbose(0) << " a - state absolute coordinates of atom" << endl;
74 cout << Verbose(0) << " b - state relative coordinates of atom wrt to reference point" << endl;
75 cout << Verbose(0) << " c - state relative coordinates of atom wrt to already placed atom" << endl;
76 cout << Verbose(0) << " d - state two atoms, two angles and a distance" << endl;
77 cout << Verbose(0) << " e - least square distance position to a set of atoms" << endl;
78 cout << Verbose(0) << "all else - go back" << endl;
79 cout << Verbose(0) << "===============================================" << endl;
80 cout << Verbose(0) << "Note: Specifiy angles in degrees not multiples of Pi!" << endl;
81 cout << Verbose(0) << "INPUT: ";
82 cin >> choice;
83
84 switch (choice) {
[1907a7]85 default:
86 cout << Verbose(0) << "Not a valid choice." << endl;
87 break;
[6ac7ee]88 case 'a': // absolute coordinates of atom
[1907a7]89 cout << Verbose(0) << "Enter absolute coordinates." << endl;
90 first = new atom;
91 first->x.AskPosition(mol->cell_size, false);
92 first->type = periode->AskElement(); // give type
93 mol->AddAtom(first); // add to molecule
[6ac7ee]94 break;
95
96 case 'b': // relative coordinates of atom wrt to reference point
[1907a7]97 first = new atom;
98 valid = true;
99 do {
100 if (!valid) cout << Verbose(0) << "Resulting position out of cell." << endl;
101 cout << Verbose(0) << "Enter reference coordinates." << endl;
102 x.AskPosition(mol->cell_size, true);
103 cout << Verbose(0) << "Enter relative coordinates." << endl;
104 first->x.AskPosition(mol->cell_size, false);
105 first->x.AddVector((const Vector *)&x);
106 cout << Verbose(0) << "\n";
107 } while (!(valid = mol->CheckBounds((const Vector *)&first->x)));
108 first->type = periode->AskElement(); // give type
109 mol->AddAtom(first); // add to molecule
[6ac7ee]110 break;
111
112 case 'c': // relative coordinates of atom wrt to already placed atom
[1907a7]113 first = new atom;
114 valid = true;
115 do {
116 if (!valid) cout << Verbose(0) << "Resulting position out of cell." << endl;
117 second = mol->AskAtom("Enter atom number: ");
118 cout << Verbose(0) << "Enter relative coordinates." << endl;
119 first->x.AskPosition(mol->cell_size, false);
120 for (int i=NDIM;i--;) {
121 first->x.x[i] += second->x.x[i];
122 }
123 } while (!(valid = mol->CheckBounds((const Vector *)&first->x)));
124 first->type = periode->AskElement(); // give type
125 mol->AddAtom(first); // add to molecule
126 break;
127
128 case 'd': // two atoms, two angles and a distance
129 first = new atom;
130 valid = true;
131 do {
132 if (!valid) {
133 cout << Verbose(0) << "Resulting coordinates out of cell - ";
134 first->x.Output((ofstream *)&cout);
135 cout << Verbose(0) << endl;
136 }
137 cout << Verbose(0) << "First, we need two atoms, the first atom is the central, while the second is the outer one." << endl;
138 second = mol->AskAtom("Enter central atom: ");
139 third = mol->AskAtom("Enter second atom (specifying the axis for first angle): ");
140 fourth = mol->AskAtom("Enter third atom (specifying a plane for second angle): ");
141 a = ask_value("Enter distance between central (first) and new atom: ");
142 b = ask_value("Enter angle between new, first and second atom (degrees): ");
143 b *= M_PI/180.;
144 bound(&b, 0., 2.*M_PI);
145 c = ask_value("Enter second angle between new and normal vector of plane defined by first, second and third atom (degrees): ");
146 c *= M_PI/180.;
147 bound(&c, -M_PI, M_PI);
148 cout << Verbose(0) << "radius: " << a << "\t phi: " << b*180./M_PI << "\t theta: " << c*180./M_PI << endl;
[14de469]149/*
[1907a7]150 second->Output(1,1,(ofstream *)&cout);
151 third->Output(1,2,(ofstream *)&cout);
152 fourth->Output(1,3,(ofstream *)&cout);
153 n.MakeNormalvector((const vector *)&second->x, (const vector *)&third->x, (const vector *)&fourth->x);
154 x.Copyvector(&second->x);
155 x.SubtractVector(&third->x);
156 x.Copyvector(&fourth->x);
157 x.SubtractVector(&third->x);
158
159 if (!z.SolveSystem(&x,&y,&n, b, c, a)) {
160 cout << Verbose(0) << "Failure solving self-dependent linear system!" << endl;
161 continue;
162 }
163 cout << Verbose(0) << "resulting relative coordinates: ";
164 z.Output((ofstream *)&cout);
165 cout << Verbose(0) << endl;
166 */
167 // calc axis vector
168 x.CopyVector(&second->x);
169 x.SubtractVector(&third->x);
170 x.Normalize();
171 cout << "x: ",
172 x.Output((ofstream *)&cout);
173 cout << endl;
174 z.MakeNormalVector(&second->x,&third->x,&fourth->x);
175 cout << "z: ",
176 z.Output((ofstream *)&cout);
177 cout << endl;
178 y.MakeNormalVector(&x,&z);
179 cout << "y: ",
180 y.Output((ofstream *)&cout);
181 cout << endl;
182
183 // rotate vector around first angle
184 first->x.CopyVector(&x);
185 first->x.RotateVector(&z,b - M_PI);
186 cout << "Rotated vector: ",
187 first->x.Output((ofstream *)&cout);
188 cout << endl;
189 // remove the projection onto the rotation plane of the second angle
190 n.CopyVector(&y);
191 n.Scale(first->x.Projection(&y));
192 cout << "N1: ",
193 n.Output((ofstream *)&cout);
194 cout << endl;
195 first->x.SubtractVector(&n);
196 cout << "Subtracted vector: ",
197 first->x.Output((ofstream *)&cout);
198 cout << endl;
199 n.CopyVector(&z);
200 n.Scale(first->x.Projection(&z));
201 cout << "N2: ",
202 n.Output((ofstream *)&cout);
203 cout << endl;
204 first->x.SubtractVector(&n);
205 cout << "2nd subtracted vector: ",
206 first->x.Output((ofstream *)&cout);
207 cout << endl;
208
209 // rotate another vector around second angle
210 n.CopyVector(&y);
211 n.RotateVector(&x,c - M_PI);
212 cout << "2nd Rotated vector: ",
213 n.Output((ofstream *)&cout);
214 cout << endl;
215
216 // add the two linear independent vectors
217 first->x.AddVector(&n);
218 first->x.Normalize();
219 first->x.Scale(a);
220 first->x.AddVector(&second->x);
221
222 cout << Verbose(0) << "resulting coordinates: ";
223 first->x.Output((ofstream *)&cout);
224 cout << Verbose(0) << endl;
225 } while (!(valid = mol->CheckBounds((const Vector *)&first->x)));
226 first->type = periode->AskElement(); // give type
227 mol->AddAtom(first); // add to molecule
[6ac7ee]228 break;
229
230 case 'e': // least square distance position to a set of atoms
[1907a7]231 first = new atom;
232 atoms = new (Vector*[128]);
233 valid = true;
234 for(int i=128;i--;)
235 atoms[i] = NULL;
236 int i=0, j=0;
237 cout << Verbose(0) << "Now we need at least three molecules.\n";
238 do {
239 cout << Verbose(0) << "Enter " << i+1 << "th atom: ";
240 cin >> j;
241 if (j != -1) {
242 second = mol->FindAtom(j);
243 atoms[i++] = &(second->x);
244 }
245 } while ((j != -1) && (i<128));
246 if (i >= 2) {
247 first->x.LSQdistance(atoms, i);
248
249 first->x.Output((ofstream *)&cout);
250 first->type = periode->AskElement(); // give type
251 mol->AddAtom(first); // add to molecule
252 } else {
253 delete first;
254 cout << Verbose(0) << "Please enter at least two vectors!\n";
255 }
[6ac7ee]256 break;
257 };
[14de469]258};
259
260/** Submenu for centering the atoms in the molecule.
[1907a7]261 * \param *mol molecule with all the atoms
[14de469]262 */
[7f3b9d]263static void CenterAtoms(molecule *mol)
[14de469]264{
[6ac7ee]265 Vector x, y, helper;
266 char choice; // menu choice char
267
268 cout << Verbose(0) << "===========CENTER ATOMS=========================" << endl;
269 cout << Verbose(0) << " a - on origin" << endl;
270 cout << Verbose(0) << " b - on center of gravity" << endl;
271 cout << Verbose(0) << " c - within box with additional boundary" << endl;
272 cout << Verbose(0) << " d - within given simulation box" << endl;
273 cout << Verbose(0) << "all else - go back" << endl;
274 cout << Verbose(0) << "===============================================" << endl;
275 cout << Verbose(0) << "INPUT: ";
276 cin >> choice;
277
278 switch (choice) {
279 default:
280 cout << Verbose(0) << "Not a valid choice." << endl;
281 break;
282 case 'a':
283 cout << Verbose(0) << "Centering atoms in config file on origin." << endl;
284 mol->CenterOrigin((ofstream *)&cout, &x);
285 break;
286 case 'b':
287 cout << Verbose(0) << "Centering atoms in config file on center of gravity." << endl;
288 mol->CenterGravity((ofstream *)&cout, &x);
289 break;
290 case 'c':
291 cout << Verbose(0) << "Centering atoms in config file within given additional boundary." << endl;
292 for (int i=0;i<NDIM;i++) {
293 cout << Verbose(0) << "Enter axis " << i << " boundary: ";
294 cin >> y.x[i];
295 }
296 mol->CenterEdge((ofstream *)&cout, &x); // make every coordinate positive
297 mol->Translate(&y); // translate by boundary
298 helper.CopyVector(&y);
299 helper.Scale(2.);
300 helper.AddVector(&x);
301 mol->SetBoxDimension(&helper); // update Box of atoms by boundary
302 break;
303 case 'd':
304 cout << Verbose(1) << "Centering atoms in config file within given simulation box." << endl;
305 for (int i=0;i<NDIM;i++) {
306 cout << Verbose(0) << "Enter axis " << i << " boundary: ";
307 cin >> x.x[i];
308 }
309 // center
310 mol->CenterInBox((ofstream *)&cout, &x);
311 // update Box of atoms by boundary
312 mol->SetBoxDimension(&x);
313 break;
314 }
[14de469]315};
316
317/** Submenu for aligning the atoms in the molecule.
318 * \param *periode periodentafel
[1907a7]319 * \param *mol molecule with all the atoms
[14de469]320 */
[7f3b9d]321static void AlignAtoms(periodentafel *periode, molecule *mol)
[14de469]322{
[6ac7ee]323 atom *first, *second, *third;
324 Vector x,n;
325 char choice; // menu choice char
326
327 cout << Verbose(0) << "===========ALIGN ATOMS=========================" << endl;
328 cout << Verbose(0) << " a - state three atoms defining align plane" << endl;
329 cout << Verbose(0) << " b - state alignment vector" << endl;
330 cout << Verbose(0) << " c - state two atoms in alignment direction" << endl;
331 cout << Verbose(0) << " d - align automatically by least square fit" << endl;
332 cout << Verbose(0) << "all else - go back" << endl;
333 cout << Verbose(0) << "===============================================" << endl;
334 cout << Verbose(0) << "INPUT: ";
335 cin >> choice;
336
337 switch (choice) {
338 default:
339 case 'a': // three atoms defining mirror plane
340 first = mol->AskAtom("Enter first atom: ");
341 second = mol->AskAtom("Enter second atom: ");
342 third = mol->AskAtom("Enter third atom: ");
343
344 n.MakeNormalVector((const Vector *)&first->x,(const Vector *)&second->x,(const Vector *)&third->x);
345 break;
346 case 'b': // normal vector of mirror plane
347 cout << Verbose(0) << "Enter normal vector of mirror plane." << endl;
348 n.AskPosition(mol->cell_size,0);
349 n.Normalize();
350 break;
351 case 'c': // three atoms defining mirror plane
352 first = mol->AskAtom("Enter first atom: ");
353 second = mol->AskAtom("Enter second atom: ");
354
355 n.CopyVector((const Vector *)&first->x);
356 n.SubtractVector((const Vector *)&second->x);
357 n.Normalize();
358 break;
359 case 'd':
360 char shorthand[4];
361 Vector a;
362 struct lsq_params param;
363 do {
364 fprintf(stdout, "Enter the element of atoms to be chosen: ");
365 fscanf(stdin, "%3s", shorthand);
366 } while ((param.type = periode->FindElement(shorthand)) == NULL);
367 cout << Verbose(0) << "Element is " << param.type->name << endl;
368 mol->GetAlignvector(&param);
369 for (int i=NDIM;i--;) {
370 x.x[i] = gsl_vector_get(param.x,i);
371 n.x[i] = gsl_vector_get(param.x,i+NDIM);
372 }
373 gsl_vector_free(param.x);
374 cout << Verbose(0) << "Offset vector: ";
375 x.Output((ofstream *)&cout);
376 cout << Verbose(0) << endl;
377 n.Normalize();
378 break;
379 };
380 cout << Verbose(0) << "Alignment vector: ";
381 n.Output((ofstream *)&cout);
382 cout << Verbose(0) << endl;
383 mol->Align(&n);
[14de469]384};
385
386/** Submenu for mirroring the atoms in the molecule.
[1907a7]387 * \param *mol molecule with all the atoms
[14de469]388 */
[7f3b9d]389static void MirrorAtoms(molecule *mol)
[14de469]390{
[6ac7ee]391 atom *first, *second, *third;
392 Vector n;
393 char choice; // menu choice char
394
395 cout << Verbose(0) << "===========MIRROR ATOMS=========================" << endl;
396 cout << Verbose(0) << " a - state three atoms defining mirror plane" << endl;
397 cout << Verbose(0) << " b - state normal vector of mirror plane" << endl;
398 cout << Verbose(0) << " c - state two atoms in normal direction" << endl;
399 cout << Verbose(0) << "all else - go back" << endl;
400 cout << Verbose(0) << "===============================================" << endl;
401 cout << Verbose(0) << "INPUT: ";
402 cin >> choice;
403
404 switch (choice) {
405 default:
406 case 'a': // three atoms defining mirror plane
407 first = mol->AskAtom("Enter first atom: ");
408 second = mol->AskAtom("Enter second atom: ");
409 third = mol->AskAtom("Enter third atom: ");
410
411 n.MakeNormalVector((const Vector *)&first->x,(const Vector *)&second->x,(const Vector *)&third->x);
412 break;
413 case 'b': // normal vector of mirror plane
414 cout << Verbose(0) << "Enter normal vector of mirror plane." << endl;
415 n.AskPosition(mol->cell_size,0);
416 n.Normalize();
417 break;
418 case 'c': // three atoms defining mirror plane
419 first = mol->AskAtom("Enter first atom: ");
420 second = mol->AskAtom("Enter second atom: ");
421
422 n.CopyVector((const Vector *)&first->x);
423 n.SubtractVector((const Vector *)&second->x);
424 n.Normalize();
425 break;
426 };
427 cout << Verbose(0) << "Normal vector: ";
428 n.Output((ofstream *)&cout);
429 cout << Verbose(0) << endl;
430 mol->Mirror((const Vector *)&n);
[14de469]431};
432
433/** Submenu for removing the atoms from the molecule.
[1907a7]434 * \param *mol molecule with all the atoms
[14de469]435 */
[7f3b9d]436static void RemoveAtoms(molecule *mol)
[14de469]437{
[c54da3]438 atom *first, *second, *third;
[6ac7ee]439 int axis;
440 double tmp1, tmp2;
441 char choice; // menu choice char
442
443 cout << Verbose(0) << "===========REMOVE ATOMS=========================" << endl;
444 cout << Verbose(0) << " a - state atom for removal by number" << endl;
445 cout << Verbose(0) << " b - keep only in radius around atom" << endl;
446 cout << Verbose(0) << " c - remove this with one axis greater value" << endl;
447 cout << Verbose(0) << "all else - go back" << endl;
448 cout << Verbose(0) << "===============================================" << endl;
449 cout << Verbose(0) << "INPUT: ";
450 cin >> choice;
451
452 switch (choice) {
453 default:
454 case 'a':
455 if (mol->RemoveAtom(mol->AskAtom("Enter number of atom within molecule: ")))
456 cout << Verbose(1) << "Atom removed." << endl;
457 else
458 cout << Verbose(1) << "Atom not found." << endl;
459 break;
460 case 'b':
[c54da3]461 third = mol->AskAtom("Enter number of atom as reference point: ");
[6ac7ee]462 cout << Verbose(0) << "Enter radius: ";
463 cin >> tmp1;
464 first = mol->start;
[c54da3]465 second = first->next;
466 while(second != mol->end) {
467 first = second;
468 second = first->next;
469 if (first->x.DistanceSquared((const Vector *)&third->x) > tmp1*tmp1) // distance to first above radius ...
[6ac7ee]470 mol->RemoveAtom(first);
471 }
472 break;
473 case 'c':
474 cout << Verbose(0) << "Which axis is it: ";
475 cin >> axis;
[a5b2c3a]476 cout << Verbose(0) << "Lower boundary: ";
[6ac7ee]477 cin >> tmp1;
[a5b2c3a]478 cout << Verbose(0) << "Upper boundary: ";
[6ac7ee]479 cin >> tmp2;
480 first = mol->start;
[a5b2c3a]481 second = first->next;
482 while(second != mol->end) {
483 first = second;
484 second = first->next;
485 if ((first->x.x[axis] < tmp1) || (first->x.x[axis] > tmp2)) {// out of boundary ...
486 //cout << "Atom " << *first << " with " << first->x.x[axis] << " on axis " << axis << " is out of bounds [" << tmp1 << "," << tmp2 << "]." << endl;
[6ac7ee]487 mol->RemoveAtom(first);
[a5b2c3a]488 }
[6ac7ee]489 }
490 break;
491 };
492 //mol->Output((ofstream *)&cout);
493 choice = 'r';
[14de469]494};
495
496/** Submenu for measuring out the atoms in the molecule.
497 * \param *periode periodentafel
[1907a7]498 * \param *mol molecule with all the atoms
[14de469]499 */
[d52ea1b]500static void MeasureAtoms(periodentafel *periode, molecule *mol, config *configuration)
[14de469]501{
[6ac7ee]502 atom *first, *second, *third;
503 Vector x,y;
504 double min[256], tmp1, tmp2, tmp3;
505 int Z;
506 char choice; // menu choice char
507
508 cout << Verbose(0) << "===========MEASURE ATOMS=========================" << endl;
509 cout << Verbose(0) << " a - calculate bond length between one atom and all others" << endl;
510 cout << Verbose(0) << " b - calculate bond length between two atoms" << endl;
511 cout << Verbose(0) << " c - calculate bond angle" << endl;
512 cout << Verbose(0) << " d - calculate principal axis of the system" << endl;
513 cout << Verbose(0) << " e - calculate volume of the convex envelope" << endl;
514 cout << Verbose(0) << " f - calculate temperature from current velocity" << endl;
515 cout << Verbose(0) << " g - output all temperatures per step from velocities" << endl;
516 cout << Verbose(0) << "all else - go back" << endl;
517 cout << Verbose(0) << "===============================================" << endl;
518 cout << Verbose(0) << "INPUT: ";
519 cin >> choice;
520
521 switch(choice) {
522 default:
523 cout << Verbose(1) << "Not a valid choice." << endl;
524 break;
525 case 'a':
526 first = mol->AskAtom("Enter first atom: ");
527 for (int i=MAX_ELEMENTS;i--;)
528 min[i] = 0.;
529
530 second = mol->start;
531 while ((second->next != mol->end)) {
532 second = second->next; // advance
533 Z = second->type->Z;
534 tmp1 = 0.;
535 if (first != second) {
536 x.CopyVector((const Vector *)&first->x);
537 x.SubtractVector((const Vector *)&second->x);
538 tmp1 = x.Norm();
539 }
540 if ((tmp1 != 0.) && ((min[Z] == 0.) || (tmp1 < min[Z]))) min[Z] = tmp1;
541 //cout << Verbose(0) << "Bond length between Atom " << first->nr << " and " << second->nr << ": " << tmp1 << " a.u." << endl;
542 }
543 for (int i=MAX_ELEMENTS;i--;)
544 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;
545 break;
546
547 case 'b':
548 first = mol->AskAtom("Enter first atom: ");
549 second = mol->AskAtom("Enter second atom: ");
550 for (int i=NDIM;i--;)
551 min[i] = 0.;
552 x.CopyVector((const Vector *)&first->x);
553 x.SubtractVector((const Vector *)&second->x);
554 tmp1 = x.Norm();
555 cout << Verbose(1) << "Distance vector is ";
556 x.Output((ofstream *)&cout);
557 cout << "." << endl << "Norm of distance is " << tmp1 << "." << endl;
558 break;
559
560 case 'c':
561 cout << Verbose(0) << "Evaluating bond angle between three - first, central, last - atoms." << endl;
562 first = mol->AskAtom("Enter first atom: ");
563 second = mol->AskAtom("Enter central atom: ");
564 third = mol->AskAtom("Enter last atom: ");
565 tmp1 = tmp2 = tmp3 = 0.;
566 x.CopyVector((const Vector *)&first->x);
567 x.SubtractVector((const Vector *)&second->x);
568 y.CopyVector((const Vector *)&third->x);
569 y.SubtractVector((const Vector *)&second->x);
570 cout << Verbose(0) << "Bond angle between first atom Nr." << first->nr << ", central atom Nr." << second->nr << " and last atom Nr." << third->nr << ": ";
571 cout << Verbose(0) << (acos(x.ScalarProduct((const Vector *)&y)/(y.Norm()*x.Norm()))/M_PI*180.) << " degrees" << endl;
572 break;
573 case 'd':
574 cout << Verbose(0) << "Evaluating prinicipal axis." << endl;
575 cout << Verbose(0) << "Shall we rotate? [0/1]: ";
576 cin >> Z;
577 if ((Z >=0) && (Z <=1))
578 mol->PrincipalAxisSystem((ofstream *)&cout, (bool)Z);
579 else
580 mol->PrincipalAxisSystem((ofstream *)&cout, false);
581 break;
582 case 'e':
583 cout << Verbose(0) << "Evaluating volume of the convex envelope.";
584 VolumeOfConvexEnvelope((ofstream *)&cout, NULL, configuration, NULL, mol);
585 break;
586 case 'f':
587 mol->OutputTemperatureFromTrajectories((ofstream *)&cout, mol->MDSteps-1, mol->MDSteps, (ofstream *)&cout);
588 break;
589 case 'g':
590 {
591 char filename[255];
592 cout << "Please enter filename: " << endl;
593 cin >> filename;
594 cout << Verbose(1) << "Storing temperatures in " << filename << "." << endl;
595 ofstream *output = new ofstream(filename, ios::trunc);
596 if (!mol->OutputTemperatureFromTrajectories((ofstream *)&cout, 0, mol->MDSteps, output))
597 cout << Verbose(2) << "File could not be written." << endl;
598 else
599 cout << Verbose(2) << "File stored." << endl;
600 output->close();
601 delete(output);
602 }
603 break;
604 }
[14de469]605};
606
607/** Submenu for measuring out the atoms in the molecule.
[1907a7]608 * \param *mol molecule with all the atoms
[14de469]609 * \param *configuration configuration structure for the to be written config files of all fragments
610 */
[7f3b9d]611static void FragmentAtoms(molecule *mol, config *configuration)
[14de469]612{
[6ac7ee]613 int Order1;
614 clock_t start, end;
615
616 cout << Verbose(0) << "Fragmenting molecule with current connection matrix ..." << endl;
617 cout << Verbose(0) << "What's the desired bond order: ";
618 cin >> Order1;
619 if (mol->first->next != mol->last) { // there are bonds
620 start = clock();
621 mol->FragmentMolecule((ofstream *)&cout, Order1, configuration);
622 end = clock();
623 cout << Verbose(0) << "Clocks for this operation: " << (end-start) << ", time: " << ((double)(end-start)/CLOCKS_PER_SEC) << "s." << endl;
624 } else
625 cout << Verbose(0) << "Connection matrix has not yet been generated!" << endl;
[14de469]626};
627
[1907a7]628/********************************************** Submenu routine **************************************/
629
630/** Submenu for manipulating atoms.
631 * \param *periode periodentafel
632 * \param *molecules list of molecules whose atoms are to be manipulated
633 */
634static void ManipulateAtoms(periodentafel *periode, MoleculeListClass *molecules, config *configuration)
635{
636 atom *first, *second, *third, *fourth;
637 Vector **atoms;
638 molecule *mol = NULL;
639 Vector x,y,z,n; // coordinates for absolute point in cell volume
640 double *factor; // unit factor if desired
641 double a,b,c;
642 double bond, min_bond;
643 char choice; // menu choice char
644 bool valid;
645
646 cout << Verbose(0) << "=========MANIPULATE ATOMS======================" << endl;
647 cout << Verbose(0) << "a - add an atom" << endl;
648 cout << Verbose(0) << "r - remove an atom" << endl;
649 cout << Verbose(0) << "b - scale a bond between atoms" << endl;
650 cout << Verbose(0) << "u - change an atoms element" << endl;
651 cout << Verbose(0) << "l - measure lengths, angles, ... for an atom" << endl;
652 cout << Verbose(0) << "all else - go back" << endl;
653 cout << Verbose(0) << "===============================================" << endl;
654 if (molecules->NumberOfActiveMolecules() > 0)
655 cout << Verbose(0) << "WARNING: There is more than one molecule active! Atoms will be added to each." << endl;
656 cout << Verbose(0) << "INPUT: ";
657 cin >> choice;
658
659 switch (choice) {
660 default:
661 cout << Verbose(0) << "Not a valid choice." << endl;
662 break;
663
664 case 'a': // add atom
665 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++) {
666 mol = *ListRunner;
667 cout << Verbose(0) << "Current molecule is: " << mol->IndexNr << "\t" << mol->name << endl;
668 AddAtoms(periode, mol);
669 }
670 break;
671
672 case 'b': // scale a bond
673 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++) {
674 mol = *ListRunner;
675 cout << Verbose(0) << "Current molecule is: " << mol->IndexNr << "\t" << mol->name << endl;
676 cout << Verbose(0) << "Scaling bond length between two atoms." << endl;
677 first = mol->AskAtom("Enter first (fixed) atom: ");
678 second = mol->AskAtom("Enter second (shifting) atom: ");
679 min_bond = 0.;
680 for (int i=NDIM;i--;)
681 min_bond += (first->x.x[i]-second->x.x[i])*(first->x.x[i] - second->x.x[i]);
682 min_bond = sqrt(min_bond);
683 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;
684 cout << Verbose(0) << "Enter new bond length [a.u.]: ";
685 cin >> bond;
686 for (int i=NDIM;i--;) {
687 second->x.x[i] -= (second->x.x[i]-first->x.x[i])/min_bond*(min_bond-bond);
688 }
689 //cout << Verbose(0) << "New coordinates of Atom " << second->nr << " are: ";
690 //second->Output(second->type->No, 1, (ofstream *)&cout);
691 }
692 break;
693
694 case 'c': // unit scaling of the metric
695 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++) {
696 mol = *ListRunner;
697 cout << Verbose(0) << "Current molecule is: " << mol->IndexNr << "\t" << mol->name << endl;
698 cout << Verbose(0) << "Angstroem -> Bohrradius: 1.8897261\t\tBohrradius -> Angstroem: 0.52917721" << endl;
699 cout << Verbose(0) << "Enter three factors: ";
700 factor = new double[NDIM];
701 cin >> factor[0];
702 cin >> factor[1];
703 cin >> factor[2];
704 valid = true;
705 mol->Scale(&factor);
706 delete[](factor);
707 }
708 break;
709
710 case 'l': // measure distances or angles
711 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++) {
712 mol = *ListRunner;
713 cout << Verbose(0) << "Current molecule is: " << mol->IndexNr << "\t" << mol->name << endl;
714 MeasureAtoms(periode, mol, configuration);
715 }
716 break;
717
718 case 'r': // remove atom
719 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++) {
720 mol = *ListRunner;
721 cout << Verbose(0) << "Current molecule is: " << mol->IndexNr << "\t" << mol->name << endl;
722 RemoveAtoms(mol);
723 }
724 break;
725
726 case 'u': // change an atom's element
727 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++) {
728 int Z;
729 mol = *ListRunner;
730 cout << Verbose(0) << "Current molecule is: " << mol->IndexNr << "\t" << mol->name << endl;
731 first = NULL;
732 do {
733 cout << Verbose(0) << "Change the element of which atom: ";
734 cin >> Z;
735 } while ((first = mol->FindAtom(Z)) == NULL);
736 cout << Verbose(0) << "New element by atomic number Z: ";
737 cin >> Z;
738 first->type = periode->FindElement(Z);
739 cout << Verbose(0) << "Atom " << first->nr << "'s element is " << first->type->name << "." << endl;
740 }
741 break;
742 }
743};
744
745/** Submenu for manipulating molecules.
746 * \param *periode periodentafel
747 * \param *molecules list of molecule to manipulate
748 */
749static void ManipulateMolecules(periodentafel *periode, MoleculeListClass *molecules, config *configuration)
750{
751 atom *first, *second, *third, *fourth;
752 Vector **atoms;
753 Vector x,y,z,n; // coordinates for absolute point in cell volume
754 double a,b,c;
755 int j, axis, count, faktor;
756 char choice; // menu choice char
757 bool valid;
758 molecule *mol = NULL;
759 element **Elements;
760 Vector **vectors;
761 MoleculeLeafClass *Subgraphs = NULL;
762
763 cout << Verbose(0) << "=========MANIPULATE GLOBALLY===================" << endl;
764 cout << Verbose(0) << "c - scale by unit transformation" << endl;
765 cout << Verbose(0) << "d - duplicate molecule/periodic cell" << endl;
766 cout << Verbose(0) << "f - fragment molecule many-body bond order style" << endl;
767 cout << Verbose(0) << "g - center atoms in box" << endl;
768 cout << Verbose(0) << "i - realign molecule" << endl;
769 cout << Verbose(0) << "m - mirror all molecules" << endl;
770 cout << Verbose(0) << "o - create connection matrix" << endl;
771 cout << Verbose(0) << "t - translate molecule by vector" << endl;
772 cout << Verbose(0) << "all else - go back" << endl;
773 cout << Verbose(0) << "===============================================" << endl;
774 if (molecules->NumberOfActiveMolecules() > 0)
775 cout << Verbose(0) << "WARNING: There is more than one molecule active! Atoms will be added to each." << endl;
776 cout << Verbose(0) << "INPUT: ";
777 cin >> choice;
778
779 switch (choice) {
780 default:
781 cout << Verbose(0) << "Not a valid choice." << endl;
782 break;
783
784 case 'd': // duplicate the periodic cell along a given axis, given times
785 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++) {
786 mol = *ListRunner;
787 cout << Verbose(0) << "Current molecule is: " << mol->IndexNr << "\t" << mol->name << endl;
788 cout << Verbose(0) << "State the axis [(+-)123]: ";
789 cin >> axis;
790 cout << Verbose(0) << "State the factor: ";
791 cin >> faktor;
792
793 mol->CountAtoms((ofstream *)&cout); // recount atoms
794 if (mol->AtomCount != 0) { // if there is more than none
795 count = mol->AtomCount; // is changed becausing of adding, thus has to be stored away beforehand
796 Elements = new element *[count];
797 vectors = new Vector *[count];
798 j = 0;
799 first = mol->start;
800 while (first->next != mol->end) { // make a list of all atoms with coordinates and element
801 first = first->next;
802 Elements[j] = first->type;
803 vectors[j] = &first->x;
804 j++;
805 }
806 if (count != j)
807 cout << Verbose(0) << "ERROR: AtomCount " << count << " is not equal to number of atoms in molecule " << j << "!" << endl;
808 x.Zero();
809 y.Zero();
810 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
811 for (int i=1;i<faktor;i++) { // then add this list with respective translation factor times
812 x.AddVector(&y); // per factor one cell width further
813 for (int k=count;k--;) { // go through every atom of the original cell
814 first = new atom(); // create a new body
815 first->x.CopyVector(vectors[k]); // use coordinate of original atom
816 first->x.AddVector(&x); // translate the coordinates
817 first->type = Elements[k]; // insert original element
818 mol->AddAtom(first); // and add to the molecule (which increments ElementsInMolecule, AtomCount, ...)
819 }
820 }
821 if (mol->first->next != mol->last) // if connect matrix is present already, redo it
822 mol->CreateAdjacencyList((ofstream *)&cout, mol->BondDistance, configuration->GetIsAngstroem());
823 // free memory
824 delete[](Elements);
825 delete[](vectors);
826 // correct cell size
827 if (axis < 0) { // if sign was negative, we have to translate everything
828 x.Zero();
829 x.AddVector(&y);
830 x.Scale(-(faktor-1));
831 mol->Translate(&x);
832 }
833 mol->cell_size[(abs(axis) == 2) ? 2 : ((abs(axis) == 3) ? 5 : 0)] *= faktor;
834 }
835 }
836 break;
837
838 case 'f':
839 FragmentAtoms(mol, configuration);
840 break;
841
842 case 'g': // center the atoms
843 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++) {
844 mol = *ListRunner;
845 cout << Verbose(0) << "Current molecule is: " << mol->IndexNr << "\t" << mol->name << endl;
846 CenterAtoms(mol);
847 }
848 break;
849
850 case 'i': // align all atoms
851 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++) {
852 mol = *ListRunner;
853 cout << Verbose(0) << "Current molecule is: " << mol->IndexNr << "\t" << mol->name << endl;
854 AlignAtoms(periode, mol);
855 }
856 break;
857
858 case 'm': // mirror atoms along a given axis
859 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++) {
860 mol = *ListRunner;
861 cout << Verbose(0) << "Current molecule is: " << mol->IndexNr << "\t" << mol->name << endl;
862 MirrorAtoms(mol);
863 }
864 break;
865
866 case 'o': // create the connection matrix
867 {
868 double bonddistance;
869 clock_t start,end;
870 cout << Verbose(0) << "What's the maximum bond distance: ";
871 cin >> bonddistance;
872 start = clock();
873 mol->CreateAdjacencyList((ofstream *)&cout, bonddistance, configuration->GetIsAngstroem());
874 mol->CreateListOfBondsPerAtom((ofstream *)&cout);
875 end = clock();
876 cout << Verbose(0) << "Clocks for this operation: " << (end-start) << ", time: " << ((double)(end-start)/CLOCKS_PER_SEC) << "s." << endl;
877 }
878 break;
879
880 case 't': // translate all atoms
881 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++) {
882 mol = *ListRunner;
883 cout << Verbose(0) << "Current molecule is: " << mol->IndexNr << "\t" << mol->name << endl;
884 cout << Verbose(0) << "Enter translation vector." << endl;
885 x.AskPosition(mol->cell_size,0);
886 mol->Translate((const Vector *)&x);
887 }
888 break;
889 }
890 // Free all
891 if (Subgraphs != NULL) { // free disconnected subgraph list of DFS analysis was performed
892 while (Subgraphs->next != NULL) {
893 Subgraphs = Subgraphs->next;
894 delete(Subgraphs->previous);
895 }
896 delete(Subgraphs);
897 }
898};
899
900
901/** Submenu for creating new molecules.
902 * \param *periode periodentafel
903 * \param *molecules list of molecules to add to
904 */
905static void EditMolecules(periodentafel *periode, MoleculeListClass *molecules)
906{
907 char choice; // menu choice char
908 bool valid;
909 Vector Center;
910 int nr, count;
911 molecule *mol = NULL;
912 char *molname = NULL;
913 int length;
914 char filename[MAXSTRINGSIZE];
915
916 cout << Verbose(0) << "==========Edit MOLECULES=====================" << endl;
917 cout << Verbose(0) << "c - create new molecule" << endl;
918 cout << Verbose(0) << "l - load molecule from xyz file" << endl;
919 cout << Verbose(0) << "n - change molecule's name" << endl;
920 cout << Verbose(0) << "N - give molecules filename" << endl;
921 cout << Verbose(0) << "p - parse xyz file into molecule" << endl;
922 cout << Verbose(0) << "r - remove a molecule" << endl;
923 cout << Verbose(0) << "all else - go back" << endl;
924 cout << Verbose(0) << "===============================================" << endl;
925 cout << Verbose(0) << "INPUT: ";
926 cin >> choice;
927
928 switch (choice) {
929 default:
930 cout << Verbose(0) << "Not a valid choice." << endl;
931 break;
932 case 'c':
933 mol = new molecule(periode);
934 molecules->insert(mol);
935 break;
936
937 case 'l': // laod from XYZ file
938 cout << Verbose(0) << "Format should be XYZ with: ShorthandOfElement\tX\tY\tZ" << endl;
939 mol = new molecule(periode);
940 do {
941 cout << Verbose(0) << "Enter file name: ";
942 cin >> filename;
943 } while (!mol->AddXYZFile(filename));
944 mol->SetNameFromFilename(filename);
945 // center at set box dimensions
946 mol->CenterEdge((ofstream *)&cout, &Center);
947 mol->cell_size[0] = Center.x[0];
948 mol->cell_size[1] = 0;
949 mol->cell_size[2] = Center.x[1];
950 mol->cell_size[3] = 0;
951 mol->cell_size[4] = 0;
952 mol->cell_size[5] = Center.x[2];
953 molecules->insert(mol);
954 break;
955
956 case 'n':
957 do {
958 cout << Verbose(0) << "Enter index of molecule: ";
959 cin >> nr;
960 mol = molecules->ReturnIndex(nr);
961 } while (mol != NULL);
962 cout << Verbose(0) << "Enter name: ";
963 cin >> filename;
964 strcpy(mol->name, filename);
965 break;
966
967 case 'N':
968 do {
969 cout << Verbose(0) << "Enter index of molecule: ";
970 cin >> nr;
971 mol = molecules->ReturnIndex(nr);
972 } while (mol != NULL);
973 cout << Verbose(0) << "Enter name: ";
974 cin >> filename;
975 mol->SetNameFromFilename(filename);
976 break;
977
978 case 'p': // parse XYZ file
979 mol = NULL;
980 do {
981 cout << Verbose(0) << "Enter index of molecule: ";
982 cin >> nr;
983 mol = molecules->ReturnIndex(nr);
984 } while (mol == NULL);
985 cout << Verbose(0) << "Format should be XYZ with: ShorthandOfElement\tX\tY\tZ" << endl;
986 do {
987 cout << Verbose(0) << "Enter file name: ";
988 cin >> filename;
989 } while (!mol->AddXYZFile(filename));
990 mol->SetNameFromFilename(filename);
991 break;
992
993 case 'r':
994 cout << Verbose(0) << "Enter index of molecule: ";
995 cin >> nr;
996 count = 1;
997 MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin();
998 for(; ((ListRunner != molecules->ListOfMolecules.end()) && (count < nr)); ListRunner++);
999 mol = *ListRunner;
1000 if (count == nr) {
1001 molecules->ListOfMolecules.erase(ListRunner);
1002 delete(mol);
1003 }
1004 break;
1005 }
1006};
1007
1008
1009/** Submenu for merging molecules.
1010 * \param *periode periodentafel
1011 * \param *molecules list of molecules to add to
1012 */
1013static void MergeMolecules(periodentafel *periode, MoleculeListClass *molecules)
1014{
1015 char choice; // menu choice char
1016 bool valid;
1017
1018 cout << Verbose(0) << "===========MERGE MOLECULES=====================" << endl;
1019 cout << Verbose(0) << "e - embedding merge of two molecules" << endl;
1020 cout << Verbose(0) << "m - multi-merge of all molecules" << endl;
1021 cout << Verbose(0) << "s - scatter merge of two molecules" << endl;
1022 cout << Verbose(0) << "t - simple merge of two molecules" << endl;
1023 cout << Verbose(0) << "all else - go back" << endl;
1024 cout << Verbose(0) << "===============================================" << endl;
1025 cout << Verbose(0) << "INPUT: ";
1026 cin >> choice;
1027
1028 switch (choice) {
1029 default:
1030 cout << Verbose(0) << "Not a valid choice." << endl;
1031 break;
1032
1033 case 'e':
1034 break;
1035
1036 case 'm':
1037 break;
1038
1039 case 's':
1040 break;
1041
1042 case 't':
1043 break;
1044 }
1045};
1046
1047
[14de469]1048/********************************************** Test routine **************************************/
1049
1050/** Is called always as option 'T' in the menu.
[1907a7]1051 * \param *molecules list of molecules
[14de469]1052 */
[1907a7]1053static void testroutine(MoleculeListClass *molecules)
[14de469]1054{
[6ac7ee]1055 // the current test routine checks the functionality of the KeySet&Graph concept:
1056 // We want to have a multiindex (the KeySet) describing a unique subgraph
[1907a7]1057 int i, comp, counter=0;
1058
1059 // create a clone
1060 molecule *mol = NULL;
1061 if (molecules->ListOfMolecules.size() != 0) // clone
1062 mol = (molecules->ListOfMolecules.front())->CopyMolecule();
1063 else {
1064 cerr << "I don't have anything to test on ... ";
1065 return;
1066 }
1067 atom *Walker = mol->start;
[6ac7ee]1068
1069 // generate some KeySets
1070 cout << "Generating KeySets." << endl;
1071 KeySet TestSets[mol->AtomCount+1];
1072 i=1;
1073 while (Walker->next != mol->end) {
1074 Walker = Walker->next;
1075 for (int j=0;j<i;j++) {
1076 TestSets[j].insert(Walker->nr);
1077 }
1078 i++;
1079 }
1080 cout << "Testing insertion of already present item in KeySets." << endl;
1081 KeySetTestPair test;
1082 test = TestSets[mol->AtomCount-1].insert(Walker->nr);
1083 if (test.second) {
1084 cout << Verbose(1) << "Insertion worked?!" << endl;
1085 } else {
1086 cout << Verbose(1) << "Insertion rejected: Present object is " << (*test.first) << "." << endl;
1087 }
1088 TestSets[mol->AtomCount].insert(mol->end->previous->nr);
1089 TestSets[mol->AtomCount].insert(mol->end->previous->previous->previous->nr);
1090
1091 // constructing Graph structure
1092 cout << "Generating Subgraph class." << endl;
1093 Graph Subgraphs;
1094
1095 // insert KeySets into Subgraphs
1096 cout << "Inserting KeySets into Subgraph class." << endl;
1097 for (int j=0;j<mol->AtomCount;j++) {
1098 Subgraphs.insert(GraphPair (TestSets[j],pair<int, double>(counter++, 1.)));
1099 }
1100 cout << "Testing insertion of already present item in Subgraph." << endl;
1101 GraphTestPair test2;
1102 test2 = Subgraphs.insert(GraphPair (TestSets[mol->AtomCount],pair<int, double>(counter++, 1.)));
1103 if (test2.second) {
1104 cout << Verbose(1) << "Insertion worked?!" << endl;
1105 } else {
1106 cout << Verbose(1) << "Insertion rejected: Present object is " << (*(test2.first)).second.first << "." << endl;
1107 }
1108
1109 // show graphs
1110 cout << "Showing Subgraph's contents, checking that it's sorted." << endl;
1111 Graph::iterator A = Subgraphs.begin();
1112 while (A != Subgraphs.end()) {
1113 cout << (*A).second.first << ": ";
1114 KeySet::iterator key = (*A).first.begin();
1115 comp = -1;
1116 while (key != (*A).first.end()) {
1117 if ((*key) > comp)
1118 cout << (*key) << " ";
1119 else
1120 cout << (*key) << "! ";
1121 comp = (*key);
1122 key++;
1123 }
1124 cout << endl;
1125 A++;
1126 }
[1907a7]1127 delete(mol);
[14de469]1128};
1129
[dbe929]1130/** Tries given filename or standard on saving the config file.
1131 * \param *ConfigFileName name of file
1132 * \param *configuration pointer to configuration structure with all the values
1133 * \param *periode pointer to periodentafel structure with all the elements
[1907a7]1134 * \param *molecules list of molecules structure with all the atoms and coordinates
[dbe929]1135 */
[1907a7]1136static void SaveConfig(char *ConfigFileName, config *configuration, periodentafel *periode, MoleculeListClass *molecules)
[dbe929]1137{
[6ac7ee]1138 char filename[MAXSTRINGSIZE];
1139 ofstream output;
[1907a7]1140 molecule *mol = new molecule(periode);
1141
1142 // merge all molecules in MoleculeListClass into this molecule
1143 int N = molecules->ListOfMolecules.size();
1144 int *src = new int(N);
1145 N=0;
1146 for (MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin(); ListRunner != molecules->ListOfMolecules.end(); ListRunner++)
1147 src[N++] = (*ListRunner)->IndexNr;
1148 molecules->SimpleMultiAdd(mol, src, N);
[6ac7ee]1149
1150 cout << Verbose(0) << "Storing configuration ... " << endl;
1151 // get correct valence orbitals
1152 mol->CalculateOrbitals(*configuration);
1153 configuration->InitMaxMinStopStep = configuration->MaxMinStopStep = configuration->MaxPsiDouble;
1154 strcpy(filename, ConfigFileName);
1155 if (ConfigFileName != NULL) { // test the file name
1156 output.open(ConfigFileName, ios::trunc);
1157 } else if (strlen(configuration->configname) != 0) {
1158 strcpy(filename, configuration->configname);
1159 output.open(configuration->configname, ios::trunc);
1160 } else {
1161 strcpy(filename, DEFAULTCONFIG);
1162 output.open(DEFAULTCONFIG, ios::trunc);
1163 }
1164 output.close();
1165 output.clear();
1166 cout << Verbose(0) << "Saving of config file ";
1167 if (configuration->Save(filename, periode, mol))
1168 cout << "successful." << endl;
1169 else
1170 cout << "failed." << endl;
1171
1172 // and save to xyz file
1173 if (ConfigFileName != NULL) {
1174 strcpy(filename, ConfigFileName);
1175 strcat(filename, ".xyz");
1176 output.open(filename, ios::trunc);
1177 }
1178 if (output == NULL) {
1179 strcpy(filename,"main_pcp_linux");
1180 strcat(filename, ".xyz");
1181 output.open(filename, ios::trunc);
1182 }
1183 cout << Verbose(0) << "Saving of XYZ file ";
1184 if (mol->MDSteps <= 1) {
1185 if (mol->OutputXYZ(&output))
1186 cout << "successful." << endl;
1187 else
1188 cout << "failed." << endl;
1189 } else {
1190 if (mol->OutputTrajectoriesXYZ(&output))
1191 cout << "successful." << endl;
1192 else
1193 cout << "failed." << endl;
1194 }
1195 output.close();
1196 output.clear();
1197
1198 // and save as MPQC configuration
1199 if (ConfigFileName != NULL)
1200 strcpy(filename, ConfigFileName);
1201 if (output == NULL)
1202 strcpy(filename,"main_pcp_linux");
1203 cout << Verbose(0) << "Saving as mpqc input ";
1204 if (configuration->SaveMPQC(filename, mol))
1205 cout << "done." << endl;
1206 else
1207 cout << "failed." << endl;
1208
1209 if (!strcmp(configuration->configpath, configuration->GetDefaultPath())) {
1210 cerr << "WARNING: config is found under different path then stated in config file::defaultpath!" << endl;
1211 }
[1907a7]1212 delete(mol);
[dbe929]1213};
1214
[ca2b83]1215/** Parses the command line options.
1216 * \param argc argument count
1217 * \param **argv arguments array
[1907a7]1218 * \param *molecules list of molecules structure
[ca2b83]1219 * \param *periode elements structure
1220 * \param configuration config file structure
1221 * \param *ConfigFileName pointer to config file name in **argv
[d7d29c]1222 * \param *PathToDatabases pointer to db's path in **argv
[ca2b83]1223 * \return exit code (0 - successful, all else - something's wrong)
1224 */
[1907a7]1225static int ParseCommandLineOptions(int argc, char **argv, MoleculeListClass *&molecules, periodentafel *&periode, config& configuration, char *&ConfigFileName, char *&PathToDatabases)
[14de469]1226{
[6ac7ee]1227 Vector x,y,z,n; // coordinates for absolute point in cell volume
1228 double *factor; // unit factor if desired
1229 ifstream test;
1230 ofstream output;
1231 string line;
1232 atom *first;
1233 bool SaveFlag = false;
1234 int ExitFlag = 0;
1235 int j;
1236 double volume = 0.;
1237 enum ConfigStatus config_present = absent;
1238 clock_t start,end;
1239 int argptr;
1240 PathToDatabases = LocalPath;
1241
[1907a7]1242 // simply create a new molecule, wherein the config file is loaded and the manipulation takes place
1243 molecule *mol = new molecule(periode);
1244 molecules->insert(mol);
1245
[6ac7ee]1246 if (argc > 1) { // config file specified as option
1247 // 1. : Parse options that just set variables or print help
1248 argptr = 1;
1249 do {
1250 if (argv[argptr][0] == '-') {
1251 cout << Verbose(0) << "Recognized command line argument: " << argv[argptr][1] << ".\n";
1252 argptr++;
1253 switch(argv[argptr-1][1]) {
1254 case 'h':
1255 case 'H':
1256 case '?':
1257 cout << "MoleCuilder suite" << endl << "==================" << endl << endl;
1258 cout << "Usage: " << argv[0] << "[config file] [-{acefpsthH?vfrp}] [further arguments]" << endl;
1259 cout << "or simply " << argv[0] << " without arguments for interactive session." << endl;
1260 cout << "\t-a Z x1 x2 x3\tAdd new atom of element Z at coordinates (x1,x2,x3)." << endl;
1261 cout << "\t-A <source>\tCreate adjacency list from bonds parsed from 'dbond'-style file." <<endl;
1262 cout << "\t-b x1 x2 x3\tCenter atoms in domain with given edge lengths of (x1,x2,x3)." << endl;
[d8b94a]1263 cout << "\t-B <basis>\tSetting basis to store to MPQC config files." << endl;
[6ac7ee]1264 cout << "\t-c x1 x2 x3\tCenter atoms in domain with a minimum distance to boundary of (x1,x2,x3)." << endl;
1265 cout << "\t-D <bond distance>\tDepth-First-Search Analysis of the molecule, giving cycles and tree/back edges." << endl;
1266 cout << "\t-O\tCenter atoms in origin." << endl;
1267 cout << "\t-d x1 x2 x3\tDuplicate cell along each axis by given factor." << endl;
1268 cout << "\t-e <file>\tSets the databases path to be parsed (default: ./)." << endl;
1269 cout << "\t-E <id> <Z>\tChange atom <id>'s element to <Z>, <id> begins at 0." << endl;
1270 cout << "\t-f/F <dist> <order>\tFragments the molecule in BOSSANOVA manner (with/out rings compressed) and stores config files in same dir as config (return code 0 - fragmented, 2 - no fragmentation necessary)." << endl;
1271 cout << "\t-h/-H/-?\tGive this help screen." << endl;
1272 cout << "\t-m <0/1>\tCalculate (0)/ Align in(1) PAS with greatest EV along z axis." << endl;
1273 cout << "\t-n\tFast parsing (i.e. no trajectories are looked for)." << endl;
[a5b2c3a]1274 cout << "\t-N <radius> <file>\tGet non-convex-envelope." << endl;
[6ac7ee]1275 cout << "\t-o <out>\tGet volume of the convex envelope (and store to tecplot file)." << endl;
1276 cout << "\t-p <file>\tParse given xyz file and create raw config file from it." << endl;
1277 cout << "\t-P <file>\tParse given forces file and append as an MD step to config file via Verlet." << endl;
1278 cout << "\t-r\t\tConvert file from an old pcp syntax." << endl;
[a5b2c3a]1279 cout << "\t-R\t\tRemove all atoms out of sphere around a given one." << endl;
[6ac7ee]1280 cout << "\t-t x1 x2 x3\tTranslate all atoms by this vector (x1,x2,x3)." << endl;
1281 cout << "\t-T <file> Store temperatures from the config file in <file>." << endl;
1282 cout << "\t-s x1 x2 x3\tScale all atom coordinates by this vector (x1,x2,x3)." << endl;
1283 cout << "\t-u rho\tsuspend in water solution and output necessary cell lengths, average density rho and repetition." << endl;
1284 cout << "\t-v/-V\t\tGives version information." << endl;
1285 cout << "Note: config files must not begin with '-' !" << endl;
1286 delete(mol);
1287 delete(periode);
1288 return (1);
1289 break;
1290 case 'v':
1291 case 'V':
1292 cout << argv[0] << " " << VERSIONSTRING << endl;
1293 cout << "Build your own molecule position set." << endl;
1294 delete(mol);
1295 delete(periode);
1296 return (1);
1297 break;
1298 case 'e':
1299 if ((argptr >= argc) || (argv[argptr][0] == '-')) {
1300 cerr << "Not enough or invalid arguments for specifying element db: -e <db file>" << endl;
1301 } else {
1302 cout << "Using " << argv[argptr] << " as elements database." << endl;
1303 PathToDatabases = argv[argptr];
1304 argptr+=1;
1305 }
1306 break;
1307 case 'n':
1308 cout << "I won't parse trajectories." << endl;
1309 configuration.FastParsing = true;
1310 break;
1311 default: // no match? Step on
1312 argptr++;
1313 break;
1314 }
1315 } else
1316 argptr++;
1317 } while (argptr < argc);
1318
1319 // 2. Parse the element database
1320 if (periode->LoadPeriodentafel(PathToDatabases)) {
1321 cout << Verbose(0) << "Element list loaded successfully." << endl;
1322 //periode->Output((ofstream *)&cout);
1323 } else {
1324 cout << Verbose(0) << "Element list loading failed." << endl;
1325 return 1;
1326 }
1327 // 3. Find config file name and parse if possible
1328 if (argv[1][0] != '-') {
1329 cout << Verbose(0) << "Config file given." << endl;
1330 test.open(argv[1], ios::in);
1331 if (test == NULL) {
1332 //return (1);
1333 output.open(argv[1], ios::out);
1334 if (output == NULL) {
1335 cout << Verbose(1) << "Specified config file " << argv[1] << " not found." << endl;
1336 config_present = absent;
1337 } else {
1338 cout << "Empty configuration file." << endl;
1339 ConfigFileName = argv[1];
1340 config_present = empty;
1341 output.close();
1342 }
1343 } else {
1344 test.close();
1345 ConfigFileName = argv[1];
1346 cout << Verbose(1) << "Specified config file found, parsing ... ";
1347 switch (configuration.TestSyntax(ConfigFileName, periode, mol)) {
1348 case 1:
1349 cout << "new syntax." << endl;
1350 configuration.Load(ConfigFileName, periode, mol);
1351 config_present = present;
1352 break;
1353 case 0:
1354 cout << "old syntax." << endl;
1355 configuration.LoadOld(ConfigFileName, periode, mol);
1356 config_present = present;
1357 break;
1358 default:
1359 cout << "Unknown syntax or empty, yet present file." << endl;
1360 config_present = empty;
1361 }
1362 }
1363 } else
1364 config_present = absent;
1365 // 4. parse again through options, now for those depending on elements db and config presence
1366 argptr = 1;
1367 do {
1368 cout << "Current Command line argument: " << argv[argptr] << "." << endl;
1369 if (argv[argptr][0] == '-') {
1370 argptr++;
1371 if ((config_present == present) || (config_present == empty)) {
1372 switch(argv[argptr-1][1]) {
1373 case 'p':
1374 ExitFlag = 1;
1375 if ((argptr >= argc) || (argv[argptr][0] == '-')) {
1376 ExitFlag = 255;
1377 cerr << "Not enough arguments for parsing: -p <xyz file>" << endl;
1378 } else {
1379 SaveFlag = true;
1380 cout << Verbose(1) << "Parsing xyz file for new atoms." << endl;
1381 if (!mol->AddXYZFile(argv[argptr]))
1382 cout << Verbose(2) << "File not found." << endl;
1383 else {
1384 cout << Verbose(2) << "File found and parsed." << endl;
1385 config_present = present;
1386 }
1387 }
1388 break;
1389 case 'a':
1390 ExitFlag = 1;
1391 if ((argptr >= argc) || (argv[argptr][0] == '-') || (!IsValidNumber(argv[argptr+1]))) {
1392 ExitFlag = 255;
1393 cerr << "Not enough or invalid arguments for adding atom: -a <element> <x> <y> <z>" << endl;
1394 } else {
1395 SaveFlag = true;
1396 cout << Verbose(1) << "Adding new atom with element " << argv[argptr] << " at (" << argv[argptr+1] << "," << argv[argptr+2] << "," << argv[argptr+3] << "), ";
1397 first = new atom;
1398 first->type = periode->FindElement(atoi(argv[argptr]));
1399 if (first->type != NULL)
1400 cout << Verbose(2) << "found element " << first->type->name << endl;
1401 for (int i=NDIM;i--;)
1402 first->x.x[i] = atof(argv[argptr+1+i]);
1403 if (first->type != NULL) {
1404 mol->AddAtom(first); // add to molecule
1405 if ((config_present == empty) && (mol->AtomCount != 0))
1406 config_present = present;
1407 } else
1408 cerr << Verbose(1) << "Could not find the specified element." << endl;
1409 argptr+=4;
1410 }
1411 break;
1412 default: // no match? Don't step on (this is done in next switch's default)
1413 break;
1414 }
1415 }
1416 if (config_present == present) {
1417 switch(argv[argptr-1][1]) {
[d8b94a]1418 case 'B':
1419 if ((argptr >= argc) || (argv[argptr][0] == '-')) {
1420 ExitFlag = 255;
1421 cerr << "Not enough or invalid arguments given for setting MPQC basis: -B <basis name>" << endl;
1422 } else {
1423 configuration.basis = argv[argptr];
1424 cout << Verbose(1) << "Setting MPQC basis to " << configuration.basis << "." << endl;
1425 argptr+=1;
1426 }
1427 break;
[6ac7ee]1428 case 'D':
1429 ExitFlag = 1;
1430 {
1431 cout << Verbose(1) << "Depth-First-Search Analysis." << endl;
1432 MoleculeLeafClass *Subgraphs = NULL; // list of subgraphs from DFS analysis
1433 int *MinimumRingSize = new int[mol->AtomCount];
1434 atom ***ListOfLocalAtoms = NULL;
1435 int FragmentCounter = 0;
1436 class StackClass<bond *> *BackEdgeStack = NULL;
1437 class StackClass<bond *> *LocalBackEdgeStack = NULL;
1438 mol->CreateAdjacencyList((ofstream *)&cout, atof(argv[argptr]), configuration.GetIsAngstroem());
1439 mol->CreateListOfBondsPerAtom((ofstream *)&cout);
1440 Subgraphs = mol->DepthFirstSearchAnalysis((ofstream *)&cout, BackEdgeStack);
1441 if (Subgraphs != NULL) {
1442 Subgraphs->next->FillBondStructureFromReference((ofstream *)&cout, mol, (FragmentCounter = 0), ListOfLocalAtoms, false); // we want to keep the created ListOfLocalAtoms
1443 while (Subgraphs->next != NULL) {
1444 Subgraphs = Subgraphs->next;
1445 LocalBackEdgeStack = new StackClass<bond *> (Subgraphs->Leaf->BondCount);
1446 Subgraphs->Leaf->PickLocalBackEdges((ofstream *)&cout, ListOfLocalAtoms[FragmentCounter++], BackEdgeStack, LocalBackEdgeStack);
1447 Subgraphs->Leaf->CyclicStructureAnalysis((ofstream *)&cout, BackEdgeStack, MinimumRingSize);
1448 delete(LocalBackEdgeStack);
1449 delete(Subgraphs->previous);
1450 }
1451 delete(Subgraphs);
1452 for (int i=0;i<FragmentCounter;i++)
1453 Free((void **)&ListOfLocalAtoms[FragmentCounter], "ParseCommandLineOptions: **ListOfLocalAtoms[]");
1454 Free((void **)&ListOfLocalAtoms, "ParseCommandLineOptions: ***ListOfLocalAtoms");
1455 }
1456 delete(BackEdgeStack);
1457 delete[](MinimumRingSize);
1458 }
1459 //argptr+=1;
1460 break;
1461 case 'E':
1462 ExitFlag = 1;
1463 if ((argptr+1 >= argc) || (!IsValidNumber(argv[argptr])) || (argv[argptr+1][0] == '-')) {
1464 ExitFlag = 255;
1465 cerr << "Not enough or invalid arguments given for changing element: -E <atom nr.> <element>" << endl;
1466 } else {
1467 SaveFlag = true;
1468 cout << Verbose(1) << "Changing atom " << argv[argptr] << " to element " << argv[argptr+1] << "." << endl;
1469 first = mol->FindAtom(atoi(argv[argptr]));
1470 first->type = periode->FindElement(atoi(argv[argptr+1]));
1471 argptr+=2;
1472 }
1473 break;
1474 case 'A':
1475 ExitFlag = 1;
1476 if ((argptr >= argc) || (argv[argptr][0] == '-')) {
1477 ExitFlag =255;
1478 cerr << "Missing source file for bonds in molecule: -A <bond sourcefile>" << endl;
1479 } else {
1480 cout << "Parsing bonds from " << argv[argptr] << "." << endl;
1481 ifstream *input = new ifstream(argv[argptr]);
1482 mol->CreateAdjacencyList2((ofstream *)&cout, input);
1483 input->close();
1484 argptr+=1;
1485 }
1486 break;
1487 case 'N':
1488 ExitFlag = 1;
1489 if ((argptr+1 >= argc) || (argv[argptr+1][0] == '-')){
1490 ExitFlag = 255;
1491 cerr << "Not enough or invalid arguments given for non-convex envelope: -o <radius> <tecplot output file>" << endl;
1492 } else {
1493 class Tesselation T;
1494 int N = 15;
1495 int number = 100;
1496 string filename(argv[argptr+1]);
1497 filename.append(".csv");
1498 cout << Verbose(0) << "Evaluating non-convex envelope.";
1499 cout << Verbose(1) << "Using rolling ball of radius " << atof(argv[argptr]) << " and storing tecplot data in " << argv[argptr+1] << "." << endl;
1500 LinkedCell LCList(mol, atof(argv[argptr])); // \NOTE not twice the radius??
1501 Find_non_convex_border((ofstream *)&cout, mol, &T, &LCList, argv[argptr+1], atof(argv[argptr]));
[a5b2c3a]1502 //FindDistributionOfEllipsoids((ofstream *)&cout, &T, &LCList, N, number, filename.c_str());
[6ac7ee]1503 argptr+=2;
1504 }
1505 break;
1506 case 'T':
1507 ExitFlag = 1;
1508 if ((argptr >= argc) || (argv[argptr][0] == '-')) {
1509 ExitFlag = 255;
1510 cerr << "Not enough or invalid arguments given for storing tempature: -T <temperature file>" << endl;
1511 } else {
1512 cout << Verbose(1) << "Storing temperatures in " << argv[argptr] << "." << endl;
1513 ofstream *output = new ofstream(argv[argptr], ios::trunc);
1514 if (!mol->OutputTemperatureFromTrajectories((ofstream *)&cout, 0, mol->MDSteps, output))
1515 cout << Verbose(2) << "File could not be written." << endl;
1516 else
1517 cout << Verbose(2) << "File stored." << endl;
1518 output->close();
1519 delete(output);
1520 argptr+=1;
1521 }
1522 break;
1523 case 'P':
1524 ExitFlag = 1;
1525 if ((argptr >= argc) || (argv[argptr][0] == '-')) {
1526 ExitFlag = 255;
1527 cerr << "Not enough or invalid arguments given for parsing and integrating forces: -P <forces file>" << endl;
1528 } else {
1529 SaveFlag = true;
1530 cout << Verbose(1) << "Parsing forces file and Verlet integrating." << endl;
1531 if (!mol->VerletForceIntegration(argv[argptr], configuration.Deltat, configuration.GetIsAngstroem()))
1532 cout << Verbose(2) << "File not found." << endl;
1533 else
1534 cout << Verbose(2) << "File found and parsed." << endl;
1535 argptr+=1;
1536 }
1537 break;
[a5b2c3a]1538 case 'R':
1539 ExitFlag = 1;
1540 if ((argptr+1 >= argc) || (!IsValidNumber(argv[argptr])) || (!IsValidNumber(argv[argptr+1]))) {
1541 ExitFlag = 255;
1542 cerr << "Not enough or invalid arguments given for removing atoms: -R <id> <distance>" << endl;
1543 } else {
1544 SaveFlag = true;
1545 cout << Verbose(1) << "Removing atoms around " << argv[argptr] << " with radius " << argv[argptr+1] << "." << endl;
1546 double tmp1 = atof(argv[argptr+1]);
1547 atom *third = mol->FindAtom(atoi(argv[argptr]));
1548 atom *first = mol->start;
1549 if ((third != NULL) && (first != mol->end)) {
1550 atom *second = first->next;
1551 while(second != mol->end) {
1552 first = second;
1553 second = first->next;
1554 if (first->x.DistanceSquared((const Vector *)&third->x) > tmp1*tmp1) // distance to first above radius ...
1555 mol->RemoveAtom(first);
1556 }
1557 } else {
1558 cerr << "Removal failed due to missing atoms on molecule or wrong id." << endl;
1559 }
1560 argptr+=2;
1561 }
1562 break;
[6ac7ee]1563 case 't':
1564 ExitFlag = 1;
1565 if ((argptr+2 >= argc) || (!IsValidNumber(argv[argptr])) || (!IsValidNumber(argv[argptr+1])) || (!IsValidNumber(argv[argptr+2])) ) {
1566 ExitFlag = 255;
1567 cerr << "Not enough or invalid arguments given for translation: -t <x> <y> <z>" << endl;
1568 } else {
1569 ExitFlag = 1;
1570 SaveFlag = true;
1571 cout << Verbose(1) << "Translating all ions to new origin." << endl;
1572 for (int i=NDIM;i--;)
1573 x.x[i] = atof(argv[argptr+i]);
1574 mol->Translate((const Vector *)&x);
1575 argptr+=3;
1576 }
1577 break;
1578 case 's':
1579 ExitFlag = 1;
1580 if ((argptr >= argc) || (!IsValidNumber(argv[argptr])) ) {
1581 ExitFlag = 255;
1582 cerr << "Not enough or invalid arguments given for scaling: -s <factor/[factor_x]> [factor_y] [factor_z]" << endl;
1583 } else {
1584 SaveFlag = true;
1585 j = -1;
1586 cout << Verbose(1) << "Scaling all ion positions by factor." << endl;
1587 factor = new double[NDIM];
1588 factor[0] = atof(argv[argptr]);
1589 if ((argptr < argc) && (IsValidNumber(argv[argptr])))
1590 argptr++;
1591 factor[1] = atof(argv[argptr]);
1592 if ((argptr < argc) && (IsValidNumber(argv[argptr])))
1593 argptr++;
1594 factor[2] = atof(argv[argptr]);
1595 mol->Scale(&factor);
1596 for (int i=0;i<NDIM;i++) {
1597 j += i+1;
1598 x.x[i] = atof(argv[NDIM+i]);
1599 mol->cell_size[j]*=factor[i];
1600 }
1601 delete[](factor);
1602 argptr+=1;
1603 }
1604 break;
1605 case 'b':
1606 ExitFlag = 1;
1607 if ((argptr+2 >= argc) || (!IsValidNumber(argv[argptr])) || (!IsValidNumber(argv[argptr+1])) || (!IsValidNumber(argv[argptr+2])) ) {
1608 ExitFlag = 255;
1609 cerr << "Not enough or invalid arguments given for centering in box: -b <length_x> <length_y> <length_z>" << endl;
1610 } else {
1611 SaveFlag = true;
1612 j = -1;
1613 cout << Verbose(1) << "Centering atoms in config file within given simulation box." << endl;
1614 j=-1;
1615 for (int i=0;i<NDIM;i++) {
1616 j += i+1;
1617 x.x[i] = atof(argv[argptr++]);
1618 mol->cell_size[j] += x.x[i]*2.;
1619 }
1620 // center
1621 mol->CenterInBox((ofstream *)&cout, &x);
1622 // update Box of atoms by boundary
1623 mol->SetBoxDimension(&x);
1624 }
1625 break;
1626 case 'c':
1627 ExitFlag = 1;
1628 if ((argptr+2 >= argc) || (!IsValidNumber(argv[argptr])) || (!IsValidNumber(argv[argptr+1])) || (!IsValidNumber(argv[argptr+2])) ) {
1629 ExitFlag = 255;
1630 cerr << "Not enough or invalid arguments given for centering with boundary: -c <boundary_x> <boundary_y> <boundary_z>" << endl;
1631 } else {
1632 SaveFlag = true;
1633 j = -1;
1634 cout << Verbose(1) << "Centering atoms in config file within given additional boundary." << endl;
1635 // make every coordinate positive
1636 mol->CenterEdge((ofstream *)&cout, &x);
1637 // update Box of atoms by boundary
1638 mol->SetBoxDimension(&x);
1639 // translate each coordinate by boundary
1640 j=-1;
1641 for (int i=0;i<NDIM;i++) {
1642 j += i+1;
1643 x.x[i] = atof(argv[argptr++]);
1644 mol->cell_size[j] += x.x[i]*2.;
1645 }
1646 mol->Translate((const Vector *)&x);
1647 }
1648 break;
1649 case 'O':
1650 ExitFlag = 1;
1651 SaveFlag = true;
1652 cout << Verbose(1) << "Centering atoms in origin." << endl;
1653 mol->CenterOrigin((ofstream *)&cout, &x);
1654 mol->SetBoxDimension(&x);
1655 break;
1656 case 'r':
1657 ExitFlag = 1;
1658 SaveFlag = true;
1659 cout << Verbose(1) << "Converting config file from supposed old to new syntax." << endl;
1660 break;
1661 case 'F':
1662 case 'f':
1663 ExitFlag = 1;
1664 if ((argptr+1 >= argc) || (!IsValidNumber(argv[argptr])) || (!IsValidNumber(argv[argptr+1]))) {
1665 ExitFlag = 255;
1666 cerr << "Not enough or invalid arguments for fragmentation: -f <max. bond distance> <bond order>" << endl;
1667 } else {
1668 cout << "Fragmenting molecule with bond distance " << argv[argptr] << " angstroem, order of " << argv[argptr+1] << "." << endl;
1669 cout << Verbose(0) << "Creating connection matrix..." << endl;
1670 start = clock();
1671 mol->CreateAdjacencyList((ofstream *)&cout, atof(argv[argptr++]), configuration.GetIsAngstroem());
1672 cout << Verbose(0) << "Fragmenting molecule with current connection matrix ..." << endl;
1673 if (mol->first->next != mol->last) {
1674 ExitFlag = mol->FragmentMolecule((ofstream *)&cout, atoi(argv[argptr]), &configuration);
1675 }
1676 end = clock();
1677 cout << Verbose(0) << "Clocks for this operation: " << (end-start) << ", time: " << ((double)(end-start)/CLOCKS_PER_SEC) << "s." << endl;
1678 argptr+=2;
1679 }
1680 break;
1681 case 'm':
1682 ExitFlag = 1;
1683 j = atoi(argv[argptr++]);
1684 if ((j<0) || (j>1)) {
1685 cerr << Verbose(1) << "ERROR: Argument of '-m' should be either 0 for no-rotate or 1 for rotate." << endl;
1686 j = 0;
1687 }
1688 if (j) {
1689 SaveFlag = true;
1690 cout << Verbose(0) << "Converting to prinicipal axis system." << endl;
1691 } else
1692 cout << Verbose(0) << "Evaluating prinicipal axis." << endl;
1693 mol->PrincipalAxisSystem((ofstream *)&cout, (bool)j);
1694 break;
1695 case 'o':
1696 ExitFlag = 1;
1697 if ((argptr >= argc) || (argv[argptr][0] == '-')){
1698 ExitFlag = 255;
1699 cerr << "Not enough or invalid arguments given for convex envelope: -o <tecplot output file>" << endl;
1700 } else {
1701 cout << Verbose(0) << "Evaluating volume of the convex envelope.";
1702 cout << Verbose(1) << "Storing tecplot data in " << argv[argptr] << "." << endl;
1703 VolumeOfConvexEnvelope((ofstream *)&cout, argv[argptr], &configuration, NULL, mol);
1704 argptr+=1;
1705 }
1706 break;
1707 case 'U':
1708 ExitFlag = 1;
1709 if ((argptr+1 >= argc) || (!IsValidNumber(argv[argptr])) || (!IsValidNumber(argv[argptr+1])) ) {
1710 ExitFlag = 255;
1711 cerr << "Not enough or invalid arguments given for suspension with specified volume: -U <volume> <density>" << endl;
1712 volume = -1; // for case 'u': don't print error again
1713 } else {
1714 volume = atof(argv[argptr++]);
1715 cout << Verbose(0) << "Using " << volume << " angstrom^3 as the volume instead of convex envelope one's." << endl;
1716 }
1717 case 'u':
1718 ExitFlag = 1;
1719 if ((argptr >= argc) || (!IsValidNumber(argv[argptr])) ) {
1720 if (volume != -1)
1721 ExitFlag = 255;
1722 cerr << "Not enough arguments given for suspension: -u <density>" << endl;
1723 } else {
1724 double density;
1725 SaveFlag = true;
1726 cout << Verbose(0) << "Evaluating necessary cell volume for a cluster suspended in water.";
1727 density = atof(argv[argptr++]);
1728 if (density < 1.0) {
1729 cerr << Verbose(0) << "Density must be greater than 1.0g/cm^3 !" << endl;
1730 density = 1.3;
1731 }
1732// for(int i=0;i<NDIM;i++) {
1733// repetition[i] = atoi(argv[argptr++]);
1734// if (repetition[i] < 1)
1735// cerr << Verbose(0) << "ERROR: repetition value must be greater 1!" << endl;
1736// repetition[i] = 1;
1737// }
1738 PrepareClustersinWater((ofstream *)&cout, &configuration, mol, volume, density); // if volume == 0, will calculate from ConvexEnvelope
1739 }
1740 break;
1741 case 'd':
1742 ExitFlag = 1;
1743 if ((argptr+2 >= argc) || (!IsValidNumber(argv[argptr])) || (!IsValidNumber(argv[argptr+1])) || (!IsValidNumber(argv[argptr+2])) ) {
1744 ExitFlag = 255;
1745 cerr << "Not enough or invalid arguments given for repeating cells: -d <repeat_x> <repeat_y> <repeat_z>" << endl;
1746 } else {
1747 SaveFlag = true;
1748 for (int axis = 1; axis <= NDIM; axis++) {
1749 int faktor = atoi(argv[argptr++]);
1750 int count;
1751 element ** Elements;
1752 Vector ** vectors;
1753 if (faktor < 1) {
1754 cerr << Verbose(0) << "ERROR: Repetition faktor mus be greater than 1!" << endl;
1755 faktor = 1;
1756 }
1757 mol->CountAtoms((ofstream *)&cout); // recount atoms
1758 if (mol->AtomCount != 0) { // if there is more than none
1759 count = mol->AtomCount; // is changed becausing of adding, thus has to be stored away beforehand
1760 Elements = new element *[count];
1761 vectors = new Vector *[count];
1762 j = 0;
1763 first = mol->start;
1764 while (first->next != mol->end) { // make a list of all atoms with coordinates and element
1765 first = first->next;
1766 Elements[j] = first->type;
1767 vectors[j] = &first->x;
1768 j++;
1769 }
1770 if (count != j)
1771 cout << Verbose(0) << "ERROR: AtomCount " << count << " is not equal to number of atoms in molecule " << j << "!" << endl;
1772 x.Zero();
1773 y.Zero();
1774 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
1775 for (int i=1;i<faktor;i++) { // then add this list with respective translation factor times
1776 x.AddVector(&y); // per factor one cell width further
1777 for (int k=count;k--;) { // go through every atom of the original cell
1778 first = new atom(); // create a new body
1779 first->x.CopyVector(vectors[k]); // use coordinate of original atom
1780 first->x.AddVector(&x); // translate the coordinates
1781 first->type = Elements[k]; // insert original element
1782 mol->AddAtom(first); // and add to the molecule (which increments ElementsInMolecule, AtomCount, ...)
1783 }
1784 }
1785 // free memory
1786 delete[](Elements);
1787 delete[](vectors);
1788 // correct cell size
1789 if (axis < 0) { // if sign was negative, we have to translate everything
1790 x.Zero();
1791 x.AddVector(&y);
1792 x.Scale(-(faktor-1));
1793 mol->Translate(&x);
1794 }
1795 mol->cell_size[(abs(axis) == 2) ? 2 : ((abs(axis) == 3) ? 5 : 0)] *= faktor;
1796 }
1797 }
1798 }
1799 break;
1800 default: // no match? Step on
1801 if ((argptr < argc) && (argv[argptr][0] != '-')) // if it started with a '-' we've already made a step!
1802 argptr++;
1803 break;
1804 }
1805 }
1806 } else argptr++;
1807 } while (argptr < argc);
1808 if (SaveFlag)
[1907a7]1809 SaveConfig(ConfigFileName, &configuration, periode, molecules);
[6ac7ee]1810 if ((ExitFlag >= 1)) {
1811 delete(mol);
1812 delete(periode);
1813 return (ExitFlag);
1814 }
1815 } else { // no arguments, hence scan the elements db
1816 if (periode->LoadPeriodentafel(PathToDatabases))
1817 cout << Verbose(0) << "Element list loaded successfully." << endl;
1818 else
1819 cout << Verbose(0) << "Element list loading failed." << endl;
1820 configuration.RetrieveConfigPathAndName("main_pcp_linux");
1821 }
1822 return(0);
[ca2b83]1823};
1824
1825/********************************************** Main routine **************************************/
[14de469]1826
[ca2b83]1827int main(int argc, char **argv)
1828{
[6ac7ee]1829 periodentafel *periode = new periodentafel; // and a period table of all elements
[1907a7]1830 MoleculeListClass *molecules = new MoleculeListClass; // list of all molecules
1831 molecule *mol = NULL;
[6ac7ee]1832 config configuration;
1833 double tmp1;
1834 atom *first, *second;
1835 char choice; // menu choice char
1836 Vector x,y,z,n; // coordinates for absolute point in cell volume
1837 bool valid; // flag if input was valid or not
1838 ifstream test;
1839 ofstream output;
1840 string line;
1841 char *ConfigFileName = NULL;
1842 char *ElementsFileName = NULL;
1843 int Z;
1844 int j, axis, count, faktor;
1845
1846 // =========================== PARSE COMMAND LINE OPTIONS ====================================
[1907a7]1847 j = ParseCommandLineOptions(argc, argv, molecules, periode, configuration, ConfigFileName, ElementsFileName);
[6ac7ee]1848 if (j == 1) return 0; // just for -v and -h options
1849 if (j) return j; // something went wrong
1850
1851 // General stuff
[1907a7]1852 if (molecules->ListOfMolecules.size() == 0) {
1853 mol = new molecule(periode);
1854 if (mol->cell_size[0] == 0.) {
1855 cout << Verbose(0) << "enter lower tridiagonal form of basis matrix" << endl << endl;
1856 for (int i=0;i<6;i++) {
1857 cout << Verbose(1) << "Cell size" << i << ": ";
1858 cin >> mol->cell_size[i];
1859 }
1860 }
1861 molecules->insert(mol);
[6ac7ee]1862 }
1863
1864 // =========================== START INTERACTIVE SESSION ====================================
1865
1866 // now the main construction loop
1867 cout << Verbose(0) << endl << "Now comes the real construction..." << endl;
1868 do {
1869 cout << Verbose(0) << endl << endl;
[1907a7]1870 cout << Verbose(0) << "============Molecule list=======================" << endl;
1871 molecules->Enumerate((ofstream *)&cout);
[6ac7ee]1872 cout << Verbose(0) << "============Menu===============================" << endl;
[1907a7]1873 cout << Verbose(0) << "a - set molecule (in)active" << endl;
1874 cout << Verbose(0) << "e - edit new molecules" << endl;
1875 cout << Verbose(0) << "g - globally manipulate atoms in molecule" << endl;
1876 cout << Verbose(0) << "M - Merge molecules" << endl;
1877 cout << Verbose(0) << "m - manipulate atoms" << endl;
1878 cout << Verbose(0) << "-----------------------------------------------" << endl;
1879 cout << Verbose(0) << "c - edit the current configuration" << endl;
1880 cout << Verbose(0) << "-----------------------------------------------" << endl;
1881 cout << Verbose(0) << "s - save current setup to config file" << endl;
1882 cout << Verbose(0) << "T - call the current test routine" << endl;
1883 cout << Verbose(0) << "q - quit" << endl;
1884 cout << Verbose(0) << "===============================================" << endl;
1885 cout << Verbose(0) << "Input: ";
1886 cin >> choice;
[6ac7ee]1887
1888 switch (choice) {
[1907a7]1889 case 'a': // (in)activate molecule
1890 {
1891 cout << "Enter index of molecule: ";
1892 cin >> j;
1893 count = 1;
1894 MoleculeList::iterator ListRunner = molecules->ListOfMolecules.begin();
1895 for(; ((ListRunner != molecules->ListOfMolecules.end()) && (count < j)); ListRunner++);
1896 if (count == j)
1897 (*ListRunner)->ActiveFlag = !(*ListRunner)->ActiveFlag;
1898 }
1899 break;
1900
1901 case 'c': // edit each field of the configuration
1902 configuration.Edit();
[6ac7ee]1903 break;
1904
[ca3ccc]1905 case 'e': // create molecule
1906 EditMolecules(periode, molecules);
1907 break;
1908
[1907a7]1909 case 'g': // manipulate molecules
1910 ManipulateMolecules(periode, molecules, &configuration);
1911 break;
[6ac7ee]1912
[1907a7]1913 case 'M': // merge molecules
1914 MergeMolecules(periode, molecules);
1915 break;
[6ac7ee]1916
[1907a7]1917 case 'm': // manipulate atoms
1918 ManipulateAtoms(periode, molecules, &configuration);
1919 break;
[6ac7ee]1920
1921 case 'q': // quit
1922 break;
1923
1924 case 's': // save to config file
[1907a7]1925 SaveConfig(ConfigFileName, &configuration, periode, molecules);
[6ac7ee]1926 break;
1927
1928 case 'T':
[1907a7]1929 testroutine(molecules);
[6ac7ee]1930 break;
1931
[1907a7]1932 default:
1933 break;
[6ac7ee]1934 };
1935 } while (choice != 'q');
1936
1937 // save element data base
1938 if (periode->StorePeriodentafel(ElementsFileName)) //ElementsFileName
1939 cout << Verbose(0) << "Saving of elements.db successful." << endl;
1940 else
1941 cout << Verbose(0) << "Saving of elements.db failed." << endl;
1942
[1907a7]1943 delete(molecules);
[6ac7ee]1944 delete(periode);
1945 return (0);
[14de469]1946}
1947
1948/********************************************** E N D **************************************************/
Note: See TracBrowser for help on using the repository browser.