MolGraphWriter.h

Go to the documentation of this file.

#ifndef INCLUDE_MOLASSEMBLER_MOL_GRAPH_WRITER_H
#define INCLUDE_MOLASSEMBLER_MOL_GRAPH_WRITER_H

#include "Molassembler/Graph/PrivateGraph.h"

namespace Scine {

namespace Molassembler {

class StereopermutatorList;
class AtomStereopermutator;
class BondStereopermutator;

struct MolGraphWriter {
  static const std::map<std::string, std::string> elementBGColorMap;
  static const std::map<std::string, std::string> elementTextColorMap;
  static const std::map<BondType, std::string> bondTypeDisplayString;

  const PrivateGraph* const graphPtr;
  const StereopermutatorList* const stereopermutatorListPtr;

  MolGraphWriter(
    const PrivateGraph* passGraphPtr,
    const StereopermutatorList* passPermutatorListPtr
  );

  virtual ~MolGraphWriter() = default;

  /* Information */
  Utils::ElementType getElementType(PrivateGraph::Vertex vertexIndex) const;

  // Global options
  void operator() (std::ostream& os) const;

  // Vertex options
  void operator() (std::ostream& os, PrivateGraph::Vertex vertexIndex) const;

  // Edge options
  void operator() (std::ostream& os, const PrivateGraph::Edge& edgeIndex) const;

  virtual std::vector<std::string> edgeTooltips(AtomIndex source, AtomIndex target) const;

  virtual std::vector<std::string> atomStereopermutatorTooltips(
    const AtomStereopermutator& permutator
  ) const;

  virtual std::vector<std::string> bondStereopermutatorTooltips(
    const BondStereopermutator& permutator
  ) const;
};

} // namespace Molassembler

} // namespace Scine

#endif