File: normalize.cpp

package info (click to toggle)
rdkit 202209.3-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 203,880 kB
  • sloc: cpp: 334,239; python: 80,247; ansic: 24,579; java: 7,667; sql: 2,123; yacc: 1,884; javascript: 1,358; lex: 1,260; makefile: 576; xml: 229; fortran: 183; cs: 181; sh: 101
file content (61 lines) | stat: -rw-r--r-- 1,922 bytes parent folder | download | duplicates (3)
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
//
//  Copyright (C) 2019-2022 NextMove Software and other RDKit contributors
//
//   @@ 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/RDKitQueries.h>
#include <GraphMol/SmilesParse/SmilesWrite.h>
#include "nmmolhash.h"

namespace RDKit {
namespace MolHash {
void Strip(RWMol *mol, unsigned int striptype) {
  // The order of these operations is significant to some degree
  // - Hydrogens should be at the end as that is the common
  //   use case

  if (striptype & static_cast<unsigned>(StripType::AtomStereo)) {
    for (auto aptr : mol->atoms()) {
      aptr->setChiralTag(RDKit::Atom::CHI_UNSPECIFIED);
    }
    if (!mol->getStereoGroups().empty()) {
      std::vector<StereoGroup> no_sgs;
      mol->setStereoGroups(std::move(no_sgs));
    }
  }
  if (striptype & static_cast<unsigned>(StripType::BondStereo)) {
    for (auto bptr : mol->bonds()) {
      if (bptr->getStereo() > RDKit::Bond::STEREOANY) {
        bptr->setStereo(RDKit::Bond::STEREOANY);
      }
    }
  }
  if (striptype & static_cast<unsigned>(StripType::Isotope)) {
    for (auto aptr : mol->atoms()) {
      aptr->setIsotope(0);
    }
  }
  if (striptype & static_cast<unsigned>(StripType::AtomMap)) {
    for (auto aptr : mol->atoms()) {
      aptr->setAtomMapNum(0);
    }
  }
  if (striptype & static_cast<unsigned>(StripType::Hydrogen)) {
    MolOps::removeHs(*mol);
  }
}

void SplitMolecule(RWMol *mol, std::vector<RWMol *> &molv) {
  auto mfrags = RDKit::MolOps::getMolFrags(*mol);
  for (const auto &frag : mfrags) {
    const auto *wrappedmol = frag.get();  // reach inside the shared pointer...
    molv.push_back(new RWMol(*wrappedmol));  // ...and make a copy
  }
}
}  // namespace MolHash
}  // namespace RDKit