source: src/World.cpp@ 5be0eb

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 5be0eb was 6e97e5, checked in by Tillmann Crueger <crueger@…>, 15 years ago

Added a generic Iterator that can be used to iterate only over certain parts of an internal data structure

  • Property mode set to 100644
File size: 6.3 KB
Line 
1/*
2 * World.cpp
3 *
4 * Created on: Feb 3, 2010
5 * Author: crueger
6 */
7
8#include "World.hpp"
9
10#include "atom.hpp"
11#include "molecule.hpp"
12#include "periodentafel.hpp"
13#include "Descriptors/AtomDescriptor.hpp"
14#include "Descriptors/AtomDescriptor_impl.hpp"
15#include "Descriptors/MoleculeDescriptor.hpp"
16#include "Descriptors/MoleculeDescriptor_impl.hpp"
17#include "Descriptors/SelectiveIterator_impl.hpp"
18#include "Actions/ManipulateAtomsProcess.hpp"
19
20#include "Patterns/Singleton_impl.hpp"
21
22using namespace std;
23
24/******************************* getter and setter ************************/
25periodentafel *&World::getPeriode(){
26 return periode;
27}
28
29// Atoms
30
31atom* World::getAtom(AtomDescriptor descriptor){
32 return descriptor.find();
33}
34
35vector<atom*> World::getAllAtoms(AtomDescriptor descriptor){
36 return descriptor.findAll();
37}
38
39vector<atom*> World::getAllAtoms(){
40 return getAllAtoms(AllAtoms());
41}
42
43int World::numAtoms(){
44 return atoms.size();
45}
46
47// Molecules
48
49molecule *World::getMolecule(MoleculeDescriptor descriptor){
50 return descriptor.find();
51}
52
53std::vector<molecule*> World::getAllMolecules(MoleculeDescriptor descriptor){
54 return descriptor.findAll();
55}
56
57int World::numMolecules(){
58 return molecules_deprecated->ListOfMolecules.size();
59}
60
61/******************** Methods to change World state *********************/
62
63molecule* World::createMolecule(){
64 OBSERVE;
65 molecule *mol = NULL;
66 mol = NewMolecule();
67 assert(!molecules.count(currMoleculeId));
68 mol->setId(currMoleculeId++);
69 // store the molecule by ID
70 molecules[mol->getId()] = mol;
71 mol->signOn(this);
72 return mol;
73}
74
75void World::destroyMolecule(molecule* mol){
76 OBSERVE;
77 destroyMolecule(mol->getId());
78}
79
80void World::destroyMolecule(moleculeId_t id){
81 OBSERVE;
82 molecule *mol = molecules[id];
83 assert(mol);
84 DeleteMolecule(mol);
85 molecules.erase(id);
86}
87
88
89atom *World::createAtom(){
90 OBSERVE;
91 atomId_t id = getNextAtomId();
92 atom *res = NewAtom(id);
93 res->setWorld(this);
94 // store the atom by ID
95 atoms[res->getId()] = res;
96 return res;
97}
98
99int World::registerAtom(atom *atom){
100 OBSERVE;
101 atomId_t id = getNextAtomId();
102 atom->setId(id);
103 atom->setWorld(this);
104 atoms[atom->getId()] = atom;
105 return atom->getId();
106}
107
108void World::destroyAtom(atom* atom){
109 OBSERVE;
110 int id = atom->getId();
111 destroyAtom(id);
112}
113
114void World::destroyAtom(atomId_t id) {
115 OBSERVE;
116 atom *atom = atoms[id];
117 assert(atom);
118 DeleteAtom(atom);
119 atoms.erase(id);
120 releaseAtomId(id);
121}
122
123bool World::changeAtomId(atomId_t oldId, atomId_t newId, atom* target){
124 OBSERVE;
125 // in case this call did not originate from inside the atom, we redirect it,
126 // to also let it know that it has changed
127 if(!target){
128 target = atoms[oldId];
129 assert(target && "Atom with that ID not found");
130 return target->changeId(newId);
131 }
132 else{
133 if(reserveAtomId(newId)){
134 atoms.erase(oldId);
135 atoms.insert(pair<atomId_t,atom*>(newId,target));
136 return true;
137 }
138 else{
139 return false;
140 }
141 }
142}
143
144ManipulateAtomsProcess* World::manipulateAtoms(boost::function<void(atom*)> op,std::string name,AtomDescriptor descr){
145 return new ManipulateAtomsProcess(op, descr,name,true);
146}
147
148ManipulateAtomsProcess* World::manipulateAtoms(boost::function<void(atom*)> op,std::string name){
149 return manipulateAtoms(op,name,AllAtoms());
150}
151
152/********************* Internal Change methods for double Callback and Observer mechanism ********/
153
154void World::doManipulate(ManipulateAtomsProcess *proc){
155 proc->signOn(this);
156 {
157 OBSERVE;
158 proc->doManipulate(this);
159 }
160 proc->signOff(this);
161}
162/******************************* IDManagement *****************************/
163
164// Atoms
165
166atomId_t World::getNextAtomId(){
167 // see if we can reuse some Id
168 if(atomIdPool.empty()){
169 return currAtomId++;
170 }
171 else{
172 // we give out the first ID from the pool
173 atomId_t id = *(atomIdPool.begin());
174 atomIdPool.erase(id);
175 return id;
176 }
177}
178
179void World::releaseAtomId(atomId_t id){
180 atomIdPool.insert(id);
181 // defragmentation of the pool
182 set<atomId_t>::reverse_iterator iter;
183 // go through all Ids in the pool that lie immediately below the border
184 while(!atomIdPool.empty() && *(atomIdPool.rbegin())==(currAtomId-1)){
185 atomIdPool.erase(--currAtomId);
186 }
187}
188
189bool World::reserveAtomId(atomId_t id){
190 if(id>=currAtomId ){
191 // add all ids between the new one and current border as available
192 for(atomId_t pos=currAtomId; pos<id; ++pos){
193 atomIdPool.insert(pos);
194 }
195 currAtomId=id+1;
196 return true;
197 }
198 else if(atomIdPool.count(id)){
199 atomIdPool.erase(id);
200 return true;
201 }
202 else{
203 // this ID could not be reserved
204 return false;
205 }
206}
207
208// Molecules
209
210/******************************* Iterators ********************************/
211
212// Build the AtomIterator from template
213CONSTRUCT_SELECTIVE_ITERATOR(atom*,World::AtomSet,AtomDescriptor);
214
215
216World::AtomIterator World::getAtomIter(AtomDescriptor descr){
217 return AtomIterator(descr,atoms);
218}
219
220World::AtomIterator World::atomEnd(){
221 return AtomIterator(AllAtoms(),atoms,atoms.end());
222}
223
224// build the MoleculeIterator from template
225CONSTRUCT_SELECTIVE_ITERATOR(molecule*,World::MoleculeSet,MoleculeDescriptor);
226
227World::MoleculeIterator World::getMoleculeIter(MoleculeDescriptor descr){
228 return MoleculeIterator(descr,molecules);
229}
230
231World::MoleculeIterator World::moleculeEnd(){
232 return MoleculeIterator(AllMolecules(),molecules,molecules.end());
233}
234
235/******************************* Singleton Stuff **************************/
236
237World::World() :
238 periode(new periodentafel),
239 atoms(),
240 currAtomId(0),
241 molecules(),
242 currMoleculeId(0),
243 molecules_deprecated(new MoleculeListClass(this))
244{
245 molecules_deprecated->signOn(this);
246}
247
248World::~World()
249{
250 molecules_deprecated->signOff(this);
251 delete molecules_deprecated;
252 delete periode;
253 MoleculeSet::iterator molIter;
254 for(molIter=molecules.begin();molIter!=molecules.end();++molIter){
255 DeleteMolecule((*molIter).second);
256 }
257 molecules.clear();
258 AtomSet::iterator atIter;
259 for(atIter=atoms.begin();atIter!=atoms.end();++atIter){
260 DeleteAtom((*atIter).second);
261 }
262 atoms.clear();
263}
264
265// Explicit instantiation of the singleton mechanism at this point
266
267CONSTRUCT_SINGLETON(World)
268
269/******************************* deprecated Legacy Stuff ***********************/
270
271MoleculeListClass *&World::getMolecules() {
272 return molecules_deprecated;
273}
Note: See TracBrowser for help on using the repository browser.