| 1 | /*
 | 
|---|
| 2 |  * Project: MoleCuilder
 | 
|---|
| 3 |  * Description: creates and alters molecular systems
 | 
|---|
| 4 |  * Copyright (C)  2010 University of Bonn. All rights reserved.
 | 
|---|
| 5 |  * Please see the LICENSE file or "Copyright notice" in builder.cpp for details.
 | 
|---|
| 6 |  */
 | 
|---|
| 7 | 
 | 
|---|
| 8 | /*
 | 
|---|
| 9 |  * ChangeElementAction.cpp
 | 
|---|
| 10 |  *
 | 
|---|
| 11 |  *  Created on: May 9, 2010
 | 
|---|
| 12 |  *      Author: heber
 | 
|---|
| 13 |  */
 | 
|---|
| 14 | 
 | 
|---|
| 15 | // include config.h
 | 
|---|
| 16 | #ifdef HAVE_CONFIG_H
 | 
|---|
| 17 | #include <config.h>
 | 
|---|
| 18 | #endif
 | 
|---|
| 19 | 
 | 
|---|
| 20 | #include "Helpers/MemDebug.hpp"
 | 
|---|
| 21 | 
 | 
|---|
| 22 | #include "Descriptors/AtomIdDescriptor.hpp"
 | 
|---|
| 23 | #include "atom.hpp"
 | 
|---|
| 24 | #include "element.hpp"
 | 
|---|
| 25 | #include "Helpers/Log.hpp"
 | 
|---|
| 26 | #include "LinearAlgebra/Vector.hpp"
 | 
|---|
| 27 | #include "Helpers/Verbose.hpp"
 | 
|---|
| 28 | #include "molecule.hpp"
 | 
|---|
| 29 | #include "World.hpp"
 | 
|---|
| 30 | 
 | 
|---|
| 31 | #include <iostream>
 | 
|---|
| 32 | #include <map>
 | 
|---|
| 33 | #include <string>
 | 
|---|
| 34 | 
 | 
|---|
| 35 | using namespace std;
 | 
|---|
| 36 | 
 | 
|---|
| 37 | #include "Actions/AtomAction/ChangeElementAction.hpp"
 | 
|---|
| 38 | 
 | 
|---|
| 39 | // and construct the stuff
 | 
|---|
| 40 | #include "ChangeElementAction.def"
 | 
|---|
| 41 | #include "Action_impl_pre.hpp"
 | 
|---|
| 42 | /** =========== define the function ====================== */
 | 
|---|
| 43 | Action::state_ptr AtomChangeElementAction::performCall() {
 | 
|---|
| 44 |   atom *first = NULL;
 | 
|---|
| 45 |   molecule *mol = NULL;
 | 
|---|
| 46 | 
 | 
|---|
| 47 |   // obtain information
 | 
|---|
| 48 |   getParametersfromValueStorage();
 | 
|---|
| 49 | 
 | 
|---|
| 50 |   // create undo state
 | 
|---|
| 51 |   ElementMap Elements;
 | 
|---|
| 52 |   for (World::AtomSelectionIterator iter = World::getInstance().beginAtomSelection(); iter != World::getInstance().endAtomSelection(); ++iter) {
 | 
|---|
| 53 |     Elements.insert(std::pair<int, const element *> (iter->second->getId(), iter->second->getType()));
 | 
|---|
| 54 |   }
 | 
|---|
| 55 |   AtomChangeElementState *UndoState = new AtomChangeElementState(Elements, params);
 | 
|---|
| 56 | 
 | 
|---|
| 57 |   for (World::AtomSelectionIterator iter = World::getInstance().beginAtomSelection(); iter != World::getInstance().endAtomSelection(); ++iter) {
 | 
|---|
| 58 |     first = iter->second;
 | 
|---|
| 59 |     DoLog(1) && (Log() << Verbose(1) << "Changing atom " << *first << " to element " << *params.elemental << "." << endl);
 | 
|---|
| 60 |     mol = first->getMolecule();
 | 
|---|
| 61 |     first->removeFromMolecule(); // remove atom
 | 
|---|
| 62 |     first->setType(params.elemental);
 | 
|---|
| 63 |     mol->AddAtom(first);  // add atom to ensure correctness of formula
 | 
|---|
| 64 |   }
 | 
|---|
| 65 |   return Action::state_ptr(UndoState);
 | 
|---|
| 66 | }
 | 
|---|
| 67 | 
 | 
|---|
| 68 | Action::state_ptr AtomChangeElementAction::performUndo(Action::state_ptr _state) {
 | 
|---|
| 69 |   AtomChangeElementState *state = assert_cast<AtomChangeElementState*>(_state.get());
 | 
|---|
| 70 |   atom *first = NULL;
 | 
|---|
| 71 |   molecule *mol = NULL;
 | 
|---|
| 72 | 
 | 
|---|
| 73 |   for(ElementMap::const_iterator iter = state->Elements.begin(); iter != state->Elements.end(); ++iter) {
 | 
|---|
| 74 |     first = World::getInstance().getAtom(AtomById(iter->first));
 | 
|---|
| 75 |     mol = first->getMolecule();
 | 
|---|
| 76 |     first->removeFromMolecule(); // remove atom
 | 
|---|
| 77 |     first->setType(iter->second);
 | 
|---|
| 78 |     mol->AddAtom(first);  // add atom to ensure correctness of formula
 | 
|---|
| 79 |   }
 | 
|---|
| 80 | 
 | 
|---|
| 81 |   return Action::state_ptr(_state);
 | 
|---|
| 82 | }
 | 
|---|
| 83 | 
 | 
|---|
| 84 | Action::state_ptr AtomChangeElementAction::performRedo(Action::state_ptr _state){
 | 
|---|
| 85 |   AtomChangeElementState *state = assert_cast<AtomChangeElementState*>(_state.get());
 | 
|---|
| 86 |   atom *first = NULL;
 | 
|---|
| 87 |   molecule *mol = NULL;
 | 
|---|
| 88 | 
 | 
|---|
| 89 |   for(ElementMap::const_iterator iter = state->Elements.begin(); iter != state->Elements.end(); ++iter) {
 | 
|---|
| 90 |     first = World::getInstance().getAtom(AtomById(iter->first));
 | 
|---|
| 91 |     mol = first->getMolecule();
 | 
|---|
| 92 |     first->removeFromMolecule(); // remove atom
 | 
|---|
| 93 |     first->setType(state->params.elemental);
 | 
|---|
| 94 |     mol->AddAtom(first);  // add atom to ensure correctness of formula
 | 
|---|
| 95 |   }
 | 
|---|
| 96 | 
 | 
|---|
| 97 |   return Action::state_ptr(_state);
 | 
|---|
| 98 | }
 | 
|---|
| 99 | 
 | 
|---|
| 100 | bool AtomChangeElementAction::canUndo() {
 | 
|---|
| 101 |   return true;
 | 
|---|
| 102 | }
 | 
|---|
| 103 | 
 | 
|---|
| 104 | bool AtomChangeElementAction::shouldUndo() {
 | 
|---|
| 105 |   return true;
 | 
|---|
| 106 | }
 | 
|---|
| 107 | /** =========== end of function ====================== */
 | 
|---|