Files
rdkit/Code/GraphMol/FileParsers/MolSGroupWriting.h

136 lines
4.5 KiB
C++

//
// Copyright (C) 2002-2021 Greg Landrum and T5 Informatics GmbH
//
// @@ 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.
//
#pragma once
#include <RDGeneral/BoostStartInclude.h>
#include <boost/algorithm/string/trim.hpp>
#include <boost/format.hpp>
#include <RDGeneral/BoostEndInclude.h>
#include <GraphMol/SubstanceGroup.h>
namespace RDKit {
namespace SGroupWriting {
typedef std::unordered_map<int, SubstanceGroup> IDX_TO_SGROUP_MAP;
/* ------------------ Inlined Formatters ------------------ */
inline std::string FormatV2000IntField(int value) {
auto fmt = boost::format(" %3d") % value;
return fmt.str();
}
inline std::string FormatV2000NumEntriesField(int value) {
auto fmt = boost::format(" %2d") % value;
return fmt.str();
}
inline std::string FormatV2000DoubleField(double value) {
auto fmt = boost::format("%10.4f") % value;
return fmt.str();
}
inline std::string FormatV2000StringField(const std::string &value,
unsigned int fieldSize, bool pad,
bool addSeparator) {
std::ostringstream os;
if (addSeparator) {
os << ' ';
}
if (value.size() >= fieldSize) {
os << value.substr(0, fieldSize);
} else if (pad) {
os << std::setw(fieldSize) << std::left << value;
} else {
os << value;
}
return os.str();
}
inline std::string FormatV3000DoubleField(double value) {
return boost::trim_copy(FormatV2000DoubleField(value));
}
/* ------------------ V2000 Utils ------------------ */
std::string BuildV2000STYLines(const ROMol &mol);
std::string BuildV2000StringPropLines(const unsigned int entriesPerLine,
const ROMol &mol,
const std::string &propName,
const std::string &propCode,
const unsigned int fieldWitdh);
std::string BuildV2000SLBLines(const ROMol &mol);
std::string BuildV2000SDSLines(const ROMol &mol);
std::string BuildV2000SPLLines(const ROMol &mol);
std::string BuildV2000SNCLines(const ROMol &mol);
std::string BuildV2000SBTLines(const ROMol &mol);
template <class T>
std::string BuildV2000IdxVectorDataLines(const unsigned int entriesPerLine,
const unsigned int sGroupId,
const std::string &code,
const T &dataVector);
std::string BuildV2000SMTLine(const int idx, const SubstanceGroup *sgroup);
std::string BuildV2000SDILine(const int idx, const SubstanceGroup *sgroup);
std::string BuildV2000SBVLine(const int idx, const SubstanceGroup *sgroup);
std::string BuildV2000SDTLine(const int idx, const SubstanceGroup *sgroup);
std::string BuildV2000SDDLine(const int idx, const SubstanceGroup *sgroup);
std::string BuildV2000SCDSEDLines(const int idx, const SubstanceGroup *sgroup);
std::string BuildV2000SAPLines(const int idx, const SubstanceGroup *sgroup);
std::string BuildV2000SCLLine(const int idx, const SubstanceGroup *sgroup);
const std::string GetMolFileSGroupInfo(const RWMol &mol);
/* ------------------ V3000 Utils ------------------ */
template <class T>
std::string BuildV3000IdxVectorDataBlock(const std::string &key,
const std::vector<T *> &dataVector);
template <class Iterator>
std::string BuildV3000IdxVectorDataBlock(const std::string &key,
const Iterator &dataVectorBegin,
const Iterator &dataVectorEnd);
/* Classify bonds between XBONDS and CBOfindP work on a copy of
* bonds vector to prevent reordering of original vector */
std::string BuildV3000BondsBlock(const SubstanceGroup &sgroup);
std::string FormatV3000StringPropertyBlock(const std::string &prop,
const SubstanceGroup &sgroup);
std::string FormatV3000ParentBlock(const SubstanceGroup &sgroup);
std::string FormatV3000CompNoBlock(const SubstanceGroup &sgroup);
std::string FormatV3000BracketBlock(
const std::vector<SubstanceGroup::Bracket> brackets);
std::string FormatV3000CStateBlock(
const std::vector<SubstanceGroup::CState> &cstates);
const std::string GetV3000MolFileSGroupLines(const unsigned int idx,
const SubstanceGroup &sgroup);
} // namespace SGroupWriting
} // namespace RDKit