1 /* This code is part of the tng binary trajectory format.
3 * Written by Magnus Lundborg
4 * Copyright (c) 2012-2013, The GROMACS development team.
5 * check out http://www.gromacs.org for more information.
8 * This program is free software; you can redistribute it and/or
9 * modify it under the terms of the Revised BSD License.
12 #ifdef USE_STD_INTTYPES_H
18 #include "tng/tng_io.h"
20 int main(int argc
, char **argv
)
22 tng_trajectory_t traj
;
23 union data_values
***positions
= 0; /* A 3-dimensional array to be populated */
24 int64_t n_particles
, n_values_per_frame
, n_frames
, tot_n_frames
;
28 /* Set a default frame range */
29 int first_frame
= 0, last_frame
= 50;
30 char atom_name
[64], res_name
[64], chain_name
[64];
34 printf("No file specified\n");
36 printf("tng_io_read_pos <tng_file> [particle number = %d] "
37 "[first_frame = %d] [last_frame = %d]\n",
38 particle
, first_frame
, last_frame
);
42 /* A reference must be passed to allocate memory */
43 if(tng_trajectory_init(&traj
) != TNG_SUCCESS
)
45 tng_trajectory_destroy(&traj
);
48 tng_input_file_set(traj
, argv
[1]);
50 /* Read the file headers */
51 tng_file_headers_read(traj
, TNG_USE_HASH
);
55 particle
= strtol(argv
[2], 0, 10);
58 first_frame
= strtol(argv
[3], 0, 10);
61 last_frame
= strtol(argv
[4], 0, 10);
66 if(tng_num_frames_get(traj
, &tot_n_frames
) != TNG_SUCCESS
)
68 printf("Cannot determine the number of frames in the file\n");
69 tng_trajectory_destroy(&traj
);
73 printf("%"PRId64
" frames in file\n", tot_n_frames
);
75 if(last_frame
> tot_n_frames
- 1)
77 last_frame
= tot_n_frames
- 1;
80 n_frames
= last_frame
- first_frame
+ 1;
82 if(tng_atom_name_of_particle_nr_get(traj
, particle
, atom_name
,
85 tng_residue_name_of_particle_nr_get(traj
, particle
, res_name
,
88 tng_chain_name_of_particle_nr_get(traj
, particle
, chain_name
,
89 sizeof(chain_name
)) ==
92 printf("Particle: %s (%s: %s)\n", atom_name
, chain_name
, res_name
);
96 printf("Particle name not found\n");
99 /* Get the positions of all particles in the requested frame range.
100 The positions are stored in the positions array.
101 N.B. No proper error checks. */
102 if(tng_particle_data_interval_get(traj
, TNG_TRAJ_POSITIONS
, first_frame
,
103 last_frame
, TNG_USE_HASH
, &positions
, &n_particles
, &n_values_per_frame
,
104 &data_type
) == TNG_SUCCESS
)
106 if(particle
>= n_particles
)
108 printf("Chosen particle out of range. Only %"PRId64
" particles in trajectory.\n", n_particles
);
112 /* Print the positions of the wanted particle (zero based) */
113 for(i
=0; i
<n_frames
; i
++)
115 printf("%d", first_frame
+ i
);
116 for(j
=0; j
<n_values_per_frame
; j
++)
121 printf("\t%"PRId64
"", positions
[i
][particle
][j
].i
);
124 printf("\t%f", positions
[i
][particle
][j
].f
);
126 case TNG_DOUBLE_DATA
:
127 printf("\t%f", positions
[i
][particle
][j
].d
);
139 printf("Cannot read positions\n");
145 tng_particle_data_values_free(traj
, positions
, n_frames
, n_particles
,
146 n_values_per_frame
, data_type
);
148 tng_trajectory_destroy(&traj
);