mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-04 21:54:27 +08:00
116 lines
3.9 KiB
C++
116 lines
3.9 KiB
C++
//
|
|
// 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 <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);
|
|
friend RDKIT_GRAPHMOL_EXPORT void removeBondFromGroups(
|
|
const Bond *bond, std::vector<StereoGroup> &groups);
|
|
};
|
|
RDKIT_GRAPHMOL_EXPORT void removeAtomFromGroups(
|
|
const Atom *atom, std::vector<StereoGroup> &groups);
|
|
RDKIT_GRAPHMOL_EXPORT void removeBondFromGroups(
|
|
const Bond *bond, 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
|