source: src/Parser/PdbParser.cpp@ ef8667

ForceAnnealing_oldresults IndependentFragmentGrids_IntegrationTest
Last change on this file since ef8667 was ef8667, checked in by Frederik Heber <frederik.heber@…>, 8 years ago

Extracted getMinMaxTrajectories() into FormatParser_common.

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