aboutsummaryrefslogtreecommitdiff
path: root/src/Write.c
blob: a6180335a5db28b867e48f315ba7ccd9ffeb5cd5 (plain)
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
/*@@
   @routine    Write.c
   @date       18th September 1999
   @author     Gabrielle Allen
   @desc 
   Writes scalar grid variable data.
   @enddesc 
   @calls    
   @calledby   
   @history
   @hauthor  @hdate 
   @hdesc 
   @hendhistory
@@*/

#include <stdio.h>
#include <stdlib.h>

#include "cctk.h"
#include "cctk_parameters.h"
#include "iobasicGH.h"

void IOBasic_Write (cGH *GH, int index, const char *alias)
{
  DECLARE_CCTK_PARAMETERS
  char *openmode;
  FILE *file;
  CCTK_REAL *data_real;
  CCTK_INT *data_int;


  /* first, check if variable has storage assigned */
  if (! CCTK_QueryGroupStorageI (GH, CCTK_GroupIndexFromVarI (index))) {
    char *msg, *fullname;

    fullname = CCTK_FullName (index);
    msg = (char *) malloc (200 * sizeof (char) + strlen (fullname));
    sprintf (msg, "No scalar output for '%s' (no storage)", fullname);
    CCTK_WARN (2, msg);
    free (fullname);
    free (msg);
    return;
  }

  /* Open the file (we write only on proc0) */
  if (CCTK_MyProc (GH) == 0) 
  {
    char *fname;
    iobasicGH *myGH;

    myGH = (iobasicGH *) GH->extensions [CCTK_GHExtensionHandle ("IOBasic")];

    fname = (char *) malloc (strlen (myGH->outdirScalar) + strlen (alias) + 5);
    sprintf (fname, "%s/%s.tl", myGH->outdirScalar, alias);

    /* see if output files for this alias name were already created */
    if (GetNamedData (myGH->filenameListScalar, fname) == NULL)
      openmode = "w";
    else
      openmode = "a";

    file = fopen(fname,openmode);
    if (file == NULL) {
      char *msg = (char *) malloc (100 + strlen (fname));

      sprintf (msg, "Could not open output file '%s'", fname);
      CCTK_WARN (3, msg);
      free (msg);
      free (fname);
      return;
    }
    
    if (*openmode == 'w') {
      char title_start_char;

      if (CCTK_Equals(outScalar_style,"gnuplot")) 
        title_start_char = '#';
      else
        title_start_char = 34;    /* this is for xgraph */

      fprintf (file,"%c%s v time\n",title_start_char,alias);

      /* just store a non-NULL pointer in database */
      StoreNamedData (&myGH->filenameListScalar, fname, (void *) 1);
    }

    switch (CCTK_VarTypeI(index)) {
    case CCTK_VARIABLE_REAL:
      data_real  = ((CCTK_REAL ***) GH->data) [index][0]; 
      fprintf(file,"%f %25.13f\n",GH->cctk_time,*data_real);
      break;
    case CCTK_VARIABLE_INT:
      data_int  = ((CCTK_INT ***) GH->data) [index][0]; 
      fprintf(file,"%f %d\n",GH->cctk_time,*data_int);
      break;
    default:
      CCTK_WARN (3, "Unsupported data type");
      break;
    }
    
    fclose(file);
    free (fname);

  }

  USE_CCTK_PARAMETERS

}