File: StereoGroup.h

package info (click to toggle)
rdkit 202503.1-5
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 220,160 kB
  • sloc: cpp: 399,240; python: 77,453; ansic: 25,517; java: 8,173; javascript: 4,005; sql: 2,389; yacc: 1,565; lex: 1,263; cs: 1,081; makefile: 580; xml: 229; fortran: 183; sh: 105
file content (112 lines) | stat: -rw-r--r-- 3,796 bytes parent folder | download | duplicates (2)
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
//
//  Copyright (C) 2018-2021 Greg Landrum 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.
//
/*! \file StereoGroup.h

  \brief Defines the class StereoGroup which stores relationships between
  the absolute configurations of atoms within a structure.

*/

#include <RDGeneral/export.h>
#ifndef RD_StereoGroup_092018
#define RD_StereoGroup_092018

#include <iostream>
#include <vector>

namespace RDKit {
class Atom;
class Bond;
class ROMol;

// OR means that it is known to be one or the other, but not both
// AND means that it is known to be a mix.
enum class StereoGroupType {
  STEREO_ABSOLUTE = 0,
  STEREO_OR = 1,
  STEREO_AND = 2
};

//! StereoGroup is a collection of atoms with a known stereochemical
//! relationship
/*!
  Used to help represent a sample with unknown stereochemistry, or that is a mix
  of diastereomers.

 */
class RDKIT_GRAPHMOL_EXPORT StereoGroup {
 private:
  StereoGroupType d_grouptype{StereoGroupType::STEREO_ABSOLUTE};
  std::vector<Atom *> d_atoms;
  std::vector<Bond *> d_bonds;

  // The group ID for AND/OR groups (it has no meaning in ABS groups).
  // 0 means no group ID is defined.
  unsigned d_readId = 0u;
  unsigned d_writeId = 0u;

 public:
  StereoGroup() {}
  // Takes control of atoms if possible.
  StereoGroup(StereoGroupType grouptype, std::vector<Atom *> &&atoms,
              std::vector<Bond *> &&bonds, unsigned readId = 0);
  StereoGroup(StereoGroupType grouptype, const std::vector<Atom *> &atoms,
              const std::vector<Bond *> &bonds, unsigned readId = 0);

  StereoGroup(const StereoGroup &other) = default;
  StereoGroup &operator=(const StereoGroup &other) = default;
  StereoGroup(StereoGroup &&other) = default;
  StereoGroup &operator=(StereoGroup &&other) = default;

  StereoGroupType getGroupType() const;
  const std::vector<Atom *> &getAtoms() const;
  const std::vector<Bond *> &getBonds() const;

  unsigned getReadId() const { return d_readId; }
  unsigned getWriteId() const { return d_writeId; }
  void setWriteId(unsigned id) { d_writeId = id; }

  // Seems odd to have to define these, but otherwise the SWIG wrappers
  // won't build
  bool operator==(const StereoGroup &other) const {
    return (d_grouptype == other.d_grouptype) && (d_atoms == other.d_atoms) &&
           (d_bonds == other.d_bonds);
  }
  bool operator!=(const StereoGroup &other) const {
    return (d_grouptype != other.d_grouptype) || (d_atoms != other.d_atoms) ||
           (d_bonds != other.d_bonds);
  }
  friend RDKIT_GRAPHMOL_EXPORT void removeAtomFromGroups(
      const Atom *atom, std::vector<StereoGroup> &groups);
};
RDKIT_GRAPHMOL_EXPORT void removeAtomFromGroups(
    const Atom *atom, std::vector<StereoGroup> &groups);
RDKIT_GRAPHMOL_EXPORT void removeGroupsWithAtom(
    const Atom *atom, std::vector<StereoGroup> &groups);
RDKIT_GRAPHMOL_EXPORT void removeGroupsWithAtoms(
    const std::vector<Atom *> &atoms, std::vector<StereoGroup> &groups);

//! Assign Group output IDs to all AND and OR StereoGroups in the vector
//! that don't already have one. The IDs are assigned based on the order
//! of the groups.
RDKIT_GRAPHMOL_EXPORT void assignStereoGroupIds(
    std::vector<StereoGroup> &groups);

//! Copy StereoGroup "read" IDs to "write" IDs so that they will be preserved
//! when the mol is exported.
RDKIT_GRAPHMOL_EXPORT void forwardStereoGroupIds(ROMol &mol);

}  // namespace RDKit

//! allows StereoGroup objects to be dumped to streams
RDKIT_GRAPHMOL_EXPORT std::ostream &operator<<(std::ostream &target,
                                               const RDKit::StereoGroup &stg);

#endif