Convert some mdrun utility code to C++
[alexxy/gromacs.git] / src / gromacs / gmxlib / viewit.cpp
1 /*
2  * This file is part of the GROMACS molecular simulation package.
3  *
4  * Copyright (c) 1991-2000, University of Groningen, The Netherlands.
5  * Copyright (c) 2001-2004, The GROMACS development team.
6  * Copyright (c) 2013,2014,2015, by the GROMACS development team, led by
7  * Mark Abraham, David van der Spoel, Berk Hess, and Erik Lindahl,
8  * and including many others, as listed in the AUTHORS file in the
9  * top-level source directory and at http://www.gromacs.org.
10  *
11  * GROMACS is free software; you can redistribute it and/or
12  * modify it under the terms of the GNU Lesser General Public License
13  * as published by the Free Software Foundation; either version 2.1
14  * of the License, or (at your option) any later version.
15  *
16  * GROMACS is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19  * Lesser General Public License for more details.
20  *
21  * You should have received a copy of the GNU Lesser General Public
22  * License along with GROMACS; if not, see
23  * http://www.gnu.org/licenses, or write to the Free Software Foundation,
24  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
25  *
26  * If you want to redistribute modifications to GROMACS, please
27  * consider that scientific software is very special. Version
28  * control is crucial - bugs must be traceable. We will be happy to
29  * consider code for inclusion in the official distribution, but
30  * derived work must not be called official GROMACS. Details are found
31  * in the README & COPYING files - if they are missing, get the
32  * official version at http://www.gromacs.org.
33  *
34  * To help us fund GROMACS development, we humbly ask that you cite
35  * the research papers on the package. Check out http://www.gromacs.org.
36  */
37 /* This file is completely threadsafe - keep it that way! */
38 #include "gmxpre.h"
39
40 #include "gromacs/legacyheaders/viewit.h"
41
42 #include <cstdlib>
43 #include <cstring>
44
45 #include "gromacs/fileio/filenm.h"
46 #include "gromacs/legacyheaders/macros.h"
47 #include "gromacs/legacyheaders/oenv.h"
48 #include "gromacs/utility/cstringutil.h"
49 #include "gromacs/utility/fatalerror.h"
50
51 static const int   can_view_ftp[] = {
52     0,
53     efEPS,           efXPM,         efXVG,          efPDB
54 };
55 #define NVIEW asize(can_view_ftp)
56 static const char* view_program[] = {
57     NULL,
58     "ghostview",    "display",      NULL,           "xterm -e rasmol"
59 };
60
61 static int can_view(int ftp)
62 {
63     int i;
64
65     for (i = 1; i < NVIEW; i++)
66     {
67         if (ftp == can_view_ftp[i])
68         {
69             return i;
70         }
71     }
72
73     return 0;
74 }
75
76 void do_view(const output_env_t oenv, const char *fn, const char *opts)
77 {
78     char        buf[STRLEN], env[STRLEN];
79     const char *cmd;
80     int         ftp, n;
81
82     if (output_env_get_view(oenv) && fn)
83     {
84         if (getenv("DISPLAY") == NULL)
85         {
86             fprintf(stderr, "Can not view %s, no DISPLAY environment variable.\n", fn);
87         }
88         else
89         {
90             ftp = fn2ftp(fn);
91             sprintf(env, "GMX_VIEW_%s", ftp2ext(ftp));
92             upstring(env);
93             switch (ftp)
94             {
95                 case efXVG:
96                     if (!(cmd = getenv(env)) )
97                     {
98                         if (getenv("GMX_USE_XMGR") )
99                         {
100                             cmd = "xmgr";
101                         }
102                         else
103                         {
104                             cmd = "xmgrace";
105                         }
106                     }
107                     break;
108                 default:
109                     if ( (n = can_view(ftp)) )
110                     {
111                         if (!(cmd = getenv(env)) )
112                         {
113                             cmd = view_program[n];
114                         }
115                     }
116                     else
117                     {
118                         fprintf(stderr, "Don't know how to view file %s", fn);
119                         return;
120                     }
121             }
122             if (strlen(cmd) )
123             {
124                 sprintf(buf, "%s %s %s &", cmd, opts ? opts : "", fn);
125                 fprintf(stderr, "Executing '%s'\n", buf);
126                 if (0 != system(buf) )
127                 {
128                     gmx_fatal(FARGS, "Failed executing command: %s", buf);
129                 }
130             }
131         }
132     }
133 }
134
135 void view_all(const output_env_t oenv, int nf, t_filenm fnm[])
136 {
137     int i;
138
139     for (i = 0; i < nf; i++)
140     {
141         if (can_view(fnm[i].ftp) && is_output(&(fnm[i])) &&
142             ( !is_optional(&(fnm[i])) || is_set(&(fnm[i])) ) )
143         {
144             do_view(oenv, fnm[i].fns[0], NULL);
145         }
146     }
147 }