Changes in src/Actions/BondAction/BondAddAction.cpp [88afc9:26b4d62]

File:

• src/Actions/BondAction/BondAddAction.cpp (modified) (3 diffs)

src/Actions/BondAction/BondAddAction.cpp

@@ -57 +57 @@
 ActionState::ptr BondAddAction::performCall() {
   // check preconditions
-  World& world = World::getInstance();
-  if (world.countSelectedAtoms() <= 1) {
-    STATUS("There must be at least two atoms selected for BondAction Add.");
+  if (World::getInstance().countSelectedAtoms() != 2) {
+    STATUS("Exactly two atoms must be selected for BondAction Add.");
     return Action::failure;
   }
-
-  bondPairIds_t bondPairIds;
-  for (World::AtomSelectionConstIterator firstiter = world.beginAtomSelection();
-      firstiter != world.endAtomSelection(); ++firstiter) {
-    for (World::AtomSelectionConstIterator seconditer = firstiter;
-        seconditer != world.endAtomSelection(); ++seconditer) {
-      if (firstiter == seconditer)
-        continue;
-      if (!(firstiter->second)->IsBondedTo(WorldTime::getTime(), seconditer->second))
-        bondPairIds.push_back(
-            std::make_pair((firstiter->second)->getId(), (seconditer->second)->getId()));
-    }
-  }
-  if (bondPairIds.empty()) {
-    STATUS("All bonds are already present.");
+  const std::vector<atom *> selected_atoms = World::getInstance().getSelectedAtoms();
+  if (selected_atoms[0]->IsBondedTo(WorldTime::getTime(), selected_atoms[1])) {
+    STATUS("There already is a bond in between the two selected atoms.");
     return Action::failure;
   }
 
   // create undo
-  BondAddState *UndoState = new BondAddState(bondPairIds, params);
+  BondAddState *UndoState = new BondAddState(selected_atoms[0]->getId(), selected_atoms[1]->getId(), params);
 
   // execute action
-  for (bondPairIds_t::const_iterator iter = bondPairIds.begin();
-      iter != bondPairIds.end(); ++iter) {
-    atom *firstatom = world.getAtom(AtomById(iter->first));
-    atom *secondatom = world.getAtom(AtomById(iter->second));
-    ASSERT((firstatom != NULL) && (secondatom != NULL),
-        "BondAddAction::performCall() - at least one of the ids "
-        +toString(iter->first)+" or "+toString(iter->second)+" is not present.");
-    firstatom->addBond(WorldTime::getTime(), secondatom);
-    ASSERT( firstatom->IsBondedTo(WorldTime::getTime(), secondatom),
+  selected_atoms[0]->addBond(WorldTime::getTime(), selected_atoms[1]);
+  ASSERT( selected_atoms[0]->IsBondedTo(WorldTime::getTime(), selected_atoms[1]),
       "BondAddAction::performCall() - adding bond in between "
-      +toString(*firstatom)+" and "+toString(*secondatom)+" failed.");
-  }
+      +toString(*selected_atoms[0])+" and "+toString(*selected_atoms[1])+" failed.");
 
   return ActionState::ptr(UndoState);
@@ -104 +83 @@
 
   // check whether bond already existed
-  World& world = World::getInstance();
-  for (bondPairIds_t::const_iterator iter = state->bondPairIds.begin();
-      iter != state->bondPairIds.end(); ++iter) {
-    atom *firstatom = world.getAtom(AtomById(iter->first));
-    atom *secondatom = world.getAtom(AtomById(iter->second));
-    ASSERT((firstatom != NULL) && (secondatom != NULL),
-        "BondAddAction::performCall() - at least one of the ids "
-        +toString(iter->first)+" or "+toString(iter->second)+" is not present.");
-    if (firstatom->IsBondedTo(WorldTime::getTime(), secondatom)) {
-      firstatom->removeBond(WorldTime::getTime(), secondatom);
-    } else {
-      ELOG(2, "There is no bond in between "+toString(iter->first)
-          +" and "+toString(iter->second)+".");
-    }
+  atom * const first = World::getInstance().getAtom(AtomById(state->firstId));
+  atom * const second = World::getInstance().getAtom(AtomById(state->secondId));
+  ASSERT((first != NULL) && (second != NULL),
+      "BondAddAction::performUndo() - at least one of the ids "
+      +toString(state->firstId)+" or "+toString(state->secondId)+" is not present.");
+  if (first->IsBondedTo(WorldTime::getTime(), second)) {
+    first->removeBond(WorldTime::getTime(), second);
+  } else {
+    ELOG(2, "There is no bond in between "+toString(state->firstId)
+        +" and "+toString(state->secondId)+".");
   }
 
@@ -127 +102 @@
 
   // check whether bond already existed
-  World& world = World::getInstance();
-  for (bondPairIds_t::const_iterator iter = state->bondPairIds.begin();
-      iter != state->bondPairIds.end(); ++iter) {
-    atom * const firstatom = world.getAtom(AtomById(iter->first));
-    atom * const secondatom = world.getAtom(AtomById(iter->second));
-    ASSERT((firstatom != NULL) && (secondatom != NULL),
-        "BondAddAction::performCall() - at least one of the ids "
-        +toString(iter->first)+" or "+toString(iter->second)+" is not present.");
-    if (!firstatom->IsBondedTo(WorldTime::getTime(), secondatom)) {
-      firstatom->addBond(WorldTime::getTime(), secondatom);
-    } else {
-      ELOG(2, "There is already a bond in between "+toString(iter->first)
-          +" and "+toString(iter->second)+".");
-    }
+  atom * const first = World::getInstance().getAtom(AtomById(state->firstId));
+  atom * const second = World::getInstance().getAtom(AtomById(state->secondId));
+  ASSERT((first != NULL) && (second != NULL),
+      "BondAddAction::performRedo() - at least one of the ids "
+      +toString(state->firstId)+" or "+toString(state->secondId)+" is not present.");
+  if (!first->IsBondedTo(WorldTime::getTime(), second)) {
+    first->addBond(WorldTime::getTime(), second);
+  } else {
+    ELOG(2, "There is already a bond in between "+toString(state->firstId)
+        +" and "+toString(state->secondId)+".");
   }