pub_soft/rdkit
3
0
Fork 0
mirror of https://github.com/rdkit/rdkit.git synced 2026-06-07 22:44:25 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
36b966917929b8b00a5a0c44476eb24b7078c8fe
rdkit/Code/GraphMol/MolDraw2D/catch_tests.cpp
Greg Landrum 36b9669179 modify the way legends are handled with the flexicanvas mode (#5082) 
* modify the way legends are handled with the flexicanvas mode

* support partially flexicanvases

* changes in response to review
also update expected hashes
2022-03-13 07:15:16 +01:00

4463 lines
142 KiB
C++
Raw Blame History

//
// Copyright (C) 2019-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.
//
#include "catch.hpp"
#include <GraphMol/RDKitBase.h>
#include <RDGeneral/hash/hash.hpp>
#include <GraphMol/SmilesParse/SmilesParse.h>
#include <GraphMol/MolDraw2D/MolDraw2D.h>
#include <GraphMol/MolDraw2D/MolDraw2DSVG.h>
#include <GraphMol/MolDraw2D/MolDraw2DUtils.h>
#include <GraphMol/MolDraw2D/MolDraw2DDetails.h>
#include <GraphMol/FileParsers/FileParsers.h>
#include <GraphMol/FileParsers/PNGParser.h>
#include <boost/algorithm/string/split.hpp>
#include <GraphMol/ChemReactions/Reaction.h>
#include <GraphMol/ChemReactions/ReactionParser.h>
#include <GraphMol/CIPLabeler/CIPLabeler.h>
#include <GraphMol/Depictor/RDDepictor.h>
#include <regex>
#ifdef RDK_BUILD_CAIRO_SUPPORT
#include <cairo.h>
#include "MolDraw2DCairo.h"
#endif
// a lot of the tests check <text> flags in the SVG. That doesn't
// happen with the Freetype versions
static const bool NO_FREETYPE = true;
namespace {
// if the generated SVG hashes to the value we're expecting, delete
// the file. That way, only the files that need inspection will be
// left at the end of the run.
// The hand-drawn pictures will fail this frequently due to the use
// of random numbers to draw the lines. As well as all the testHandDrawn
// files, this includes testBrackets-5a.svg and testPositionVariation-1b.svg
static const bool DELETE_WITH_GOOD_HASH = true;
// The expected hash code for a file may be included in these maps, or
// provided in the call to check_file_hash().
// These values are for a build with FreeType, so expect them all to be
// wrong when building without.
static const std::map<std::string, std::hash_result_t> SVG_HASHES = {
{"testAtomTags_1.svg", 758653557U},
{"testAtomTags_2.svg", 1404202323U},
{"testAtomTags_3.svg", 1502891459U},
{"contourMol_1.svg", 3182387283U},
{"contourMol_2.svg", 1391972140U},
{"contourMol_3.svg", 2924226869U},
{"contourMol_4.svg", 2505505660U},
{"testDativeBonds_1.svg", 221028862U},
{"testDativeBonds_2.svg", 3778105012U},
{"testDativeBonds_3.svg", 1514383728U},
{"testDativeBonds_2a.svg", 2707752518U},
{"testDativeBonds_2b.svg", 2705796074U},
{"testDativeBonds_2c.svg", 2254149823U},
{"testDativeBonds_2d.svg", 2005943601U},
{"testZeroOrderBonds_1.svg", 582365640U},
{"testFoundations_1.svg", 224887173U},
{"testFoundations_2.svg", 3267131106U},
{"testTest_1.svg", 3267131106U},
{"testKekulizationProblems_1.svg", 2400077772U},
{"testAtomBondIndices_1.svg", 1911177499U},
{"testAtomBondIndices_2.svg", 1242142533U},
{"testAtomBondIndices_3.svg", 3503077013U},
{"testAtomBondIndices_4.svg", 505101730U},
{"testAtomBondIndices_5.svg", 216678044U},
{"testAtomBondIndices_6.svg", 3548449169U},
{"testGithub3226_1.svg", 4133242855U},
{"testGithub3226_2.svg", 1284710968U},
{"testGithub3226_3.svg", 3576515669U},
{"testGithub3369_1.svg", 3427946604U},
{"testIncludeRadicals_1a.svg", 2528551797U},
{"testIncludeRadicals_1b.svg", 3075507489U},
{"testLegendsAndDrawing-1.svg", 507104010U},
{"testGithub3577-1.svg", 3396617064U},
{"testHandDrawn-1.svg", 3735619026U},
{"testHandDrawn-2.svg", 1486085851U},
{"testHandDrawn-3.svg", 2338246554U},
{"testHandDrawn-4.svg", 1675229670U},
{"testHandDrawn-5a.svg", 3506479782U},
{"testHandDrawn-5b.svg", 1416355320U},
{"testBrackets-1a.svg", 4177217402U},
{"testBrackets-1b.svg", 326236098U},
{"testBrackets-1c.svg", 4177217402U},
{"testBrackets-1d.svg", 326236098U},
{"testBrackets-1e.svg", 3315056920U},
{"testBrackets-2a.svg", 99973183U},
{"testBrackets-2b.svg", 2970080980U},
{"testBrackets-2c.svg", 99973183U},
{"testBrackets-2d.svg", 2970080980U},
{"testBrackets-3a.svg", 3420234396U},
{"testBrackets-4a.svg", 2503277409U},
{"testBrackets-4b.svg", 3642264248U},
{"testBrackets-5a.svg", 2658328205U},
{"testSGroupData-1a.svg", 85295362U},
{"testSGroupData-1b.svg", 266947944U},
{"testSGroupData-3a.svg", 2917194552U},
{"testSGroupData-2a.svg", 2674491786U},
{"testSGroupData-2b.svg", 1043374999U},
{"testSGroupData-3a.svg", 2917194552U},
{"testPositionVariation-1.svg", 3747164807U},
{"testPositionVariation-1b.svg", 431391342U},
{"testPositionVariation-2.svg", 2598790992U},
{"testPositionVariation-3.svg", 3729075587U},
{"testPositionVariation-4.svg", 1108067269U},
{"testNoAtomLabels-1.svg", 684646985U},
{"testQueryBonds-1a.svg", 2738181855U},
{"testQueryBonds-1b.svg", 2040401048U},
{"testQueryBonds-1c.svg", 2047026108U},
{"testQueryBonds-2.svg", 187017791U},
{"testLinkNodes-2-0.svg", 1075968419U},
{"testLinkNodes-2-30.svg", 2946114571U},
{"testLinkNodes-2-60.svg", 459222785U},
{"testLinkNodes-2-90.svg", 1824183852U},
{"testLinkNodes-2-120.svg", 1813968296U},
{"testLinkNodes-2-150.svg", 2018864833U},
{"testLinkNodes-2-180.svg", 4098055961U},
{"testMolAnnotations-1.svg", 3644455962U},
{"testMolAnnotations-2a.svg", 2171709669U},
{"testMolAnnotations-2b.svg", 3176628370U},
{"testMolAnnotations-2c.svg", 2599683212U},
{"testMolAnnotations-3a.svg", 2177320081U},
{"testMolAnnotations-3b.svg", 2901914273U},
{"testMolAnnotations-3c.svg", 3200609195U},
{"testMolAnnotations-3d.svg", 944839085U},
{"testMolAnnotations-4a.svg", 606674354U},
{"testLinkNodes-1-0.svg", 1438620683U},
{"testLinkNodes-1-30.svg", 437416524U},
{"testLinkNodes-1-60.svg", 2479208702U},
{"testLinkNodes-1-90.svg", 197579058U},
{"testLinkNodes-1-120.svg", 2490429618U},
{"testLinkNodes-1-150.svg", 3088708815U},
{"testLinkNodes-1-180.svg", 2965553806U},
{"testGithub3744.svg", 2551850875U},
{"testAtomLists-1.svg", 2371311862U},
{"testAtomLists-2.svg", 1913642107U},
{"testIsoDummyIso.svg", 1431506277U},
{"testNoIsoDummyIso.svg", 4212560651U},
{"testIsoNoDummyIso.svg", 2329441786U},
{"testNoIsoNoDummyIso.svg", 1733251404U},
{"testDeuteriumTritium.svg", 2634768249U},
{"testHydrogenBonds1.svg", 2843283161U},
{"testHydrogenBonds2.svg", 3829372791U},
{"testGithub3912.1.svg", 3081580881U},
{"testGithub3912.2.svg", 1662866562U},
{"testGithub2976.svg", 716635560U},
{"testReactionCoords.svg", 258153716U},
{"testAnnotationColors.svg", 11788540U},
{"testGithub4323_1.svg", 1407816475U},
{"testGithub4323_2.svg", 2915352032U},
{"testGithub4323_3.svg", 2725419149U},
{"testGithub4323_4.svg", 3108996984U},
{"testGithub4238_1.svg", 3072374075U},
{"testGithub4508_1.svg", 4261373064U},
{"testGithub4508_1b.svg", 3090854313U},
{"testGithub4508_2.svg", 1114092U},
{"testGithub4508_2b.svg", 2711827082U},
{"testGithub4538.svg", 2194628460U},
{"testDarkMode.1.svg", 196650833U},
{"testMonochrome.1.svg", 1026689840U},
{"testMonochrome.2.svg", 2372276214U},
{"testAvalon.1.svg", 2220658203U},
{"testCDK.1.svg", 2319494973U},
{"testGithub4519_1.svg", 2772276044U},
{"testGithub4519_2.svg", 3326639754U},
{"testGithub4519_3.svg", 4206757259U},
{"testGithub4519_4.svg", 3436332962U},
{"testBaseFontSize.1a.svg", 3985093664U},
{"testBaseFontSize.1b.svg", 2606958576U},
{"testBaseFontSize.2a.svg", 1813461063U},
{"testBaseFontSize.2b.svg", 3124815325U},
{"testFlexiCanvas.1a.svg", 4203229062U},
{"testFlexiCanvas.1b.svg", 1952591508U},
{"testFlexiCanvas.1c.svg", 2165912813U},
{"testFlexiCanvas.1d.svg", 1964345717U},
{"testFlexiCanvas.2.svg", 2237869593U},
{"testSemiFlexiCanvas.1a.svg", 1387116588U},
{"testSemiFlexiCanvas.1b.svg", 4011474384U},
{"testSemiFlexiCanvas.1c.svg", 3129493597U},
{"testFlexiCanvas.3.svg", 1503902249U},
{"testFlexiCanvas.4a.svg", 428287035U},
{"testFlexiCanvas.4b.svg", 3643035210U},
{"testFlexiCanvas.4c.svg", 377546443U},
{"testFlexiCanvas.4d.svg", 2960126120U},
{"testFlexiCanvas.5a.svg", 948987396U},
{"testFlexiCanvas.5b.svg", 768214783U},
{"testFlexiCanvas.5c.svg", 1950949146U},
{"testFlexiCanvas.5d.svg", 4273629948U},
{"testFlexiCanvas.6a.svg", 1705496796U},
{"testFlexiCanvas.6b.svg", 1850606159U},
{"testFlexiCanvas.6c.svg", 1705496796U},
{"testFlexiCanvas.6d.svg", 1705496796U},
{"testFlexiCanvas.7a.svg", 1790658127U},
{"testFlexiCanvas.7b.svg", 1178964791U},
{"testFlexiCanvas.7c.svg", 1790658127U},
{"testFlexiCanvas.7d.svg", 1790658127U},
{"testGithub4764.sz1.svg", 2195931596U},
{"testGithub4764.sz2.svg", 3477099305U},
{"testGithub4764.sz3.svg", 3324176273U},
{"testDrawArc1.svg", 898508369U},
{"testMetalWedges.svg", 3741293579U},
{"testVariableLegend_1.svg", 2181678815U},
{"testVariableLegend_2.svg", 2392644674U},
{"testVariableLegend_3.svg", 2954965314U},
{"testGithub_5061.svg", 632991478U},
};
// These PNG hashes aren't completely reliable due to floating point cruft,
// but they can still reduce the number of drawings that need visual
// inspection. At present, the files
// testPNGMetadata_2.png
// give different results on my MBP and Ubuntu 20.04 VM. The SVGs work
// better because the floats are all output to only 1 decimal place so there
// is a much smaller chance of different systems producing different files.
static const std::map<std::string, std::hash_result_t> PNG_HASHES = {
{"testGithub3226_1.png", 3932085076U},
{"testGithub3226_2.png", 2515387942U},
{"testGithub3226_3.png", 913109891U},
{"testPNGMetadata_1.png", 2398100960U},
{"testPNGMetadata_2.png", 1766095719U},
{"testHandDrawn-1.png", 1894099095U},
{"testHandDrawn-2.png", 2745741502U},
{"testHandDrawn-3.png", 1157236326U},
{"testHandDrawn-4.png", 1813349867U},
{"testHandDrawn-5.png", 3312994675U},
{"testGithub4323_1.png", 4185841490U},
{"testGithub4323_3.png", 2906371027U},
{"testFlexiCanvas.2a.png", 2356975333U},
{"testFlexiCanvas.2b.png", 3107706417U},
{"testGithub4764.sz1.png", 1677681492U},
{"testGithub4764.sz2.png", 468390008U},
{"testGithub4764.sz3.png", 552619356U},
};
std::hash_result_t hash_file(const std::string &filename) {
std::ifstream ifs(filename, std::ios_base::binary);
std::string file_contents(std::istreambuf_iterator<char>{ifs}, {});
if (filename.substr(filename.length() - 4) == ".svg") {
// deal with MSDOS newlines.
file_contents.erase(
remove(file_contents.begin(), file_contents.end(), '\r'),
file_contents.end());
}
return gboost::hash_range(file_contents.begin(), file_contents.end());
}
void check_file_hash(const std::string &filename,
std::hash_result_t exp_hash = 0U) {
// std::cout << filename << " : " << hash_file(filename) << "U" <<
// std::endl;
std::map<std::string, std::hash_result_t>::const_iterator it;
if (filename.substr(filename.length() - 4) == ".svg") {
it = SVG_HASHES.find(filename);
} else {
it = PNG_HASHES.find(filename);
}
std::hash_result_t file_hash = hash_file(filename);
if (exp_hash == 0U) {
exp_hash = it == SVG_HASHES.end() ? 0U : it->second;
}
if (it != SVG_HASHES.end() && file_hash == exp_hash) {
if (DELETE_WITH_GOOD_HASH) {
std::remove(filename.c_str());
}
} else {
std::cout << "file " << filename << " gave hash " << file_hash
<< "U not the expected " << exp_hash << "U" << std::endl;
}
}
} // namespace
using namespace RDKit;
TEST_CASE("prepareAndDrawMolecule", "[drawing]") {
SECTION("basics") {
auto m1 = "C1N[C@@H]2OCC12"_smiles;
REQUIRE(m1);
// we will be able to recognize that the prep worked because there
// will be an H in the output:
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareAndDrawMolecule(drawer, *m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
CHECK(text.find(">H</text>") != std::string::npos);
}
}
TEST_CASE("tag atoms in SVG", "[drawing][SVG]") {
SECTION("basics") {
auto m1 = "C1N[C@@H]2OCC12"_smiles;
REQUIRE(m1);
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawMolecule(*m1);
std::map<std::string, std::string> actions;
actions["onclick"] = "alert";
double radius = 0.2;
drawer.tagAtoms(*m1, radius, actions);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testAtomTags_1.svg");
outs << text;
outs.close();
check_file_hash("testAtomTags_1.svg");
CHECK(text.find("<circle") != std::string::npos);
CHECK(text.find("<circle") != std::string::npos);
CHECK(text.find("atom-selector") != std::string::npos);
CHECK(text.find("bond-selector") != std::string::npos);
}
SECTION("inject prop to class") {
auto m1 = "C1N[C@@H]2OCC12"_smiles;
REQUIRE(m1);
for (auto atom : m1->atoms()) {
auto prop = boost::format("__prop_class_atom_%d") % atom->getIdx();
atom->setProp("_tagClass", prop.str());
}
for (auto bond : m1->bonds()) {
auto prop = boost::format("__prop_class_bond_%d") % bond->getIdx();
bond->setProp("_tagClass", prop.str());
}
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawMolecule(*m1);
drawer.tagAtoms(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testAtomTags_2.svg");
outs << text;
outs.close();
check_file_hash("testAtomTags_2.svg");
size_t i = 0;
size_t c = 0;
while (true) {
auto i2 = text.find("__prop_class_atom_", i);
if (i2 == std::string::npos) {
break;
}
i = i2 + 1;
c++;
}
CHECK(c == 6);
i = 0;
c = 0;
while (true) {
auto i2 = text.find("__prop_class_bond_", i);
if (i2 == std::string::npos) {
break;
}
i = i2 + 1;
c++;
}
CHECK(c == 7);
}
}
TEST_CASE("metadata in SVG", "[drawing][SVG]") {
SECTION("inject prop to metada") {
auto m1 = "C1N[C@@H]2OCC12"_smiles;
REQUIRE(m1);
for (auto atom : m1->atoms()) {
auto prop = boost::format("__prop_metadata_atom_%d") % atom->getIdx();
atom->setProp("_metaData-atom-inject-prop", prop.str());
}
for (auto bond : m1->bonds()) {
auto prop = boost::format("__prop_metadata_bond_%d") % bond->getIdx();
bond->setProp("_metaData-bond-inject-prop", prop.str());
}
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawMolecule(*m1);
drawer.addMoleculeMetadata(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testAtomTags_3.svg");
outs << text;
outs.close();
check_file_hash("testAtomTags_3.svg");
size_t i = 0;
size_t c = 0;
while (true) {
auto i2 = text.find("atom-inject-prop=\"__prop_metadata_atom_", i);
if (i2 == std::string::npos) {
break;
}
i = i2 + 1;
c++;
}
CHECK(c == 6);
i = 0;
c = 0;
while (true) {
auto i2 = text.find("bond-inject-prop=\"__prop_metadata_bond_", i);
if (i2 == std::string::npos) {
break;
}
i = i2 + 1;
c++;
}
CHECK(c == 7);
}
}
TEST_CASE("contour data", "[drawing][conrec]") {
auto m1 = "C1N[C@@H]2OCC12"_smiles;
REQUIRE(m1);
SECTION("grid basics") {
MolDraw2DSVG drawer(250, 250, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
const size_t gridSz = 100;
auto *grid = new double[gridSz * gridSz];
std::vector<double> xps(gridSz);
std::vector<double> yps(gridSz);
double minX = 1000, minY = 1000, maxX = -1000, maxY = -1000;
const auto conf = m1->getConformer();
for (size_t i = 0; i < conf.getNumAtoms(); ++i) {
minX = std::min(minX, conf.getAtomPos(i).x);
minY = std::min(minY, conf.getAtomPos(i).y);
maxX = std::max(maxX, conf.getAtomPos(i).x);
maxY = std::max(maxY, conf.getAtomPos(i).y);
}
double x1 = minX - 0.5, y1 = minY - 0.5, x2 = maxX + 0.5, y2 = maxY + 0.5;
double dx = (x2 - x1) / gridSz, dy = (y2 - y1) / gridSz;
double maxV = 0.0;
for (size_t ix = 0; ix < gridSz; ++ix) {
auto px = x1 + ix * dx;
xps[ix] = px;
for (size_t iy = 0; iy < gridSz; ++iy) {
auto py = y1 + iy * dy;
if (ix == 0) {
yps[iy] = py;
}
RDGeom::Point2D loc(px, py);
double val = 0.0;
for (size_t ia = 0; ia < conf.getNumAtoms(); ++ia) {
auto dv = loc - RDGeom::Point2D(conf.getAtomPos(ia).x,
conf.getAtomPos(ia).y);
auto r = dv.length();
if (r > 0.1) {
val += 1 / r;
}
}
maxV = std::max(val, maxV);
grid[ix * gridSz + iy] = val;
}
}
std::vector<double> levels;
drawer.clearDrawing();
MolDraw2DUtils::contourAndDrawGrid(drawer, grid, xps, yps, 10, levels,
MolDraw2DUtils::ContourParams(),
m1.get());
drawer.drawOptions().clearBackground = false;
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("contourMol_1.svg");
outs << text;
outs.close();
check_file_hash("contourMol_1.svg");
delete[] grid;
}
SECTION("gaussian basics") {
MolDraw2DSVG drawer(250, 250, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawOptions().padding = 0.1;
const auto conf = m1->getConformer();
std::vector<Point2D> cents(conf.getNumAtoms());
std::vector<double> weights(conf.getNumAtoms());
std::vector<double> widths(conf.getNumAtoms());
for (size_t i = 0; i < conf.getNumAtoms(); ++i) {
cents[i] = Point2D(conf.getAtomPos(i).x, conf.getAtomPos(i).y);
weights[i] = 1;
widths[i] = 0.4 * PeriodicTable::getTable()->getRcovalent(
m1->getAtomWithIdx(i)->getAtomicNum());
}
std::vector<double> levels;
drawer.clearDrawing();
MolDraw2DUtils::contourAndDrawGaussians(
drawer, cents, weights, widths, 10, levels,
MolDraw2DUtils::ContourParams(), m1.get());
drawer.drawOptions().clearBackground = false;
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("contourMol_2.svg");
outs << text;
outs.close();
check_file_hash("contourMol_2.svg");
}
SECTION("gaussian fill") {
MolDraw2DSVG drawer(250, 250, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawOptions().padding = 0.1;
const auto conf = m1->getConformer();
std::vector<Point2D> cents(conf.getNumAtoms());
std::vector<double> weights(conf.getNumAtoms());
std::vector<double> widths(conf.getNumAtoms());
for (size_t i = 0; i < conf.getNumAtoms(); ++i) {
cents[i] = Point2D(conf.getAtomPos(i).x, conf.getAtomPos(i).y);
weights[i] = i % 2 ? -0.5 : 1;
widths[i] = 0.4 * PeriodicTable::getTable()->getRcovalent(
m1->getAtomWithIdx(i)->getAtomicNum());
}
std::vector<double> levels;
MolDraw2DUtils::ContourParams cps;
cps.fillGrid = true;
drawer.clearDrawing();
MolDraw2DUtils::contourAndDrawGaussians(drawer, cents, weights, widths, 10,
levels, cps, m1.get());
drawer.drawOptions().clearBackground = false;
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("contourMol_3.svg");
outs << text;
outs.close();
check_file_hash("contourMol_3.svg");
}
SECTION("gaussian fill 2") {
auto m2 = "C1N[C@@H]2OCC12C=CC"_smiles;
REQUIRE(m2);
MolDraw2DSVG drawer(450, 250, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m2);
drawer.drawOptions().padding = 0.1;
const auto conf = m2->getConformer();
std::vector<Point2D> cents(conf.getNumAtoms());
std::vector<double> weights(conf.getNumAtoms());
std::vector<double> widths(conf.getNumAtoms());
for (size_t i = 0; i < conf.getNumAtoms(); ++i) {
cents[i] = Point2D(conf.getAtomPos(i).x, conf.getAtomPos(i).y);
weights[i] = i % 2 ? -0.5 : 1;
widths[i] = 0.3 * PeriodicTable::getTable()->getRcovalent(
m2->getAtomWithIdx(i)->getAtomicNum());
}
std::vector<double> levels;
MolDraw2DUtils::ContourParams cps;
cps.fillGrid = true;
cps.gridResolution = 0.5;
drawer.clearDrawing();
MolDraw2DUtils::contourAndDrawGaussians(drawer, cents, weights, widths, 10,
levels, cps, m2.get());
drawer.drawOptions().clearBackground = false;
drawer.drawMolecule(*m2);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("contourMol_4.svg");
outs << text;
outs.close();
check_file_hash("contourMol_4.svg");
}
}
TEST_CASE("dative bonds", "[drawing][organometallics]") {
SECTION("basics") {
auto m1 = "N->[Pt]"_smiles;
REQUIRE(m1);
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testDativeBonds_1.svg");
outs << text;
outs.close();
check_file_hash("testDativeBonds_1.svg");
CHECK(text.find("d='M 122.5,88.4 L 85.6,88.4' "
"style='fill:none;fill-rule:evenodd;stroke:#0000FF") !=
std::string::npos);
}
SECTION("more complex") {
auto m1 = "N->1[C@@H]2CCCC[C@H]2N->[Pt]11OC(=O)C(=O)O1"_smiles;
REQUIRE(m1);
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testDativeBonds_2.svg");
outs << text;
outs.close();
check_file_hash("testDativeBonds_2.svg");
CHECK(text.find("-8' d='M 101.1,79.8 L 95.8,87.1' "
"style='fill:none;fill-rule:evenodd;stroke:#0000FF;") !=
std::string::npos);
}
SECTION("test colours") {
// the dative bonds point the wrong way, but the point is to test
// if the tip of the arrow is blue.
auto m1 = "[Cu++]->1->2.N1CCN2"_smiles;
REQUIRE(m1);
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testDativeBonds_3.svg");
outs << text;
outs.close();
check_file_hash("testDativeBonds_3.svg");
CHECK(
text.find(
"<path class='bond-2 atom-3 atom-4' d='M 50.4,140.6 L 77.9,149.5' "
"style='fill:none;fill-rule:evenodd;stroke:#0000FF;") !=
std::string::npos);
}
SECTION("dative series") {
auto m1 = "N->1[C@@H]2CCCC[C@H]2N->[Pt]11OC(=O)C(=O)O1"_smiles;
REQUIRE(m1);
{
MolDraw2DSVG drawer(150, 150, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testDativeBonds_2a.svg");
outs << text;
outs.close();
check_file_hash("testDativeBonds_2a.svg");
}
{
MolDraw2DSVG drawer(250, 250, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testDativeBonds_2b.svg");
outs << text;
outs.close();
check_file_hash("testDativeBonds_2b.svg");
}
{
MolDraw2DSVG drawer(350, 350, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testDativeBonds_2c.svg");
outs << text;
outs.close();
check_file_hash("testDativeBonds_2c.svg");
}
{
MolDraw2DSVG drawer(450, 450, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testDativeBonds_2d.svg");
outs << text;
outs.close();
check_file_hash("testDativeBonds_2d.svg");
}
}
}
TEST_CASE("zero-order bonds", "[drawing][organometallics]") {
SECTION("basics") {
auto m1 = "N-[Pt]"_smiles;
REQUIRE(m1);
m1->getBondWithIdx(0)->setBondType(Bond::ZERO);
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareMolForDrawing(*m1);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testZeroOrderBonds_1.svg");
outs << text;
outs.close();
check_file_hash("testZeroOrderBonds_1.svg");
CHECK(text.find("stroke-dasharray:2,2") != std::string::npos);
}
}
TEST_CASE("copying drawing options", "[drawing]") {
auto m1 = "C1N[C@@H]2OCC12"_smiles;
REQUIRE(m1);
SECTION("foundations") {
{
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareAndDrawMolecule(drawer, *m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testFoundations_1.svg");
outs << text;
outs.close();
check_file_hash("testFoundations_1.svg");
CHECK(text.find("fill:#0000FF' >N</text>") != std::string::npos);
}
{
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
assignBWPalette(drawer.drawOptions().atomColourPalette);
MolDraw2DUtils::prepareAndDrawMolecule(drawer, *m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testFoundations_2.svg");
outs << text;
outs.close();
check_file_hash("testFoundations_2.svg");
CHECK(text.find("fill:#0000FF' >N</text>") == std::string::npos);
CHECK(text.find("fill:#000000' >N</text>") != std::string::npos);
}
}
SECTION("test") {
{
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
MolDrawOptions options = drawer.drawOptions();
assignBWPalette(options.atomColourPalette);
drawer.drawOptions() = options;
MolDraw2DUtils::prepareAndDrawMolecule(drawer, *m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testTest_1.svg");
outs << text;
outs.close();
check_file_hash("testTest_1.svg");
CHECK(text.find("fill:#0000FF' >N</text>") == std::string::npos);
CHECK(text.find("fill:#000000' >N</text>") != std::string::npos);
}
}
}
TEST_CASE("bad DrawMolecules() when molecules are not kekulized",
"[drawing][bug]") {
auto m1 = "CCN(CC)CCn1nc2c3ccccc3sc3c(CNS(C)(=O)=O)ccc1c32"_smiles;
REQUIRE(m1);
SECTION("foundations") {
MolDraw2DSVG drawer(500, 200, 250, 200, NO_FREETYPE);
drawer.drawOptions().prepareMolsBeforeDrawing = false;
RWMol dm1(*m1);
RWMol dm2(*m1);
bool kekulize = false;
MolDraw2DUtils::prepareMolForDrawing(dm1, kekulize);
kekulize = true;
MolDraw2DUtils::prepareMolForDrawing(dm2, kekulize);
MOL_PTR_VECT ms{&dm1, &dm2};
drawer.drawMolecule(dm1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testKekulizationProblems_1.svg");
outs << text;
outs.close();
check_file_hash("testKekulizationProblems_1.svg");
// this is a very crude test - really we just need to look at the SVG - but
// it's better than nothing.
CHECK(text.find(
"<path class='bond-18' d='M 169.076,79.056 L 191.285,69.2653' "
"style='fill:none;fill-rule:evenodd;stroke:#000000;stroke-width:"
"2px;stroke-linecap:butt;stroke-linejoin:miter;stroke-opacity:1;"
"stroke-dasharray:6,6' />") == std::string::npos);
}
}
TEST_CASE("draw atom/bond indices", "[drawing]") {
auto m1 = "C[C@H](F)N"_smiles;
auto m2 = "C[C@@H](F)N"_smiles;
REQUIRE(m1);
REQUIRE(m2);
SECTION("foundations") {
{
MolDraw2DSVG drawer(250, 200, -1, -1, NO_FREETYPE);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testAtomBondIndices_1.svg");
outs << text;
outs.close();
check_file_hash("testAtomBondIndices_1.svg");
CHECK(text.find(">1</text>") == std::string::npos);
CHECK(text.find(">(</text>") == std::string::npos);
CHECK(text.find(">S</text>") == std::string::npos);
CHECK(text.find(">)</text>") == std::string::npos);
}
{
MolDraw2DSVG drawer(250, 200, -1, -1, NO_FREETYPE);
drawer.drawOptions().addAtomIndices = true;
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testAtomBondIndices_2.svg");
outs << text;
outs.close();
check_file_hash("testAtomBondIndices_2.svg");
CHECK(text.find(">1</text>") != std::string::npos);
// it only appears once though:
CHECK(text.find(">1</text>", text.find(">1</text>") + 1) ==
std::string::npos);
CHECK(text.find("1,(S)") == std::string::npos);
}
{
MolDraw2DSVG drawer(250, 200, -1, -1, NO_FREETYPE);
drawer.drawOptions().addBondIndices = true;
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testAtomBondIndices_3.svg");
outs << text;
outs.close();
check_file_hash("testAtomBondIndices_3.svg");
CHECK(text.find(">1</text>") != std::string::npos);
// it only appears once though:
CHECK(text.find(">1</text>", text.find(">1</text>") + 1) ==
std::string::npos);
}
{
MolDraw2DSVG drawer(250, 200, -1, -1, NO_FREETYPE);
drawer.drawOptions().addAtomIndices = true;
drawer.drawOptions().addBondIndices = true;
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testAtomBondIndices_4.svg");
outs << text;
outs.close();
check_file_hash("testAtomBondIndices_4.svg");
CHECK(text.find(">1</text>") != std::string::npos);
// it appears twice:
CHECK(text.find(">1</text>", text.find(">1</text>") + 1) !=
std::string::npos);
}
{
MolDraw2DSVG drawer(250, 200, -1, -1, NO_FREETYPE);
m1->getAtomWithIdx(2)->setProp(common_properties::atomNote, "foo");
drawer.drawOptions().addAtomIndices = true;
drawer.drawOptions().addStereoAnnotation = true;
drawer.drawMolecule(*m1);
m1->getAtomWithIdx(2)->clearProp(common_properties::atomNote);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testAtomBondIndices_5.svg");
outs << text;
outs.close();
check_file_hash("testAtomBondIndices_5.svg");
CHECK(text.find(">1</text>") != std::string::npos);
CHECK(text.find(">,</text>") != std::string::npos);
CHECK(text.find(">(</text>") != std::string::npos);
CHECK(text.find(">S</text>") != std::string::npos);
CHECK(text.find(")</text>") != std::string::npos);
CHECK(text.find(">2</text>") != std::string::npos);
CHECK(text.find(">f</text>") != std::string::npos);
CHECK(text.find(">o</text>") != std::string::npos);
}
{
// Make sure it works for solid wedges as well.
MolDraw2DSVG drawer(250, 200, -1, -1, NO_FREETYPE);
drawer.drawOptions().addAtomIndices = true;
drawer.drawMolecule(*m2);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testAtomBondIndices_6.svg");
outs << text;
outs.close();
check_file_hash("testAtomBondIndices_6.svg");
CHECK(text.find(">1</text>") != std::string::npos);
// it only appears once though:
CHECK(text.find(">1</text>", text.find(">1</text>") + 1) ==
std::string::npos);
CHECK(text.find("1,(S)") == std::string::npos);
}
}
}
TEST_CASE("Github #3226: Lines in wedge bonds being drawn too closely together",
"[drawing]") {
auto m1 =
"C[C@H](C1=C(C=CC(=C1Cl)F)Cl)OC2=C(N=CC(=C2)C3=CN(N=C3)C4CCNCC4)N"_smiles;
REQUIRE(m1);
SECTION("larger SVG") {
{
MolDraw2DSVG drawer(450, 400);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub3226_1.svg");
outs << text;
outs.close();
check_file_hash("testGithub3226_1.svg");
std::vector<std::string> tkns;
boost::algorithm::find_all(tkns, text, "bond-0");
CHECK(tkns.size() == 6);
}
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
SECTION("larger PNG") {
{
MolDraw2DCairo drawer(450, 400);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
drawer.writeDrawingText("testGithub3226_1.png");
check_file_hash("testGithub3226_1.png");
}
}
#endif
SECTION("smaller SVG") {
{
MolDraw2DSVG drawer(200, 150);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub3226_2.svg");
outs << text;
outs.close();
check_file_hash("testGithub3226_2.svg");
std::vector<std::string> tkns;
boost::algorithm::find_all(tkns, text, "bond-0");
CHECK(tkns.size() == 3);
}
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
SECTION("smaller PNG") {
{
MolDraw2DCairo drawer(200, 150);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
drawer.writeDrawingText("testGithub3226_2.png");
check_file_hash("testGithub3226_2.png");
}
}
#endif
SECTION("middle SVG") {
{
MolDraw2DSVG drawer(300, 200);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub3226_3.svg");
outs << text;
outs.close();
check_file_hash("testGithub3226_3.svg");
std::vector<std::string> tkns;
boost::algorithm::find_all(tkns, text, "bond-0");
CHECK(tkns.size() == 4);
}
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
SECTION("middle PNG") {
{
MolDraw2DCairo drawer(250, 200);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
drawer.writeDrawingText("testGithub3226_3.png");
check_file_hash("testGithub3226_3.png");
}
}
#endif
}
TEST_CASE("github #3258: ", "[drawing][bug]") {
auto m1 = "CCN"_smiles;
REQUIRE(m1);
SECTION("foundations") {
MolDraw2DSVG drawer(500, 200, 250, 200, NO_FREETYPE);
drawer.drawOptions().addAtomIndices = true;
drawer.drawOptions().addBondIndices = true;
RWMol dm1(*m1);
RWMol dm2(*m1);
MOL_PTR_VECT ms{&dm1, &dm2};
drawer.drawMolecules(ms);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
CHECK(text.find(">,</text>") == std::string::npos);
CHECK(!dm1.hasProp("_atomIndicesAdded"));
CHECK(!dm1.hasProp("_bondIndicesAdded"));
}
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
TEST_CASE("adding png metadata", "[drawing][png]") {
SECTION("molecule") {
auto m1 = R"CTAB(
Mrv2014 08172015242D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 3 2 0 0 0
M V30 BEGIN ATOM
M V30 1 C 2.31 -1.3337 0 0
M V30 2 C 3.6437 -2.1037 0 0
M V30 3 O 4.9774 -1.3337 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 2 3
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m1);
{
MolDraw2DCairo drawer(250, 200);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
auto png = drawer.getDrawingText();
drawer.writeDrawingText("testPNGMetadata_1.png");
check_file_hash("testPNGMetadata_1.png");
CHECK(png.find(PNGData::smilesTag) != std::string::npos);
CHECK(png.find(PNGData::molTag) != std::string::npos);
CHECK(png.find(PNGData::pklTag) != std::string::npos);
std::unique_ptr<ROMol> newmol(PNGStringToMol(png));
REQUIRE(newmol);
CHECK(MolToCXSmiles(*m1) == MolToCXSmiles(*newmol));
}
{ // disable metadata output
MolDraw2DCairo drawer(250, 200);
drawer.drawOptions().includeMetadata = false;
drawer.drawMolecule(*m1);
drawer.finishDrawing();
auto png = drawer.getDrawingText();
CHECK(png.find(PNGData::smilesTag) == std::string::npos);
CHECK(png.find(PNGData::molTag) == std::string::npos);
CHECK(png.find(PNGData::pklTag) == std::string::npos);
}
{ // draw multiple molecules
MolDraw2DCairo drawer(250, 200);
drawer.drawMolecule(*m1);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
auto png = drawer.getDrawingText();
CHECK(png.find(PNGData::smilesTag) != std::string::npos);
CHECK(png.find(PNGData::molTag) != std::string::npos);
CHECK(png.find(PNGData::pklTag) != std::string::npos);
CHECK(png.find(PNGData::smilesTag + "1") != std::string::npos);
CHECK(png.find(PNGData::molTag + "1") != std::string::npos);
CHECK(png.find(PNGData::pklTag + "1") != std::string::npos);
}
}
SECTION("reaction") {
std::unique_ptr<ChemicalReaction> rxn(RxnSmartsToChemicalReaction(
"[N:1][C:2][C:3](=[O:4])[O:5].[N:6][C:7][C:8](=[O:9])[O:10]>>[N:1]1[C:"
"2][C:3](=[O:4])[N:6][C:7][C:8]1=[O:9].[O:5][O:10]"));
REQUIRE(rxn);
{
MolDraw2DCairo drawer(600, 200);
drawer.drawReaction(*rxn);
drawer.finishDrawing();
auto png = drawer.getDrawingText();
drawer.writeDrawingText("testPNGMetadata_2.png");
check_file_hash("testPNGMetadata_2.png");
CHECK(png.find(PNGData::smilesTag) == std::string::npos);
CHECK(png.find(PNGData::molTag) == std::string::npos);
CHECK(png.find(PNGData::pklTag) == std::string::npos);
CHECK(png.find(PNGData::rxnPklTag) != std::string::npos);
CHECK(png.find(PNGData::rxnSmartsTag) != std::string::npos);
std::unique_ptr<ChemicalReaction> rxn2(PNGStringToChemicalReaction(png));
REQUIRE(rxn2);
CHECK(ChemicalReactionToRxnSmarts(*rxn) ==
ChemicalReactionToRxnSmarts(*rxn2));
}
{ // disable metadata
MolDraw2DCairo drawer(600, 200);
drawer.drawOptions().includeMetadata = false;
drawer.drawReaction(*rxn);
drawer.finishDrawing();
auto png = drawer.getDrawingText();
CHECK(png.find(PNGData::smilesTag) == std::string::npos);
CHECK(png.find(PNGData::molTag) == std::string::npos);
CHECK(png.find(PNGData::pklTag) == std::string::npos);
CHECK(png.find(PNGData::rxnPklTag) == std::string::npos);
CHECK(png.find(PNGData::rxnSmartsTag) == std::string::npos);
}
}
}
#endif
TEST_CASE(
"github #3392: prepareMolForDrawing() incorrectly adds chiral Hs if no "
"ring info is present",
"[bug]") {
SECTION("foundations") {
SmilesParserParams ps;
ps.sanitize = false;
ps.removeHs = false;
std::unique_ptr<RWMol> m1(SmilesToMol("C[C@H](F)Cl", ps));
REQUIRE(m1);
m1->updatePropertyCache();
CHECK(m1->getNumAtoms() == 4);
const bool kekulize = false;
const bool addChiralHs = true;
MolDraw2DUtils::prepareMolForDrawing(*m1, kekulize, addChiralHs);
CHECK(m1->getNumAtoms() == 4);
}
}
TEST_CASE(
"github #3369: support new CIP code and StereoGroups in "
"addStereoAnnotation()",
"[chirality]") {
auto m1 =
"C[C@@H]1N[C@H](C)[C@@H]([C@H](C)[C@@H]1C)C1[C@@H](C)O[C@@H](C)[C@@H](C)[C@H]1C/C=C/C |a:5,o1:1,8,o2:14,16,&1:11,18,&2:3,6,r|"_smiles;
REQUIRE(m1);
SECTION("defaults") {
ROMol m2(*m1);
MolDraw2D_detail::addStereoAnnotation(m2);
std::string txt;
CHECK(m2.getAtomWithIdx(5)->getPropIfPresent(common_properties::atomNote,
txt));
CHECK(txt == "abs (S)");
CHECK(m2.getAtomWithIdx(3)->getPropIfPresent(common_properties::atomNote,
txt));
CHECK(txt == "and4");
}
SECTION("including CIP with relative stereo") {
ROMol m2(*m1);
bool includeRelativeCIP = true;
MolDraw2D_detail::addStereoAnnotation(m2, includeRelativeCIP);
std::string txt;
CHECK(m2.getAtomWithIdx(5)->getPropIfPresent(common_properties::atomNote,
txt));
CHECK(txt == "abs (S)");
CHECK(m2.getAtomWithIdx(3)->getPropIfPresent(common_properties::atomNote,
txt));
CHECK(txt == "and4 (R)");
}
SECTION("new CIP labels") {
ROMol m2(*m1);
REQUIRE(m2.getBondBetweenAtoms(20, 21));
m2.getBondBetweenAtoms(20, 21)->setStereo(Bond::BondStereo::STEREOTRANS);
// initially no label is assigned since we have TRANS
MolDraw2D_detail::addStereoAnnotation(m2);
CHECK(
!m2.getBondBetweenAtoms(20, 21)->hasProp(common_properties::bondNote));
CIPLabeler::assignCIPLabels(m2);
std::string txt;
CHECK(m2.getBondBetweenAtoms(20, 21)->getPropIfPresent(
common_properties::_CIPCode, txt));
CHECK(txt == "E");
MolDraw2D_detail::addStereoAnnotation(m2);
CHECK(m2.getBondBetweenAtoms(20, 21)->getPropIfPresent(
common_properties::bondNote, txt));
CHECK(txt == "(E)");
}
SECTION("works with the drawing code") {
MolDraw2DSVG drawer(300, 250);
RWMol dm1(*m1);
bool includeRelativeCIP = true;
MolDraw2D_detail::addStereoAnnotation(dm1, includeRelativeCIP);
drawer.drawMolecule(dm1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub3369_1.svg");
outs << text;
outs.close();
check_file_hash("testGithub3369_1.svg");
}
}
TEST_CASE("includeRadicals", "[options]") {
SECTION("basics") {
auto m = "[O][C]"_smiles;
REQUIRE(m);
int panelHeight = -1;
int panelWidth = -1;
bool noFreeType = true;
{
MolDraw2DSVG drawer(250, 200, panelWidth, panelHeight, noFreeType);
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testIncludeRadicals_1a.svg");
outs << text;
outs.close();
check_file_hash("testIncludeRadicals_1a.svg");
CHECK(text.find("<path class='atom-0' d='M") != std::string::npos);
}
{
MolDraw2DSVG drawer(250, 200, panelWidth, panelHeight, noFreeType);
drawer.drawOptions().includeRadicals = false;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testIncludeRadicals_1b.svg");
outs << text;
outs.close();
check_file_hash("testIncludeRadicals_1b.svg");
CHECK(text.find("<path class='atom-0' d='M") == std::string::npos);
}
}
}
TEST_CASE("including legend in drawing results in offset drawing later",
"[bug]") {
SECTION("basics") {
auto m = "c1ccccc1"_smiles;
REQUIRE(m);
MolDraw2DUtils::prepareMolForDrawing(*m);
auto &conf = m->getConformer();
std::vector<Point2D> polyg;
for (const auto &pt : conf.getPositions()) {
polyg.emplace_back(pt);
}
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m, "molecule legend");
drawer.setFillPolys(true);
drawer.setColour(DrawColour(1.0, 0.3, 1.0));
drawer.drawPolygon(polyg);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testLegendsAndDrawing-1.svg");
outs << text;
outs.close();
outs.close();
check_file_hash("testLegendsAndDrawing-1.svg");
// make sure the polygon starts at a bond
CHECK(text.find("<path class='bond-0 atom-0 atom-1' d='M 316.7,135.0") !=
std::string::npos);
CHECK(text.find("<path d='M 316.7,135.0") != std::string::npos);
}
}
TEST_CASE("Github #3577", "[bug]") {
SECTION("basics") {
auto m = "CCC"_smiles;
REQUIRE(m);
MolDraw2DUtils::prepareMolForDrawing(*m);
m->getAtomWithIdx(1)->setProp("atomNote", "CCC");
m->getAtomWithIdx(2)->setProp("atomNote", "ccc");
m->getBondWithIdx(0)->setProp("bondNote", "CCC");
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testGithub3577-1.svg");
outs << text;
outs.close();
check_file_hash("testGithub3577-1.svg");
}
}
TEST_CASE("hand drawn", "[play]") {
SECTION("basics") {
auto m =
"CC[CH](C)[CH]1NC(=O)[CH](Cc2ccc(O)cc2)NC(=O)[CH](N)CSSC[CH](C(=O)N2CCC[CH]2C(=O)N[CH](CC(C)C)C(=O)NCC(N)=O)NC(=O)[CH](CC(N)=O)NC(=O)[CH](CCC(N)=O)NC1=O"_smiles;
REQUIRE(m);
RDDepict::preferCoordGen = true;
MolDraw2DUtils::prepareMolForDrawing(*m);
std::string fName = getenv("RDBASE");
fName += "/Data/Fonts/ComicNeue-Regular.ttf";
{
MolDraw2DSVG drawer(450, 400);
drawer.drawOptions().fontFile = fName;
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m, "Oxytocin (flat)");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testHandDrawn-1.svg");
outs << text;
outs.close();
check_file_hash("testHandDrawn-1.svg");
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
{
MolDraw2DCairo drawer(450, 400);
drawer.drawOptions().fontFile = fName;
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m, "Oxytocin (flat)");
drawer.finishDrawing();
drawer.writeDrawingText("testHandDrawn-1.png");
check_file_hash("testHandDrawn-1.png");
}
#endif
}
SECTION("with chirality") {
auto m =
"CC[C@H](C)[C@@H]1NC(=O)[C@H](Cc2ccc(O)cc2)NC(=O)[C@@H](N)CSSC[C@@H](C(=O)N2CCC[C@H]2C(=O)N[C@@H](CC(C)C)C(=O)NCC(N)=O)NC(=O)[C@H](CC(N)=O)NC(=O)[C@H](CCC(N)=O)NC1=O"_smiles;
REQUIRE(m);
RDDepict::preferCoordGen = true;
MolDraw2DUtils::prepareMolForDrawing(*m);
std::string fName = getenv("RDBASE");
fName += "/Data/Fonts/ComicNeue-Regular.ttf";
{
MolDraw2DSVG drawer(450, 400);
drawer.drawOptions().fontFile = fName;
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m, "Oxytocin");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testHandDrawn-2.svg");
outs << text;
outs.close();
check_file_hash("testHandDrawn-2.svg");
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
{
MolDraw2DCairo drawer(450, 400);
drawer.drawOptions().fontFile = fName;
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m, "Oxytocin");
drawer.finishDrawing();
drawer.writeDrawingText("testHandDrawn-2.png");
check_file_hash("testHandDrawn-2.png");
}
#endif
}
SECTION("smaller") {
auto m = "N=c1nc([C@H]2NCCCC2)cc(N)n1O"_smiles;
REQUIRE(m);
RDDepict::preferCoordGen = true;
MolDraw2DUtils::prepareMolForDrawing(*m);
std::string fName = getenv("RDBASE");
fName += "/Data/Fonts/ComicNeue-Regular.ttf";
{
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().fontFile = fName;
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testHandDrawn-3.svg");
outs << text;
outs.close();
check_file_hash("testHandDrawn-3.svg");
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
{
MolDraw2DCairo drawer(350, 300);
drawer.drawOptions().fontFile = fName;
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m);
drawer.finishDrawing();
drawer.writeDrawingText("testHandDrawn-3.png");
check_file_hash("testHandDrawn-3.png");
}
#endif
}
SECTION("another one") {
auto m =
"CCCc1nn(C)c2c(=O)nc(-c3cc(S(=O)(=O)N4CCN(C)CC4)ccc3OCC)[nH]c12"_smiles;
REQUIRE(m);
RDDepict::preferCoordGen = true;
MolDraw2DUtils::prepareMolForDrawing(*m);
std::string fName = getenv("RDBASE");
fName += "/Data/Fonts/ComicNeue-Regular.ttf";
{
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().fontFile = fName;
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testHandDrawn-4.svg");
outs << text;
outs.close();
check_file_hash("testHandDrawn-4.svg");
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
{
MolDraw2DCairo drawer(350, 300);
drawer.drawOptions().fontFile = fName;
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m);
drawer.finishDrawing();
drawer.writeDrawingText("testHandDrawn-4.png");
check_file_hash("testHandDrawn-4.png");
}
#endif
}
SECTION("large") {
auto m =
"CC[C@H](C)[C@@H](C(=O)N[C@@H]([C@@H](C)CC)C(=O)N[C@@H](CCCCN)C(=O)N[C@@H](CC(=O)N)C(=O)N[C@@H](C)C(=O)N[C@@H](Cc1ccc(cc1)O)C(=O)N[C@@H](CCCCN)C(=O)N[C@@H](CCCCN)C(=O)NCC(=O)N[C@@H](CCC(=O)N)C(=O)O)NC(=O)[C@H](C)NC(=O)[C@H](CC(=O)N)NC(=O)[C@H](CCCCN)NC(=O)[C@H](Cc2ccccc2)NC(=O)[C@H](CC(C)C)NC(=O)[C@H]([C@@H](C)O)NC(=O)[C@H](C(C)C)NC(=O)[C@H](CC(C)C)NC(=O)[C@@H]3CCCN3C(=O)[C@H]([C@@H](C)O)NC(=O)[C@H](CCC(=O)N)NC(=O)[C@H](CO)NC(=O)[C@H](CCCCN)NC(=O)[C@H](CCC(=O)N)NC(=O)[C@H](CO)NC(=O)[C@H]([C@@H](C)O)NC(=O)[C@H](CCSC)NC(=O)[C@H](Cc4ccccc4)NC(=O)CNC(=O)CNC(=O)[C@H](Cc5ccc(cc5)O)N"_smiles;
REQUIRE(m);
RDDepict::preferCoordGen = true;
MolDraw2DUtils::prepareMolForDrawing(*m);
std::string fName = getenv("RDBASE");
fName += "/Data/Fonts/ComicNeue-Regular.ttf";
{
MolDraw2DSVG drawer(900, 450);
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testHandDrawn-5a.svg");
outs << text;
outs.close();
check_file_hash("testHandDrawn-5a.svg");
}
{
MolDraw2DSVG drawer(900, 450);
drawer.drawOptions().fontFile = fName;
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testHandDrawn-5b.svg");
outs << text;
outs.close();
check_file_hash("testHandDrawn-5b.svg");
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
{
MolDraw2DCairo drawer(900, 450);
drawer.drawOptions().fontFile = fName;
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m);
drawer.finishDrawing();
drawer.writeDrawingText("testHandDrawn-5.png");
check_file_hash("testHandDrawn-5.png");
}
#endif
}
}
TEST_CASE("drawMoleculeBrackets", "[extras]") {
SECTION("basics") {
auto m = R"CTAB(
ACCLDraw11042015112D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 5 4 1 0 0
M V30 BEGIN ATOM
M V30 1 C 7 -6.7813 0 0
M V30 2 C 8.0229 -6.1907 0 0 CFG=3
M V30 3 C 8.0229 -5.0092 0 0
M V30 4 C 9.046 -6.7814 0 0
M V30 5 C 10.0692 -6.1907 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 2 3
M V30 3 1 2 4
M V30 4 1 4 5
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 SRU 1 ATOMS=(3 3 2 4) XBONDS=(2 1 4) BRKXYZ=(9 7.51 -7.08 0 7.51 -
M V30 -5.9 0 0 0 0) BRKXYZ=(9 9.56 -5.9 0 9.56 -7.08 0 0 0 0) -
M V30 CONNECT=HT LABEL=n
M V30 END SGROUP
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-1a.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-1a.svg");
}
{ // rotation
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().rotate = 90;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-1b.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-1b.svg");
}
{ // centering
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().centreMoleculesBeforeDrawing = true;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-1c.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-1c.svg");
}
{ // rotation + centering
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().centreMoleculesBeforeDrawing = true;
drawer.drawOptions().rotate = 90;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-1d.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-1d.svg");
}
{ // rotation
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().rotate = 180;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-1e.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-1e.svg");
}
}
SECTION("three brackets") {
auto m = R"CTAB(three brackets
Mrv2014 11052006542D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 6 5 1 0 0
M V30 BEGIN ATOM
M V30 1 * -1.375 3.1667 0 0
M V30 2 C -0.0413 3.9367 0 0
M V30 3 C 1.2924 3.1667 0 0
M V30 4 * 2.626 3.9367 0 0
M V30 5 C 0.0003 5.6017 0 0
M V30 6 * 1.334 6.3717 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 2 3
M V30 3 1 3 4
M V30 4 1 2 5
M V30 5 1 5 6
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 SRU 0 ATOMS=(3 2 3 5) XBONDS=(3 1 3 5) BRKXYZ=(9 0.0875 6.7189 0 -
M V30 1.0115 5.1185 0 0 0 0) BRKXYZ=(9 1.3795 4.2839 0 2.3035 2.6835 0 0 0 -
M V30 0) BRKXYZ=(9 -0.1285 2.8194 0 -1.0525 4.4198 0 0 0 0) CONNECT=HT -
M V30 LABEL=n
M V30 END SGROUP
M V30 END CTAB
M END)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-2a.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-2a.svg");
}
{ // rotation
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().rotate = 90;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-2b.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-2b.svg");
}
{ // centering
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().centreMoleculesBeforeDrawing = true;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-2c.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-2c.svg");
}
{ // rotation + centering
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().centreMoleculesBeforeDrawing = true;
drawer.drawOptions().rotate = 90;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-2d.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-2d.svg");
}
}
SECTION("ChEBI 59342") {
// thanks to John Mayfield for pointing out the example
auto m = R"CTAB(ChEBI59342
Marvin 05041012302D
29 30 0 0 1 0 999 V2000
10.1615 -7.7974 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
8.7305 -6.9763 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
8.7309 -7.8004 0.0000 C 0 0 2 0 0 0 0 0 0 0 0 0
9.4464 -8.2109 0.0000 C 0 0 2 0 0 0 0 0 0 0 0 0
8.0153 -8.2225 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
9.4464 -9.0437 0.0000 C 0 0 2 0 0 0 0 0 0 0 0 0
8.0138 -9.0500 0.0000 C 0 0 1 0 0 0 0 0 0 0 0 0
8.7293 -9.4606 0.0000 C 0 0 1 0 0 0 0 0 0 0 0 0
10.1669 -9.4529 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
7.3058 -9.4590 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
8.7368 -10.2801 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
8.0263 -10.6992 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
8.0339 -11.5241 0.0000 H 0 0 0 0 0 0 0 0 0 0 0 0
7.3081 -10.2933 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
8.7305 -5.3264 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
8.0159 -5.7369 0.0000 C 0 0 2 0 0 0 0 0 0 0 0 0
8.0159 -6.5618 0.0000 C 0 0 2 0 0 0 0 0 0 0 0 0
7.2936 -5.3263 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
7.2936 -6.9762 0.0000 C 0 0 2 0 0 0 0 0 0 0 0 0
6.5751 -5.7368 0.0000 C 0 0 1 0 0 0 0 0 0 0 0 0
6.5751 -6.5618 0.0000 C 0 0 1 0 0 0 0 0 0 0 0 0
7.2973 -7.8049 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
5.8681 -5.3263 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
5.8680 -6.9762 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
5.1510 -6.5684 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
4.4392 -6.9856 0.0000 H 0 0 0 0 0 0 0 0 0 0 0 0
5.1455 -5.7435 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
10.4142 -5.3560 0.0000 * 0 0 0 0 0 0 0 0 0 0 0 0
11.5590 -7.8297 0.0000 * 0 0 0 0 0 0 0 0 0 0 0 0
3 2 1 6 0 0 0
3 4 1 0 0 0 0
3 5 1 0 0 0 0
4 6 1 0 0 0 0
4 1 1 1 0 0 0
5 7 1 0 0 0 0
6 8 1 0 0 0 0
6 9 1 1 0 0 0
7 10 1 1 0 0 0
8 11 1 6 0 0 0
7 8 1 0 0 0 0
13 12 1 0 0 0 0
14 12 2 0 0 0 0
11 12 1 0 0 0 0
16 15 1 6 0 0 0
16 17 1 0 0 0 0
16 18 1 0 0 0 0
17 19 1 0 0 0 0
17 2 1 1 0 0 0
18 20 1 0 0 0 0
19 21 1 0 0 0 0
19 22 1 1 0 0 0
20 23 1 1 0 0 0
21 24 1 6 0 0 0
20 21 1 0 0 0 0
26 25 1 0 0 0 0
27 25 2 0 0 0 0
24 25 1 0 0 0 0
15 28 1 0 0 0 0
1 29 1 0 0 0 0
M STY 1 1 SRU
M SCN 1 1 HT
M SAL 1 15 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
M SAL 1 12 16 17 18 19 20 21 22 23 24 25 26 27
M SDI 1 4 9.4310 -4.9261 9.4165 -5.7510
M SDI 1 4 10.7464 -7.3983 10.7274 -8.2231
M SBL 1 2 30 29
M SMT 1 n
M END)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-3a.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-3a.svg");
}
}
SECTION("pathological bracket orientation") {
{ // including the bonds
auto m = R"CTAB(bogus
Mrv2014 11202009512D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 9 8 1 0 1
M V30 BEGIN ATOM
M V30 1 C 23.5462 -14.464 0 0
M V30 2 C 20.8231 -13.0254 0 0
M V30 3 C 20.8776 -14.5628 0 0
M V30 4 C 22.2391 -15.2819 0 0
M V30 5 C 16.2969 -9.9426 0 0
M V30 6 C 14.963 -10.7089 0 0
M V30 7 C 19.463 -12.2987 0 0
M V30 8 * 19.4398 -9.9979 0 0
M V30 9 * 26.1554 -14.4332 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 3 4
M V30 2 1 6 7
M V30 3 1 5 8
M V30 4 1 1 9
M V30 5 1 7 2
M V30 6 1 6 5
M V30 7 1 4 1
M V30 8 1 3 2
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 SRU 0 ATOMS=(7 4 3 7 6 5 2 1) XBONDS=(2 3 4) BRKXYZ=(9 17.6045 -
M V30 -9.1954 0 17.5775 -10.7352 0 0 0 0) BRKXYZ=(9 24.6113 -13.6813 0 -
M V30 24.6296 -15.2213 0 0 0 0) CONNECT=HT LABEL=n
M V30 END SGROUP
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-4a.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-4a.svg");
}
{ // no bonds in the sgroup, the bracket should point the other way
// (towards the majority of the atoms in the sgroup)
auto m = R"CTAB(bogus
Mrv2014 11202009512D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 9 8 1 0 1
M V30 BEGIN ATOM
M V30 1 C 23.5462 -14.464 0 0
M V30 2 C 20.8231 -13.0254 0 0
M V30 3 C 20.8776 -14.5628 0 0
M V30 4 C 22.2391 -15.2819 0 0
M V30 5 C 16.2969 -9.9426 0 0
M V30 6 C 14.963 -10.7089 0 0
M V30 7 C 19.463 -12.2987 0 0
M V30 8 * 19.4398 -9.9979 0 0
M V30 9 * 26.1554 -14.4332 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 3 4
M V30 2 1 6 7
M V30 3 1 5 8
M V30 4 1 1 9
M V30 5 1 7 2
M V30 6 1 6 5
M V30 7 1 4 1
M V30 8 1 3 2
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 SRU 0 ATOMS=(7 4 3 7 6 5 2 1) BRKXYZ=(9 17.6045 -
M V30 -9.1954 0 17.5775 -10.7352 0 0 0 0) BRKXYZ=(9 24.6113 -13.6813 0 -
M V30 24.6296 -15.2213 0 0 0 0) CONNECT=HT LABEL=n
M V30 END SGROUP
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-4b.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-4b.svg");
}
}
SECTION("comic brackets (no font though)") {
auto m = R"CTAB(
ACCLDraw11042015112D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 5 4 1 0 0
M V30 BEGIN ATOM
M V30 1 C 7 -6.7813 0 0
M V30 2 C 8.0229 -6.1907 0 0 CFG=3
M V30 3 C 8.0229 -5.0092 0 0
M V30 4 C 9.046 -6.7814 0 0
M V30 5 C 10.0692 -6.1907 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 2 3
M V30 3 1 2 4
M V30 4 1 4 5
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 SRU 1 ATOMS=(3 3 2 4) XBONDS=(2 1 4) BRKXYZ=(9 7.51 -7.08 0 7.51 -
M V30 -5.9 0 0 0 0) BRKXYZ=(9 9.56 -5.9 0 9.56 -7.08 0 0 0 0) -
M V30 CONNECT=HT LABEL=n
M V30 END SGROUP
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testBrackets-5a.svg");
outs << text;
outs.close();
check_file_hash("testBrackets-5a.svg");
}
}
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
TEST_CASE("github #3543: Error adding PNG metadata when kekulize=False",
"[bug][metadata][png]") {
SECTION("basics") {
auto m = "n1cccc1"_smarts;
m->updatePropertyCache(false);
MolDraw2DCairo drawer(350, 300);
bool kekulize = false;
MolDraw2DUtils::prepareMolForDrawing(*m, kekulize);
drawer.drawOptions().prepareMolsBeforeDrawing = false;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto png = drawer.getDrawingText();
}
SECTION("as reported") {
auto m = "n1cnc2c(n)ncnc12"_smarts;
m->updatePropertyCache(false);
MolDraw2DCairo drawer(350, 300);
bool kekulize = false;
MolDraw2DUtils::prepareMolForDrawing(*m, kekulize);
drawer.drawOptions().prepareMolsBeforeDrawing = false;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto png = drawer.getDrawingText();
}
}
#endif
TEST_CASE("SGroup Data") {
SECTION("ABS") {
auto m = R"CTAB(
Mrv2014 12072015352D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 9 9 1 0 0
M V30 BEGIN ATOM
M V30 1 C -6.5833 4.3317 0 0
M V30 2 C -7.917 3.5617 0 0
M V30 3 C -7.917 2.0216 0 0
M V30 4 C -6.5833 1.2516 0 0
M V30 5 C -5.2497 2.0216 0 0
M V30 6 C -5.2497 3.5617 0 0
M V30 7 C -3.916 4.3317 0 0
M V30 8 O -3.916 5.8717 0 0
M V30 9 O -2.5823 3.5617 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 2 2 3
M V30 3 1 3 4
M V30 4 2 4 5
M V30 5 1 5 6
M V30 6 2 1 6
M V30 7 1 6 7
M V30 8 2 7 8
M V30 9 1 7 9
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 DAT 0 ATOMS=(1 9) FIELDNAME=pKa -
M V30 FIELDDISP=" -2.2073 2.3950 DAU ALL 0 0" -
M V30 MRV_FIELDDISP=0 FIELDDATA=4.2
M V30 END SGROUP
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m, "abs");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testSGroupData-1a.svg");
outs << text;
outs.close();
check_file_hash("testSGroupData-1a.svg");
}
{
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().centreMoleculesBeforeDrawing = true;
drawer.drawOptions().rotate = 90;
drawer.drawMolecule(*m, "centered, rotated");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testSGroupData-1b.svg");
outs << text;
outs.close();
check_file_hash("testSGroupData-1b.svg");
}
}
SECTION("REL") {
auto m = R"CTAB(
Mrv2014 12072015352D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 9 9 1 0 0
M V30 BEGIN ATOM
M V30 1 C -6.5833 4.3317 0 0
M V30 2 C -7.917 3.5617 0 0
M V30 3 C -7.917 2.0216 0 0
M V30 4 C -6.5833 1.2516 0 0
M V30 5 C -5.2497 2.0216 0 0
M V30 6 C -5.2497 3.5617 0 0
M V30 7 C -3.916 4.3317 0 0
M V30 8 O -3.916 5.8717 0 0
M V30 9 O -2.5823 3.5617 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 2 2 3
M V30 3 1 3 4
M V30 4 2 4 5
M V30 5 1 5 6
M V30 6 2 1 6
M V30 7 1 6 7
M V30 8 2 7 8
M V30 9 1 7 9
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 DAT 0 ATOMS=(1 9) FIELDNAME=pKa -
M V30 FIELDDISP=" 0.2000 0.2000 DRU ALL 0 0" -
M V30 MRV_FIELDDISP=0 FIELDDATA=4.2
M V30 END SGROUP
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m, "rel");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testSGroupData-2a.svg");
outs << text;
outs.close();
check_file_hash("testSGroupData-2a.svg");
}
{
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().centreMoleculesBeforeDrawing = true;
drawer.drawOptions().rotate = 90;
drawer.drawMolecule(*m, "rel, centered, rotated");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testSGroupData-2b.svg");
outs << text;
outs.close();
check_file_hash("testSGroupData-2b.svg");
}
}
{
auto m = R"CTAB(random example found on internet
JSDraw204221719232D
20 21 0 0 0 0 0 V2000
10.1710 -5.6553 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
10.9428 -4.2996 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
8.6110 -5.6647 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
10.9591 -7.0015 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
12.5190 -6.9921 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
13.3072 -8.3384 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
13.2909 -5.6364 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
12.5028 -4.2902 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
13.2746 -2.9345 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
14.8508 -5.6270 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
15.6226 -4.2713 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
20.3026 -4.2431 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
19.5307 -5.5987 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
21.8625 -4.2336 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
19.5144 -2.8968 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
17.9544 -2.9062 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
17.1663 -1.5600 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
17.1826 -4.2619 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
17.9708 -5.6082 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
17.1989 -6.9638 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
1 2 1 0 0 0 0
1 3 2 0 0 0 0
1 4 1 0 0 0 0
4 5 2 0 0 0 0
5 6 1 0 0 0 0
5 7 1 0 0 0 0
7 8 1 0 0 0 0
8 9 2 0 0 0 0
8 2 1 0 0 0 0
7 10 1 0 0 0 0
10 11 2 0 0 0 0
12 13 1 0 0 0 0
12 14 2 0 0 0 0
12 15 1 0 0 0 0
15 16 1 0 0 0 0
16 17 2 0 0 0 0
16 18 1 0 0 0 0
18 19 1 0 0 0 0
19 20 1 0 0 0 0
19 13 2 0 0 0 0
11 18 1 0 0 0 0
M STY 1 1 DAT
M SDT 1 UNKNOWN F
M SDD 1 16.0856 -8.1573 DA ALL 1 5
M SED 1 Ni-complex
M END)CTAB"_ctab;
{
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testSGroupData-3a.svg");
outs << text;
outs.close();
check_file_hash("testSGroupData-3a.svg");
}
}
}
TEST_CASE("position variation bonds", "[extras]") {
SECTION("simple") {
auto m = R"CTAB(
Mrv2014 12092006072D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 9 8 0 0 0
M V30 BEGIN ATOM
M V30 1 C -4.7083 4.915 0 0
M V30 2 C -6.042 4.145 0 0
M V30 3 C -6.042 2.605 0 0
M V30 4 C -4.7083 1.835 0 0
M V30 5 C -3.3747 2.605 0 0
M V30 6 C -3.3747 4.145 0 0
M V30 7 * -3.8192 3.8883 0 0
M V30 8 O -3.8192 6.1983 0 0
M V30 9 C -2.4855 6.9683 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 2 2 3
M V30 3 1 3 4
M V30 4 2 4 5
M V30 5 1 5 6
M V30 6 2 1 6
M V30 7 1 7 8 ENDPTS=(3 1 6 5) ATTACH=ANY
M V30 8 1 8 9
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m, "variations");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testPositionVariation-1.svg");
outs << text;
outs.close();
check_file_hash("testPositionVariation-1.svg");
}
{ // make sure comic mode doesn't screw this up
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().comicMode = true;
drawer.drawMolecule(*m, "comic variations");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testPositionVariation-1b.svg");
outs << text;
outs.close();
check_file_hash("testPositionVariation-1b.svg");
}
}
SECTION("multiple") {
auto m = R"CTAB(
Mrv2014 12092006082D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 15 14 0 0 0
M V30 BEGIN ATOM
M V30 1 C -4.7083 4.915 0 0
M V30 2 C -6.042 4.145 0 0
M V30 3 C -6.042 2.605 0 0
M V30 4 C -4.7083 1.835 0 0
M V30 5 C -3.3747 2.605 0 0
M V30 6 C -3.3747 4.145 0 0
M V30 7 * -3.8192 3.8883 0 0
M V30 8 O -3.8192 6.1983 0 0
M V30 9 C -2.4855 6.9683 0 0
M V30 10 C -7.3757 4.915 0 0
M V30 11 C -8.7093 4.145 0 0
M V30 12 C -8.7093 2.605 0 0
M V30 13 C -7.3757 1.835 0 0
M V30 14 * -8.7093 3.375 0 0
M V30 15 O -10.2922 3.375 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 2 3
M V30 3 1 3 4
M V30 4 2 4 5
M V30 5 1 5 6
M V30 6 2 1 6
M V30 7 1 7 8 ENDPTS=(3 1 6 5) ATTACH=ANY
M V30 8 1 8 9
M V30 9 1 10 11
M V30 10 2 11 12
M V30 11 1 12 13
M V30 12 2 10 2
M V30 13 2 13 3
M V30 14 1 14 15 ENDPTS=(2 11 12) ATTACH=ANY
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m, "multiple variations");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testPositionVariation-2.svg");
outs << text;
outs.close();
check_file_hash("testPositionVariation-2.svg");
}
}
SECTION("non-contiguous") {
auto m = R"CTAB(
Mrv2014 12092006102D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 9 8 0 0 0
M V30 BEGIN ATOM
M V30 1 C -0.875 8.7484 0 0
M V30 2 C -2.2087 7.9784 0 0
M V30 3 C -2.2087 6.4383 0 0
M V30 4 C -0.875 5.6683 0 0
M V30 5 C 0.4587 6.4383 0 0
M V30 6 C 0.4587 7.9784 0 0
M V30 7 * -0.4304 6.9517 0 0
M V30 8 O -0.4304 4.6417 0 0
M V30 9 C -1.7641 3.8717 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 2 2 3
M V30 3 1 3 4
M V30 4 2 4 5
M V30 5 1 5 6
M V30 6 2 1 6
M V30 7 1 7 8 ENDPTS=(3 1 5 4) ATTACH=ANY
M V30 8 1 8 9
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m, "non-contiguous atoms");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testPositionVariation-3.svg");
outs << text;
outs.close();
check_file_hash("testPositionVariation-3.svg");
}
}
SECTION("larger mol") {
auto m = R"CTAB(
Mrv2014 12092009152D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 23 24 0 0 0
M V30 BEGIN ATOM
M V30 1 C -0.875 8.7484 0 0
M V30 2 C -2.2087 7.9784 0 0
M V30 3 C -2.2087 6.4383 0 0
M V30 4 C -0.875 5.6683 0 0
M V30 5 N 0.4587 6.4383 0 0
M V30 6 C 0.4587 7.9784 0 0
M V30 7 * -0.4304 6.9517 0 0
M V30 8 O -0.4304 4.6417 0 0
M V30 9 C -1.7641 3.8717 0 0
M V30 10 C -3.5423 8.7484 0 0
M V30 11 C -4.876 7.9784 0 0
M V30 12 C -4.876 6.4383 0 0
M V30 13 C -3.5423 5.6683 0 0
M V30 14 C -4.876 11.0584 0 0
M V30 15 C -6.2097 10.2884 0 0
M V30 16 C -6.2097 8.7484 0 0
M V30 17 C -3.5423 10.2884 0 0
M V30 18 C -6.2097 13.3685 0 0
M V30 19 C -7.5433 12.5985 0 0
M V30 20 C -7.5433 11.0584 0 0
M V30 21 C -4.876 12.5985 0 0
M V30 22 * -5.5428 9.1334 0 0
M V30 23 C -7.3712 7.7304 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 2 2 3
M V30 3 1 3 4
M V30 4 2 4 5
M V30 5 1 5 6
M V30 6 2 1 6
M V30 7 1 7 8 ENDPTS=(3 1 4 5) ATTACH=ANY
M V30 8 1 8 9
M V30 9 2 10 11
M V30 10 1 11 12
M V30 11 2 12 13
M V30 12 1 10 2
M V30 13 1 13 3
M V30 14 1 14 15
M V30 15 2 15 16
M V30 16 2 14 17
M V30 17 1 10 17
M V30 18 1 16 11
M V30 19 1 18 19
M V30 20 2 19 20
M V30 21 2 18 21
M V30 22 1 14 21
M V30 23 1 20 15
M V30 24 1 22 23 ENDPTS=(2 15 11) ATTACH=ANY
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(250, 200);
drawer.drawMolecule(*m, "smaller");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testPositionVariation-4.svg");
outs << text;
outs.close();
check_file_hash("testPositionVariation-4.svg");
}
}
}
TEST_CASE("disable atom labels", "[feature]") {
SECTION("basics") {
auto m = "NCC(=O)O"_smiles;
MolDraw2DSVG drawer(350, 300);
MolDraw2DUtils::prepareMolForDrawing(*m);
drawer.drawOptions().noAtomLabels = true;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testNoAtomLabels-1.svg");
outs << text;
outs.close();
check_file_hash("testNoAtomLabels-1.svg");
CHECK(text.find("class='atom-0") == std::string::npos);
CHECK(text.find("class='atom-3") == std::string::npos);
}
}
TEST_CASE("drawing query bonds", "[queries]") {
SECTION("basics") {
auto m = R"CTAB(
Mrv2014 12072005332D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 14 14 0 0 0
M V30 BEGIN ATOM
M V30 1 C 3.7917 -2.96 0 0
M V30 2 C 2.458 -3.73 0 0
M V30 3 C 2.458 -5.27 0 0
M V30 4 C 3.7917 -6.04 0 0
M V30 5 C 5.1253 -5.27 0 0
M V30 6 C 5.1253 -3.73 0 0
M V30 7 C 6.459 -2.96 0 0
M V30 8 C 3.7917 -7.58 0 0
M V30 9 C 4.8806 -8.669 0 0
M V30 10 C 4.482 -10.1565 0 0
M V30 11 C 6.459 -6.04 0 0
M V30 12 C 7.7927 -5.27 0 0
M V30 13 C 9.1263 -6.0399 0 0
M V30 14 C 9.1263 -7.5799 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 2 3
M V30 2 1 4 5
M V30 3 1 1 6
M V30 4 5 1 2
M V30 5 6 5 6
M V30 6 7 3 4
M V30 7 8 6 7
M V30 8 1 4 8
M V30 9 1 8 9 TOPO=1
M V30 10 1 9 10 TOPO=2
M V30 11 1 5 11
M V30 12 1 12 13
M V30 13 2 11 12 TOPO=1
M V30 14 2 13 14 TOPO=2
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testQueryBonds-1a.svg");
outs << text;
outs.close();
check_file_hash("testQueryBonds-1a.svg");
}
{
MolDraw2DSVG drawer(350, 300);
m->getBondWithIdx(3)->setProp("bondNote", "S/D");
m->getBondWithIdx(4)->setProp("bondNote", "S/A");
m->getBondWithIdx(5)->setProp("bondNote", "D/A");
m->getBondWithIdx(6)->setProp("bondNote", "Any");
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testQueryBonds-1b.svg");
outs << text;
outs.close();
check_file_hash("testQueryBonds-1b.svg");
}
{
MolDraw2DSVG drawer(350, 300);
std::vector<int> highlightAtoms = {0, 1, 2, 3, 4, 5, 7, 8, 9};
std::vector<int> highlightBonds = {0, 3, 2, 4, 1, 5, 8, 9};
drawer.drawMolecule(*m, "", &highlightAtoms, &highlightBonds);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testQueryBonds-1c.svg");
outs << text;
outs.close();
check_file_hash("testQueryBonds-1c.svg");
}
}
SECTION("smaller drawing") {
auto m = R"CTAB(
Mrv2014 12012004302D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 26 29 0 0 0
M V30 BEGIN ATOM
M V30 1 O 3.7917 -2.96 0 0
M V30 2 C 2.458 -3.73 0 0
M V30 3 C 2.458 -5.27 0 0
M V30 4 N 3.7917 -6.04 0 0
M V30 5 N 5.1253 -5.27 0 0
M V30 6 C 5.1253 -3.73 0 0
M V30 7 C 6.459 -2.96 0 0
M V30 8 C 3.7917 -7.58 0 0
M V30 9 C 4.8806 -8.669 0 0
M V30 10 C 4.482 -10.1565 0 0
M V30 11 C 1.1243 -2.9599 0 0
M V30 12 C -0.2093 -3.73 0 0
M V30 13 C -0.2093 -5.27 0 0
M V30 14 C 1.1243 -6.04 0 0
M V30 15 C -0.2093 -0.6499 0 0
M V30 16 C -1.543 -1.4199 0 0
M V30 17 C -1.543 -2.9599 0 0
M V30 18 C 1.1243 -1.4199 0 0
M V30 19 C -2.8767 -0.6499 0 0
M V30 20 C -4.2103 -1.4199 0 0
M V30 21 C -4.2103 -2.9599 0 0
M V30 22 C -2.8767 -3.73 0 0
M V30 23 C -5.544 -3.7299 0 0
M V30 24 C -6.8777 -2.9599 0 0
M V30 25 C -8.2114 -3.7299 0 0
M V30 26 C -9.5451 -2.9599 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 2 3
M V30 2 1 4 5
M V30 3 1 1 6
M V30 4 5 1 2
M V30 5 6 5 6
M V30 6 7 3 4
M V30 7 8 6 7
M V30 8 1 4 8
M V30 9 1 8 9 TOPO=1
M V30 10 1 9 10 TOPO=2
M V30 11 1 12 13
M V30 12 1 13 14
M V30 13 1 14 3
M V30 14 1 11 2
M V30 15 1 15 16
M V30 16 1 16 17
M V30 17 2 15 18
M V30 18 1 11 18
M V30 19 1 17 12
M V30 20 2 12 11
M V30 21 1 19 20
M V30 22 2 20 21
M V30 23 1 21 22
M V30 24 2 19 16
M V30 25 2 22 17
M V30 26 1 21 23
M V30 27 1 23 24
M V30 28 1 24 25
M V30 29 1 25 26
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
{
MolDraw2DSVG drawer(250, 200);
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testQueryBonds-2.svg");
outs << text;
outs.close();
check_file_hash("testQueryBonds-2.svg");
}
}
SECTION("two linknodes") {
auto m = R"CTAB(two linknodes
Mrv2014 07072016412D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 7 7 0 0 0
M V30 BEGIN ATOM
M V30 1 C 8.25 12.1847 0 0
M V30 2 C 6.9164 12.9547 0 0
M V30 3 C 7.2366 14.4611 0 0
M V30 4 C 8.7681 14.622 0 0
M V30 5 C 9.3945 13.2151 0 0
M V30 6 O 8.25 10.6447 0 0
M V30 7 F 9.5382 15.9557 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 2 3
M V30 3 1 4 5
M V30 4 1 1 5
M V30 5 1 3 4
M V30 6 1 1 6
M V30 7 1 4 7
M V30 END BOND
M V30 LINKNODE 1 3 2 1 2 1 5
M V30 LINKNODE 1 4 2 4 3 4 5
M V30 END CTAB
M END)CTAB"_ctab;
std::vector<int> rotns = {0, 30, 60, 90, 120, 150, 180};
for (auto rotn : rotns) {
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().rotate = (double)rotn;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::string filename(
(boost::format("testLinkNodes-2-%d.svg") % rotn).str());
std::ofstream outs(filename);
outs << text;
outs.close();
check_file_hash(filename);
}
}
}
TEST_CASE("molecule annotations", "[extra]") {
int panelHeight = -1;
int panelWidth = -1;
bool noFreeType = false;
SECTION("basics") {
auto m = "NCC(=O)O"_smiles;
MolDraw2DSVG drawer(350, 300, panelHeight, panelWidth, noFreeType);
MolDraw2DUtils::prepareMolForDrawing(*m);
m->setProp(common_properties::molNote, "molecule note");
drawer.drawMolecule(*m, "with note");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMolAnnotations-1.svg");
outs << text;
outs.close();
check_file_hash("testMolAnnotations-1.svg");
CHECK(text.find("class='note'") != std::string::npos);
}
SECTION("chiral flag") {
auto m = R"CTAB(
Mrv2014 12152012512D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 8 8 0 0 1
M V30 BEGIN ATOM
M V30 1 C -0.6317 0.6787 0 0 CFG=2
M V30 2 C -1.7207 1.7677 0 0
M V30 3 C 0.4571 1.7677 0 0
M V30 4 C -0.6317 2.8566 0 0 CFG=1
M V30 5 C 0.1729 4.1698 0 0
M V30 6 N -0.5619 5.5231 0 0
M V30 7 C -1.4364 4.1698 0 0
M V30 8 C -0.6316 -0.8613 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2 CFG=3
M V30 2 1 1 3
M V30 3 1 4 3
M V30 4 1 4 2
M V30 5 1 4 5
M V30 6 1 5 6
M V30 7 1 4 7 CFG=1
M V30 8 1 1 8
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
{
MolDraw2DSVG drawer(350, 300, panelHeight, panelWidth, noFreeType);
drawer.drawMolecule(*m, "chiral flag set, option disabled");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMolAnnotations-2a.svg");
outs << text;
outs.close();
check_file_hash("testMolAnnotations-2a.svg");
CHECK(text.find("class='note'") == std::string::npos);
}
{
MolDraw2DSVG drawer(350, 300, panelHeight, panelWidth, noFreeType);
drawer.drawOptions().includeChiralFlagLabel = true;
drawer.drawMolecule(*m, "chiral flag set, option enabled");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMolAnnotations-2b.svg");
outs << text;
outs.close();
check_file_hash("testMolAnnotations-2b.svg");
CHECK(text.find("class='note'") != std::string::npos);
}
{
MolDraw2DSVG drawer(350, 300, panelHeight, panelWidth, noFreeType);
drawer.drawOptions().includeChiralFlagLabel = true;
m->clearProp(common_properties::_MolFileChiralFlag);
drawer.drawMolecule(*m, "chiral flag not set, option enabled");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMolAnnotations-2c.svg");
outs << text;
outs.close();
check_file_hash("testMolAnnotations-2c.svg");
CHECK(text.find("class='note'") == std::string::npos);
}
}
SECTION("simplified stereo 1") {
{
auto m = "C[C@H](F)[C@@H](F)[C@@H](C)Cl |o1:3,5,1|"_smiles;
MolDraw2DSVG drawer(350, 300, panelHeight, panelWidth, noFreeType);
MolDraw2DUtils::prepareMolForDrawing(*m);
drawer.drawOptions().addStereoAnnotation = true;
drawer.drawMolecule(*m, "enhanced no flag");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMolAnnotations-3a.svg");
outs << text;
outs.close();
check_file_hash("testMolAnnotations-3a.svg");
}
{
auto m = "C[C@H](F)[C@@H](F)[C@@H](C)Cl |o1:3,5,1|"_smiles;
MolDraw2DSVG drawer(350, 300, panelHeight, panelWidth, noFreeType);
MolDraw2DUtils::prepareMolForDrawing(*m);
drawer.drawOptions().addStereoAnnotation = true;
drawer.drawOptions().simplifiedStereoGroupLabel = true;
drawer.drawMolecule(*m, "enhanced with flag");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMolAnnotations-3b.svg");
outs << text;
outs.close();
check_file_hash("testMolAnnotations-3b.svg");
}
{
auto m = "C[C@H](F)[C@@H](F)[C@@H](C)Cl |&1:3,5,1|"_smiles;
MolDraw2DSVG drawer(350, 300, panelHeight, panelWidth, noFreeType);
MolDraw2DUtils::prepareMolForDrawing(*m);
drawer.drawOptions().addStereoAnnotation = true;
drawer.drawOptions().simplifiedStereoGroupLabel = true;
drawer.drawMolecule(*m, "enhanced & with flag");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMolAnnotations-3c.svg");
outs << text;
outs.close();
check_file_hash("testMolAnnotations-3c.svg");
}
}
SECTION("simplified stereo 2") {
auto m = "C[C@H](F)[C@@H](F)[C@@H](C)Cl |o1:3,5,o2:1|"_smiles;
MolDraw2DSVG drawer(350, 300, panelHeight, panelWidth, noFreeType);
drawer.drawOptions().addStereoAnnotation = true;
drawer.drawOptions().simplifiedStereoGroupLabel = true;
MolDraw2DUtils::prepareMolForDrawing(*m);
drawer.drawMolecule(*m, "multi-groups");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMolAnnotations-3d.svg");
outs << text;
outs.close();
check_file_hash("testMolAnnotations-3d.svg");
}
SECTION("label placement") {
auto m = R"CTAB(
Mrv2014 12162004412D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 16 15 0 0 0
M V30 BEGIN ATOM
M V30 1 C -9.2917 3.5833 0 0
M V30 2 C -7.958 4.3533 0 0 CFG=2
M V30 3 C -6.6243 3.5833 0 0 CFG=1
M V30 4 C -5.2906 4.3533 0 0 CFG=2
M V30 5 Cl -7.958 5.8933 0 0
M V30 6 F -6.6243 2.0433 0 0
M V30 7 F -3.957 3.5833 0 0
M V30 8 C -5.2906 5.8933 0 0
M V30 9 C -3.957 6.6633 0 0
M V30 10 C -3.957 8.2033 0 0
M V30 11 C -2.6233 8.9733 0 0
M V30 12 C -2.6233 5.8933 0 0
M V30 13 C -5.2906 8.9733 0 0
M V30 14 C -2.6233 10.5133 0 0
M V30 15 C -1.2896 8.2033 0 0
M V30 16 C -1.2896 6.6633 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 2 3
M V30 3 1 3 4
M V30 4 1 2 5 CFG=1
M V30 5 1 3 6 CFG=1
M V30 6 1 4 7 CFG=1
M V30 7 1 4 8
M V30 8 1 8 9
M V30 9 1 9 10
M V30 10 1 10 11
M V30 11 1 9 12
M V30 12 1 10 13
M V30 13 1 11 14
M V30 14 1 11 15
M V30 15 1 12 16
M V30 END BOND
M V30 BEGIN COLLECTION
M V30 MDLV30/STEREL1 ATOMS=(3 2 3 4)
M V30 END COLLECTION
M V30 END CTAB
M END
)CTAB"_ctab;
MolDraw2DSVG drawer(350, 300, panelHeight, panelWidth, noFreeType);
drawer.drawOptions().addStereoAnnotation = true;
drawer.drawOptions().simplifiedStereoGroupLabel = true;
drawer.drawMolecule(*m, "label crowding");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMolAnnotations-4a.svg");
outs << text;
outs.close();
check_file_hash("testMolAnnotations-4a.svg");
}
}
TEST_CASE("draw link nodes", "[extras]") {
SECTION("one linknode") {
auto m = R"CTAB(one linknode
Mrv2007 06222005102D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 6 6 0 0 0
M V30 BEGIN ATOM
M V30 1 C 8.25 12.1847 0 0
M V30 2 C 6.9164 12.9547 0 0
M V30 3 C 6.9164 14.4947 0 0
M V30 4 C 9.5836 14.4947 0 0
M V30 5 C 9.5836 12.9547 0 0
M V30 6 O 8.25 10.6447 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 2 3
M V30 3 1 4 5
M V30 4 1 1 5
M V30 5 1 3 4
M V30 6 1 1 6
M V30 END BOND
M V30 LINKNODE 1 4 2 1 2 1 5
M V30 END CTAB
M END)CTAB"_ctab;
std::vector<int> rotns = {0, 30, 60, 90, 120, 150, 180};
for (auto rotn : rotns) {
MolDraw2DSVG drawer(350, 300);
drawer.drawOptions().rotate = (double)rotn;
drawer.drawMolecule(*m);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs(
(boost::format("testLinkNodes-1-%d.svg") % rotn).str());
outs << text;
outs.close();
check_file_hash((boost::format("testLinkNodes-1-%d.svg") % rotn).str());
}
}
}
TEST_CASE("Github #3744: Double bonds incorrectly drawn outside the ring",
"[drawing]") {
SECTION("SVG") {
ROMOL_SPTR m1(MolBlockToMol(R"CTAB(
RDKit 2D
6 6 0 0 0 0 0 0 0 0999 V2000
0.0684 -1.2135 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
1.4949 -0.7500 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
1.4949 0.7500 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
0.0684 1.2135 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
-0.8133 0.0000 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
-2.3133 -0.0000 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
1 2 2 0
2 3 1 0
3 4 2 0
4 5 1 0
5 6 1 0
5 1 1 0
M END)CTAB"));
REQUIRE(m1);
MolDraw2DSVG drawer(400, 300);
drawer.drawMolecule(*m1);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub3744.svg");
outs << text;
outs.close();
check_file_hash("testGithub3744.svg");
std::vector<std::string> bond0;
std::vector<std::string> bond2;
std::istringstream ss(text);
std::string line;
while (std::getline(ss, line)) {
if (line.find("bond-0") != std::string::npos) {
bond0.push_back(line);
} else if (line.find("bond-2") != std::string::npos) {
bond2.push_back(line);
}
}
CHECK(bond0.size() == 2);
CHECK(bond2.size() == 2);
std::regex regex(
"^.*d='M\\s+(\\d+\\.\\d+),(\\d+\\.\\d+)\\s+L\\s+(\\d+\\.\\d+),(\\d+\\."
"\\d+)'.*$");
std::smatch bond0OuterMatch;
REQUIRE(std::regex_match(bond0[0], bond0OuterMatch, regex));
REQUIRE(bond0OuterMatch.size() == 5);
std::smatch bond0InnerMatch;
REQUIRE(std::regex_match(bond0[1], bond0InnerMatch, regex));
REQUIRE(bond0InnerMatch.size() == 5);
std::smatch bond2OuterMatch;
REQUIRE(std::regex_match(bond2[0], bond2OuterMatch, regex));
REQUIRE(bond2OuterMatch.size() == 5);
std::smatch bond2InnerMatch;
REQUIRE(std::regex_match(bond2[1], bond2InnerMatch, regex));
REQUIRE(bond2InnerMatch.size() == 5);
RDGeom::Point2D bond0InnerCtd(
RDGeom::Point2D(std::stof(bond0InnerMatch[1]),
std::stof(bond0InnerMatch[2])) +
RDGeom::Point2D(std::stof(bond0InnerMatch[3]),
std::stof(bond0InnerMatch[4])) /
2.0);
RDGeom::Point2D bond0OuterCtd(
RDGeom::Point2D(std::stof(bond0OuterMatch[1]),
std::stof(bond0OuterMatch[2])) +
RDGeom::Point2D(std::stof(bond0OuterMatch[3]),
std::stof(bond0OuterMatch[4])) /
2.0);
RDGeom::Point2D bond2InnerCtd(
RDGeom::Point2D(std::stof(bond2InnerMatch[1]),
std::stof(bond2InnerMatch[2])) +
RDGeom::Point2D(std::stof(bond2InnerMatch[3]),
std::stof(bond2InnerMatch[4])) /
2.0);
RDGeom::Point2D bond2OuterCtd(
RDGeom::Point2D(std::stof(bond2OuterMatch[1]),
std::stof(bond2OuterMatch[2])) +
RDGeom::Point2D(std::stof(bond2OuterMatch[3]),
std::stof(bond2OuterMatch[4])) /
2.0);
// we look at the two double bonds of pyrrole
// we check that the ratio between the distance of the centroids of the
// outer bonds and the distance of the centroids of the inner bonds is at
// least 1.3, otherwise the inner bonds are not actually inside the ring.
float outerBondsDistance = (bond0OuterCtd - bond2OuterCtd).length();
float innerBondsDistance = (bond0InnerCtd - bond2InnerCtd).length();
CHECK(outerBondsDistance / innerBondsDistance > 1.3f);
}
}
TEST_CASE("draw atom list queries", "[extras]") {
SECTION("atom list") {
auto m = R"CTAB(
Mrv2102 02112115002D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 3 3 0 0 0
M V30 BEGIN ATOM
M V30 1 [N,O,S] 9.2083 12.8058 0 0
M V30 2 C 8.4383 11.4721 0 0
M V30 3 C 9.9783 11.4721 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 3 1
M V30 3 1 2 3
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m, "atom list");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testAtomLists-1.svg");
outs << text;
outs.close();
check_file_hash("testAtomLists-1.svg");
}
SECTION("NOT atom list") {
auto m = R"CTAB(
Mrv2102 02112115032D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 3 3 0 0 0
M V30 BEGIN ATOM
M V30 1 "NOT [N,O,S]" 9.2083 12.8058 0 0
M V30 2 C 8.4383 11.4721 0 0
M V30 3 C 9.9783 11.4721 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 3 1
M V30 3 1 2 3
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
MolDraw2DSVG drawer(350, 300);
drawer.drawMolecule(*m, "NOT atom list");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testAtomLists-2.svg");
outs << text;
outs.close();
check_file_hash("testAtomLists-2.svg");
}
}
TEST_CASE("test the options that toggle isotope labels", "[drawing]") {
SECTION("test all permutations") {
auto m = "[1*]c1cc([2*])c([3*])c[14c]1"_smiles;
REQUIRE(m);
std::regex regex(R"regex(<text\s+.*>\d</text>)regex");
std::smatch match;
std::string line;
{
MolDraw2DSVG drawer(300, 300, -1, -1, true);
drawer.drawMolecule(*m);
drawer.finishDrawing();
std::string textIsoDummyIso = drawer.getDrawingText();
std::ofstream outs("testIsoDummyIso.svg");
outs << textIsoDummyIso;
outs.close();
check_file_hash("testIsoDummyIso.svg");
size_t nIsoDummyIso = std::distance(
std::sregex_token_iterator(textIsoDummyIso.begin(),
textIsoDummyIso.end(), regex),
std::sregex_token_iterator());
CHECK(nIsoDummyIso == 5);
}
{
MolDraw2DSVG drawer(300, 300, -1, -1, true);
drawer.drawOptions().isotopeLabels = false;
drawer.drawMolecule(*m);
drawer.finishDrawing();
std::string textNoIsoDummyIso = drawer.getDrawingText();
std::ofstream outs("testNoIsoDummyIso.svg");
outs << textNoIsoDummyIso;
outs.close();
check_file_hash("testNoIsoDummyIso.svg");
size_t nNoIsoDummyIso = std::distance(
std::sregex_token_iterator(textNoIsoDummyIso.begin(),
textNoIsoDummyIso.end(), regex, 1),
std::sregex_token_iterator());
CHECK(nNoIsoDummyIso == 3);
}
{
MolDraw2DSVG drawer(300, 300, -1, -1, true);
drawer.drawOptions().dummyIsotopeLabels = false;
drawer.drawMolecule(*m);
drawer.finishDrawing();
std::string textIsoNoDummyIso = drawer.getDrawingText();
std::ofstream outs("testIsoNoDummyIso.svg");
outs << textIsoNoDummyIso;
outs.close();
check_file_hash("testIsoNoDummyIso.svg");
size_t nIsoNoDummyIso = std::distance(
std::sregex_token_iterator(textIsoNoDummyIso.begin(),
textIsoNoDummyIso.end(), regex, 1),
std::sregex_token_iterator());
CHECK(nIsoNoDummyIso == 2);
}
{
MolDraw2DSVG drawer(300, 300, -1, -1, true);
drawer.drawOptions().isotopeLabels = false;
drawer.drawOptions().dummyIsotopeLabels = false;
drawer.drawMolecule(*m);
drawer.finishDrawing();
std::string textNoIsoNoDummyIso = drawer.getDrawingText();
std::ofstream outs("testNoIsoNoDummyIso.svg");
outs << textNoIsoNoDummyIso;
outs.close();
check_file_hash("testNoIsoNoDummyIso.svg");
size_t nNoIsoNoDummyIso = std::distance(
std::sregex_token_iterator(textNoIsoNoDummyIso.begin(),
textNoIsoNoDummyIso.end(), regex, 1),
std::sregex_token_iterator());
CHECK(nNoIsoNoDummyIso == 0);
}
}
SECTION("test that D/T show up even if isotope labels are hidden") {
auto m = "C([1H])([2H])([3H])[H]"_smiles;
std::regex regex(R"regex(<text\s+.*>[DT]</text>)regex");
std::smatch match;
REQUIRE(m);
std::string line;
MolDraw2DSVG drawer(300, 300, -1, -1, true);
drawer.drawOptions().isotopeLabels = false;
drawer.drawOptions().dummyIsotopeLabels = false;
drawer.drawOptions().atomLabelDeuteriumTritium = true;
drawer.drawMolecule(*m);
drawer.finishDrawing();
std::string textDeuteriumTritium = drawer.getDrawingText();
std::ofstream outs("testDeuteriumTritium.svg");
outs << textDeuteriumTritium;
outs.close();
check_file_hash("testDeuteriumTritium.svg");
size_t nDeuteriumTritium = std::distance(
std::sregex_token_iterator(textDeuteriumTritium.begin(),
textDeuteriumTritium.end(), regex, 1),
std::sregex_token_iterator());
CHECK(nDeuteriumTritium == 2);
}
}
TEST_CASE("draw hydrogen bonds", "[drawing]") {
SECTION("basics") {
auto m = R"CTAB(
Mrv2014 03022114422D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 8 8 0 0 0
M V30 BEGIN ATOM
M V30 1 C -5.4583 -0.125 0 0
M V30 2 C -4.1247 0.645 0 0
M V30 3 C -2.791 -0.125 0 0
M V30 4 C -1.4573 0.645 0 0
M V30 5 O -2.791 -1.665 0 0
M V30 6 C -6.792 0.645 0 0
M V30 7 O -5.4583 -1.665 0 0
M V30 8 H -4.1247 -2.435 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 2 3
M V30 3 1 3 4
M V30 4 2 3 5
M V30 5 1 1 6
M V30 6 1 1 7
M V30 7 1 7 8
M V30 8 10 5 8
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(m);
MolDraw2DSVG drawer(300, 300);
drawer.drawMolecule(*m);
drawer.finishDrawing();
std::ofstream outs("testHydrogenBonds1.svg");
outs << drawer.getDrawingText();
outs.close();
check_file_hash("testHydrogenBonds1.svg");
}
SECTION("from CXSMILES") {
auto m = "CC1O[H]O=C(C)C1 |H:4.3|"_smiles;
REQUIRE(m);
MolDraw2DSVG drawer(300, 300);
drawer.drawMolecule(*m);
drawer.finishDrawing();
std::ofstream outs("testHydrogenBonds2.svg");
outs << drawer.getDrawingText();
outs.close();
check_file_hash("testHydrogenBonds2.svg");
}
}
TEST_CASE("github #3912: cannot draw atom lists from SMARTS", "[query][bug]") {
SECTION("original") {
auto m = "C-[N,O]"_smarts;
REQUIRE(m);
int panelWidth = -1;
int panelHeight = -1;
bool noFreeType = true;
MolDraw2DSVG drawer(300, 300, panelWidth, panelHeight, noFreeType);
drawer.drawMolecule(*m);
drawer.finishDrawing();
std::ofstream outs("testGithub3912.1.svg");
auto txt = drawer.getDrawingText();
outs << txt;
outs.close();
check_file_hash("testGithub3912.1.svg");
CHECK(txt.find(">N<") != std::string::npos);
CHECK(txt.find(">O<") != std::string::npos);
CHECK(txt.find(">!<") == std::string::npos);
}
SECTION("negated") {
auto m = "C-[N,O]"_smarts;
REQUIRE(m);
REQUIRE(m->getAtomWithIdx(1)->hasQuery());
m->getAtomWithIdx(1)->getQuery()->setNegation(true);
int panelWidth = -1;
int panelHeight = -1;
bool noFreeType = true;
MolDraw2DSVG drawer(300, 300, panelWidth, panelHeight, noFreeType);
drawer.drawMolecule(*m);
drawer.finishDrawing();
std::ofstream outs("testGithub3912.2.svg");
auto txt = drawer.getDrawingText();
outs << txt;
outs.close();
check_file_hash("testGithub3912.2.svg");
CHECK(txt.find(">N<") != std::string::npos);
CHECK(txt.find(">O<") != std::string::npos);
CHECK(txt.find(">!<") != std::string::npos);
}
}
TEST_CASE("github #2976: kekulizing reactions when drawing", "[reactions]") {
SECTION("basics") {
bool asSmiles = true;
std::unique_ptr<ChemicalReaction> rxn{
RxnSmartsToChemicalReaction("c1ccccc1>>c1ncccc1", nullptr, asSmiles)};
MolDraw2DSVG drawer(450, 200);
drawer.drawReaction(*rxn);
drawer.finishDrawing();
std::ofstream outs("testGithub2976.svg");
auto txt = drawer.getDrawingText();
outs << txt;
outs.close();
check_file_hash("testGithub2976.svg");
}
}
TEST_CASE("preserve Reaction coordinates", "[reactions]") {
SECTION("basics") {
std::string data = R"RXN($RXN
Mrv16822 031301211645
2 2 1
$MOL
Mrv1682203132116452D
3 2 0 0 0 0 999 V2000
-4.3304 2.5893 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
-4.3304 1.7643 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
-3.5054 1.7643 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
1 2 1 0 0 0 0
2 3 1 0 0 0 0
M END
$MOL
Mrv1682203132116452D
2 1 0 0 0 0 999 V2000
-2.1652 2.6339 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
-2.1652 1.8089 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
1 2 1 0 0 0 0
M END
$MOL
Mrv1682203132116452D
3 2 0 0 0 0 999 V2000
3.6109 1.9512 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
2.7859 1.9512 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
2.7859 2.7762 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
2 1 1 0 0 0 0
3 2 1 0 0 0 0
M END
$MOL
Mrv1682203132116452D
2 1 0 0 0 0 999 V2000
4.9511 1.9959 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
4.9511 2.8209 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
2 1 1 0 0 0 0
M END
$MOL
Mrv1682203132116452D
2 1 0 0 0 0 999 V2000
-0.3571 2.7232 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
-0.4003 3.5471 0.0000 O 0 0 0 0 0 0 0 0 0 0 0 0
1 2 1 0 0 0 0
M END
)RXN";
std::unique_ptr<ChemicalReaction> rxn{RxnBlockToChemicalReaction(data)};
MolDraw2DSVG drawer(450, 200);
drawer.drawReaction(*rxn);
drawer.finishDrawing();
std::ofstream outs("testReactionCoords.svg");
auto txt = drawer.getDrawingText();
outs << txt;
outs.close();
check_file_hash("testReactionCoords.svg");
// the reaction is drawn with some bonds vertical, make sure they remain
// vertical
{
std::regex regex("class='bond-0.*? d='M (\\d+\\.\\d+).* L (\\d+\\.\\d+)");
std::smatch bondMatch;
REQUIRE(std::regex_search(txt, bondMatch, regex));
REQUIRE(bondMatch.size() == 3); // match both halves of the bond
CHECK(bondMatch[1].str() == bondMatch[2].str());
}
{
std::regex regex("class='bond-2.*? d='M (\\d+\\.\\d+).* L (\\d+\\.\\d+)");
std::smatch bondMatch;
REQUIRE(std::regex_search(txt, bondMatch, regex));
REQUIRE(bondMatch.size() == 3); // match both halves of the bond
CHECK(bondMatch[1].str() == bondMatch[2].str());
}
{
std::regex regex("class='bond-4.*? d='M (\\d+\\.\\d+).* L (\\d+\\.\\d+)");
std::smatch bondMatch;
REQUIRE(std::regex_search(txt, bondMatch, regex));
REQUIRE(bondMatch.size() == 3); // match both halves of the bond
CHECK(bondMatch[1].str() == bondMatch[2].str());
}
}
}
TEST_CASE("support annotation colors", "[drawing]") {
SECTION("basics") {
auto m = "CCCO"_smiles;
REQUIRE(m);
int panelWidth = -1;
int panelHeight = -1;
bool noFreeType = true;
MolDraw2DSVG drawer(300, 300, panelWidth, panelHeight, noFreeType);
drawer.drawOptions().annotationColour = DrawColour{0, 0, 1, 1};
drawer.drawOptions().addAtomIndices = true;
drawer.drawMolecule(*m, "blue annotations");
drawer.finishDrawing();
std::ofstream outs("testAnnotationColors.svg");
auto txt = drawer.getDrawingText();
outs << txt;
outs.close();
check_file_hash("testAnnotationColors.svg");
CHECK(txt.find("fill:#0000FF' >2<") != std::string::npos);
}
}
TEST_CASE("Github #4238: prepareMolForDrawing and wavy bonds") {
{
auto mol = "CC=CC"_smiles;
REQUIRE(mol);
mol->getBondWithIdx(1)->setStereoAtoms(0, 3);
mol->getBondWithIdx(1)->setStereo(Bond::BondStereo::STEREOANY);
bool kekulize = true;
bool addChiralHs = true;
bool wedgeBonds = true;
bool forceCoords = true;
bool wavyBonds = false;
MolDraw2DUtils::prepareMolForDrawing(*mol, kekulize, addChiralHs,
wedgeBonds, forceCoords, wavyBonds);
CHECK(mol->getBondWithIdx(0)->getBondDir() == Bond::BondDir::NONE);
CHECK(mol->getBondWithIdx(1)->getStereo() == Bond::BondStereo::STEREOANY);
RWMol mol2(*mol);
wavyBonds = true;
MolDraw2DUtils::prepareMolForDrawing(mol2, kekulize, addChiralHs,
wedgeBonds, forceCoords, wavyBonds);
CHECK(mol2.getBondWithIdx(0)->getBondDir() == Bond::BondDir::UNKNOWN);
CHECK(mol2.getBondWithIdx(1)->getStereo() == Bond::BondStereo::STEREONONE);
MolDraw2DSVG drawer(500, 200, 250, 200);
// drawer.drawOptions().prepareMolsBeforeDrawing = false;
MOL_PTR_VECT ms{mol.get(), &mol2};
std::vector<std::string> legends = {"before", "after"};
drawer.drawMolecules(ms, &legends);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4238_1.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4238_1.svg");
}
}
TEST_CASE("Github #4323: support providing RGBA colors") {
auto mol = "CCCO"_smiles;
REQUIRE(mol);
#ifdef RDK_BUILD_FREETYPE_SUPPORT
SECTION("with alpha") {
MolDraw2DSVG drawer(200, 150);
drawer.drawOptions().legendColour = DrawColour(1, 0, 1, 0.3);
drawer.drawOptions().backgroundColour = DrawColour(0.5, 0.5, 0.5, 0.3);
drawer.drawMolecule(*mol, "partially transparent legend/background");
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4323_1.svg");
outs << text;
outs.flush();
// background
CHECK(text.find("fill:#7F7F7F4C;") != std::string::npos);
CHECK(text.find("fill:#7F7F7F;") == std::string::npos);
// legend
CHECK(text.find("fill='#FF00FF4C'") != std::string::npos);
CHECK(text.find("fill='#FF00FF'") == std::string::npos);
check_file_hash("testGithub4323_1.svg");
}
SECTION("without alpha") {
MolDraw2DSVG drawer(200, 150);
drawer.drawOptions().legendColour = DrawColour(1, 0, 1);
drawer.drawOptions().backgroundColour = DrawColour(0.5, 0.5, 0.5);
drawer.drawMolecule(*mol, "no transparency");
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4323_2.svg");
outs << text;
outs.flush();
// background
CHECK(text.find("fill:#7F7F7F4C;") == std::string::npos);
CHECK(text.find("fill:#7F7F7F;") != std::string::npos);
// legend
CHECK(text.find("fill='#FF00FF4C'") == std::string::npos);
CHECK(text.find("fill='#FF00FF'") != std::string::npos);
check_file_hash("testGithub4323_2.svg");
}
#endif
SECTION("no FT with alpha") {
MolDraw2DSVG drawer(200, 150, -1, -1, NO_FREETYPE);
drawer.drawOptions().legendColour = DrawColour(1, 0, 1, 0.3);
drawer.drawOptions().backgroundColour = DrawColour(0.5, 0.5, 0.5, 0.3);
drawer.drawMolecule(*mol, "partially transparent legend/background");
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4323_3.svg");
outs << text;
outs.flush();
// background
CHECK(text.find("fill:#7F7F7F4C;") != std::string::npos);
CHECK(text.find("fill:#7F7F7F;") == std::string::npos);
// legend
CHECK(text.find("fill:#FF00FF4C'") != std::string::npos);
CHECK(text.find("fill:#FF00FF'") == std::string::npos);
check_file_hash("testGithub4323_3.svg");
}
SECTION("no FT without alpha") {
MolDraw2DSVG drawer(200, 150, -1, -1, NO_FREETYPE);
drawer.drawOptions().legendColour = DrawColour(1, 0, 1);
drawer.drawOptions().backgroundColour = DrawColour(0.5, 0.5, 0.5);
drawer.drawMolecule(*mol, "no transparency");
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4323_4.svg");
outs << text;
outs.flush();
// background
CHECK(text.find("fill:#7F7F7F4C;") == std::string::npos);
CHECK(text.find("fill:#7F7F7F;") != std::string::npos);
// legend
CHECK(text.find("fill:#FF00FF4C'") == std::string::npos);
CHECK(text.find("fill:#FF00FF'") != std::string::npos);
check_file_hash("testGithub4323_4.svg");
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
#ifdef RDK_BUILD_FREETYPE_SUPPORT
SECTION("Cairo with alpha") {
MolDraw2DCairo drawer(200, 150);
drawer.drawOptions().legendColour = DrawColour(1, 0, 1, 0.3);
drawer.drawOptions().backgroundColour = DrawColour(0.5, 0.5, 0.5, 0.3);
drawer.drawMolecule(*mol, "partially transparent legend/background");
drawer.finishDrawing();
drawer.writeDrawingText("testGithub4323_1.png");
check_file_hash("testGithub4323_1.png");
}
#endif
SECTION("No FT Cairo with alpha") {
MolDraw2DCairo drawer(200, 150, -1, -1, NO_FREETYPE);
drawer.drawOptions().legendColour = DrawColour(1, 0, 1, 0.3);
drawer.drawOptions().backgroundColour = DrawColour(0.5, 0.5, 0.5, 0.3);
drawer.drawMolecule(*mol, "partially transparent legend/background");
drawer.finishDrawing();
drawer.writeDrawingText("testGithub4323_3.png");
check_file_hash("testGithub4323_3.png");
}
#endif
}
TEST_CASE(
"Github #4508: SubstanceGroup labels sometimes overlap with atoms in image "
"generation") {
SECTION("Basics") {
auto mol = R"CTAB(
Mrv2114 09132120172D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 8 8 1 0 1
M V30 BEGIN ATOM
M V30 1 C -0.5878 0.8085 0 0
M V30 2 C -1.9434 0.078 0 0
M V30 3 C -1.9884 -1.4614 0 0
M V30 4 C -0.6778 -2.2702 0 0
M V30 5 C 0.6778 -1.5394 0 0
M V30 6 C 0.7228 -0.0001 0 0
M V30 7 N -0.5428 2.3478 0 0
M V30 8 O 1.9884 -2.3479 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 2 1 2
M V30 2 1 2 3
M V30 3 2 3 4
M V30 4 1 4 5
M V30 5 2 5 6
M V30 6 1 6 1
M V30 7 1 1 7
M V30 8 1 5 8
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 DAT 0 ATOMS=(1 7) FIELDNAME=UV FIELDINFO=nm -
M V30 FIELDDISP=" 0.0000 0.0000 DRU ALL 0 0" -
M V30 MRV_FIELDDISP=0 FIELDDATA=340
M V30 END SGROUP
M V30 END CTAB
M END)CTAB"_ctab;
REQUIRE(mol);
{
MolDraw2DSVG drawer(300, 250);
drawer.drawMolecule(*mol, "data label with DRU");
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4508_1.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4508_1.svg");
}
// remove the sgroup-atom atom... the SGroup will not be drawn
auto &sgs = getSubstanceGroups(*mol);
REQUIRE(sgs.size() == 1);
sgs[0].setAtoms(std::vector<unsigned int>());
{
MolDraw2DSVG drawer(300, 250);
drawer.drawMolecule(*mol, "no data label drawn");
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4508_1b.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4508_1b.svg");
}
}
SECTION("Absolute") {
auto mol = R"CTAB(
Mrv2114 09132120172D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 8 8 1 0 1
M V30 BEGIN ATOM
M V30 1 C -0.5878 0.8085 0 0
M V30 2 C -1.9434 0.078 0 0
M V30 3 C -1.9884 -1.4614 0 0
M V30 4 C -0.6778 -2.2702 0 0
M V30 5 C 0.6778 -1.5394 0 0
M V30 6 C 0.7228 -0.0001 0 0
M V30 7 N -0.5428 2.3478 0 0
M V30 8 O 1.9884 -2.3479 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 2 1 2
M V30 2 1 2 3
M V30 3 2 3 4
M V30 4 1 4 5
M V30 5 2 5 6
M V30 6 1 6 1
M V30 7 1 1 7
M V30 8 1 5 8
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 DAT 0 ATOMS=(1 7) FIELDNAME=UV FIELDINFO=nm -
M V30 FIELDDISP=" 0.0000 0.0000 DAU ALL 0 0" -
M V30 MRV_FIELDDISP=0 FIELDDATA=340
M V30 END SGROUP
M V30 END CTAB
M END)CTAB"_ctab;
REQUIRE(mol);
{
MolDraw2DSVG drawer(300, 250);
drawer.drawMolecule(*mol, "data label with DAU\n(expect odd placement)");
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4508_2.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4508_2.svg");
}
// remove the sgroup-atom atom... the SGroup will still be drawn
auto &sgs = getSubstanceGroups(*mol);
REQUIRE(sgs.size() == 1);
sgs[0].setAtoms(std::vector<unsigned int>());
{
MolDraw2DSVG drawer(300, 250);
drawer.drawMolecule(*mol,
"DAU, no associated atom\n(expect odd placement)");
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4508_2b.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4508_2b.svg");
}
}
}
TEST_CASE("Github #4538 drawMolecules crash") {
auto m = "CCc1ccccc1"_smiles;
REQUIRE(m);
RDDepict::compute2DCoords(*m);
ROMol m1(*m);
ROMol m2(*m);
std::vector<ROMol *> mols{&m1, &m2};
SECTION("basics") {
MolDraw2DSVG drawer(500, 200, 250, 200);
drawer.drawOptions().prepareMolsBeforeDrawing = false;
drawer.drawMolecules(mols);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testGithub4538.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4538.svg");
}
}
TEST_CASE("dark mode mol drawing") {
SECTION("Basics") {
auto m =
"CS(=O)(=O)COC(=N)c1cc(Cl)cnc1[NH3+] |SgD:7:note:some extra text:=:::|"_smiles;
REQUIRE(m);
MolDraw2DSVG drawer(350, 300);
setDarkMode(drawer);
drawer.drawMolecule(*m, "dark mode!");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testDarkMode.1.svg");
outs << text;
outs.flush();
check_file_hash("testDarkMode.1.svg");
}
}
TEST_CASE("monochrome mol drawing") {
SECTION("Basics") {
auto m =
"CS(=O)(=O)COC(=N)c1cc(Cl)cnc1[NH3+] |SgD:7:note:some extra text:=:::|"_smiles;
REQUIRE(m);
MolDraw2DSVG drawer(350, 300);
setMonochromeMode(drawer, DrawColour{0.1, 0.1, 0.6},
DrawColour{0.75, 0.75, 0.75});
drawer.drawMolecule(*m, "monochrome");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMonochrome.1.svg");
outs << text;
outs.flush();
check_file_hash("testMonochrome.1.svg");
}
SECTION("Basics inverted") {
auto m =
"CS(=O)(=O)COC(=N)c1cc(Cl)cnc1[NH3+] |SgD:7:note:some extra text:=:::|"_smiles;
REQUIRE(m);
MolDraw2DSVG drawer(350, 300);
setMonochromeMode(drawer, DrawColour{0.75, 0.75, 0.75},
DrawColour{0.1, 0.1, 0.6});
drawer.drawMolecule(*m, "monochrome");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMonochrome.2.svg");
outs << text;
outs.flush();
check_file_hash("testMonochrome.2.svg");
}
}
TEST_CASE("other palettes") {
auto m =
"CS(=O)(=O)COC(=N)c1c(I)c(Cl)c(Br)nc1[NH2+]CP(=O) |SgD:7:note:some extra text:=:::|"_smiles;
REQUIRE(m);
SECTION("Avalon") {
MolDraw2DSVG drawer(350, 300);
assignAvalonPalette(drawer.drawOptions().atomColourPalette);
drawer.drawMolecule(*m, "Avalon");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testAvalon.1.svg");
outs << text;
outs.flush();
check_file_hash("testAvalon.1.svg");
}
SECTION("CDK") {
MolDraw2DSVG drawer(350, 300);
assignCDKPalette(drawer.drawOptions().atomColourPalette);
drawer.drawMolecule(*m, "CDK");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testCDK.1.svg");
outs << text;
outs.flush();
check_file_hash("testCDK.1.svg");
}
}
TEST_CASE("SDD record parsing") {
auto mol = R"CTAB(
Mrv2008 11122110292D
6 6 0 0 0 0 999 V2000
9.3527 2.5661 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
8.6382 2.1536 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
8.6382 1.3286 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
9.3527 0.9161 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
10.0671 1.3286 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
10.0671 2.1536 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
1 2 1 0 0 0 0
2 3 2 0 0 0 0
3 4 1 0 0 0 0
4 5 2 0 0 0 0
5 6 1 0 0 0 0
1 6 2 0 0 0 0
M STY 1 1 DAT
M SLB 1 1 1
M SAL 1 1 1
M SDT 1 NAME
M SDD 1 -2345.1234-2345.1234 DR ALL 1 0
M SED 1 Hello World
M END
)CTAB"_ctab;
// SDD record has format
// M SDD sss xxxxx.xxxxyyyyy.yyyy eeefgh i jjjkkk ll m noo
MolDraw2DSVG drawer(350, 300, -1, -1, 1);
drawer.drawMolecule(*mol);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::string name("Hello World");
for (auto &c : name) {
std::stringstream ss;
ss << " >" << c << "</text>";
auto pos = text.find(ss.str());
CHECK(pos != std::string::npos);
}
}
TEST_CASE("Github #4519 bad placement of datafield labels") {
auto mol1 = R"CTAB(
RDKit 2D
0 0 0 0 0 0 0 0 0 0999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 5 4 1 0 0
M V30 BEGIN ATOM
M V30 1 C 0.000000 0.000000 0.000000 0
M V30 2 C 1.299038 0.750000 0.000000 0
M V30 3 C 2.598076 -0.000000 0.000000 0
M V30 4 C 1.299038 2.250000 0.000000 0
M V30 5 C 2.598076 3.000000 0.000000 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 2 2 3
M V30 3 1 2 4
M V30 4 2 4 5
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 DAT 0 ATOMS=(5 2 4 5 3 1) FIELDNAME="Lambda Max" FIELDINFO=nm -
M V30 FIELDDATA="2222"
M V30 END SGROUP
M V30 END CTAB
M END)CTAB"_ctab;
REQUIRE(mol1);
auto mol2 = R"CTAB(
RDKit 2D
0 0 0 0 0 0 0 0 0 0999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 8 8 1 0 0
M V30 BEGIN ATOM
M V30 1 N 3.000000 0.000000 0.000000 0
M V30 2 C 1.500000 0.000000 0.000000 0
M V30 3 C 0.750000 -1.299038 0.000000 0
M V30 4 C -0.750000 -1.299038 0.000000 0
M V30 5 C -1.500000 0.000000 0.000000 0
M V30 6 C -0.750000 1.299038 0.000000 0
M V30 7 O -1.500000 2.598076 0.000000 0
M V30 8 C 0.750000 1.299038 0.000000 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 2 2 3
M V30 3 1 3 4
M V30 4 2 4 5
M V30 5 1 5 6
M V30 6 1 6 7
M V30 7 2 6 8
M V30 8 1 8 2
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 DAT 0 ATOMS=(1 1) FIELDNAME=UV FIELDINFO=nm -
M V30 FIELDDISP=" 0.0000 0.0000 DR ALL 0 0" -
M V30 FIELDDATA="340"
M V30 END SGROUP
M V30 END CTAB
M END)CTAB"_ctab;
REQUIRE(mol2);
auto mol3 = R"CTAB(
RDKit 2D
0 0 0 0 0 0 0 0 0 0999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 4 3 1 0 0
M V30 BEGIN ATOM
M V30 1 C -0.750000 -1.299038 0.000000 0
M V30 2 C 0.000000 0.000000 0.000000 0
M V30 3 C 1.500000 0.000000 0.000000 0
M V30 4 C 2.250000 1.299038 0.000000 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 2 2 3
M V30 3 1 3 4
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 DAT 0 ATOMS=(1 3) FIELDNAME=Stereo -
M V30 FIELDDATA="Cis"
M V30 END SGROUP
M V30 END CTAB
M END)CTAB"_ctab;
REQUIRE(mol3);
std::vector<std::string> legends = {
"datafield label bad placement1", "datafield label bad placement2",
"datafield label bad placement3"}; // std::vector<std::string> legends =
// {"datafield label bad
// placement2"};
{
MolDraw2DSVG drawer(300, 250);
drawer.drawMolecule(*mol1, legends[0]);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4519_1.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4519_1.svg");
}
{
MolDraw2DSVG drawer(300, 250);
drawer.drawMolecule(*mol2, legends[1]);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4519_2.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4519_2.svg");
}
{
MolDraw2DSVG drawer(300, 250);
drawer.drawMolecule(*mol3, legends[2]);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4519_3.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4519_3.svg");
}
{
std::vector<ROMol *> mols;
mols.push_back(mol1.get());
mols.push_back(mol2.get());
mols.push_back(mol3.get());
MolDraw2DSVG drawer(900, 250, 300, 250);
drawer.drawMolecules(mols, &legends);
drawer.finishDrawing();
std::string text = drawer.getDrawingText();
std::ofstream outs("testGithub4519_4.svg");
outs << text;
outs.flush();
outs.close();
check_file_hash("testGithub4519_4.svg");
}
}
TEST_CASE("changing baseFontSize") {
RDDepict::preferCoordGen = false;
auto mol1 =
"CC(C)C[C@H](NC(=O)[C@H](CCCCN)NC(=O)[C@H](CS)NC(=O)CNC(=O)[C@H](C)NC(=O)[C@H](CCCCN)NC(=O)[C@H](CC(C)C)NC(=O)CNC(=O)[C@H](C)NC(=O)[C@H](CS)NC(=O)[C@H](CCCCN)NC(=O)[C@H](C)NC(=O)[C@@H](NC(=O)[C@H](CS)NC(=O)CNC(=O)[C@H](C)NC(=O)[C@H](CCCCN)NC(=O)CNC(=O)[C@H](C)NC(=O)[C@H](CCCCN)NC(=O)[C@H](C)N)[C@@H](C)O)C(=O)O"_smiles;
REQUIRE(mol1);
MolDraw2DUtils::prepareMolForDrawing(*mol1);
auto mol2 = "C[C@H](N)C(=O)N[C@@H](CCCCN)C(=O)N[C@@H](C)C(=O)NCC(=O)O"_smiles;
REQUIRE(mol2);
MolDraw2DUtils::prepareMolForDrawing(*mol2);
SECTION("basics-large") {
MolDraw2DSVG drawer(350, 300, -1, -1, 1);
drawer.drawMolecule(*mol1);
drawer.finishDrawing();
CHECK(drawer.fontSize() == Approx(6.0).margin(0.1));
auto text = drawer.getDrawingText();
std::ofstream outs("testBaseFontSize.1a.svg");
outs << text;
outs.flush();
check_file_hash("testBaseFontSize.1a.svg");
}
SECTION("increase size - large") {
// here we change the base font size, but it doesn't matter since the
// structure is big enough we end up stuck with the minimum font size.
MolDraw2DSVG drawer(350, 300, -1, -1, 1);
drawer.drawOptions().baseFontSize = 0.9;
drawer.drawMolecule(*mol1);
drawer.finishDrawing();
CHECK(drawer.fontSize() == Approx(5.5).margin(.1));
auto text = drawer.getDrawingText();
std::ofstream outs("testBaseFontSize.1b.svg");
outs << text;
outs.flush();
check_file_hash("testBaseFontSize.1b.svg");
}
SECTION("basics-small") {
MolDraw2DSVG drawer(350, 300, -1, -1, 1);
drawer.drawMolecule(*mol2);
drawer.finishDrawing();
CHECK(drawer.fontSize() == Approx(14.0).margin(0.1));
auto text = drawer.getDrawingText();
std::ofstream outs("testBaseFontSize.2a.svg");
outs << text;
outs.flush();
check_file_hash("testBaseFontSize.2a.svg");
}
SECTION("increase size - smaller") {
MolDraw2DSVG drawer(350, 300, -1, -1, 1);
drawer.drawOptions().baseFontSize = 0.9;
drawer.drawMolecule(*mol2);
drawer.finishDrawing();
CHECK(drawer.fontSize() == Approx(20.4).margin(0.1));
auto text = drawer.getDrawingText();
std::ofstream outs("testBaseFontSize.2b.svg");
outs << text;
outs.flush();
check_file_hash("testBaseFontSize.2b.svg");
}
}
TEST_CASE("flexicanvas: set canvas size automatically") {
// note that these examples use Freetype if it's available.
auto mol1 = "CCN(CC)CCn1nc2c3ccccc3sc3c(CNS(C)(=O)=O)ccc1c32"_smiles;
REQUIRE(mol1);
MolDraw2DUtils::prepareMolForDrawing(*mol1);
auto mol2 = R"CTAB(
Mrv2108 11192104292D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 5 5 0 0 0
M V30 BEGIN ATOM
M V30 1 C -5.2 -1.4 0 0
M V30 2 O -5.2 -2.8 0 0
M V30 3 C -3.7 -1.4 0 0
M V30 4 C -3.7 -2.8 0 0 CFG=1
M V30 5 N -2.5994 -3.9839 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 1 3
M V30 3 1 2 4
M V30 4 1 3 4
M V30 5 1 4 5 CFG=1
M V30 END BOND
M V30 END CTAB
M END
)CTAB"_ctab;
REQUIRE(mol2);
MolDraw2DUtils::prepareMolForDrawing(*mol2);
SECTION("fixed canvas") {
MolDraw2DSVG drawer(308, 223, -1, -1);
drawer.drawMolecule(*mol1);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.1a.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.1a.svg");
}
SECTION("flexicanvas1") {
MolDraw2DSVG drawer(-1, -1, -1, -1);
drawer.drawMolecule(*mol1);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.1b.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.1b.svg");
}
SECTION("flexicanvas1") {
MolDraw2DSVG drawer(-1, -1, -1, -1);
drawer.drawOptions().scalingFactor = 30;
drawer.drawOptions().baseFontSize = 0.6;
drawer.drawMolecule(*mol1);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.1c.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.1c.svg");
}
SECTION("flexicanvas1") {
MolDraw2DSVG drawer(-1, -1, -1, -1);
drawer.drawOptions().scalingFactor = 30;
drawer.drawOptions().fixedFontSize = 32;
drawer.drawMolecule(*mol1);
drawer.finishDrawing();
CHECK(drawer.fontSize() == Approx(32).margin(0.1));
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.1d.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.1d.svg");
}
SECTION("square") {
MolDraw2DSVG drawer(-1, -1, -1, -1);
drawer.drawOptions().baseFontSize = 0.8;
drawer.drawMolecule(*mol2);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.2.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.2.svg");
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
SECTION("square PNG no freetype") {
MolDraw2DCairo drawer(-1, -1, -1, -1, true);
drawer.drawOptions().baseFontSize = 0.8;
drawer.drawMolecule(*mol2);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.2a.png");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.2a.png");
}
SECTION("square PNG with freetype") {
MolDraw2DCairo drawer(-1, -1, -1, -1, false);
drawer.drawOptions().baseFontSize = 0.8;
drawer.drawMolecule(*mol2);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.2b.png");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.2b.png");
}
#endif
// semiflexicanvas - with freetype
SECTION("semiflexicanvas1") {
MolDraw2DSVG drawer(308, -1, -1, -1, false);
drawer.drawOptions().scalingFactor = 30;
drawer.drawOptions().baseFontSize = 0.6;
drawer.drawMolecule(*mol1);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testSemiFlexiCanvas.1a.svg");
outs << text;
outs.flush();
check_file_hash("testSemiFlexiCanvas.1a.svg");
}
SECTION("semiflexicanvas2") {
MolDraw2DSVG drawer(-1, 223, -1, -1, false);
drawer.drawOptions().scalingFactor = 30;
drawer.drawOptions().baseFontSize = 0.6;
drawer.drawMolecule(*mol1);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testSemiFlexiCanvas.1b.svg");
outs << text;
outs.flush();
check_file_hash("testSemiFlexiCanvas.1b.svg");
}
SECTION("semiflexicanvas3") {
auto mol3 = "ON"_smiles;
REQUIRE(mol3);
MolDraw2DSVG drawer(-1, 150, -1, -1, false);
drawer.drawOptions().scalingFactor = 30;
drawer.drawOptions().baseFontSize = 0.6;
drawer.drawMolecule(*mol3);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testSemiFlexiCanvas.1c.svg");
outs << text;
outs.flush();
check_file_hash("testSemiFlexiCanvas.1c.svg");
}
SECTION("reaction") {
std::unique_ptr<ChemicalReaction> rxn(RxnSmartsToChemicalReaction(
"[N:1]-[C:2]-[C:3](=[O:4])-[O:5].[N:6]-[C:7]-[C:8](=[O:9])-[O:10]>>[N:"
"1]1-[C:2]-[C:3](=[O:4])-[N:6]-[C:7]-[C:8]-1=[O:9].[O:5]=[O:10]"));
MolDraw2DSVG drawer(-1, -1, -1, -1, true);
drawer.drawReaction(*rxn);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.3.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.3.svg");
}
SECTION("data labels") {
auto mol1 = R"CTAB(
RDKit 2D
0 0 0 0 0 0 0 0 0 0999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 5 4 1 0 0
M V30 BEGIN ATOM
M V30 1 C 0.000000 0.000000 0.000000 0
M V30 2 C 1.299038 0.750000 0.000000 0
M V30 3 C 2.598076 -0.000000 0.000000 0
M V30 4 C 1.299038 2.250000 0.000000 0
M V30 5 C 2.598076 3.000000 0.000000 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 2 2 3
M V30 3 1 2 4
M V30 4 2 4 5
M V30 END BOND
M V30 BEGIN SGROUP
M V30 1 DAT 0 ATOMS=(5 2 4 5 3 1) FIELDNAME="Lambda Max" FIELDINFO=nm -
M V30 FIELDDATA="2222"
M V30 END SGROUP
M V30 END CTAB
M END)CTAB"_ctab;
REQUIRE(mol1);
{
MolDraw2DSVG drawer(-1, -1);
drawer.drawMolecule(*mol1);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.4a.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.4a.svg");
}
{
MolDraw2DSVG drawer(-1, -1);
drawer.drawMolecule(*mol1, "legendary");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.4b.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.4b.svg");
}
{
MolDraw2DSVG drawer(-1, -1);
drawer.drawMolecule(*mol1, "doubly\nlegendary");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.4c.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.4c.svg");
}
{
MolDraw2DSVG drawer(-1, -1);
drawer.drawOptions().legendFraction = 0.25;
drawer.drawOptions().legendFontSize = 32;
drawer.drawMolecule(*mol1, "Hugely\nLegendary");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.4d.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.4d.svg");
}
}
SECTION("including legends") {
// add an atomNote so that we can compare font sizes
mol1->getAtomWithIdx(0)->setProp(common_properties::atomNote, "n1");
{
MolDraw2DSVG drawer(-1, -1);
drawer.drawMolecule(*mol1, "legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.5a.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.5a.svg");
}
{
MolDraw2DSVG drawer(-1, -1);
drawer.drawMolecule(*mol1, "legend\nwith two lines");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.5b.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.5b.svg");
}
{
MolDraw2DSVG drawer(-1, -1);
drawer.drawOptions().scalingFactor = 45;
drawer.drawMolecule(*mol1, "legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.5c.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.5c.svg");
}
{
MolDraw2DSVG drawer(-1, -1);
drawer.drawOptions().scalingFactor = 10;
drawer.drawMolecule(*mol1, "legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.5d.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.5d.svg");
}
}
SECTION("partially flexicanvas (height) + legends") {
// add an atomNote so that we can compare font sizes
mol1->getAtomWithIdx(0)->setProp(common_properties::atomNote, "n1");
{
MolDraw2DSVG drawer(-1, 200);
drawer.drawMolecule(*mol1, "legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.6a.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.6a.svg");
}
{
MolDraw2DSVG drawer(-1, 200);
drawer.drawMolecule(*mol1, "legend\nwith two lines");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.6b.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.6b.svg");
}
{
MolDraw2DSVG drawer(-1, 200);
drawer.drawOptions().scalingFactor = 45;
drawer.drawMolecule(*mol1, "legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.6c.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.6c.svg");
}
{
MolDraw2DSVG drawer(-1, 200);
drawer.drawOptions().scalingFactor = 10;
drawer.drawMolecule(*mol1, "legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.6d.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.6d.svg");
}
}
SECTION("partially flexicanvas (width) + legends") {
// add an atomNote so that we can compare font sizes
mol1->getAtomWithIdx(0)->setProp(common_properties::atomNote, "n1");
{
MolDraw2DSVG drawer(300, -1);
drawer.drawMolecule(*mol1, "legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.7a.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.7a.svg");
}
{
MolDraw2DSVG drawer(300, -1);
drawer.drawMolecule(*mol1, "legend\nwith two lines");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.7b.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.7b.svg");
}
{
MolDraw2DSVG drawer(300, -1);
drawer.drawOptions().scalingFactor = 45;
drawer.drawMolecule(*mol1, "legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.7c.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.7c.svg");
}
{
MolDraw2DSVG drawer(300, -1);
drawer.drawOptions().scalingFactor = 10;
drawer.drawMolecule(*mol1, "legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testFlexiCanvas.7d.svg");
outs << text;
outs.flush();
check_file_hash("testFlexiCanvas.7d.svg");
}
}
}
TEST_CASE("Github #4764") {
SECTION("basics") {
auto mol = "c1ccccc1-C1CCCCC1"_smiles;
REQUIRE(mol);
std::vector<int> highlights{6, 7, 8, 9, 10, 11};
{
MolDraw2DSVG drawer(200, 150);
drawer.drawMolecule(*mol, "highlight", &highlights);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testGithub4764.sz1.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4764.sz1.svg");
}
{
MolDraw2DSVG drawer(400, 350);
drawer.drawMolecule(*mol, "highlight", &highlights);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testGithub4764.sz2.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4764.sz2.svg");
}
{
MolDraw2DSVG drawer(800, 700);
drawer.drawMolecule(*mol, "highlight", &highlights);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testGithub4764.sz3.svg");
outs << text;
outs.flush();
check_file_hash("testGithub4764.sz3.svg");
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
{
MolDraw2DCairo drawer(200, 150);
drawer.drawMolecule(*mol, "highlight", &highlights);
drawer.finishDrawing();
drawer.writeDrawingText("testGithub4764.sz1.png");
check_file_hash("testGithub4764.sz1.png");
}
{
MolDraw2DCairo drawer(400, 350);
drawer.drawMolecule(*mol, "highlight", &highlights);
drawer.finishDrawing();
drawer.writeDrawingText("testGithub4764.sz2.png");
check_file_hash("testGithub4764.sz2.png");
}
{
MolDraw2DCairo drawer(800, 700);
drawer.drawMolecule(*mol, "highlight", &highlights);
drawer.finishDrawing();
drawer.writeDrawingText("testGithub4764.sz3.png");
check_file_hash("testGithub4764.sz3.png");
}
#endif
// check_file_hash("testGithub4538.svg");
}
}
TEST_CASE("drawArc starting from wrong angle") {
SECTION("basics") {
auto mol = R"CTAB(
RDKit 2D
9 9 0 0 0 0 0 0 0 0999 V2000
-1.2135 -0.7027 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
-0.0000 -1.5844 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
1.2135 -0.7027 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
0.7500 0.7238 0.0000 N 0 0 0 0 0 0 0 0 0 0 0 0
-0.7500 0.7238 0.0000 C 0 0 0 0 0 0 0 0 0 0 0 0
-1.6317 1.9374 0.0000 F 0 0 0 0 0 0 0 0 0 0 0 0
-2.6401 -1.1663 0.0000 F 0 0 0 0 0 0 0 0 0 0 0 0
2.6401 -1.1663 0.0000 F 0 0 0 0 0 0 0 0 0 0 0 0
1.6317 1.9374 0.0000 F 0 0 0 0 0 0 0 0 0 0 0 0
1 2 1 0
2 3 2 0
3 4 1 0
4 5 1 0
5 6 1 0
5 1 2 0
1 7 1 0
3 8 1 0
4 9 1 0
M END)CTAB"_ctab;
REQUIRE(mol);
{
MolDraw2DSVG drawer(400, 350);
drawer.drawOptions().noAtomLabels = true;
drawer.drawMolecule(*mol, "drawArc");
drawer.setFillPolys(false);
drawer.setColour({1, 0, 0});
drawer.drawArc(mol->getConformer().getAtomPos(3), 0.3, -72, 54);
drawer.drawArc(mol->getConformer().getAtomPos(0), 0.3, -162, -36);
drawer.drawArc(mol->getConformer().getAtomPos(4), 0.3, 126, 252);
drawer.drawArc(mol->getConformer().getAtomPos(2), 0.3, -18, 108);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testDrawArc1.svg");
outs << text;
outs.flush();
check_file_hash("testDrawArc1.svg");
}
}
}
TEST_CASE("wedged bonds to metals drawn in the wrong direction") {
SECTION("basics") {
auto m = R"CTAB(
Mrv2108 01092205442D
0 0 0 0 0 999 V3000
M V30 BEGIN CTAB
M V30 COUNTS 5 4 0 0 0
M V30 BEGIN ATOM
M V30 1 F 10.6667 -0.75 0 0
M V30 2 Pt 10.6667 -2.29 0 0 CFG=1
M V30 3 Cl 12.2067 -2.29 0 0
M V30 4 C 10.6667 -3.83 0 0
M V30 5 O 9.1267 -2.29 0 0
M V30 END ATOM
M V30 BEGIN BOND
M V30 1 1 1 2
M V30 2 1 2 3
M V30 3 1 2 4 CFG=1
M V30 4 1 2 5 CFG=3
M V30 END BOND
M V30 END CTAB
M END)CTAB"_ctab;
m->getBondWithIdx(2)->setBondDir(Bond::BondDir::BEGINWEDGE);
m->getBondWithIdx(3)->setBondDir(Bond::BondDir::BEGINDASH);
MolDraw2DSVG drawer(250, 200);
assignBWPalette(drawer.drawOptions().atomColourPalette);
drawer.drawMolecule(*m, "check wedges");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testMetalWedges.svg");
outs << text;
outs.flush();
check_file_hash("testMetalWedges.svg");
}
}
TEST_CASE("vary proporition of panel for legend", "[drawing]") {
SECTION("basics") {
auto m1 = "C1N[C@@H]2OCC12"_smiles;
REQUIRE(m1);
// These look a bit pants with NO_FREETYPE=true, but much better with
// Freetype.
{
// default legend
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
MolDraw2DUtils::prepareAndDrawMolecule(drawer, *m1, "default legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testVariableLegend_1.svg");
outs << text;
outs.flush();
CHECK(text.find("<text x='34.5' y='195.0' class='legend' "
"style='font-size:16px;") != std::string::npos);
check_file_hash("testVariableLegend_1.svg");
}
{
// 1/4 of panel
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
drawer.drawOptions().legendFraction = 0.25;
drawer.drawOptions().legendFontSize = 32;
MolDraw2DUtils::prepareAndDrawMolecule(drawer, *m1, "massive legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testVariableLegend_2.svg");
outs << text;
outs.flush();
CHECK(text.find("<text x='1.6' y='195.0' class='legend' "
"style='font-size:31px;") != std::string::npos);
check_file_hash("testVariableLegend_2.svg");
}
{
// tiny
MolDraw2DSVG drawer(200, 200, -1, -1, NO_FREETYPE);
drawer.drawOptions().legendFraction = 0.05;
MolDraw2DUtils::prepareAndDrawMolecule(drawer, *m1, "small legend");
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testVariableLegend_3.svg");
outs << text;
outs.flush();
CHECK(text.find("<text x='84.7' y='195.0' class='legend' "
"style='font-size:6px;") != std::string::npos);
check_file_hash("testVariableLegend_3.svg");
}
}
}
TEST_CASE(
"Github 5061 - draw reaction with no reagents and scaleBondWidth true") {
SECTION("basics") {
std::string data = R"RXN($RXN
Mrv16425 091201171606
0 1
$MOL
Mrv1642509121716062D
2 1 0 0 0 0 999 V2000
3.5357 0.0000 0.0000 R# 0 0 0 0 0 0 0 0 0 0 0 0
2.7107 0.0000 0.0000 R# 0 0 0 0 0 0 0 0 0 0 0 0
1 2 1 0 0 0 0
M RGP 2 1 1 2 2
M END)RXN";
{
std::unique_ptr<ChemicalReaction> rxn{RxnBlockToChemicalReaction(data)};
MolDraw2DSVG drawer(450, 200);
drawer.drawOptions().scaleBondWidth = true;
drawer.drawReaction(*rxn);
drawer.finishDrawing();
auto text = drawer.getDrawingText();
std::ofstream outs("testGithub_5061.svg");
outs << text;
outs.flush();
check_file_hash("testGithub_5061.svg");
}
}
}
#ifdef RDK_BUILD_CAIRO_SUPPORT
TEST_CASE("drawing doesn't destroy reaction properties", "[drawing]") {
auto rxn = "[CH3:1][OH:2]>>[CH2:1]=[OH0:2]"_rxnsmarts;
REQUIRE(rxn);
MolDraw2DCairo drawer(400, 200);
bool highlightByReactant = true;
drawer.drawReaction(*rxn, highlightByReactant);
drawer.finishDrawing();
auto png = drawer.getDrawingText();
std::unique_ptr<ChemicalReaction> rxn2{PNGStringToChemicalReaction(png)};
REQUIRE(rxn2);
CHECK(rxn->getReactants()[0]->getAtomWithIdx(0)->getAtomMapNum() == 1);
CHECK(rxn->getReactants()[0]->getAtomWithIdx(1)->getAtomMapNum() == 2);
CHECK(rxn2->getReactants()[0]->getAtomWithIdx(0)->getAtomMapNum() == 1);
CHECK(rxn2->getReactants()[0]->getAtomWithIdx(1)->getAtomMapNum() == 2);
}
#endif
Reference in New Issue View Git Blame Copy Permalink