File: MolChemicalFeature.cpp

package info (click to toggle)
rdkit 201809.1%2Bdfsg-6
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 123,688 kB
  • sloc: cpp: 230,509; python: 70,501; java: 6,329; ansic: 5,427; sql: 1,899; yacc: 1,739; lex: 1,243; makefile: 445; xml: 229; fortran: 183; sh: 123; cs: 93
file content (76 lines) | stat: -rw-r--r-- 2,195 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
// $Id$
//
//  Copyright (C) 2004-2006 Rational Discovery LLC
//
//   @@ All Rights Reserved @@
//  This file is part of the RDKit.
//  The contents are covered by the terms of the BSD license
//  which is included in the file license.txt, found at the root
//  of the RDKit source tree.
//
#include <GraphMol/RDKitBase.h>
#include <GraphMol/SmilesParse/SmilesParse.h>
#include "MolChemicalFeature.h"
#include "MolChemicalFeatureDef.h"

namespace RDKit {

const std::string &MolChemicalFeature::getFamily() const {
  return dp_def->getFamily();  // return d_family;
};

const std::string &MolChemicalFeature::getType() const {
  return dp_def->getType();  // return d_type;
};

void MolChemicalFeature::setActiveConformer(int confId) {
  PRECONDITION(dp_mol, "bad molecule");
  d_activeConf = confId;
}

RDGeom::Point3D MolChemicalFeature::getPos() const {
  return this->getPos(d_activeConf);
}

RDGeom::Point3D MolChemicalFeature::getPos(int confId) const {
  PRECONDITION(dp_mol, "bad molecule");
  PRECONDITION(dp_mol->getNumConformers(), "molecule has no conformers");
  if (confId == -1) confId = (*dp_mol->beginConformers())->getId();

  // -------------
  // Check to see if we've got the value cached:
  PointCacheType::const_iterator cacheIt = d_locs.find(confId);
  if (cacheIt != d_locs.end()) {
    return cacheIt->second;
  }

  // --------------
  // Nope, we have to figure it out on our own:
  RDGeom::Point3D res(0, 0, 0);
  bool setNeg1 = false;
  RDUNUSED_PARAM(setNeg1);
  if (confId == -1) {
    setNeg1 = true;
  }

  if (d_atoms.size() == 1) {
    res = dp_mol->getConformer(confId).getAtomPos((*d_atoms.begin())->getIdx());
  } else {
    PRECONDITION(dp_def, "bad definition");
    PRECONDITION(dp_def->getNumWeights() == this->getNumAtoms(),
                 "weight/atom mismatch");
    auto weightIt = dp_def->beginWeights();
    const Conformer &conf = dp_mol->getConformer(confId);
    for (auto atomIt = d_atoms.begin(); atomIt != d_atoms.end();
         atomIt++, weightIt++) {
      const Atom *atom = *atomIt;
      RDGeom::Point3D p = conf.getAtomPos(atom->getIdx());
      p *= *weightIt;
      res += p;
    }
  }
  d_locs[confId] = res;

  return res;
}
}