7c9f2a3309188c12cbf4f7fd1b7f725b25c6355a
[alexxy/gromacs.git] / src / gromacs / fileio / mdoutf.h
1 /*
2  * This file is part of the GROMACS molecular simulation package.
3  *
4  * Copyright (c) 2013,2014, by the GROMACS development team, led by
5  * Mark Abraham, David van der Spoel, Berk Hess, and Erik Lindahl,
6  * and including many others, as listed in the AUTHORS file in the
7  * top-level source directory and at http://www.gromacs.org.
8  *
9  * GROMACS is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public License
11  * as published by the Free Software Foundation; either version 2.1
12  * of the License, or (at your option) any later version.
13  *
14  * GROMACS is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with GROMACS; if not, see
21  * http://www.gnu.org/licenses, or write to the Free Software Foundation,
22  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
23  *
24  * If you want to redistribute modifications to GROMACS, please
25  * consider that scientific software is very special. Version
26  * control is crucial - bugs must be traceable. We will be happy to
27  * consider code for inclusion in the official distribution, but
28  * derived work must not be called official GROMACS. Details are found
29  * in the README & COPYING files - if they are missing, get the
30  * official version at http://www.gromacs.org.
31  *
32  * To help us fund GROMACS development, we humbly ask that you cite
33  * the research papers on the package. Check out http://www.gromacs.org.
34  */
35
36 #ifndef GMX_FILEIO_MDOUTF_H
37 #define GMX_FILEIO_MDOUTF_H
38
39 #include <stdio.h>
40
41 #include "../legacyheaders/types/inputrec.h"
42 #include "../legacyheaders/types/oenv.h"
43 #include "../legacyheaders/network.h"
44
45 #include "../math/vectypes.h"
46 #include "../timing/wallcycle.h"
47 #include "../utility/basedefinitions.h"
48 #include "filenm.h"
49 #include "enxio.h"
50
51 #ifdef __cplusplus
52 extern "C" {
53 #endif
54
55 struct gmx_mtop_t;
56
57 typedef struct gmx_mdoutf *gmx_mdoutf_t;
58
59 /*! \brief Allocate and initialize object to manager trajectory writing output
60  *
61  * Returns a pointer to a data structure with all output file pointers
62  * and names required by mdrun.
63  */
64 gmx_mdoutf_t init_mdoutf(FILE              *fplog,
65                          int                nfile,
66                          const t_filenm     fnm[],
67                          int                mdrun_flags,
68                          const t_commrec   *cr,
69                          const t_inputrec  *ir,
70                          struct gmx_mtop_t *mtop,
71                          const output_env_t oenv,
72                          gmx_wallcycle_t    wcycle);
73
74 /*! \brief Getter for file pointer */
75 FILE *mdoutf_get_fp_field(gmx_mdoutf_t of);
76
77 /*! \brief Getter for file pointer */
78 ener_file_t mdoutf_get_fp_ene(gmx_mdoutf_t of);
79
80 /*! \brief Getter for file pointer */
81 FILE *mdoutf_get_fp_dhdl(gmx_mdoutf_t of);
82
83 /*! \brief Getter for wallcycle timer */
84 gmx_wallcycle_t mdoutf_get_wcycle(gmx_mdoutf_t of);
85
86 /*! \brief Close TNG files if they are open.
87  *
88  * This also measures the time it takes to close the TNG
89  * files.
90  */
91 void mdoutf_tng_close(gmx_mdoutf_t of);
92
93 /*! \brief Close all open output files and free the of pointer */
94 void done_mdoutf(gmx_mdoutf_t of);
95
96 /*! \brief Routine that writes trajectory-like frames.
97  *
98  * Writes data to trn, xtc and/or checkpoint. What is written is
99  * determined by the mdof_flags defined below. Data is collected to
100  * the master node only when necessary.
101  */
102 void mdoutf_write_to_trajectory_files(FILE *fplog, t_commrec *cr,
103                                       gmx_mdoutf_t of,
104                                       int mdof_flags,
105                                       struct gmx_mtop_t *top_global,
106                                       gmx_int64_t step, double t,
107                                       t_state *state_local, t_state *state_global,
108                                       rvec *f_local, rvec *f_global);
109
110 #define MDOF_X            (1<<0)
111 #define MDOF_V            (1<<1)
112 #define MDOF_F            (1<<2)
113 #define MDOF_X_COMPRESSED (1<<3)
114 #define MDOF_CPT          (1<<4)
115 #define MDOF_IMD          (1<<5)
116
117 #ifdef __cplusplus
118 }
119 #endif
120
121 #endif /* GMX_FILEIO_MDOUTF_H */