source: src/Analysis/analysis_correlation.cpp@ d619338

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 d619338 was 99db9b, checked in by Frederik Heber <heber@…>, 10 years ago

Replaced all World::getSelected...() to const version where possible.

  • also added const version of World::getSelectedAtoms().
  • Property mode set to 100644
File size: 29.1 KB
RevLine 
[bcf653]1/*
2 * Project: MoleCuilder
3 * Description: creates and alters molecular systems
[0aa122]4 * Copyright (C) 2010-2012 University of Bonn. All rights reserved.
[5aaa43]5 * Copyright (C) 2013 Frederik Heber. All rights reserved.
[94d5ac6]6 *
7 *
8 * This file is part of MoleCuilder.
9 *
10 * MoleCuilder is free software: you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation, either version 2 of the License, or
13 * (at your option) any later version.
14 *
15 * MoleCuilder is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
19 *
20 * You should have received a copy of the GNU General Public License
21 * along with MoleCuilder. If not, see <http://www.gnu.org/licenses/>.
[bcf653]22 */
23
[c4d4df]24/*
25 * analysis.cpp
26 *
27 * Created on: Oct 13, 2009
28 * Author: heber
29 */
30
[bf3817]31// include config.h
32#ifdef HAVE_CONFIG_H
33#include <config.h>
34#endif
35
[ad011c]36#include "CodePatterns/MemDebug.hpp"
[112b09]37
[c1a9d6]38#include <algorithm>
[c4d4df]39#include <iostream>
[36166d]40#include <iomanip>
[505d05]41#include <limits>
[c4d4df]42
[6f0841]43#include "Atom/atom.hpp"
[129204]44#include "Bond/bond.hpp"
[d127c8]45#include "Tesselation/BoundaryTriangleSet.hpp"
[be945c]46#include "Box.hpp"
[3bdb6d]47#include "Element/element.hpp"
[ad011c]48#include "CodePatterns/Info.hpp"
49#include "CodePatterns/Log.hpp"
[208237b]50#include "CodePatterns/Verbose.hpp"
[e65878]51#include "Descriptors/AtomOfMoleculeSelectionDescriptor.hpp"
52#include "Descriptors/MoleculeFormulaDescriptor.hpp"
[4b8630]53#include "Descriptors/MoleculeOfAtomSelectionDescriptor.hpp"
[ea430a]54#include "Formula.hpp"
[208237b]55#include "LinearAlgebra/Vector.hpp"
56#include "LinearAlgebra/RealSpaceMatrix.hpp"
[c1a9d6]57#include "LinkedCell/LinkedCell_View.hpp"
[c4d4df]58#include "molecule.hpp"
[d127c8]59#include "Tesselation/tesselation.hpp"
60#include "Tesselation/tesselationhelpers.hpp"
61#include "Tesselation/triangleintersectionlist.hpp"
[be945c]62#include "World.hpp"
[208237b]63#include "WorldTime.hpp"
[c4d4df]64
[be945c]65#include "analysis_correlation.hpp"
66
67/** Calculates the dipole vector of a given atomSet.
68 *
69 * Note that we use the following procedure as rule of thumb:
70 * -# go through every bond of the atom
[d1912f]71 * -# calculate the difference of electronegativities \f$\Delta\mathrm{EN}\f$
72 * -# if \f$\Delta\mathrm{EN} > 0.5\f$, we align the bond vector in direction of the more negative element
[be945c]73 * -# sum up all vectors
74 * -# finally, divide by the number of summed vectors
75 *
76 * @param atomsbegin begin iterator of atomSet
77 * @param atomsend end iterator of atomset
78 * @return dipole vector
79 */
80Vector getDipole(molecule::const_iterator atomsbegin, molecule::const_iterator atomsend)
81{
82 Vector DipoleVector;
83 size_t SumOfVectors = 0;
[8fc1a6]84 Box &domain = World::getInstance().getDomain();
85
86 // go through all atoms
[be945c]87 for (molecule::const_iterator atomiter = atomsbegin;
88 atomiter != atomsend;
89 ++atomiter) {
90 // go through all bonds
[9d83b6]91 const BondList& ListOfBonds = (*atomiter)->getListOfBonds();
[4fc828]92 ASSERT(ListOfBonds.begin() != ListOfBonds.end(),
93 "getDipole() - no bonds in molecule!");
[9d83b6]94 for (BondList::const_iterator bonditer = ListOfBonds.begin();
95 bonditer != ListOfBonds.end();
[be945c]96 ++bonditer) {
97 const atom * Otheratom = (*bonditer)->GetOtherAtom(*atomiter);
98 if (Otheratom->getId() > (*atomiter)->getId()) {
99 const double DeltaEN = (*atomiter)->getType()->getElectronegativity()
100 -Otheratom->getType()->getElectronegativity();
[8fc1a6]101 // get distance and correct for boundary conditions
102 Vector BondDipoleVector = domain.periodicDistanceVector(
103 (*atomiter)->getPosition(),
104 Otheratom->getPosition());
[be945c]105 // DeltaEN is always positive, gives correct orientation of vector
106 BondDipoleVector.Normalize();
107 BondDipoleVector *= DeltaEN;
[4fc828]108 LOG(3,"INFO: Dipole vector from bond " << **bonditer << " is " << BondDipoleVector);
[be945c]109 DipoleVector += BondDipoleVector;
110 SumOfVectors++;
111 }
112 }
113 }
[4fc828]114 LOG(3,"INFO: Sum over all bond dipole vectors is "
115 << DipoleVector << " with " << SumOfVectors << " in total.");
116 if (SumOfVectors != 0)
117 DipoleVector *= 1./(double)SumOfVectors;
[44f53e]118 LOG(2, "INFO: Resulting dipole vector is " << DipoleVector);
[be945c]119
120 return DipoleVector;
121};
122
[1cc661]123/** Calculate minimum and maximum amount of trajectory steps by going through given atomic trajectories.
124 * \param vector of atoms whose trajectories to check for [min,max]
125 * \return range with [min, max]
126 */
[99db9b]127range<size_t> getMaximumTrajectoryBounds(const std::vector<const atom *> &atoms)
[1cc661]128{
129 // get highest trajectory size
130 LOG(0,"STATUS: Retrieving maximum amount of time steps ...");
[505d05]131 if (atoms.size() == 0)
132 return range<size_t>(0,0);
133 size_t max_timesteps = std::numeric_limits<size_t>::min();
134 size_t min_timesteps = std::numeric_limits<size_t>::max();
[99db9b]135 BOOST_FOREACH(const atom *_atom, atoms) {
[1cc661]136 if (_atom->getTrajectorySize() > max_timesteps)
137 max_timesteps = _atom->getTrajectorySize();
[505d05]138 if (_atom->getTrajectorySize() < min_timesteps)
[1cc661]139 min_timesteps = _atom->getTrajectorySize();
140 }
141 LOG(1,"INFO: Minimum number of time steps found is " << min_timesteps);
142 LOG(1,"INFO: Maximum number of time steps found is " << max_timesteps);
143
144 return range<size_t>(min_timesteps, max_timesteps);
145}
146
[0a7fad]147/** Calculates the angular dipole zero orientation from current time step.
[e65878]148 * \param molecules vector of molecules to calculate dipoles of
[0a7fad]149 * \return map with orientation vector for each atomic id given in \a atoms.
150 */
[99db9b]151std::map<atomId_t, Vector> CalculateZeroAngularDipole(const std::vector<const molecule *> &molecules)
[0a7fad]152{
153 // get zero orientation for each molecule.
[e65878]154 LOG(0,"STATUS: Calculating dipoles for current time step ...");
[0a7fad]155 std::map<atomId_t, Vector> ZeroVector;
[99db9b]156 BOOST_FOREACH(const molecule *_mol, molecules) {
157 const Vector Dipole = getDipole(_mol->begin(),_mol->end());
158 for(molecule::const_iterator iter = _mol->begin(); iter != _mol->end(); ++iter)
[0a7fad]159 ZeroVector[(*iter)->getId()] = Dipole;
160 LOG(2,"INFO: Zero alignment for molecule " << _mol->getId() << " is " << Dipole);
161 }
162 LOG(1,"INFO: We calculated zero orientation for a total of " << molecules.size() << " molecule(s).");
163
164 return ZeroVector;
165}
[1cc661]166
[ea430a]167/** Calculates the dipole angular correlation for given molecule type.
[208237b]168 * Calculate the change of the dipole orientation angle over time.
[ea430a]169 * Note given element order is unimportant (i.e. g(Si, O) === g(O, Si))
[be945c]170 * Angles are given in degrees.
[4b8630]171 * \param &atoms list of atoms of the molecules taking part (Note: molecules may
172 * change over time as bond structure is recalculated, hence we need the atoms)
[cda81d]173 * \param timestep time step to calculate angular correlation for (relative to
174 * \a ZeroVector)
[325687]175 * \param ZeroVector map with Zero orientation vector for each atom in \a atoms.
[99b87a]176 * \param DontResetTime don't reset time to old value (triggers re-creation of bond system)
[ea430a]177 * \return Map of doubles with values the pair of the two atoms.
178 */
[325687]179DipoleAngularCorrelationMap *DipoleAngularCorrelation(
[e65878]180 const Formula &DipoleFormula,
[cda81d]181 const size_t timestep,
[e65878]182 const std::map<atomId_t, Vector> &ZeroVector,
[99b87a]183 const enum ResetWorldTime DoTimeReset
[325687]184 )
[ea430a]185{
186 Info FunctionInfo(__func__);
[caa30b]187 DipoleAngularCorrelationMap *outmap = new DipoleAngularCorrelationMap;
[be945c]188
[99b87a]189 unsigned int oldtime = 0;
190 if (DoTimeReset == DoResetTime) {
191 // store original time step
192 oldtime = WorldTime::getTime();
193 }
[0a7fad]194
[cda81d]195 // set time step
[505d05]196 LOG(0,"STATUS: Stepping onto to time step " << timestep << ".");
[cda81d]197 World::getInstance().setTime(timestep);
198
199 // get all molecules for this time step
[e65878]200 World::getInstance().clearMoleculeSelection();
201 World::getInstance().selectAllMolecules(MoleculeByFormula(DipoleFormula));
[99db9b]202 std::vector<const molecule *> molecules =
203 const_cast<const World &>(World::getInstance()).getSelectedMolecules();
[870b4b]204 LOG(1,"INFO: There are " << molecules.size() << " molecules for time step " << timestep << ".");
[208237b]205
[cda81d]206 // calculate dipoles for each
[870b4b]207 LOG(0,"STATUS: Calculating dipoles for time step " << timestep << " ...");
[cda81d]208 size_t i=0;
[870b4b]209 size_t Counter_rejections = 0;
[99db9b]210 BOOST_FOREACH(const molecule *_mol, molecules) {
211 const Vector Dipole = getDipole(_mol->begin(),_mol->end());
[e65878]212 LOG(3,"INFO: Dipole vector at time step " << timestep << " for for molecule "
[cda81d]213 << _mol->getId() << " is " << Dipole);
[e65878]214 // check that all atoms are valid (zeroVector known)
[99db9b]215 molecule::const_iterator iter = _mol->begin();
216 for(; iter != _mol->end(); ++iter) {
[e65878]217 if (!ZeroVector.count((*iter)->getId()))
218 break;
219 }
[99db9b]220 if (iter != _mol->end()) {
[e65878]221 ELOG(2, "Skipping molecule " << _mol->getName() << " as not all atoms have a valid zeroVector.");
[870b4b]222 ++Counter_rejections;
[e65878]223 continue;
224 } else
[99db9b]225 iter = _mol->begin();
[e65878]226 std::map<atomId_t, Vector>::const_iterator zeroValue = ZeroVector.find((*iter)->getId()); //due to iter is const
[cda81d]227 double angle = 0.;
228 LOG(2, "INFO: ZeroVector of first atom " << **iter << " is "
[e65878]229 << zeroValue->second << ".");
[cda81d]230 LOG(4, "INFO: Squared norm of difference vector is "
[e65878]231 << (zeroValue->second - Dipole).NormSquared() << ".");
232 if ((zeroValue->second - Dipole).NormSquared() > MYEPSILON)
233 angle = Dipole.Angle(zeroValue->second) * (180./M_PI);
[cda81d]234 else
235 LOG(2, "INFO: Both vectors (almost) coincide, numerically unstable, angle set to zero.");
[e9bdef]236 // we print six digits, hence round up to six digit precision
237 const double precision = 1e-6;
238 angle = precision*floor(angle/precision);
[cda81d]239 LOG(1,"INFO: Resulting relative angle for molecule " << _mol->getName()
240 << " is " << angle << ".");
[59fff1]241 outmap->insert ( std::make_pair (angle, *iter ) );
[cda81d]242 ++i;
[208237b]243 }
[870b4b]244 ASSERT(Counter_rejections <= molecules.size(),
245 "DipoleAngularCorrelation() - more rejections ("+toString(Counter_rejections)
246 +") than there are molecules ("+toString(molecules.size())+").");
247 LOG(1,"INFO: " << Counter_rejections << " molecules have been rejected in time step " << timestep << ".");
248
249 LOG(0,"STATUS: Done with calculating dipoles.");
[208237b]250
[99b87a]251 if (DoTimeReset == DoResetTime) {
252 // re-set to original time step again
253 World::getInstance().setTime(oldtime);
254 }
[208237b]255
256 // and return results
257 return outmap;
258};
259
260/** Calculates the dipole correlation for given molecule type.
261 * I.e. we calculate how the angle between any two given dipoles in the
262 * systems behaves. Sort of pair correlation but distance is replaced by
263 * the orientation distance, i.e. an angle.
264 * Note given element order is unimportant (i.e. g(Si, O) === g(O, Si))
265 * Angles are given in degrees.
266 * \param *molecules vector of molecules
267 * \return Map of doubles with values the pair of the two atoms.
268 */
[99db9b]269DipoleCorrelationMap *DipoleCorrelation(
270 const std::vector<const molecule *> &molecules)
[208237b]271{
272 Info FunctionInfo(__func__);
273 DipoleCorrelationMap *outmap = new DipoleCorrelationMap;
274// double distance = 0.;
275// Box &domain = World::getInstance().getDomain();
276//
277 if (molecules.empty()) {
[47d041]278 ELOG(1, "No molecule given.");
[208237b]279 return outmap;
280 }
281
[99db9b]282 for (std::vector<const molecule *>::const_iterator MolWalker = molecules.begin();
[92e5cb]283 MolWalker != molecules.end(); ++MolWalker) {
[47d041]284 LOG(2, "INFO: Current molecule is " << (*MolWalker)->getId() << ".");
[99db9b]285 const Vector Dipole = getDipole((*MolWalker)->begin(), (*MolWalker)->end());
286 std::vector<const molecule *>::const_iterator MolOtherWalker = MolWalker;
287 for (++MolOtherWalker; MolOtherWalker != molecules.end();
[92e5cb]288 ++MolOtherWalker) {
[47d041]289 LOG(2, "INFO: Current other molecule is " << (*MolOtherWalker)->getId() << ".");
[99db9b]290 const Vector OtherDipole = getDipole((*MolOtherWalker)->begin(), (*MolOtherWalker)->end());
[be945c]291 const double angle = Dipole.Angle(OtherDipole) * (180./M_PI);
[47d041]292 LOG(1, "Angle is " << angle << ".");
[be945c]293 outmap->insert ( make_pair (angle, make_pair ((*MolWalker), (*MolOtherWalker)) ) );
294 }
295 }
[ea430a]296 return outmap;
297};
298
[c1a9d6]299/** Calculates the pair correlation between given atom sets.
300 *
301 * Note we correlate each of the \a &atomsfirst with each of the second set
302 * \a &atoms_second. However, we are aware of double counting. If an atom is
303 * in either set, the pair is counted only once.
304 *
305 * \param &atoms_first vector of atoms
306 * \param &atoms_second vector of atoms
307 * \param max_distance maximum distance for the correlation
[c4d4df]308 * \return Map of doubles with values the pair of the two atoms.
309 */
[c1a9d6]310PairCorrelationMap *PairCorrelation(
[a58c16]311 const World::ConstAtomComposite &atoms_first,
312 const World::ConstAtomComposite &atoms_second,
[c1a9d6]313 const double max_distance)
[c4d4df]314{
[3930eb]315 Info FunctionInfo(__func__);
[caa30b]316 PairCorrelationMap *outmap = new PairCorrelationMap;
[e791dc]317 //double distance = 0.;
[014475]318 Box &domain = World::getInstance().getDomain();
[c4d4df]319
[c1a9d6]320 if (atoms_first.empty() || atoms_second.empty()) {
321 ELOG(1, "No atoms given.");
[c4d4df]322 return outmap;
323 }
[c78d44]324
[c1a9d6]325 //!> typedef for an unsorted container, (output) compatible with STL algorithms
326 typedef std::vector<const TesselPoint *> LinkedVector;
[c4d4df]327
[c1a9d6]328 // create intersection (to know when to check for double-counting)
329 LinkedVector intersected_atoms(atoms_second.size(), NULL);
330 LinkedVector::iterator intersected_atoms_end =
331 std::set_intersection(
332 atoms_first.begin(),atoms_first.end(),
333 atoms_second.begin(), atoms_second.end(),
334 intersected_atoms.begin());
[a2a2f7]335 const LinkedCell::LinkedList intersected_atoms_set(intersected_atoms.begin(), intersected_atoms_end);
[c78d44]336
[c1a9d6]337 // get linked cell view
338 LinkedCell::LinkedCell_View LC = World::getInstance().getLinkedCell(max_distance);
339
340 // convert second to _sorted_ set
341 LinkedCell::LinkedList atoms_second_set(atoms_second.begin(), atoms_second.end());
342 LOG(2, "INFO: first set has " << atoms_first.size()
343 << " and second set has " << atoms_second_set.size() << " atoms.");
344
345 // fill map
[a58c16]346 for (World::ConstAtomComposite::const_iterator iter = atoms_first.begin();
[c1a9d6]347 iter != atoms_first.end();
348 ++iter) {
349 const TesselPoint * const Walker = *iter;
350 LOG(3, "INFO: Current point is " << Walker->getName() << ".");
351 // obtain all possible neighbors (that is a sorted set)
352 LinkedCell::LinkedList ListOfNeighbors = LC.getPointsInsideSphere(
353 max_distance,
354 Walker->getPosition());
355 LOG(2, "INFO: There are " << ListOfNeighbors.size() << " neighbors.");
356
357 // create intersection with second set
358 // NOTE: STL algorithms do mostly not work on sorted container because reassignment
359 // of a value may also require changing its position.
360 LinkedVector intersected_set(atoms_second.size(), NULL);
361 LinkedVector::iterator intersected_end =
362 std::set_intersection(
363 ListOfNeighbors.begin(),ListOfNeighbors.end(),
364 atoms_second_set.begin(), atoms_second_set.end(),
365 intersected_set.begin());
366 // count remaining elements
367 LOG(2, "INFO: Intersection with second set has " << int(intersected_end - intersected_set.begin()) << " elements.");
368 // we have some possible candidates, go through each
369 for (LinkedVector::const_iterator neighboriter = intersected_set.begin();
370 neighboriter != intersected_end;
371 ++neighboriter) {
372 const TesselPoint * const OtherWalker = (*neighboriter);
373 LinkedCell::LinkedList::const_iterator equaliter = intersected_atoms_set.find(OtherWalker);
374 if ((equaliter != intersected_atoms_set.end()) && (OtherWalker <= Walker)) {
375 // present in both sets, assure that we are larger
376 continue;
[7ea9e6]377 }
[c1a9d6]378 LOG(3, "INFO: Current other point is " << *OtherWalker << ".");
379 const double distance = domain.periodicDistance(OtherWalker->getPosition(),Walker->getPosition());
380 LOG(3, "INFO: Resulting distance is " << distance << ".");
381 outmap->insert (
382 std::pair<double, std::pair <const TesselPoint *, const TesselPoint*> > (
383 distance,
384 std::make_pair (Walker, OtherWalker)
385 )
386 );
[7ea9e6]387 }
[c78d44]388 }
[c1a9d6]389 // and return
[7ea9e6]390 return outmap;
391};
392
[c4d4df]393/** Calculates the distance (pair) correlation between a given element and a point.
[a5551b]394 * \param *molecules list of molecules structure
[c78d44]395 * \param &elements vector of elements to correlate with point
[c4d4df]396 * \param *point vector to the correlation point
397 * \return Map of dobules with values as pairs of atom and the vector
398 */
[99db9b]399CorrelationToPointMap *CorrelationToPoint(
400 const std::vector<const molecule *> &molecules,
401 const std::vector<const element *> &elements,
402 const Vector *point )
[c4d4df]403{
[3930eb]404 Info FunctionInfo(__func__);
[caa30b]405 CorrelationToPointMap *outmap = new CorrelationToPointMap;
[c4d4df]406 double distance = 0.;
[014475]407 Box &domain = World::getInstance().getDomain();
[c4d4df]408
[e65de8]409 if (molecules.empty()) {
[47d041]410 LOG(1, "No molecule given.");
[c4d4df]411 return outmap;
412 }
[e791dc]413
[99db9b]414 for (std::vector<const molecule *>::const_iterator MolWalker = molecules.begin();
415 MolWalker != molecules.end();
416 MolWalker++) {
[47d041]417 LOG(2, "Current molecule is " << *MolWalker << ".");
[99db9b]418 for (molecule::const_iterator iter = (*MolWalker)->begin(); iter != (*MolWalker)->end(); ++iter) {
[47d041]419 LOG(3, "Current atom is " << **iter << ".");
[e5c0a1]420 for (vector<const element *>::const_iterator type = elements.begin(); type != elements.end(); ++type)
[d74077]421 if ((*type == NULL) || ((*iter)->getType() == *type)) {
422 distance = domain.periodicDistance((*iter)->getPosition(),*point);
[47d041]423 LOG(4, "Current distance is " << distance << ".");
[59fff1]424 outmap->insert (
425 std::pair<double, std::pair<const atom *, const Vector*> >(
426 distance,
427 std::pair<const atom *, const Vector*> (
428 (*iter),
429 point)
430 )
431 );
[e65de8]432 }
[c4d4df]433 }
[e65de8]434 }
[c4d4df]435
436 return outmap;
437};
438
[7ea9e6]439/** Calculates the distance (pair) correlation between a given element, all its periodic images and a point.
440 * \param *molecules list of molecules structure
[c78d44]441 * \param &elements vector of elements to correlate to point
[7ea9e6]442 * \param *point vector to the correlation point
443 * \param ranges[NDIM] interval boundaries for the periodic images to scan also
444 * \return Map of dobules with values as pairs of atom and the vector
445 */
[99db9b]446CorrelationToPointMap *PeriodicCorrelationToPoint(
447 const std::vector<const molecule *> &molecules,
448 const std::vector<const element *> &elements,
449 const Vector *point,
450 const int ranges[NDIM] )
[7ea9e6]451{
[3930eb]452 Info FunctionInfo(__func__);
[caa30b]453 CorrelationToPointMap *outmap = new CorrelationToPointMap;
[7ea9e6]454 double distance = 0.;
455 int n[NDIM];
456 Vector periodicX;
457 Vector checkX;
458
[e65de8]459 if (molecules.empty()) {
[47d041]460 LOG(1, "No molecule given.");
[7ea9e6]461 return outmap;
462 }
[e791dc]463
[99db9b]464 for (std::vector<const molecule *>::const_iterator MolWalker = molecules.begin();
465 MolWalker != molecules.end();
466 MolWalker++) {
[cca9ef]467 RealSpaceMatrix FullMatrix = World::getInstance().getDomain().getM();
468 RealSpaceMatrix FullInverseMatrix = World::getInstance().getDomain().getMinv();
[47d041]469 LOG(2, "Current molecule is " << *MolWalker << ".");
[99db9b]470 for (molecule::const_iterator iter = (*MolWalker)->begin(); iter != (*MolWalker)->end(); ++iter) {
[47d041]471 LOG(3, "Current atom is " << **iter << ".");
[e5c0a1]472 for (vector<const element *>::const_iterator type = elements.begin(); type != elements.end(); ++type)
[d74077]473 if ((*type == NULL) || ((*iter)->getType() == *type)) {
474 periodicX = FullInverseMatrix * ((*iter)->getPosition()); // x now in [0,1)^3
[e65de8]475 // go through every range in xyz and get distance
476 for (n[0]=-ranges[0]; n[0] <= ranges[0]; n[0]++)
477 for (n[1]=-ranges[1]; n[1] <= ranges[1]; n[1]++)
478 for (n[2]=-ranges[2]; n[2] <= ranges[2]; n[2]++) {
479 checkX = FullMatrix * (Vector(n[0], n[1], n[2]) + periodicX);
480 distance = checkX.distance(*point);
[47d041]481 LOG(4, "Current distance is " << distance << ".");
[59fff1]482 outmap->insert (
483 std::pair<double,
484 std::pair<const atom *, const Vector*> >(
485 distance,
486 std::pair<const atom *, const Vector*> (
487 *iter,
488 point)
489 )
490 );
[e65de8]491 }
492 }
[7ea9e6]493 }
[e65de8]494 }
[7ea9e6]495
496 return outmap;
497};
498
[c4d4df]499/** Calculates the distance (pair) correlation between a given element and a surface.
[a5551b]500 * \param *molecules list of molecules structure
[c78d44]501 * \param &elements vector of elements to correlate to surface
[c4d4df]502 * \param *Surface pointer to Tesselation class surface
[6bd7e0]503 * \param *LC LinkedCell_deprecated structure to quickly find neighbouring atoms
[c4d4df]504 * \return Map of doubles with values as pairs of atom and the BoundaryTriangleSet that's closest
505 */
[99db9b]506CorrelationToSurfaceMap *CorrelationToSurface(
507 const std::vector<const molecule *> &molecules,
508 const std::vector<const element *> &elements,
509 const Tesselation * const Surface,
510 const LinkedCell_deprecated *LC )
[c4d4df]511{
[3930eb]512 Info FunctionInfo(__func__);
[caa30b]513 CorrelationToSurfaceMap *outmap = new CorrelationToSurfaceMap;
[99593f]514 double distance = 0;
[c4d4df]515 class BoundaryTriangleSet *triangle = NULL;
516 Vector centroid;
[7ea9e6]517
[e65de8]518 if ((Surface == NULL) || (LC == NULL) || (molecules.empty())) {
[47d041]519 ELOG(1, "No Tesselation, no LinkedCell or no molecule given.");
[7ea9e6]520 return outmap;
521 }
[e791dc]522
[99db9b]523 for (std::vector<const molecule *>::const_iterator MolWalker = molecules.begin();
524 MolWalker != molecules.end();
525 MolWalker++) {
[47d041]526 LOG(2, "Current molecule is " << (*MolWalker)->name << ".");
[e65de8]527 if ((*MolWalker)->empty())
[47d041]528 LOG(2, "\t is empty.");
[99db9b]529 for (molecule::const_iterator iter = (*MolWalker)->begin(); iter != (*MolWalker)->end(); ++iter) {
[47d041]530 LOG(3, "\tCurrent atom is " << *(*iter) << ".");
[e5c0a1]531 for (vector<const element *>::const_iterator type = elements.begin(); type != elements.end(); ++type)
[d74077]532 if ((*type == NULL) || ((*iter)->getType() == *type)) {
533 TriangleIntersectionList Intersections((*iter)->getPosition(),Surface,LC);
[e65de8]534 distance = Intersections.GetSmallestDistance();
535 triangle = Intersections.GetClosestTriangle();
[59fff1]536 outmap->insert (
537 std::pair<double,
538 std::pair<const atom *, BoundaryTriangleSet*> >(
539 distance,
540 std::pair<const atom *, BoundaryTriangleSet*> (
541 (*iter),
542 triangle)
543 )
544 );
[e65de8]545 }
[7fd416]546 }
[e65de8]547 }
[7ea9e6]548
549 return outmap;
550};
551
552/** Calculates the distance (pair) correlation between a given element, all its periodic images and and a surface.
553 * Note that we also put all periodic images found in the cells given by [ -ranges[i], ranges[i] ] and i=0,...,NDIM-1.
554 * I.e. We multiply the atom::node with the inverse of the domain matrix, i.e. transform it to \f$[0,0^3\f$, then add per
555 * axis an integer from [ -ranges[i], ranges[i] ] onto it and multiply with the domain matrix to bring it back into
556 * the real space. Then, we Tesselation::FindClosestTriangleToPoint() and DistanceToTrianglePlane().
557 * \param *molecules list of molecules structure
[c78d44]558 * \param &elements vector of elements to correlate to surface
[7ea9e6]559 * \param *Surface pointer to Tesselation class surface
[6bd7e0]560 * \param *LC LinkedCell_deprecated structure to quickly find neighbouring atoms
[7ea9e6]561 * \param ranges[NDIM] interval boundaries for the periodic images to scan also
562 * \return Map of doubles with values as pairs of atom and the BoundaryTriangleSet that's closest
563 */
[99db9b]564CorrelationToSurfaceMap *PeriodicCorrelationToSurface(
565 const std::vector<const molecule *> &molecules,
566 const std::vector<const element *> &elements,
567 const Tesselation * const Surface,
568 const LinkedCell_deprecated *LC,
569 const int ranges[NDIM] )
[7ea9e6]570{
[3930eb]571 Info FunctionInfo(__func__);
[caa30b]572 CorrelationToSurfaceMap *outmap = new CorrelationToSurfaceMap;
[7ea9e6]573 double distance = 0;
574 class BoundaryTriangleSet *triangle = NULL;
575 Vector centroid;
[99593f]576 int n[NDIM];
577 Vector periodicX;
578 Vector checkX;
[c4d4df]579
[e65de8]580 if ((Surface == NULL) || (LC == NULL) || (molecules.empty())) {
[47d041]581 LOG(1, "No Tesselation, no LinkedCell or no molecule given.");
[c4d4df]582 return outmap;
583 }
[e791dc]584
[244a84]585 double ShortestDistance = 0.;
586 BoundaryTriangleSet *ShortestTriangle = NULL;
[99db9b]587 for (std::vector<const molecule *>::const_iterator MolWalker = molecules.begin();
588 MolWalker != molecules.end();
589 MolWalker++) {
[cca9ef]590 RealSpaceMatrix FullMatrix = World::getInstance().getDomain().getM();
591 RealSpaceMatrix FullInverseMatrix = World::getInstance().getDomain().getMinv();
[47d041]592 LOG(2, "Current molecule is " << *MolWalker << ".");
[99db9b]593 for (molecule::const_iterator iter = (*MolWalker)->begin(); iter != (*MolWalker)->end(); ++iter) {
[47d041]594 LOG(3, "Current atom is " << **iter << ".");
[e5c0a1]595 for (vector<const element *>::const_iterator type = elements.begin(); type != elements.end(); ++type)
[d74077]596 if ((*type == NULL) || ((*iter)->getType() == *type)) {
597 periodicX = FullInverseMatrix * ((*iter)->getPosition()); // x now in [0,1)^3
[e65de8]598 // go through every range in xyz and get distance
599 ShortestDistance = -1.;
600 for (n[0]=-ranges[0]; n[0] <= ranges[0]; n[0]++)
601 for (n[1]=-ranges[1]; n[1] <= ranges[1]; n[1]++)
602 for (n[2]=-ranges[2]; n[2] <= ranges[2]; n[2]++) {
603 checkX = FullMatrix * (Vector(n[0], n[1], n[2]) + periodicX);
[d74077]604 TriangleIntersectionList Intersections(checkX,Surface,LC);
[e65de8]605 distance = Intersections.GetSmallestDistance();
606 triangle = Intersections.GetClosestTriangle();
607 if ((ShortestDistance == -1.) || (distance < ShortestDistance)) {
608 ShortestDistance = distance;
609 ShortestTriangle = triangle;
[99593f]610 }
[e65de8]611 }
612 // insert
[59fff1]613 outmap->insert (
614 std::pair<double,
615 std::pair<const atom *, BoundaryTriangleSet*> >(
616 ShortestDistance,
617 std::pair<const atom *, BoundaryTriangleSet*> (
618 *iter,
619 ShortestTriangle)
620 )
621 );
[47d041]622 //LOG(1, "INFO: Inserting " << Walker << " with distance " << ShortestDistance << " to " << *ShortestTriangle << ".");
[e65de8]623 }
[c4d4df]624 }
[e65de8]625 }
[c4d4df]626
627 return outmap;
628};
629
[bd61b41]630/** Returns the index of the bin for a given value.
[c4d4df]631 * \param value value whose bin to look for
632 * \param BinWidth width of bin
633 * \param BinStart first bin
634 */
[bd61b41]635int GetBin ( const double value, const double BinWidth, const double BinStart )
[c4d4df]636{
[92e5cb]637 //Info FunctionInfo(__func__);
[bd61b41]638 int bin =(int) (floor((value - BinStart)/BinWidth));
639 return (bin);
[c4d4df]640};
641
642
[92e5cb]643/** Adds header part that is unique to BinPairMap.
644 *
645 * @param file stream to print to
[c4d4df]646 */
[92e5cb]647void OutputCorrelation_Header( ofstream * const file )
[c4d4df]648{
[92e5cb]649 *file << "\tCount";
[c4d4df]650};
[b1f254]651
[92e5cb]652/** Prints values stored in BinPairMap iterator.
653 *
654 * @param file stream to print to
655 * @param runner iterator pointing at values to print
[be945c]656 */
[92e5cb]657void OutputCorrelation_Value( ofstream * const file, BinPairMap::const_iterator &runner )
[be945c]658{
[92e5cb]659 *file << runner->second;
[be945c]660};
661
[92e5cb]662
663/** Adds header part that is unique to DipoleAngularCorrelationMap.
664 *
665 * @param file stream to print to
[b1f254]666 */
[92e5cb]667void OutputDipoleAngularCorrelation_Header( ofstream * const file )
[b1f254]668{
[4b8630]669 *file << "\tFirstAtomOfMolecule";
[b1f254]670};
671
[208237b]672/** Prints values stored in DipoleCorrelationMap iterator.
[92e5cb]673 *
674 * @param file stream to print to
675 * @param runner iterator pointing at values to print
[b1f254]676 */
[92e5cb]677void OutputDipoleAngularCorrelation_Value( ofstream * const file, DipoleAngularCorrelationMap::const_iterator &runner )
[208237b]678{
[505d05]679 *file << *(runner->second);
[208237b]680};
681
682
683/** Adds header part that is unique to DipoleAngularCorrelationMap.
684 *
685 * @param file stream to print to
686 */
687void OutputDipoleCorrelation_Header( ofstream * const file )
688{
689 *file << "\tMolecule";
690};
691
692/** Prints values stored in DipoleCorrelationMap iterator.
693 *
694 * @param file stream to print to
695 * @param runner iterator pointing at values to print
696 */
697void OutputDipoleCorrelation_Value( ofstream * const file, DipoleCorrelationMap::const_iterator &runner )
[b1f254]698{
[92e5cb]699 *file << runner->second.first->getId() << "\t" << runner->second.second->getId();
[b1f254]700};
701
[92e5cb]702
703/** Adds header part that is unique to PairCorrelationMap.
704 *
705 * @param file stream to print to
[b1f254]706 */
[92e5cb]707void OutputPairCorrelation_Header( ofstream * const file )
[b1f254]708{
[92e5cb]709 *file << "\tAtom1\tAtom2";
710};
711
712/** Prints values stored in PairCorrelationMap iterator.
713 *
714 * @param file stream to print to
715 * @param runner iterator pointing at values to print
716 */
717void OutputPairCorrelation_Value( ofstream * const file, PairCorrelationMap::const_iterator &runner )
718{
719 *file << *(runner->second.first) << "\t" << *(runner->second.second);
720};
721
722
723/** Adds header part that is unique to CorrelationToPointMap.
724 *
725 * @param file stream to print to
726 */
727void OutputCorrelationToPoint_Header( ofstream * const file )
728{
729 *file << "\tAtom::x[i]-point.x[i]";
730};
731
732/** Prints values stored in CorrelationToPointMap iterator.
733 *
734 * @param file stream to print to
735 * @param runner iterator pointing at values to print
736 */
737void OutputCorrelationToPoint_Value( ofstream * const file, CorrelationToPointMap::const_iterator &runner )
738{
739 for (int i=0;i<NDIM;i++)
740 *file << "\t" << setprecision(8) << (runner->second.first->at(i) - runner->second.second->at(i));
[b1f254]741};
742
[92e5cb]743
744/** Adds header part that is unique to CorrelationToSurfaceMap.
745 *
746 * @param file stream to print to
747 */
748void OutputCorrelationToSurface_Header( ofstream * const file )
749{
750 *file << "\tTriangle";
751};
752
753/** Prints values stored in CorrelationToSurfaceMap iterator.
754 *
755 * @param file stream to print to
756 * @param runner iterator pointing at values to print
757 */
758void OutputCorrelationToSurface_Value( ofstream * const file, CorrelationToSurfaceMap::const_iterator &runner )
759{
760 *file << *(runner->second.first) << "\t" << *(runner->second.second);
761};
Note: See TracBrowser for help on using the repository browser.