source: src/UIElements/Views/Qt4/Qt3D/GLWorldScene.cpp@ f99714

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 f99714 was d103d3, checked in by Frederik Heber <heber@…>, 13 years ago

Added additional check in CodeCheck's project disclaimer for current year in copyright.

  • had to modify all .cpp files to update to 2011.
  • Property mode set to 100644
File size: 9.1 KB
RevLine 
[907636]1/*
2 * Project: MoleCuilder
3 * Description: creates and alters molecular systems
[d103d3]4 * Copyright (C) 2010-2011 University of Bonn. All rights reserved.
[907636]5 * Please see the LICENSE file or "Copyright notice" in builder.cpp for details.
6 */
7
8/*
9 * GLWorldScene.cpp
10 *
11 * This is based on the Qt3D example "teaservice", specifically parts of teaservice.cpp.
12 *
13 * Created on: Aug 17, 2011
14 * Author: heber
15 */
16
17// include config.h
18#ifdef HAVE_CONFIG_H
19#include <config.h>
20#endif
21
22#include "GLWorldScene.hpp"
23
24#include "GLMoleculeObject.hpp"
[7188b1]25#include "GLMoleculeObject_atom.hpp"
26#include "GLMoleculeObject_bond.hpp"
[907636]27
28#include "CodePatterns/MemDebug.hpp"
29
[7188b1]30#include "CodePatterns/Log.hpp"
31
[0e9ffe]32#include "Actions/SelectionAction/Atoms/AtomByIdAction.hpp"
[89643d]33#include "Actions/SelectionAction/Atoms/NotAtomByIdAction.hpp"
[907636]34#include "atom.hpp"
[7188b1]35#include "Bond/bond.hpp"
[89643d]36#include "Descriptors/AtomIdDescriptor.hpp"
[37b2575]37#include "Helpers/helpers.hpp"
[907636]38#include "molecule.hpp"
39#include "World.hpp"
40
[ce7fdc]41using namespace MoleCuilder;
[907636]42
43GLWorldScene::GLWorldScene(QObject *parent)
44 : QObject(parent)
45{
46 init();
47
48 //changeMaterials(false);
49}
50
51GLWorldScene::~GLWorldScene()
[7188b1]52{
53 // remove all elements
54 GLMoleculeObject::cleanMaterialMap();
55}
[907636]56
57/** Initialise the WorldScene with molecules and atoms from World.
58 *
59 */
60void GLWorldScene::init()
61{
62 const std::vector<molecule*> &molecules = World::getInstance().getAllMolecules();
63
64 if (molecules.size() > 0) {
65 for (std::vector<molecule*>::const_iterator Runner = molecules.begin();
66 Runner != molecules.end();
67 Runner++) {
68 // create molecule
[7188b1]69 for (molecule::const_iterator atomiter = (*Runner)->begin();
70 atomiter != (*Runner)->end();
71 ++atomiter) {
72 // create atom objects in scene
73 atomInserted(*atomiter);
74
[9c18e4]75 // create bond objects in scene
76 bondsChanged(*atomiter);
[7188b1]77 }
[907636]78 }
79 }
80}
81
[7188b1]82/** Adds an atom to the scene.
83 *
84 * @param _atom atom to add
85 */
86void GLWorldScene::atomInserted(const atom *_atom)
[907636]87{
[7188b1]88 LOG(0, "GLWorldScene: Received signal atomInserted for atom "+toString(_atom->getId())+".");
89 GLMoleculeObject_atom *atomObject = new GLMoleculeObject_atom(this, _atom);
[37b2575]90 AtomNodeMap::iterator iter = AtomsinSceneMap.find(_atom->getId());
91 ASSERT(iter == AtomsinSceneMap.end(),
[7188b1]92 "GLWorldScene::atomAdded() - same atom "+_atom->getName()+" added again.");
[37b2575]93 AtomsinSceneMap.insert( make_pair(_atom->getId(), atomObject) );
[7188b1]94 connect (atomObject, SIGNAL(clicked(atomId_t)), this, SLOT(atomClicked(atomId_t)));
[a099d3]95 connect (atomObject, SIGNAL(hoverChanged()), this, SIGNAL(changed()));
[37b2575]96 connect (atomObject, SIGNAL(BondsChanged(const atom *)), this, SLOT(bondsChanged(const atom *)));
97 bondsChanged(_atom);
[65487f]98 emit changeOccured();
[7188b1]99}
100
101/** Removes an atom from the scene.
102 *
103 * @param _atom atom to remove
104 */
105void GLWorldScene::atomRemoved(const atom *_atom)
106{
107 LOG(0, "GLWorldScene: Received signal atomRemoved for atom "+toString(_atom->getId())+".");
108 // remove all its bonds
109 const BondList& bondlist = _atom->getListOfBonds();
110 for (BondList::const_iterator iter = bondlist.begin(); iter != bondlist.end(); ++iter) {
[37b2575]111 bondRemoved((*iter)->leftatom->getId(), (*iter)->rightatom->getId());
112 bondRemoved((*iter)->rightatom->getId(), (*iter)->leftatom->getId());
[907636]113 }
[7188b1]114 // remove atoms
[37b2575]115 AtomNodeMap::iterator iter = AtomsinSceneMap.find(_atom->getId());
116 ASSERT(iter != AtomsinSceneMap.end(),
[7188b1]117 "GLWorldScene::atomRemoved() - atom "+_atom->getName()+" not on display.");
118 GLMoleculeObject_atom *atomObject = iter->second;
119 atomObject->disconnect();
[37b2575]120 AtomsinSceneMap.erase(iter);
[7188b1]121 delete atomObject;
[65487f]122 emit changeOccured();
[907636]123}
124
[37b2575]125/** Updates the bond structure of the signaled \a _atom.
126 *
127 * @param _atom atom whose bonds changed.
128 */
129void GLWorldScene::bondsChanged(const atom *_atom)
130{
131 const atomId_t id = _atom->getId();
132
133 // create list with all present bonds
134 std::set< atomId_t > presentBonds;
135 std::pair< BondIdsMap::const_iterator, BondIdsMap::const_iterator> range =
136 BondIdsinSceneMap.equal_range( id );
137 for (BondIdsMap::const_iterator iter = range.first; iter != range.second; ++iter) {
138 const atomId_t otherid = iter->second;
139#ifndef NDEBUG
140 std::set< atomId_t >::const_iterator iter = presentBonds.find(otherid);
141 ASSERT(iter == presentBonds.end(),
142 "GLWorldScene::BondsChanged() - bond id "+toString(otherid)+" for atom "
143 +toString(id)+" present twice.");
144#endif
145 presentBonds.insert( otherid );
146 }
147 LOG(0, "We have the following bonds: "+toString(presentBonds)+".");
148
149 // search for added bonds
150 const BondList &bondlist = _atom->getListOfBonds();
151 for (BondList::const_iterator bonditer = bondlist.begin();
152 bonditer != bondlist.end();
153 ++bonditer) {
154 const bond *_bond = *bonditer;
155 const atomId_t otherid = _bond->GetOtherAtom(_atom)->getId();
156 const BondIds ids = std::make_pair( id, otherid );
157 BondNodeMap::const_iterator iter = BondsinSceneMap.find(ids);
158 if (iter != BondsinSceneMap.end()) {
159 // bond is already present
160 std::set< atomId_t >::const_iterator iter = presentBonds.find(otherid);
161 ASSERT(iter != presentBonds.end(),
162 "GLWorldScene::BondsChanged() - other id "+toString(otherid)+" for atom "
163 +toString(_atom->getId())+" not present in BondIdsinSceneMap.");
164 presentBonds.erase(otherid);
165 LOG(0, "Removing "+toString(otherid)+" from presentBonds.");
166 } else {
167 // insert new bond
168 bondInserted(_bond);
169 }
170 }
171 LOG(0, "The following bonds should not be present: "+toString(presentBonds)+".");
172
173 // remove all still presentBonds
174 for (std::set< atomId_t >::iterator iter = presentBonds.begin();
175 !presentBonds.empty(); iter = presentBonds.begin()) {
176 bondRemoved( id, *iter );
177 }
178}
179
180/** Adds a bond to the scene.
[7188b1]181 *
182 * @param _bond bond to add
183 */
184void GLWorldScene::bondInserted(const bond *_bond)
185{
[37b2575]186 LOG(0, "GLWorldScene::bondInserted() - Adding bond "+toString(*_bond)+".");
[7188b1]187 const double distance =
188 _bond->leftatom->getPosition().distance(_bond->rightatom->getPosition())/2.;
189 {
190 // left bond
191 const BondIds Leftids( make_pair(_bond->leftatom->getId(), _bond->rightatom->getId()) );
[37b2575]192 BondNodeMap::iterator iter = BondsinSceneMap.find(Leftids);
193 ASSERT(iter == BondsinSceneMap.end(),
[7188b1]194 "GLWorldScene::bondAdded() - same left-sided bond "+toString(*_bond)+" added again.");
195 GLMoleculeObject_bond *bondObject =
196 new GLMoleculeObject_bond(this, _bond, distance, GLMoleculeObject_bond::left);
[37b2575]197 BondsinSceneMap.insert( make_pair(Leftids, bondObject) );
198 BondIdsinSceneMap.insert( Leftids );
[7188b1]199 }
200 {
201 // right bond
202 const BondIds Rightids( make_pair(_bond->rightatom->getId(), _bond->leftatom->getId()) );
[37b2575]203 BondNodeMap::iterator iter = BondsinSceneMap.find(Rightids);
204 ASSERT(iter == BondsinSceneMap.end(),
[7188b1]205 "GLWorldScene::bondAdded() - same right-sided bond "+toString(*_bond)+" added again.");
206 GLMoleculeObject_bond *bondObject =
207 new GLMoleculeObject_bond(this, _bond, distance, GLMoleculeObject_bond::right);
[37b2575]208 BondsinSceneMap.insert( make_pair(Rightids, bondObject) );
209 BondIdsinSceneMap.insert( Rightids );
[7188b1]210 }
[65487f]211 emit changeOccured();
[7188b1]212}
[907636]213
[37b2575]214/** Removes a bond from the scene.
[7188b1]215 *
216 * @param _bond bond to remove
217 */
[37b2575]218void GLWorldScene::bondRemoved(const atomId_t leftnr, const atomId_t rightnr)
[907636]219{
[37b2575]220 LOG(0, "GLWorldScene::bondRemoved() - Removing bond between "+toString(leftnr)+" and "+toString(leftnr)+".");
[7188b1]221 {
222 // left bond
[37b2575]223 const BondIds Leftids( make_pair(leftnr, rightnr) );
224 BondNodeMap::iterator leftiter = BondsinSceneMap.find( Leftids );
225 ASSERT(leftiter != BondsinSceneMap.end(),
226 "GLWorldScene::bondRemoved() - bond "+toString(leftnr)+"-"
227 +toString(rightnr)+" not on display.");
[7188b1]228 GLMoleculeObject_bond *bondObject = leftiter->second;
[37b2575]229 BondsinSceneMap.erase(leftiter);
[7188b1]230 delete bondObject;
231 }
[37b2575]232 // remove from bond ids
233 std::pair<BondIdsMap::iterator, BondIdsMap::iterator> leftrange =
234 BondIdsinSceneMap.equal_range(leftnr);
235 BondIdsMap::iterator iter;
236 for (iter = leftrange.first; iter != leftrange.second; ++iter) {
237 if (iter->second == rightnr) {
238 BondIdsinSceneMap.erase(iter);
239 break;
240 }
[907636]241 }
[37b2575]242 ASSERT(iter != leftrange.second,
243 "GLWorldScene::bondRemoved() - could not find ("
244 +toString(leftnr)+"-"+toString(rightnr)+" in BondIdsinSceneMap.");
[65487f]245 emit changeOccured();
[7188b1]246}
247
248void GLWorldScene::initialize(QGLView *view, QGLPainter *painter) const
249{
250 // Initialize all of the mesh objects that we have as children.
251 foreach (QObject *obj, children()) {
252 GLMoleculeObject *meshobj = qobject_cast<GLMoleculeObject *>(obj);
253 if (meshobj)
254 meshobj->initialize(view, painter);
255 }
256}
257
258void GLWorldScene::draw(QGLPainter *painter) const
259{
260 // Draw all of the mesh objects that we have as children.
261 foreach (QObject *obj, children()) {
262 GLMoleculeObject *meshobj = qobject_cast<GLMoleculeObject *>(obj);
263 if (meshobj)
264 meshobj->draw(painter);
265 }
[907636]266}
[06ebf5]267
[7188b1]268void GLWorldScene::atomClicked(atomId_t no)
[907636]269{
[7188b1]270 qDebug("GLWorldScene: atom %d has been clicked.", no);
[89643d]271 const atom *Walker = World::getInstance().getAtom(AtomById(no));
272 if (!World::getInstance().isSelected(Walker))
[0e9ffe]273 SelectionAtomById(no);
[89643d]274 else
275 SelectionNotAtomById(no);
[7188b1]276 emit clicked(no);
[907636]277}
[029bb4]278
Note: See TracBrowser for help on using the repository browser.