File: ScoringFunction_test.C

package info (click to toggle)
ball 1.5.0%2Bgit20180813.37fc53c-3
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 239,848 kB
  • sloc: cpp: 326,149; ansic: 4,208; python: 2,303; yacc: 1,778; lex: 1,099; xml: 958; sh: 322; makefile: 93
file content (143 lines) | stat: -rwxr-xr-x 4,345 bytes parent folder | download | duplicates (7)
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:

#include <BALL/CONCEPT/classTest.h>
#include <BALLTestConfig.h>

#include <BALL/SCORING/COMMON/gridBasedScoring.h>
#include <BALL/SCORING/FUNCTIONS/MMScoring.h>
#include <BALL/DOCKING/COMMON/structurePreparer.h>
#include <BALL/DOCKING/IMGDOCK/IMGDock.h>
#include <BALL/FORMAT/PDBFile.h>
#include <BALL/FORMAT/MOL2File.h>


using namespace std;
using namespace BALL;

///////////////////////////

START_TEST(ScoringFunction)

/////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////


PRECISION(1E-3)

Log.remove(cout);

PDBFile pdb(BALL_TEST_DATA_PATH(1b5i_pocket.pdb));
MOL2File mol2(BALL_TEST_DATA_PATH(1b5i_ligand.mol2));
System pocket; pdb >> pocket;
System ligand; mol2 >> ligand;

StructurePreparer sp;
sp.prepare(&pocket,"Amber/amber96-docking.ini");
sp.prepare(&ligand,"Amber/amber96-docking.ini");


CHECK(MMScoring)
	Options options;
	MMScoring scoring_function(pocket,ligand,options);
	scoring_function.setLigand(ligand);
	scoring_function.update();
	scoring_function.updateScore();

	vector<double> score_contributions;
	vector<String> names;
	scoring_function.getScoreContributions(score_contributions, names);

	TEST_EQUAL(names.size(),5)
	TEST_EQUAL(score_contributions.size(),5)

	TEST_REAL_EQUAL(scoring_function.getScore(),-68.7552);
	TEST_REAL_EQUAL(score_contributions[0],-8.52705) // ES
	TEST_REAL_EQUAL(score_contributions[1],-19.7856) // vdW
	TEST_REAL_EQUAL(score_contributions[2],-20.5528) // frag. solvation
	TEST_REAL_EQUAL(score_contributions[3],-24.8898) // hydroggen bonds
	TEST_REAL_EQUAL(score_contributions[4],5) // nRot
RESULT


// Do a fast re-docking, just in order to check whether it works. Note, that the below settings are therefore _not_ useful for a normal docking.
Options options;
options.set("iterations",1);
options.set("no_solutions",20);
options.set("step_width",90);
options.set("scoring_type","GridedMM");
options.set("scoregrid_resolution",1.0);
set<String> types;
types.insert("C");types.insert("H");
types.insert("O");types.insert("N");
IMGDock docker(pocket,ligand,options);
GridBasedScoring* grid_scoring = dynamic_cast<GridBasedScoring*>(docker.getScoringFunction());
TEST_NOT_EQUAL(grid_scoring,0);
grid_scoring->setAtomTypeNames(types);


CHECK(Grid Precalculation)
	grid_scoring->precalculateGrids();
	grid_scoring->update();
	grid_scoring->updateScore(); // evaluate binding of reference ligand
//	grid_scoring->printResult();
	TEST_REAL_EQUAL(grid_scoring->getScore(),-57.424)
RESULT


CHECK(Scoregrid storing and loading)
	grid_scoring->saveGridSetsToFile("test.grd","1b5i_pocket");
	grid_scoring->replaceGridSetFromFile("test.grd");
	grid_scoring->update();
	grid_scoring->updateScore();
	TEST_REAL_EQUAL(grid_scoring->getScore(),-57.424)
RESULT

CHECK(Compressed scoregrid storing and loading)
	grid_scoring->saveGridSetsToFile("test.grd.gz","1b5i_pocket");
	grid_scoring->replaceGridSetFromFile("test.grd.gz");
	grid_scoring->update();
	grid_scoring->updateScore();
	TEST_REAL_EQUAL(grid_scoring->getScore(),-57.424)
RESULT

CHECK(Binary scoregrid storing and loading)
	grid_scoring->saveGridSetsToFile("test.bngrd","1b5i_pocket");
	grid_scoring->replaceGridSetFromFile("test.bngrd");
	grid_scoring->update();
	grid_scoring->updateScore();
	TEST_REAL_EQUAL(grid_scoring->getScore(),-57.424)
RESULT


CHECK(IMeedyDock)
	System ligand2 = ligand; // copy reference ligand for this simple test

	// move ligand2 out of pocket
	for(AtomIterator it=ligand2.beginAtom(); +it; it++)
	{
		it->setPosition(it->getPosition()+Vector3(20,20,20));
	}

	docker.dockLigand(ligand2);
	TEST_REAL_EQUAL(docker.getScoringFunction()->getScore(),-59.7147);
	MOL2File mol2res(BALL_TEST_DATA_PATH(1b5i_ligand-dockOut.mol2));
	System expected_dockres;
	mol2res >> expected_dockres;

	AtomConstIterator it1 = ligand2.beginAtom();
	AtomConstIterator it2 = expected_dockres.beginAtom();
	for(; +it1 && +it2; it1++,it2++)
	{
		Vector3 position1 = it1->getPosition();
		Vector3 position2 = it2->getPosition();
		TEST_REAL_EQUAL(position1[0],position2[0])
		TEST_REAL_EQUAL(position1[1],position2[1])
		TEST_REAL_EQUAL(position1[2],position2[2])
	}
RESULT

/////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////

END_TEST