File: SASAreas.C

package info (click to toggle)
ball 1.5.0%2Bgit20180813.37fc53c-11
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 239,924 kB
  • sloc: cpp: 326,149; ansic: 4,208; python: 2,303; yacc: 1,778; lex: 1,099; xml: 958; sh: 322; javascript: 164; makefile: 88
file content (91 lines) | stat: -rw-r--r-- 2,271 bytes parent folder | download | duplicates (8)
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
// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:
//

// Calculate the solvent accessible surface (SAS) areas of the
// residues of a protein.

#include <BALL/STRUCTURE/numericalSAS.h>
#include <BALL/STRUCTURE/defaultProcessors.h>
#include <BALL/FORMAT/PDBFile.h>

using namespace BALL;
using namespace std;

int main(int argc, char** argv)
{
	// prefix all output by the current time
	Log.setPrefix(cout, "[%T]: ");
	Log.setPrefix(cerr, "[%T]: ERROR: ");

	// print usage
	if (argc != 2)
	{
		Log.error() << argv[0] << " <pdb file>" << endl;
		return 1;
	}

	// open the PDB file
	PDBFile pdb(argv[1]);
	if (!pdb)
	{
		// if the file could not be opened, print error message and exit
		Log.error() << "cannot open PBD file " << argv[1] << endl;
		return 2;
	}

	// read the contents of the file into a system
	System S;
	pdb >> S;
	pdb.close();

	// assign the radii defining the solvent accessible surface
	AssignRadiusProcessor rp("radii/PARSE.siz");
	S.apply(rp);

	// calculate the contribution of every single
	// atom of the surface and store it in a hash map
	HashMap<const Atom*, float> atom_areas;
	double total_area = calculateSASAtomAreas(S, atom_areas, 1.5, 400);
	Log.info() <<  "total area: " << total_area << "A^2" << endl;
	Log.info() << "# of SAS atoms: " << atom_areas.size() << endl;

	Size number_of_residues = S.countResidues();
	Log.info() << "# of residues: " << number_of_residues << endl;
	
	// we store the accumulated areas of each residue
	// (i.e., the sum of all surface areas of the residue's atoms)
	// in a vector
	vector<float> residue_areas(number_of_residues);

	// iterate over all residues...
	ResidueIterator res_it = S.beginResidue();
	AtomIterator atom_it;
	for (; +res_it; ++res_it)
	{
		//...and sum up the atom areas.
		float residue_area = 0.0;
		for (atom_it = res_it->beginAtom(); +atom_it; ++atom_it)
		{
			if (atom_areas.has(&*atom_it))
			{
				// retrieve the atom area by its pointer
				residue_area += atom_areas[&*atom_it];
			}
		}

		// print the surface area along with the residue name and ID
		Log.info() << res_it->getName() << res_it->getID() << " ";
		if (residue_area > 0.0)
		{
			 Log.info() << residue_area << endl;
		} 
		else 
		{
			 Log.info() << "buried" << endl;
		}
	}

	// done
	return 0;
}