Changeset cdac1d for src/Actions/AtomAction

Timestamp:

Sep 10, 2016, 4:01:20 PM (9 years ago)

Author:

Frederik Heber <heber@…>

Branches:

SaturateAtoms_singleDegree

Children:

19bfb5

Parents:

64cafb2

git-author:

Frederik Heber <heber@…> (05/10/16 20:04:47)

git-committer:

Frederik Heber <heber@…> (09/10/16 16:01:20)

Message:

Extended SaturateAction to be able to deal with bonded atoms.

• TESTFIX: Removed XFAIL from SaturateAction regression tests again.

File:

• src/Actions/AtomAction/SaturateAction.cpp (modified) (1 diff)

src/Actions/AtomAction/SaturateAction.cpp

@@ -75 +75 @@
     }
     ASSERT( mol != NULL,
-        "ChangeElementAction::performCall() - mol is still NULL." );
+        "AtomSaturateAction::performCall() - mol is still NULL." );
 
-//    // add the hydrogens
-//    const Vector AtomsPosition = _atom->getPosition();
-//    double typical_distance = _atom->getType()->getHBondDistance(0);
-//    if (typical_distance == -1.)
-//      typical_distance = 1.;
-//    SphericalPointDistribution PointSphere(typical_distance);
-//    PointSphere.initSelf(_atom->getType()->getNoValenceOrbitals());
-//    for (SphericalPointDistribution::Polygon_t::const_iterator iter = PointSphere.points.begin();
-//        iter != PointSphere.points.end(); ++iter) {
-//      // for every Vector add a point relative to atom's position
-//      atom * hydrogenAtom = World::getInstance().createAtom();
-//      hydrogenAtom->setType(hydrogen);
-//      hydrogenAtom->setPosition(AtomsPosition + *iter);
-//      mol->AddAtom(hydrogenAtom);
-//      // add bond
-//      _atom->addBond(WorldTime::getTime(), hydrogenAtom);
-//      // mark down for undo state
-//      addedHydrogens.push_back(AtomicInfo(*(hydrogenAtom)));
-//    }
+    // radius of polygon on sphere
+    double typical_distance = _atom->getType()->getHBondDistance(0);
+    if (typical_distance == -1.)
+      typical_distance = 1.;
+
+    // check for any bonds and get vacant positions
+    SphericalPointDistribution::Polygon_t vacant_positions;
+    const BondList& ListOfBonds = _atom->getListOfBonds();
+    SphericalPointDistribution PointSphere(typical_distance);
+    if (ListOfBonds.size() == 0) {
+      vacant_positions = PointSphere.getSimplePolygon(_atom->getType()->getNoValenceOrbitals());
+      LOG(3, "DEBUG: Using ideal positions as " << vacant_positions);
+    } else {
+      // get ideal polygon and currently occupied positions
+      const SphericalPointDistribution::Polygon_t ideal_positions =
+          PointSphere.getSimplePolygon(_atom->getType()->getNoValenceOrbitals());
+      LOG(3, "DEBUG: ideal positions are " << ideal_positions);
+      SphericalPointDistribution::Polygon_t current_positions;
+      for (BondList::const_iterator bonditer = ListOfBonds.begin();
+          bonditer != ListOfBonds.end(); ++bonditer) {
+        const Vector position =
+            (*bonditer)->GetOtherAtom(_atom)->getPosition().getVectorToPoint(_atom->getPosition());
+        current_positions.push_back((1./position.Norm())*position);
+      }
+      LOG(3, "DEBUG: current occupied positions are " << current_positions);
+
+      // find the best matching rotated polygon
+      vacant_positions = SphericalPointDistribution::matchSphericalPointDistributions(
+          current_positions,
+          ideal_positions);
+      LOG(3, "DEBUG: Resulting vacant positions are " << vacant_positions);
+
+      // scale vacant positions to typical_distance
+      std::for_each(
+          vacant_positions.begin(), vacant_positions.end(),
+          boost::bind(&Vector::Scale, _1, boost::cref(typical_distance)));
+    }
+
+    // add the hydrogens
+    const Vector AtomsPosition = _atom->getPosition();
+    for (SphericalPointDistribution::Polygon_t::const_iterator iter = vacant_positions.begin();
+        iter != vacant_positions.end(); ++iter) {
+      // for every Vector add a point relative to atom's position
+      atom * hydrogenAtom = World::getInstance().createAtom();
+      hydrogenAtom->setType(hydrogen);
+      hydrogenAtom->setPosition(AtomsPosition + *iter);
+      mol->AddAtom(hydrogenAtom);
+      // add bond
+      _atom->addBond(WorldTime::getTime(), hydrogenAtom);
+      // mark down for undo state
+      addedHydrogens.push_back(AtomicInfo(*(hydrogenAtom)));
+    }
   }