File: reading.C

package info (click to toggle)
ball 1.5.0%2Bgit20180813.37fc53c-6
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 239,888 kB
  • sloc: cpp: 326,149; ansic: 4,208; python: 2,303; yacc: 1,778; lex: 1,099; xml: 958; sh: 322; makefile: 95
file content (220 lines) | stat: -rw-r--r-- 4,621 bytes parent folder | download | duplicates (9)
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
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:
//
// $Id: reading.C,v 1.7 2002/02/27 12:20:32 sturm Exp $

#include <BALL/FORMAT/PDBFile.h>
#include <BALL/FORMAT/HINFile.h>
#include <BALL/FORMAT/INIFile.h>
#include <BALL/KERNEL/PTE.h>
#include "global.h"
#include "reading.h"
#include "assignment.h"
#include <fstream>

using namespace std;

void addToSystem(System& system)
{
	// splice the system into the global system
	S.splice(system);
}

void readPDBFile(const String& filename)
{
	// read a PDB file
	PDBFile f;
	System system;
	f.open(filename);
	if (f.bad())
	{
		Log.error() << "Could not open HIN file " << filename << " for reading." << endl;
		Log.error() << "Aborted." << endl;
		exit(4);
	}
	f >> system;
	f.close();
	
	// print the number of atoms read in verbose mod
	if (verbose)
	{
		Log.info() << "read " << system.countAtoms() << " atoms from " << filename << endl;
	}
	
	// normalize the names
	normalizeNames(system);

	// build the bonds
	buildBonds(system);

	// assign charges and radii
	assignCharges(system);
	assignRadii(system);
	
	// insert into the system
	addToSystem(system);
}

void readSystemFromHINFile(const String& filename, System& system)
{
	// read a HIN file
	HINFile f;
	f.open(filename);
	if (f.bad())
	{
		Log.error() << "Could not open HIN file " << filename << " for reading." << endl;
		Log.error() << "Aborted." << endl;
		exit(4);
	}
	f >> system;
	f.close();
	
	// print the number of atoms read in verbose mod
	if (verbose)
	{
		Log.info() << "read " << system.countAtoms() << " atoms from " << filename << endl;
	}

	// normalize the names
	normalizeNames(system);

	// assign radii
	assignRadii(system);
}

void readHINFile(const String& filename)
{
	System system;
	readSystemFromHINFile(filename, system);
	assignCharges(system);
	addToSystem(system);
}

void readHINFileNoAssignment(const String& filename)
{
	System system;
	readSystemFromHINFile(filename, system);
	addToSystem(system);
}

void readOptionFile(const String& filename)
{
	if (verbose)
	{
		Log.info() << "reading options from " << filename << endl;
	}
	options.readOptionFile(filename);
}

void readChargeFile(const String& filename)
{
	if (verbose)
	{
		Log.info() << "reading charges from " << filename << endl;
	}
	charges.setFilename(filename);
	
	// ignore the rule-based processors
	use_charge_rules = false;
}

void readRadiusFile(const String& filename)
{
	if (verbose)
	{
		Log.info() << "reading radii from " << filename << endl;
	}
	radii.setFilename(filename);
	
	// ignore the rule-based processors
	use_radius_rules = false;
}

void readRuleFile(const String& filename, RuleType rule_type)
{
	if (verbose)
	{
		switch (rule_type)
		{
			case CHARGES_AND_RADII:
				Log.info() << "reading charge and radius rules from " << filename << endl;
				break;
			case CHARGES:
				Log.info() << "reading charge rules from " << filename << endl;
				break;
			case RADII:
				Log.info() << "reading radius rules from " << filename << endl;
		}
	}

	// open the rules file
	INIFile ini(filename);
	
	// read the rules
	ini.read();

	switch (rule_type)
	{
		case CHARGES_AND_RADII:
			charge_rules.initialize(ini, "ChargeRules");
			radius_rules.initialize(ini, "RadiusRules");
			use_charge_rules = true;
			use_radius_rules = true;
			break;

		case RADII:
			radius_rules.initialize(ini, "RadiusRules");
			use_radius_rules = true;
			break;

		case CHARGES:
			charge_rules.initialize(ini, "ChargeRules");
			use_charge_rules = true;
	}
}

void dumpFile()
{
	if (verbose)
	{
		Log.info() << "dumping atom charges, radii, and surface areas to " << dump_file << endl;
	}

	ofstream outfile(dump_file.c_str(), ios::out);
	
	outfile << "#   PB dump file" << endl;
	outfile << "# atom  element   charge[e0]  radius[A]   SAS[A^2] " << endl;
	outfile << "#---------------------------------------------------" << endl;
	
	AtomIterator it = S.beginAtom();
	double total_charge = 0.0;
	for (; +it; ++it)
	{
		total_charge += it->getCharge();
		outfile << setprecision(5) 
						<< it->getFullName() 
						<< " " << it->getElement().getSymbol()
						<< " " << it->getCharge()
						<< " " << it->getRadius();

		if (sas_calculation && surface_map.has(&*it))
		{	
			outfile << " " << surface_map[&*it];
		} 
		else 
		{
			outfile << " -";
		}
		outfile	<< endl;
	}
	outfile << "# total charge: " << total_charge << " e0" << endl;
	if (ses_calculation)
	{
		outfile << "# total solvent excluded surface area: " << total_SES_area << " A^2" << endl;
	}
	if (sas_calculation)
	{
		outfile << "# total solvent accessible surface area: " << total_SAS_area << " A^2" << endl;
	}
	outfile.close();
}