Changeset ea9696

Timestamp:

Jul 3, 2009, 2:25:17 PM (16 years ago)

Author:

Frederik Heber <heber@…>

Children:

1f6efb

Parents:

13088d

Message:

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

File:

• molecuilder/src/builder.cpp (modified) (4 diffs)

molecuilder/src/builder.cpp

@@ -474 +474 @@
                         cout << Verbose(0) << "Which axis is it: ";
                         cin >> axis;
-                        cout << Verbose(0) << "Left inward boundary: ";
+                        cout << Verbose(0) << "Lower boundary: ";
                         cin >> tmp1;
-                        cout << Verbose(0) << "Right inward boundary: ";
+                        cout << Verbose(0) << "Upper boundary: ";
                         cin >> tmp2;
                         first = mol->start;
-                        while(first->next != mol->end) {
-                                first = first->next;
-                                if ((first->x.x[axis] > tmp2) || (first->x.x[axis] < tmp1)) // out of boundary ...
+      second = first->next;
+      while(second != mol->end) {
+        first = second;
+        second = first->next;
+                                if ((first->x.x[axis] < tmp1) || (first->x.x[axis] > tmp2)) {// out of boundary ...
+                                  //cout << "Atom " << *first << " with " << first->x.x[axis] << " on axis " << axis << " is out of bounds [" << tmp1 << "," << tmp2 << "]." << endl;
                                         mol->RemoveAtom(first);
+                                }
                         }
                         break;
@@ -1268 +1272 @@
                                                 cout << "\t-m <0/1>\tCalculate (0)/ Align in(1) PAS with greatest EV along z axis." << endl;
                                                 cout << "\t-n\tFast parsing (i.e. no trajectories are looked for)." << endl;
-                                                cout << "\t-N\tGet non-convex-envelope." << endl;
+                                                cout << "\t-N <radius> <file>\tGet non-convex-envelope." << endl;
                                                 cout << "\t-o <out>\tGet volume of the convex envelope (and store to tecplot file)." << endl;
                                                 cout << "\t-p <file>\tParse given xyz file and create raw config file from it." << endl;
                                                 cout << "\t-P <file>\tParse given forces file and append as an MD step to config file via Verlet." << endl;
                                                 cout << "\t-r\t\tConvert file from an old pcp syntax." << endl;
+                                                cout << "\t-R\t\tRemove all atoms out of sphere around a given one." << endl;
                                                 cout << "\t-t x1 x2 x3\tTranslate all atoms by this vector (x1,x2,x3)." << endl;
                                                 cout << "\t-T <file> Store temperatures from the config file in <file>." << endl;
@@ -1495 +1500 @@
                                                                 LinkedCell LCList(mol, atof(argv[argptr]));             // \NOTE not twice the radius??
                                                                 Find_non_convex_border((ofstream *)&cout, mol, &T, &LCList, argv[argptr+1], atof(argv[argptr]));
-                                                                FindDistributionOfEllipsoids((ofstream *)&cout, &T, &LCList, N, number, filename.c_str());
+                                                                //FindDistributionOfEllipsoids((ofstream *)&cout, &T, &LCList, N, number, filename.c_str());
                                                                 argptr+=2;
                                                         }
@@ -1531 +1536 @@
                                                         }
                                                         break;
+            case 'R':
+              ExitFlag = 1;
+              if ((argptr+1 >= argc) || (!IsValidNumber(argv[argptr])) || (!IsValidNumber(argv[argptr+1])))  {
+                ExitFlag = 255;
+                cerr << "Not enough or invalid arguments given for removing atoms: -R <id> <distance>" << endl;
+              } else {
+                SaveFlag = true;
+                cout << Verbose(1) << "Removing atoms around " << argv[argptr] << " with radius " << argv[argptr+1] << "." << endl;
+                double tmp1 = atof(argv[argptr+1]);
+                atom *third = mol->FindAtom(atoi(argv[argptr]));
+                atom *first = mol->start;
+                if ((third != NULL) && (first != mol->end)) {
+                  atom *second = first->next;
+                  while(second != mol->end) {
+                    first = second;
+                    second = first->next;
+                    if (first->x.DistanceSquared((const Vector *)&third->x) > tmp1*tmp1) // distance to first above radius ...
+                      mol->RemoveAtom(first);
+                  }
+                } else {
+                  cerr << "Removal failed due to missing atoms on molecule or wrong id." << endl;
+                }
+                argptr+=2;
+              }
+              break;
                                                 case 't':
                                                         ExitFlag = 1;