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
|
/* This code is part of the tng binary trajectory format.
*
* VERSION 1.0
*
* Written by Magnus Lundborg
* Copyright (c) 2012, The GROMACS development team.
* check out http://www.gromacs.org for more information.
*
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public License
* as published by the Free Software Foundation; either version 2.1
* of the License, or (at your option) any later version.
*/
#include <stdlib.h>
#include <stdio.h>
#include <tng_io.h>
int main(int argc, char **argv)
{
tng_trajectory_t traj;
union data_values ***positions = 0; // A 3-dimensional array to be populated
int64_t n_particles, n_values_per_frame, n_frames, tot_n_frames;
tng_data_type data_type;
int i, j;
int64_t particle = 0;
// Set a default frame range
int first_frame = 0, last_frame = 50;
char atom_name[64], res_name[64], chain_name[64];
if(argc <= 1)
{
printf("No file specified\n");
printf("Usage:\n");
printf("tng_io_read_pos <tng_file> [particle number = %"PRId64"] "
"[first_frame = %d] [last_frame = %d]\n",
particle, first_frame, last_frame);
exit(1);
}
// A reference must be passed to allocate memory
if(tng_trajectory_init(&traj) != TNG_SUCCESS)
{
tng_trajectory_destroy(&traj);
exit(1);
}
tng_input_file_set(traj, argv[1]);
// Read the file headers
tng_file_headers_read(traj, TNG_USE_HASH);
if(argc >= 3)
{
particle = strtoll(argv[2], 0, 10);
if(argc >= 4)
{
first_frame = strtoll(argv[3], 0, 10);
if(argc >= 5)
{
last_frame = strtoll(argv[4], 0, 10);
}
}
}
if(tng_num_frames_get(traj, &tot_n_frames) != TNG_SUCCESS)
{
printf("Cannot determine the number of frames in the file\n");
tng_trajectory_destroy(&traj);
exit(1);
}
printf("%"PRId64" frames in file\n", tot_n_frames);
if(last_frame > tot_n_frames - 1)
{
last_frame = tot_n_frames - 1;
}
n_frames = last_frame - first_frame + 1;
if(tng_atom_name_of_particle_nr_get(traj, particle, atom_name,
sizeof(atom_name)) ==
TNG_SUCCESS &&
tng_residue_name_of_particle_nr_get(traj, particle, res_name,
sizeof(res_name)) ==
TNG_SUCCESS &&
tng_chain_name_of_particle_nr_get(traj, particle, chain_name,
sizeof(chain_name)) ==
TNG_SUCCESS)
{
printf("Particle: %s (%s: %s)\n", atom_name, chain_name, res_name);
}
else
{
printf("Particle name not found\n");
}
// Get the positions of all particles in the requested frame range.
// The positions are stored in the positions array.
// N.B. No proper error checks.
if(tng_particle_data_interval_get(traj, TNG_TRAJ_POSITIONS, first_frame,
last_frame, TNG_USE_HASH, &positions, &n_particles, &n_values_per_frame,
&data_type) == TNG_SUCCESS)
{
if(particle >= n_particles)
{
printf("Chosen particle out of range. Only %"PRId64" particles in trajectory.\n", n_particles);
}
else
{
// Print the positions of the wanted particle (zero based)
for(i=0; i<n_frames; i++)
{
printf("%d", first_frame + i);
for(j=0; j<n_values_per_frame; j++)
{
switch(data_type)
{
case TNG_INT_DATA:
printf("\t%"PRId64"", positions[i][particle][j].i);
break;
case TNG_FLOAT_DATA:
printf("\t%f", positions[i][particle][j].f);
break;
case TNG_DOUBLE_DATA:
printf("\t%f", positions[i][particle][j].d);
break;
default:
break;
}
printf("\n");
}
}
}
}
else
{
printf("Cannot read positions\n");
}
// Free memory
if(positions)
{
tng_particle_data_values_free(traj, positions, n_frames, n_particles,
n_values_per_frame, data_type);
}
tng_trajectory_destroy(&traj);
return(0);
}
|
