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
|
//
// Copyright (C) 2021-2022 David Cosgrove 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.
//
// Original author: David Cosgrove (CozChemIx Limited)
//
// This class, derived from DrawMol, does multi-coloured highlights.
// Like DrawMol, it is not part of the public API and should only be
// used by MolDraw2D.
#ifndef RDKIT_DRAWMOLMCH_H
#define RDKIT_DRAWMOLMCH_H
#include <GraphMol/MolDraw2D/AtomSymbol.h>
#include <GraphMol/MolDraw2D/DrawMol.h>
namespace RDKit {
namespace MolDraw2D_detail {
class DrawMolMCH : public DrawMol {
public:
/*!
Make a DrawMol that does multi-coloured highlights. Both atoms and
bonds can have more than 1 highlight. There's no maximum although more
than 4 is very cluttered.
\param mol : the molecule to draw
\param legend : the legend (to be drawn under the molecule)
\param width : width (in pixels) of the rendering
set this to -1 to have the canvas size set automatically
\param height : height (in pixels) of the rendering
set this to -1 to have the canvas size set automatically
\param drawOptions : a MolDrawOptions object from the owning MolDraw2D
\param textDrawer : a DrawText object from the owning MolDraw2D
\param highlight_atom_map : indexed on atom idx, the colours to be used to
highlight atoms. Not all atoms need to be mentioned.
\param highlight_bond_map : indexed on bond idx, the colours to be used to
highlight bonds. Not all bonds need to be mentioned.
\param highlightRadii : map from atomId -> radius (in molecule coordinates)
for the radii of atomic highlights. If not provided for an atom, the default
value from \c drawOptions() will be used.
\param highlight_linewidth_multipliers : map from bondId -> int, to change
the thickness of the lines used for the highlighting.
\param confId : (optional) conformer ID to be used for atomic
coordinates
*/
DrawMolMCH(const ROMol &mol, const std::string &legend, int width, int height,
MolDrawOptions &drawOptions, DrawText &textDrawer,
const std::map<int, std::vector<DrawColour>> &highlight_atom_map,
const std::map<int, std::vector<DrawColour>> &highlight_bond_map,
const std::map<int, double> &highlight_radii,
const std::map<int, int> &highlight_linewidth_multipliers,
int confId = -1);
DrawMolMCH(const DrawMol &) = delete;
DrawMolMCH(DrawMol &&) = delete;
DrawMolMCH &operator=(const DrawMol &) = delete;
DrawMolMCH &operator=(DrawMol &&) = delete;
void extractHighlights(double scale) override;
virtual void extractMCHighlights() = 0;
void getAtomRadius(unsigned int atomIdx, double &xradius,
double &yradius) const;
const std::map<int, std::vector<DrawColour>> &mcHighlightAtomMap_;
const std::map<int, std::vector<DrawColour>> &mcHighlightBondMap_;
const std::map<int, int> &highlightLinewidthMultipliers_;
};
} // namespace MolDraw2D_detail
} // namespace RDKit
#endif // RDKIT_DRAWMOLMCH_H
|