File: export_fragment.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 (110 lines) | stat: -rw-r--r-- 2,877 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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:
//
// $Id: export_fragment.C,v 1.4.20.1 2007/08/07 18:26:18 oliver Exp $
//

#include <fstream>

#include <BALL/KERNEL/system.h>
#include <BALL/KERNEL/bond.h>
#include <BALL/KERNEL/PTE.h>
#include <BALL/KERNEL/forEach.h>
#include <BALL/DATATYPE/string.h>
#include <BALL/FORMAT/HINFile.h>
#include <BALL/FORMAT/PDBFile.h>

using namespace BALL;

int main(int argc, char** argv)
{
	if (argc != 3)
	{
		Log.info() << "Usage :" << argv[0] << "<HIN/PDB input>  <FragmentDB output>." 
			<< std::endl;
		return(1);
	}

	String infilename(argv[1]);

	std::vector<String> substrings;
	infilename.split(substrings, ".");

	System system;

	if ((substrings[substrings.size()-1] == "pdb")
		|| (substrings[substrings.size()-1] == "PDB"))
	{
		PDBFile infile(infilename);	
		if (!infile)
		{
			Log.error() << "error opening " << argv[1] << " for input." << std::endl;
			return 2;
		}

		infile >> system;
		infile.close();
	}
	else
	{
		if ((substrings[substrings.size()-1] == "hin")
				|| (substrings[substrings.size()-1] == "HIN"))
		{
			HINFile infile(infilename);	
			if (!infile)
			{
				Log.error() << "error opening " << argv[1] << " for input." << std::endl;
				return 2;
			}

			infile >> system;
			infile.close();
		}
		else
		{
			Log.error() << "I only understand HyperChem .hin and PDB .pdb files."
				<< std::endl;
		}
	}

	String outfilename(argv[2]);

	std::ofstream outfile(outfilename.c_str(), std::ios::out);
	outfile << "<node>Unknown Compound" << std::endl;
	outfile << "\t<node>Names" << std::endl;
	outfile << "\t\t<node>Unknown Name</node>" << std::endl;
	outfile << "\t</node>" << std::endl;
	outfile << "\t<node>Atoms" << std::endl;
	AtomConstIterator atom_it = system.beginAtom();
	for (; +atom_it; ++atom_it)
	{
		outfile << "\t\t<node>" << atom_it->getName() << "<value>"
			<< atom_it->getElement().getSymbol() << " "
			<< atom_it->getPosition().x << " "
			<< atom_it->getPosition().y << " "
			<< atom_it->getPosition().z << "</value></node>" << std::endl;
	}
	outfile << "\t</node>" << std::endl;
	outfile << "\t<node>Bonds" << std::endl;
	Size counter = 1;
	char bond_order_char = '\0';
	AtomBondConstIterator bond_it;
	BALL_FOREACH_BOND(system, atom_it, bond_it)
	{
		switch (bond_it->getOrder())
		{
			case Bond::ORDER__SINGLE: bond_order_char = 's'; break;
			case Bond::ORDER__DOUBLE: bond_order_char = 'd'; break;
			case Bond::ORDER__TRIPLE: bond_order_char = 't'; break;
			case Bond::ORDER__QUADRUPLE: bond_order_char = 'q'; break;
			case Bond::ORDER__AROMATIC: bond_order_char = 'a'; break;
		}
		outfile << "\t\t<node>" << counter++ << "<value>"
			<< bond_it->getFirstAtom()->getName() << " "
			<< bond_it->getSecondAtom()->getName() << " "
			<< bond_order_char << "</value></node>" << std::endl;
	}
	outfile << "\t</node>" << std::endl;
	outfile << "</node>" << std::endl;

}