source: src/Parser/PdbParser.cpp@ 5aaa43

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 5aaa43 was 5aaa43, checked in by Frederik Heber <heber@…>, 12 years ago

FIX: Fixed new copyright line since start of 2013 in CodeChecks test.

  • we must look for either Uni Bonn or myself.
  • added second copyright line since from 1st of Jan 2013 I am not employed by University of Bonn anymore, hence changes to the code are my own copyright.
  • Property mode set to 100644
File size: 33.9 KB
RevLine 
[3ae731]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/>.
[3ae731]22 */
23
24/*
25 * PdbParser.cpp
26 *
27 * Created on: Aug 17, 2010
28 * Author: heber
29 */
30
31// include config.h
32#ifdef HAVE_CONFIG_H
33#include <config.h>
34#endif
35
[ad011c]36#include "CodePatterns/MemDebug.hpp"
[3ae731]37
[ad011c]38#include "CodePatterns/Assert.hpp"
39#include "CodePatterns/Log.hpp"
40#include "CodePatterns/toString.hpp"
41#include "CodePatterns/Verbose.hpp"
[42127c]42
[6f0841]43#include "Atom/atom.hpp"
[129204]44#include "Bond/bond.hpp"
[42127c]45#include "Descriptors/AtomIdDescriptor.hpp"
[3bdb6d]46#include "Element/element.hpp"
47#include "Element/periodentafel.hpp"
[42127c]48#include "molecule.hpp"
49#include "MoleculeListClass.hpp"
[4fbca9c]50#include "Parser/PdbParser.hpp"
[073a9e4]51#include "World.hpp"
52#include "WorldTime.hpp"
[bb6193]53
[105b72]54#include <algorithm>
55#include <cmath>
[3ae731]56#include <map>
57#include <vector>
58
[bb6193]59#include <iostream>
60#include <iomanip>
[3ae731]61
62using namespace std;
63
[765f16]64// declare specialized static variables
65const std::string FormatParserTrait<pdb>::name = "pdb";
66const std::string FormatParserTrait<pdb>::suffix = "pdb";
67const ParserTypes FormatParserTrait<pdb>::type = pdb;
68
[3ae731]69/**
70 * Constructor.
71 */
[765f16]72FormatParser< pdb >::FormatParser() :
73 FormatParser_common(NULL)
74{
[4fbca9c]75 knownTokens["ATOM"] = PdbKey::Atom;
[16462f]76 knownTokens["HETATM"] = PdbKey::Atom;
[4fbca9c]77 knownTokens["TER"] = PdbKey::Filler;
[9dba5f]78 knownTokens["END"] = PdbKey::EndOfTimestep;
[4fbca9c]79 knownTokens["CONECT"] = PdbKey::Connect;
80 knownTokens["REMARK"] = PdbKey::Remark;
[9dba5f]81 knownTokens[""] = PdbKey::EndOfTimestep;
[16462f]82
83 // argh, why can't just PdbKey::X+(size_t)i
84 PositionEnumMap[0] = PdbKey::X;
85 PositionEnumMap[1] = PdbKey::Y;
86 PositionEnumMap[2] = PdbKey::Z;
[3ae731]87}
88
89/**
90 * Destructor.
91 */
[765f16]92FormatParser< pdb >::~FormatParser()
93{
[873037]94 PdbAtomInfoContainer::clearknownDataKeys();
[3ae731]95 additionalAtomData.clear();
[4fbca9c]96}
97
98
99/** Parses the initial word of the given \a line and returns the token type.
100 *
101 * @param line line to scan
102 * @return token type
103 */
[955b91]104enum PdbKey::KnownTokens FormatParser< pdb >::getToken(std::string &line)
[4fbca9c]105{
106 // look for first space
[3dfd9c]107 std::string token = line.substr(0,6);
108 const size_t space_location = token.find(' ');
109 const size_t tab_location = token.find('\t');
[4fbca9c]110 size_t location = space_location < tab_location ? space_location : tab_location;
111 if (location != string::npos) {
[47d041]112 //LOG(1, "Found space at position " << space_location);
[3dfd9c]113 token = token.substr(0,space_location);
[4fbca9c]114 }
115
[47d041]116 //LOG(1, "Token is " << token);
[4fbca9c]117 if (knownTokens.count(token) == 0)
118 return PdbKey::NoToken;
119 else
120 return knownTokens[token];
121
122 return PdbKey::NoToken;
[3ae731]123}
124
125/**
[4fbca9c]126 * Loads atoms from a PDB-formatted file.
[3ae731]127 *
[4fbca9c]128 * \param PDB file
[3ae731]129 */
[765f16]130void FormatParser< pdb >::load(istream* file) {
[4fbca9c]131 string line;
132 size_t linecount = 0;
133 enum PdbKey::KnownTokens token;
134
[c0e28c]135 // reset id maps for this file (to correctly parse CONECT entries)
136 resetIdAssociations();
[16462f]137
[9dba5f]138 bool NotEndOfFile = true;
[4fbca9c]139 molecule *newmol = World::getInstance().createMolecule();
140 newmol->ActiveFlag = true;
[b0a2e3]141 unsigned int step = 0;
[4fbca9c]142 // TODO: Remove the insertion into molecule when saving does not depend on them anymore. Also, remove molecule.hpp include
143 World::getInstance().getMolecules()->insert(newmol);
144 while (NotEndOfFile) {
[9dba5f]145 bool NotEndOfTimestep = true;
[b0a2e3]146 while (NotEndOfTimestep && NotEndOfFile) {
[9dba5f]147 std::getline(*file, line, '\n');
[b0a2e3]148 if (!line.empty()) {
149 // extract first token
150 token = getToken(line);
151 switch (token) {
152 case PdbKey::Atom:
153 LOG(3,"INFO: Parsing ATOM entry for time step " << step << ".");
154 readAtomDataLine(step, line, newmol);
155 break;
156 case PdbKey::Remark:
157 LOG(3,"INFO: Parsing REM entry for time step " << step << ".");
158 break;
159 case PdbKey::Connect:
160 LOG(3,"INFO: Parsing CONECT entry for time step " << step << ".");
161 readNeighbors(step, line);
162 break;
163 case PdbKey::Filler:
164 LOG(3,"INFO: Stumbled upon Filler entry for time step " << step << ".");
165 break;
166 case PdbKey::EndOfTimestep:
[44f53e]167 LOG(1,"INFO: Parsing END entry or empty line for time step " << step << ".");
[b0a2e3]168 NotEndOfTimestep = false;
169 break;
170 default:
171 // TODO: put a throw here
[47d041]172 ELOG(2, "Unknown token: '" << line << "' for time step " << step << ".");
[765f16]173 //ASSERT(0, "FormatParser< pdb >::load() - Unknown token in line "+toString(linecount)+": "+line+".");
[b0a2e3]174 break;
175 }
[9dba5f]176 }
177 NotEndOfFile = NotEndOfFile && (file->good());
178 linecount++;
[4fbca9c]179 }
[b0a2e3]180 ++step;
181 }
[3dfd9c]182 LOG(4, "INFO: Listing all newly parsed atoms.");
183 BOOST_FOREACH(atom *_atom, *newmol)
[48801a]184 LOG(4, "INFO: Atom " << _atom->getName() << " " << *dynamic_cast<AtomInfo *>(_atom) << ".");
[4afa46]185
186 // refresh atom::nr and atom::name
187 newmol->getAtomCount();
[3ae731]188}
189
190/**
[73916f]191 * Saves the \a atoms into as a PDB file.
[3ae731]192 *
193 * \param file where to save the state
[73916f]194 * \param atoms atoms to store
[3ae731]195 */
[765f16]196void FormatParser< pdb >::save(ostream* file, const std::vector<atom *> &AtomList)
[73916f]197{
[830b3e]198 LOG(2, "DEBUG: Saving changes to pdb.");
[9dba5f]199
200 // check for maximum number of time steps
201 size_t max_timesteps = 0;
202 BOOST_FOREACH(atom *_atom, World::getInstance().getAllAtoms()) {
[48801a]203 LOG(4, "INFO: Atom " << _atom->getName() << " " << *dynamic_cast<AtomInfo *>(_atom) << ".");
[9dba5f]204 if (_atom->getTrajectorySize() > max_timesteps)
205 max_timesteps = _atom->getTrajectorySize();
[bb6193]206 }
[9dba5f]207 LOG(2,"INFO: Found a maximum of " << max_timesteps << " time steps to store.");
[3ae731]208
[9dba5f]209 // re-distribute serials
[c0e28c]210 resetIdAssociations();
[9dba5f]211 // (new atoms might have been added)
212 int AtomNo = 1; // serial number starts at 1 in pdb
213 for (vector<atom *>::const_iterator atomIt = AtomList.begin(); atomIt != AtomList.end(); atomIt++) {
214 PdbAtomInfoContainer &atomInfo = getadditionalAtomData(*atomIt);
[c0e28c]215 associateLocaltoGlobalId(AtomNo, (*atomIt)->getId());
[9dba5f]216 atomInfo.set(PdbKey::serial, toString(AtomNo));
[c0e28c]217 ++AtomNo;
[9dba5f]218 }
219
[5c5472]220 // store all time steps (always do first step)
221 for (size_t step = 0; (step == 0) || (step < max_timesteps); ++step) {
[9dba5f]222 {
223 // add initial remark
224 *file << "REMARK created by molecuilder on ";
225 time_t now = time((time_t *)NULL); // Get the system time and put it into 'now' as 'calender time'
226 // ctime ends in \n\0, we have to cut away the newline
227 std::string time(ctime(&now));
228 size_t pos = time.find('\n');
229 if (pos != 0)
230 *file << time.substr(0,pos);
231 else
232 *file << time;
233 *file << ", time step " << step;
234 *file << endl;
[16462f]235 }
[9dba5f]236
237 {
238 std::map<size_t,size_t> MolIdMap;
239 size_t MolNo = 1; // residue number starts at 1 in pdb
240 for (vector<atom *>::const_iterator atomIt = AtomList.begin(); atomIt != AtomList.end(); atomIt++) {
241 const molecule *mol = (*atomIt)->getMolecule();
242 if ((mol != NULL) && (MolIdMap.find(mol->getId()) == MolIdMap.end())) {
243 MolIdMap[mol->getId()] = MolNo++;
244 }
[bb6193]245 }
[9dba5f]246 const size_t MaxMol = MolNo;
247
248 // have a count per element and per molecule (0 is for all homeless atoms)
249 std::vector<int> **elementNo = new std::vector<int>*[MaxMol];
250 for (size_t i = 0; i < MaxMol; ++i)
251 elementNo[i] = new std::vector<int>(MAX_ELEMENTS,1);
252 char name[MAXSTRINGSIZE];
253 std::string ResidueName;
254
255 // write ATOMs
256 for (vector<atom *>::const_iterator atomIt = AtomList.begin(); atomIt != AtomList.end(); atomIt++) {
257 PdbAtomInfoContainer &atomInfo = getadditionalAtomData(*atomIt);
258 // gather info about residue
259 const molecule *mol = (*atomIt)->getMolecule();
260 if (mol == NULL) {
261 MolNo = 0;
262 atomInfo.set(PdbKey::resSeq, "0");
263 } else {
264 ASSERT(MolIdMap.find(mol->getId()) != MolIdMap.end(),
[765f16]265 "FormatParser< pdb >::save() - Mol id "+toString(mol->getId())+" not present despite we set it?!");
[9dba5f]266 MolNo = MolIdMap[mol->getId()];
267 atomInfo.set(PdbKey::resSeq, toString(MolIdMap[mol->getId()]));
268 if (atomInfo.get<std::string>(PdbKey::resName) == "-")
269 atomInfo.set(PdbKey::resName, mol->getName().substr(0,3));
270 }
271 // get info about atom
272 const size_t Z = (*atomIt)->getType()->getAtomicNumber();
273 if (atomInfo.get<std::string>(PdbKey::name) == "-") { // if no name set, give it a new name
274 sprintf(name, "%2s%02d",(*atomIt)->getType()->getSymbol().c_str(), (*elementNo[MolNo])[Z]);
275 (*elementNo[MolNo])[Z] = ((*elementNo[MolNo])[Z]+1) % 100; // confine to two digits
276 atomInfo.set(PdbKey::name, name);
277 }
278 // set position
279 for (size_t i=0; i<NDIM;++i) {
280 stringstream position;
[105b72]281 position << setprecision(7) << (*atomIt)->getPositionAtStep(step).at(i);
[9dba5f]282 atomInfo.set(PositionEnumMap[i], position.str());
283 }
284 // change element and charge if changed
[8990879]285 if (atomInfo.get<std::string>(PdbKey::element) != (*atomIt)->getType()->getSymbol()) {
286 std::string symbol = (*atomIt)->getType()->getSymbol();
287 if ((symbol[1] >= 'a') && (symbol[1] <= 'z'))
288 symbol[1] = (symbol[1] - 'a') + 'A';
289 atomInfo.set(PdbKey::element, symbol);
290 }
[9dba5f]291
292 // finally save the line
293 saveLine(file, atomInfo);
[16462f]294 }
[9dba5f]295 for (size_t i = 0; i < MaxMol; ++i)
296 delete elementNo[i];
297 delete elementNo;
[3ae731]298
[9dba5f]299 // write CONECTs
300 for (vector<atom *>::const_iterator atomIt = AtomList.begin(); atomIt != AtomList.end(); atomIt++) {
301 writeNeighbors(file, 4, *atomIt);
302 }
[bb6193]303 }
[9dba5f]304 // END
305 *file << "END" << endl;
[3ae731]306 }
307
308}
309
[6bc86c]310/** Add default info, when new atom is added to World.
311 *
312 * @param id of atom
313 */
[765f16]314void FormatParser< pdb >::AtomInserted(atomId_t id)
[6bc86c]315{
[765f16]316 //LOG(3, "FormatParser< pdb >::AtomInserted() - notified of atom " << id << "'s insertion.");
[6bc86c]317 ASSERT(!isPresentadditionalAtomData(id),
[765f16]318 "FormatParser< pdb >::AtomInserted() - additionalAtomData already present for newly added atom "
[6bc86c]319 +toString(id)+".");
320 // don't insert here as this is our check whether we are in the first time step
321 //additionalAtomData.insert( std::make_pair(id, defaultAdditionalData) );
322}
323
324/** Remove additional AtomData info, when atom has been removed from World.
325 *
326 * @param id of atom
327 */
[765f16]328void FormatParser< pdb >::AtomRemoved(atomId_t id)
[6bc86c]329{
[765f16]330 //LOG(3, "FormatParser< pdb >::AtomRemoved() - notified of atom " << id << "'s removal.");
[8bf9c6]331 std::map<const atomId_t, PdbAtomInfoContainer>::iterator iter = additionalAtomData.find(id);
[6bc86c]332 // as we do not insert AtomData on AtomInserted, we cannot be assured of its presence
333// ASSERT(iter != additionalAtomData.end(),
[765f16]334// "FormatParser< pdb >::AtomRemoved() - additionalAtomData is not present for atom "
[6bc86c]335// +toString(id)+" to remove.");
336 if (iter != additionalAtomData.end()) {
337 additionalAtomData.erase(iter);
338 }
339}
340
341
[9dba5f]342/** Checks whether there is an entry for the given atom's \a _id.
343 *
344 * @param _id atom's id we wish to check on
345 * @return true - entry present, false - only for atom's father or no entry
346 */
[8bf9c6]347bool FormatParser< pdb >::isPresentadditionalAtomData(const atomId_t _id) const
[9dba5f]348{
[8bf9c6]349 std::map<const atomId_t, PdbAtomInfoContainer>::const_iterator iter = additionalAtomData.find(_id);
350 return (iter != additionalAtomData.end());
[9dba5f]351}
352
353
[93fd43e]354/** Either returns reference to present entry or creates new with default values.
355 *
356 * @param _atom atom whose entry we desire
357 * @return
358 */
[765f16]359PdbAtomInfoContainer& FormatParser< pdb >::getadditionalAtomData(atom *_atom)
[93fd43e]360{
361 if (additionalAtomData.find(_atom->getId()) != additionalAtomData.end()) {
362 } else if (additionalAtomData.find(_atom->father->getId()) != additionalAtomData.end()) {
363 // use info from direct father
364 additionalAtomData[_atom->getId()] = additionalAtomData[_atom->father->getId()];
365 } else if (additionalAtomData.find(_atom->GetTrueFather()->getId()) != additionalAtomData.end()) {
366 // use info from topmost father
367 additionalAtomData[_atom->getId()] = additionalAtomData[_atom->GetTrueFather()->getId()];
368 } else {
369 // create new entry use default values if nothing else is known
370 additionalAtomData[_atom->getId()] = defaultAdditionalData;
371 }
372 return additionalAtomData[_atom->getId()];
373}
374
[105b72]375/** Tiny helper function to print a float with a most 8 digits.
376 *
377 * A few examples best give the picture:
378 * 1234.678
379 * 1.234
380 * 0.001
381 * 0.100
382 * 1234567.
383 * 123456.7
384 * -1234.56
385 *
386 * \param value
387 * \return string representation
388 */
389const std::string FormatParser< pdb >::printCoordinate(
390 const double value)
391{
392 size_t meaningful_bits=7; // one for decimal dot
393 if (value < 0) //one for the minus sign
394 --meaningful_bits;
395 // count digits before dot (without minus and round towards zero!)
396 const int full = floor(fabs(value));
397 size_t bits_before_dot = 1;
398 {
399 int tmp = full;
400 for (;bits_before_dot < meaningful_bits;++bits_before_dot) {
401 // even if value is 0...somethingish, we still must start at one digit
402 tmp = tmp/10;
403 if (tmp == 0)
404 break;
405 }
406 }
407 // this fixes bits available after dot
408 const size_t bits_after_dot = std::min((int)meaningful_bits - (int)bits_before_dot, 3);
409 stringstream position;
410 if (bits_after_dot > 0) {
411 if (value < 0)
412 position << "-";
413 const int remainder = round((abs(value)-full)*pow(10,bits_after_dot));
414 position << full << "." << setfill('0') << setw(bits_after_dot) << remainder;
415 if (bits_after_dot == 2)
416 ELOG(2, "PdbParser is writing coordinates with just a two decimal places.");
417 if (bits_after_dot == 1)
418 ELOG(1, "PdbParser is writing coordinates with just a single decimal places.");
419 } else {
420 ELOG(0, "PdbParser is writing coordinates without any decimal places.");
421 position << full << ".";
422 }
423 return position.str();
424}
425
[3ae731]426/**
[4fbca9c]427 * Writes one line of PDB-formatted data to the provided stream.
[3ae731]428 *
429 * \param stream where to write the line to
[bb6193]430 * \param *currentAtom the atom of which information should be written
431 * \param AtomNo serial number of atom
[16462f]432 * \param *name name of atom, i.e. H01
433 * \param ResidueName Name of molecule
[bb6193]434 * \param ResidueNo number of residue
[3ae731]435 */
[765f16]436void FormatParser< pdb >::saveLine(
[16462f]437 ostream* file,
438 const PdbAtomInfoContainer &atomInfo)
439{
440 *file << setfill(' ') << left << setw(6)
441 << atomInfo.get<std::string>(PdbKey::token);
442 *file << setfill(' ') << right << setw(5)
[1f5e97]443 << (atomInfo.get<int>(PdbKey::serial) % 100000); /* atom serial number */
[16462f]444 *file << " "; /* char 12 is empty */
445 *file << setfill(' ') << left << setw(4)
446 << atomInfo.get<std::string>(PdbKey::name); /* atom name */
447 *file << setfill(' ') << left << setw(1)
448 << atomInfo.get<std::string>(PdbKey::altLoc); /* alternate location/conformation */
449 *file << setfill(' ') << left << setw(3)
450 << atomInfo.get<std::string>(PdbKey::resName); /* residue name */
451 *file << " "; /* char 21 is empty */
452 *file << setfill(' ') << left << setw(1)
453 << atomInfo.get<std::string>(PdbKey::chainID); /* chain identifier */
454 *file << setfill(' ') << left << setw(4)
[1f5e97]455 << (atomInfo.get<int>(PdbKey::resSeq) % 10000); /* residue sequence number */
[16462f]456 *file << setfill(' ') << left << setw(1)
457 << atomInfo.get<std::string>(PdbKey::iCode); /* iCode */
458 *file << " "; /* char 28-30 are empty */
459 // have the following operate on stringstreams such that format specifiers
460 // only act on these
461 for (size_t i=0;i<NDIM;++i) {
[105b72]462 *file << setfill(' ') << right << setw(8)
463 << printCoordinate(atomInfo.get<double>(PositionEnumMap[i]));
[16462f]464 }
465 {
466 stringstream occupancy;
467 occupancy << fixed << setprecision(2) << showpoint
468 << atomInfo.get<double>(PdbKey::occupancy); /* occupancy */
469 *file << setfill(' ') << right << setw(6) << occupancy.str();
[3ae731]470 }
[16462f]471 {
472 stringstream tempFactor;
473 tempFactor << fixed << setprecision(2) << showpoint
474 << atomInfo.get<double>(PdbKey::tempFactor); /* temperature factor */
475 *file << setfill(' ') << right << setw(6) << tempFactor.str();
476 }
477 *file << " "; /* char 68-76 are empty */
478 *file << setfill(' ') << right << setw(2) << atomInfo.get<std::string>(PdbKey::element); /* element */
479 *file << setfill(' ') << right << setw(2) << atomInfo.get<int>(PdbKey::charge); /* charge */
[3ae731]480
481 *file << endl;
482}
483
484/**
485 * Writes the neighbor information of one atom to the provided stream.
486 *
[9d83b6]487 * Note that ListOfBonds of WorldTime::CurrentTime is used.
488 *
[473237]489 * Also, we fill up the CONECT line to extend over 80 chars.
490 *
[bb6193]491 * \param *file where to write neighbor information to
492 * \param MaxnumberOfNeighbors of neighbors
493 * \param *currentAtom to the atom of which to take the neighbor information
[3ae731]494 */
[765f16]495void FormatParser< pdb >::writeNeighbors(ostream* file, int MaxnumberOfNeighbors, atom* currentAtom) {
[4c1230]496 int MaxNo = MaxnumberOfNeighbors;
[473237]497 int charsleft = 80;
[9d83b6]498 const BondList & ListOfBonds = currentAtom->getListOfBonds();
499 if (!ListOfBonds.empty()) {
500 for(BondList::const_iterator currentBond = ListOfBonds.begin(); currentBond != ListOfBonds.end(); ++currentBond) {
[4c1230]501 if (MaxNo >= MaxnumberOfNeighbors) {
502 *file << "CONECT";
[c0e28c]503 *file << setw(5) << getLocalId(currentAtom->getId());
[473237]504 charsleft = 80-6-5;
[4c1230]505 MaxNo = 0;
[bb6193]506 }
[c0e28c]507 *file << setw(5) << getLocalId((*currentBond)->GetOtherAtom(currentAtom)->getId());
[473237]508 charsleft -= 5;
[bb6193]509 MaxNo++;
[473237]510 if (MaxNo == MaxnumberOfNeighbors) {
511 for (;charsleft > 0; charsleft--)
512 *file << ' ';
[4c1230]513 *file << "\n";
[473237]514 }
[3ae731]515 }
[473237]516 if (MaxNo != MaxnumberOfNeighbors) {
517 for (;charsleft > 0; charsleft--)
518 *file << ' ';
[4c1230]519 *file << "\n";
[473237]520 }
[3ae731]521 }
522}
523
[9dba5f]524/** Either returns present atom with given id or a newly created one.
525 *
526 * @param id_string
527 * @return
528 */
[c0e28c]529atom* FormatParser< pdb >::getAtomToParse(std::string id_string)
[9dba5f]530{
531 // get the local ID
532 ConvertTo<int> toInt;
[c0e28c]533 const unsigned int AtomID_local = toInt(id_string);
534 LOG(4, "INFO: Local id is "+toString(AtomID_local)+".");
[9dba5f]535 // get the atomic ID if present
536 atom* newAtom = NULL;
[c0e28c]537 if (getGlobalId(AtomID_local) != -1) {
538 const unsigned int AtomID_global = getGlobalId(AtomID_local);
539 LOG(4, "INFO: Global id present as " << AtomID_global << ".");
[9dba5f]540 // check if atom exists
[c0e28c]541 newAtom = World::getInstance().getAtom(AtomById(AtomID_global));
[9dba5f]542 LOG(5, "INFO: Listing all present atoms with id.");
543 BOOST_FOREACH(atom *_atom, World::getInstance().getAllAtoms())
544 LOG(5, "INFO: " << *_atom << " with id " << _atom->getId());
545 }
546 // if not exists, create
547 if (newAtom == NULL) {
548 newAtom = World::getInstance().createAtom();
[8bf9c6]549 //const unsigned int AtomID_global = newAtom->getId();
[9dba5f]550 LOG(4, "INFO: No association to global id present, creating atom.");
551 } else {
552 LOG(4, "INFO: Existing atom found: " << *newAtom << ".");
553 }
554 return newAtom;
555}
556
[5fa2ba]557/** read a line starting with key ATOM.
558 *
559 * We check for line's length and parse only up to this value.
560 *
561 * @param atomInfo container to put information in
562 * @param line line containing key ATOM
563 */
[765f16]564void FormatParser< pdb >::readPdbAtomInfoContainer(PdbAtomInfoContainer &atomInfo, std::string &line) const
[9dba5f]565{
[5fa2ba]566 const size_t length = line.length();
567 if (length < 80)
[765f16]568 ELOG(2, "FormatParser< pdb >::readPdbAtomInfoContainer() - pdb is mal-formed, containing less than 80 chars!");
[5fa2ba]569 if (length >= 6) {
570 LOG(4,"INFO: Parsing token from "+line.substr(0,6)+".");
571 atomInfo.set(PdbKey::token, line.substr(0,6));
572 }
573 if (length >= 11) {
574 LOG(4,"INFO: Parsing serial from "+line.substr(6,5)+".");
575 atomInfo.set(PdbKey::serial, line.substr(6,5));
576 ASSERT(atomInfo.get<int>(PdbKey::serial) != 0,
[765f16]577 "FormatParser< pdb >::readPdbAtomInfoContainer() - serial 0 is invalid (filler id for conect entries).");
[5fa2ba]578 }
579
580 if (length >= 16) {
581 LOG(4,"INFO: Parsing name from "+line.substr(12,4)+".");
582 atomInfo.set(PdbKey::name, line.substr(12,4));
583 }
584 if (length >= 17) {
585 LOG(4,"INFO: Parsing altLoc from "+line.substr(16,1)+".");
586 atomInfo.set(PdbKey::altLoc, line.substr(16,1));
587 }
588 if (length >= 20) {
589 LOG(4,"INFO: Parsing resName from "+line.substr(17,3)+".");
590 atomInfo.set(PdbKey::resName, line.substr(17,3));
591 }
592 if (length >= 22) {
593 LOG(4,"INFO: Parsing chainID from "+line.substr(21,1)+".");
594 atomInfo.set(PdbKey::chainID, line.substr(21,1));
595 }
596 if (length >= 26) {
597 LOG(4,"INFO: Parsing resSeq from "+line.substr(22,4)+".");
598 atomInfo.set(PdbKey::resSeq, line.substr(22,4));
599 }
600 if (length >= 27) {
601 LOG(4,"INFO: Parsing iCode from "+line.substr(26,1)+".");
602 atomInfo.set(PdbKey::iCode, line.substr(26,1));
603 }
604
605 if (length >= 60) {
606 LOG(4,"INFO: Parsing occupancy from "+line.substr(54,6)+".");
607 atomInfo.set(PdbKey::occupancy, line.substr(54,6));
608 }
609 if (length >= 66) {
610 LOG(4,"INFO: Parsing tempFactor from "+line.substr(60,6)+".");
611 atomInfo.set(PdbKey::tempFactor, line.substr(60,6));
612 }
613 if (length >= 80) {
614 LOG(4,"INFO: Parsing charge from "+line.substr(78,2)+".");
615 atomInfo.set(PdbKey::charge, line.substr(78,2));
616 }
617 if (length >= 78) {
618 LOG(4,"INFO: Parsing element from "+line.substr(76,2)+".");
619 atomInfo.set(PdbKey::element, line.substr(76,2));
620 } else {
621 LOG(4,"INFO: Trying to parse alternative element from name "+line.substr(12,4)+".");
622 atomInfo.set(PdbKey::element, line.substr(12,4));
623 }
[9dba5f]624}
625
[4fbca9c]626/** Parse an ATOM line from a PDB file.
627 *
628 * Reads one data line of a pdstatus file and interprets it according to the
629 * specifications of the PDB 3.2 format: http://www.wwpdb.org/docs.html
630 *
631 * A new atom is created and filled with available information, non-
632 * standard information is placed in additionalAtomData at the atom's id.
[3ae731]633 *
[b0a2e3]634 * \param _step time step to use
[3ae731]635 * \param line to parse as an atom
[4fbca9c]636 * \param newmol molecule to add parsed atoms to
[3ae731]637 */
[765f16]638void FormatParser< pdb >::readAtomDataLine(const unsigned int _step, std::string &line, molecule *newmol = NULL) {
[4fbca9c]639 vector<string>::iterator it;
[9dba5f]640
641 atom* newAtom = getAtomToParse(line.substr(6,5));
642 LOG(3,"INFO: Parsing END entry or empty line.");
643 bool FirstTimestep = isPresentadditionalAtomData(newAtom->getId()) ? false : true;
[b0a2e3]644 ASSERT((FirstTimestep && (_step == 0)) || (!FirstTimestep && (_step !=0)),
[765f16]645 "FormatParser< pdb >::readAtomDataLine() - additionalAtomData present though atom is newly parsed.");
[9dba5f]646 if (FirstTimestep) {
[c0e28c]647 LOG(3,"INFO: Parsing new atom "+toString(*newAtom)+" "+toString(newAtom->getId())+".");
[9dba5f]648 } else {
649 LOG(3,"INFO: Parsing present atom "+toString(*newAtom)+".");
650 }
[93fd43e]651 PdbAtomInfoContainer &atomInfo = getadditionalAtomData(newAtom);
[9dba5f]652 LOG(4,"INFO: Information in container is "+toString(atomInfo)+".");
653
[4fbca9c]654 string word;
655 ConvertTo<size_t> toSize_t;
656
657 // check whether serial exists, if so, assign next available
658
[47d041]659// LOG(2, "Split line:"
[4fbca9c]660// << line.substr(6,5) << "|"
661// << line.substr(12,4) << "|"
662// << line.substr(16,1) << "|"
663// << line.substr(17,3) << "|"
664// << line.substr(21,1) << "|"
665// << line.substr(22,4) << "|"
666// << line.substr(26,1) << "|"
667// << line.substr(30,8) << "|"
668// << line.substr(38,8) << "|"
669// << line.substr(46,8) << "|"
670// << line.substr(54,6) << "|"
671// << line.substr(60,6) << "|"
672// << line.substr(76,2) << "|"
[47d041]673// << line.substr(78,2));
[4fbca9c]674
[9dba5f]675 if (FirstTimestep) {
676 // first time step
677 // then fill info container
678 readPdbAtomInfoContainer(atomInfo, line);
[c0e28c]679 // associate local with global id
680 associateLocaltoGlobalId(toSize_t(atomInfo.get<std::string>(PdbKey::serial)), newAtom->getId());
[9dba5f]681 // set position
682 Vector tempVector;
683 LOG(4,"INFO: Parsing position from ("
684 +line.substr(30,8)+","
685 +line.substr(38,8)+","
686 +line.substr(46,8)+").");
687 PdbAtomInfoContainer::ScanKey(tempVector[0], line.substr(30,8));
688 PdbAtomInfoContainer::ScanKey(tempVector[1], line.substr(38,8));
689 PdbAtomInfoContainer::ScanKey(tempVector[2], line.substr(46,8));
690 newAtom->setPosition(tempVector);
691 // set element
[8990879]692 std::string value = atomInfo.get<std::string>(PdbKey::element);
693 // make second character lower case if not
694 if ((value[1] >= 'A') && (value[1] <= 'Z'))
695 value[1] = (value[1] - 'A') + 'a';
[9dba5f]696 const element *elem = World::getInstance().getPeriode()
[8990879]697 ->FindElement(value);
[9dba5f]698 ASSERT(elem != NULL,
[765f16]699 "FormatParser< pdb >::readAtomDataLine() - element "+atomInfo.get<std::string>(PdbKey::element)+" is unknown!");
[9dba5f]700 newAtom->setType(elem);
701
702 if (newmol != NULL)
703 newmol->AddAtom(newAtom);
704 } else {
705 // not first time step
706 // then parse into different container
707 PdbAtomInfoContainer consistencyInfo;
708 readPdbAtomInfoContainer(consistencyInfo, line);
709 // then check additional info for consistency
710 ASSERT(atomInfo.get<std::string>(PdbKey::token) == consistencyInfo.get<std::string>(PdbKey::token),
[765f16]711 "FormatParser< pdb >::readAtomDataLine() - difference in token on multiple time step for atom with id "
[9dba5f]712 +atomInfo.get<std::string>(PdbKey::serial)+"!");
713 ASSERT(atomInfo.get<std::string>(PdbKey::name) == consistencyInfo.get<std::string>(PdbKey::name),
[765f16]714 "FormatParser< pdb >::readAtomDataLine() - difference in name on multiple time step for atom with id "
[9dba5f]715 +atomInfo.get<std::string>(PdbKey::serial)+":"
716 +atomInfo.get<std::string>(PdbKey::name)+"!="
717 +consistencyInfo.get<std::string>(PdbKey::name)+".");
718 ASSERT(atomInfo.get<std::string>(PdbKey::altLoc) == consistencyInfo.get<std::string>(PdbKey::altLoc),
[765f16]719 "FormatParser< pdb >::readAtomDataLine() - difference in altLoc on multiple time step for atom with id "
[9dba5f]720 +atomInfo.get<std::string>(PdbKey::serial)+"!");
721 ASSERT(atomInfo.get<std::string>(PdbKey::resName) == consistencyInfo.get<std::string>(PdbKey::resName),
[765f16]722 "FormatParser< pdb >::readAtomDataLine() - difference in resName on multiple time step for atom with id "
[9dba5f]723 +atomInfo.get<std::string>(PdbKey::serial)+"!");
724 ASSERT(atomInfo.get<std::string>(PdbKey::chainID) == consistencyInfo.get<std::string>(PdbKey::chainID),
[765f16]725 "FormatParser< pdb >::readAtomDataLine() - difference in chainID on multiple time step for atom with id "
[9dba5f]726 +atomInfo.get<std::string>(PdbKey::serial)+"!");
727 ASSERT(atomInfo.get<std::string>(PdbKey::resSeq) == consistencyInfo.get<std::string>(PdbKey::resSeq),
[765f16]728 "FormatParser< pdb >::readAtomDataLine() - difference in resSeq on multiple time step for atom with id "
[9dba5f]729 +atomInfo.get<std::string>(PdbKey::serial)+"!");
730 ASSERT(atomInfo.get<std::string>(PdbKey::iCode) == consistencyInfo.get<std::string>(PdbKey::iCode),
[765f16]731 "FormatParser< pdb >::readAtomDataLine() - difference in iCode on multiple time step for atom with id "
[9dba5f]732 +atomInfo.get<std::string>(PdbKey::serial)+"!");
733 ASSERT(atomInfo.get<std::string>(PdbKey::occupancy) == consistencyInfo.get<std::string>(PdbKey::occupancy),
[765f16]734 "FormatParser< pdb >::readAtomDataLine() - difference in occupancy on multiple time step for atom with id "
[9dba5f]735 +atomInfo.get<std::string>(PdbKey::serial)+"!");
736 ASSERT(atomInfo.get<std::string>(PdbKey::tempFactor) == consistencyInfo.get<std::string>(PdbKey::tempFactor),
[765f16]737 "FormatParser< pdb >::readAtomDataLine() - difference in tempFactor on multiple time step for atom with id "
[9dba5f]738 +atomInfo.get<std::string>(PdbKey::serial)+"!");
739 ASSERT(atomInfo.get<std::string>(PdbKey::charge) == consistencyInfo.get<std::string>(PdbKey::charge),
[765f16]740 "FormatParser< pdb >::readAtomDataLine() - difference in charge on multiple time step for atom with id "
[9dba5f]741 +atomInfo.get<std::string>(PdbKey::serial)+"!");
742 ASSERT(atomInfo.get<std::string>(PdbKey::element) == consistencyInfo.get<std::string>(PdbKey::element),
[765f16]743 "FormatParser< pdb >::readAtomDataLine() - difference in element on multiple time step for atom with id "
[9dba5f]744 +atomInfo.get<std::string>(PdbKey::serial)+"!");
745 // and parse in trajectory
746 Vector tempVector;
747 LOG(4,"INFO: Parsing trajectory position from ("
748 +line.substr(30,8)+","
749 +line.substr(38,8)+","
750 +line.substr(46,8)+").");
751 PdbAtomInfoContainer::ScanKey(tempVector[0], line.substr(30,8));
752 PdbAtomInfoContainer::ScanKey(tempVector[1], line.substr(38,8));
753 PdbAtomInfoContainer::ScanKey(tempVector[2], line.substr(46,8));
[b0a2e3]754 LOG(4,"INFO: Adding trajectory point to time step "+toString(_step)+".");
[9dba5f]755 // and set position at new time step
[b0a2e3]756 newAtom->setPositionAtStep(_step, tempVector);
[9dba5f]757 }
758
[4fbca9c]759
760// printAtomInfo(newAtom);
[3ae731]761}
762
[4fbca9c]763/** Prints all PDB-specific information known about an atom.
[3ae731]764 *
765 */
[765f16]766void FormatParser< pdb >::printAtomInfo(const atom * const newAtom) const
[4fbca9c]767{
768 const PdbAtomInfoContainer &atomInfo = additionalAtomData.at(newAtom->getId()); // operator[] const does not exist
769
[47d041]770 LOG(1, "We know about atom " << newAtom->getId() << ":");
771 LOG(1, "\ttoken is " << atomInfo.get<std::string>(PdbKey::token));
772 LOG(1, "\tserial is " << atomInfo.get<int>(PdbKey::serial));
773 LOG(1, "\tname is " << atomInfo.get<std::string>(PdbKey::name));
774 LOG(1, "\taltLoc is " << atomInfo.get<std::string>(PdbKey::altLoc));
775 LOG(1, "\tresName is " << atomInfo.get<std::string>(PdbKey::resName));
776 LOG(1, "\tchainID is " << atomInfo.get<std::string>(PdbKey::chainID));
777 LOG(1, "\tresSeq is " << atomInfo.get<int>(PdbKey::resSeq));
778 LOG(1, "\tiCode is " << atomInfo.get<std::string>(PdbKey::iCode));
779 LOG(1, "\tX is " << atomInfo.get<double>(PdbKey::X));
780 LOG(1, "\tY is " << atomInfo.get<double>(PdbKey::Y));
781 LOG(1, "\tZ is " << atomInfo.get<double>(PdbKey::Z));
782 LOG(1, "\toccupancy is " << atomInfo.get<double>(PdbKey::occupancy));
783 LOG(1, "\ttempFactor is " << atomInfo.get<double>(PdbKey::tempFactor));
784 LOG(1, "\telement is '" << *(newAtom->getType()) << "'");
785 LOG(1, "\tcharge is " << atomInfo.get<int>(PdbKey::charge));
[3ae731]786}
787
788/**
[4fbca9c]789 * Reads neighbor information for one atom from the input.
790 *
[b0a2e3]791 * \param _step time step to use
[4fbca9c]792 * \param line to parse as an atom
[3ae731]793 */
[765f16]794void FormatParser< pdb >::readNeighbors(const unsigned int _step, std::string &line)
[4fbca9c]795{
796 const size_t length = line.length();
797 std::list<size_t> ListOfNeighbors;
798 ConvertTo<size_t> toSize_t;
799
800 // obtain neighbours
801 // show split line for debugging
802 string output;
803 ASSERT(length >=16,
[765f16]804 "FormatParser< pdb >::readNeighbors() - CONECT entry has not enough entries: "+line+"!");
805// output = "Split line:|";
806// output += line.substr(6,5) + "|";
[4fbca9c]807 const size_t id = toSize_t(line.substr(6,5));
808 for (size_t index = 11; index <= 26; index+=5) {
809 if (index+5 <= length) {
[473237]810 output += line.substr(index,5) + "|";
811 // search for digits
812 int otherid = -1;
813 PdbAtomInfoContainer::ScanKey(otherid, line.substr(index,5));
[5fa2ba]814 if (otherid > 0)
815 ListOfNeighbors.push_back(otherid);
816 else
[44f53e]817 ELOG(3, "FormatParser< pdb >::readNeighbors() - discarding CONECT entry with id 0.");
[4fbca9c]818 } else {
819 break;
820 }
821 }
[473237]822 LOG(4, output);
[4fbca9c]823
824 // add neighbours
[c0e28c]825 atom *_atom = World::getInstance().getAtom(AtomById(getGlobalId(id)));
[473237]826 LOG(2, "STATUS: Atom " << _atom->getId() << " gets " << ListOfNeighbors.size() << " more neighbours.");
[4fbca9c]827 for (std::list<size_t>::const_iterator iter = ListOfNeighbors.begin();
828 iter != ListOfNeighbors.end();
829 ++iter) {
[c0e28c]830 atom * const _Otheratom = World::getInstance().getAtom(AtomById(getGlobalId(*iter)));
[473237]831 LOG(3, "INFO: Adding Bond (" << *_atom << "," << *_Otheratom << ")");
[b0a2e3]832 _atom->addBond(_step, _Otheratom);
[4fbca9c]833 }
[3ae731]834}
835
[765f16]836bool FormatParser< pdb >::operator==(const FormatParser< pdb >& b) const
[4fbca9c]837{
838 bool status = true;
839 World::AtomComposite atoms = World::getInstance().getAllAtoms();
840 for (World::AtomComposite::const_iterator iter = atoms.begin(); iter != atoms.end(); ++iter) {
841 if ((additionalAtomData.find((*iter)->getId()) != additionalAtomData.end())
842 && (b.additionalAtomData.find((*iter)->getId()) != b.additionalAtomData.end())) {
843 const PdbAtomInfoContainer &atomInfo = additionalAtomData.at((*iter)->getId());
844 const PdbAtomInfoContainer &OtheratomInfo = b.additionalAtomData.at((*iter)->getId());
845
[16462f]846 status = status && (atomInfo.get<std::string>(PdbKey::serial) == OtheratomInfo.get<std::string>(PdbKey::serial));
[47d041]847 if (!status) ELOG(1, "Mismatch in serials!");
[16462f]848 status = status && (atomInfo.get<std::string>(PdbKey::name) == OtheratomInfo.get<std::string>(PdbKey::name));
[47d041]849 if (!status) ELOG(1, "Mismatch in names!");
[16462f]850 status = status && (atomInfo.get<std::string>(PdbKey::altLoc) == OtheratomInfo.get<std::string>(PdbKey::altLoc));
[47d041]851 if (!status) ELOG(1, "Mismatch in altLocs!");
[16462f]852 status = status && (atomInfo.get<std::string>(PdbKey::resName) == OtheratomInfo.get<std::string>(PdbKey::resName));
[47d041]853 if (!status) ELOG(1, "Mismatch in resNames!");
[16462f]854 status = status && (atomInfo.get<std::string>(PdbKey::chainID) == OtheratomInfo.get<std::string>(PdbKey::chainID));
[47d041]855 if (!status) ELOG(1, "Mismatch in chainIDs!");
[16462f]856 status = status && (atomInfo.get<std::string>(PdbKey::resSeq) == OtheratomInfo.get<std::string>(PdbKey::resSeq));
[47d041]857 if (!status) ELOG(1, "Mismatch in resSeqs!");
[16462f]858 status = status && (atomInfo.get<std::string>(PdbKey::iCode) == OtheratomInfo.get<std::string>(PdbKey::iCode));
[47d041]859 if (!status) ELOG(1, "Mismatch in iCodes!");
[16462f]860 status = status && (atomInfo.get<std::string>(PdbKey::occupancy) == OtheratomInfo.get<std::string>(PdbKey::occupancy));
[47d041]861 if (!status) ELOG(1, "Mismatch in occupancies!");
[16462f]862 status = status && (atomInfo.get<std::string>(PdbKey::tempFactor) == OtheratomInfo.get<std::string>(PdbKey::tempFactor));
[47d041]863 if (!status) ELOG(1, "Mismatch in tempFactors!");
[16462f]864 status = status && (atomInfo.get<std::string>(PdbKey::charge) == OtheratomInfo.get<std::string>(PdbKey::charge));
[47d041]865 if (!status) ELOG(1, "Mismatch in charges!");
[4fbca9c]866 }
[3ae731]867 }
868
[4fbca9c]869 return status;
[3ae731]870}
871
Note: See TracBrowser for help on using the repository browser.