1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
//
// Copyright (C) 2005-2019 Rational Discovery LLC
//
// @@ 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 <RDGeneral/test.h>
#include <Geometry/UniformGrid3D.h>
#include "ShapeEncoder.h"
#include "ShapeUtils.h"
#include <GraphMol/RDKitBase.h>
#include <GraphMol/FileParsers/FileParsers.h>
#include <GraphMol/SmilesParse/SmilesParse.h>
#include <Geometry/GridUtils.h>
#include <GraphMol/MolAlign/AlignMolecules.h>
#include <GraphMol/MolTransforms/MolTransforms.h>
#include <GraphMol/Substruct/SubstructMatch.h>
using namespace RDKit;
void test1Encode() {
RDGeom::UniformGrid3D grd(30.0, 16.0, 10.0);
std::string rdbase = getenv("RDBASE");
std::string fname1 =
rdbase + "/Code/GraphMol/ShapeHelpers/test_data/1oir.mol";
ROMol *m = MolFileToMol(fname1);
MolTransforms::canonicalizeMol(*m);
MolShapes::EncodeShape(*m, grd, 0);
delete m;
CHECK_INVARIANT(grd.getOccupancyVect()->getTotalVal() == 7405, "");
}
void test2Compare() {
RDGeom::UniformGrid3D grd(30.0, 16.0, 10.0);
std::string rdbase = getenv("RDBASE");
std::string fname1 =
rdbase + "/Code/GraphMol/ShapeHelpers/test_data/1oir.mol";
ROMol *m = MolFileToMol(fname1);
MolTransforms::canonicalizeMol(*m);
ROMol *mdup = MolFileToMol(fname1);
MolTransforms::canonicalizeMol(*mdup);
double dist = MolShapes::tanimotoDistance(*m, *mdup);
CHECK_INVARIANT(dist == 0.0, "");
dist = MolShapes::tverskyIndex(*m, *mdup, 1.0, 1.0);
CHECK_INVARIANT(dist == 1.0, "");
delete m;
delete mdup;
m = MolFileToMol(fname1);
std::string fname2 =
rdbase + "/Code/GraphMol/ShapeHelpers/test_data/1oir_conf.mol";
ROMol *m2 = MolFileToMol(fname2);
double rmsd = MolAlign::alignMol(*m, *m2);
CHECK_INVARIANT(rmsd >= 0.0, "");
dist = MolShapes::tanimotoDistance(*m, *m2);
CHECK_INVARIANT(RDKit::feq(dist, 0.31, 0.01), "");
dist = MolShapes::tverskyIndex(*m, *m2, 1.0, 1.0);
CHECK_INVARIANT(RDKit::feq(dist, 0.68, 0.01), "");
delete m2;
m2 = MolFileToMol(fname2);
MatchVectType atomMap;
atomMap.push_back(std::pair<int, int>(18, 27));
atomMap.push_back(std::pair<int, int>(13, 23));
atomMap.push_back(std::pair<int, int>(21, 14));
atomMap.push_back(std::pair<int, int>(24, 7));
atomMap.push_back(std::pair<int, int>(9, 19));
atomMap.push_back(std::pair<int, int>(16, 30));
rmsd = MolAlign::alignMol(*m2, *m, 0, 0, &atomMap);
dist = MolShapes::tanimotoDistance(*m, *m2);
CHECK_INVARIANT(RDKit::feq(dist, 0.3593), "");
delete m;
delete m2;
}
int main() {
#if 1
std::cout << "***********************************************************\n";
std::cout << "Testing ShapeHelpers\n";
std::cout << "\t---------------------------------\n";
std::cout << "\t test1Encode \n\n";
test1Encode();
std::cout << "\t---------------------------------\n";
std::cout << "\t test2Compare \n\n";
test2Compare();
std::cout << "***********************************************************\n";
#endif
return 0;
}
|
