path: root/src/Output2D.c

 /*@@
   @file      Output2D.c
   @date      Thu May 11 2000
   @author    Thomas Radke
   @desc 
   Functions to deal with 2D ASCII output of variables
   @enddesc 
 @@*/
 
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "cctk.h"
#include "cctk_Parameters.h"
#include "ioASCIIGH.h"

/*#define IOASCII_DEBUG*/

/* function prototypes */
int IOASCII_TimeFor2D (cGH *GH, int index);
static int CheckOutputVar (int index);
static void CheckSteerableParameters (asciiioGH *myGH);
static void SetOutputFlag (int index, const char *optstring, void *arg);


/*@@
   @routine    IOASCII_Output2DGH
   @date       Sat March 6 1999
   @author     Gabrielle Allen
   @desc
   Loops over all variables and outputs them if necessary
   @enddesc
   @calledby   CCTK_OutputGH ("IOASCII")
   @history

   @endhistory
   @var     GH
   @vdesc   Pointer to CCTK GH
   @vtype   cGH
   @vio     in
   @endvar
@@*/
int IOASCII_Output2DGH (cGH *GH)
{
  int index;
  asciiioGH *myGH;
  const char *name;


  /* Get the GH extensions for IOASCII */
  myGH = (asciiioGH *) GH->extensions [CCTK_GHExtensionHandle ("IOASCII")];

  CheckSteerableParameters (myGH);

  /* Return if no output is required */
  if (myGH->out2D_every <= 0)
  {
    return (0);
  }

  /* Loop over all variables */
  for (index = 0; index < CCTK_NumVars (); index++) 
  {

    /* Check if this variable should be output */
    if (! IOASCII_TimeFor2D (GH, index))
    {
       continue;
    }

    /* Get the variable name for this index (for filename) */
    name = CCTK_VarName (index);

#ifdef IO_DEBUG
    printf ("\nIn IOASCII Output2DGH\n---------------------\n");
    printf ("  Index = %d\n", index);
    printf ("  Variable = -%s-\n", name);
    printf ("  On iteration %d\n", GH->cctk_iteration);
    printf ("  Last output iteration was = %d\n", myGH->out2D_last [index]);
#endif
  
    /* Make the IO call */
    IOASCII_Write2D (GH, index, name);
        
    /* Register GF as having 2D output this iteration */
    myGH->out2D_last [index] = GH->cctk_iteration;

  } /* end of loop over all variables */

  return (0);
}


/*@@
   @routine    IOASCII_Output2DVarAs
   @date       Sat March 6 1999
   @author     Gabrielle Allen
   @desc
   unconditional output of a variable using the IOASCII output method
   @enddesc
   @calledby IOASCII_Output2DGH, CCTK_OutputVarAsByMethod ("IOASCII")
   @var     GH
   @vdesc   Pointer to CCTK GH
   @vtype   cGH
   @vio     in
   @vcomment
   @endvar
   @var     fullname
   @vdesc   complete name of variable to output
   @vtype   const char *
   @vio     in
   @vcomment
   @endvar
   @var     alias
   @vdesc   alias name of variable to output (used to generate output filename)
   @vtype   const char *
   @vio     in
   @vcomment
   @endvar
@@*/
int IOASCII_Output2DVarAs (cGH *GH, const char *fullname, const char *alias)
{
  int index;


  index = CCTK_VarIndex (fullname);

#ifdef IO_DEBUG
  printf ("\nIn IOASCII Output2DVarAs\n-------------------\n");
  printf ("  Fullname = -%s-\n", fullname);
  printf ("  Alias = -%s-\n", alias);
  printf ("  Index = %d\n", index);
#endif

  if (CheckOutputVar (index) == 0)
  {
    IOASCII_Write2D (GH, index, alias);
  }

  return (0);
}


 /*@@
   @routine    IOASCII_TimeFor2D
   @date       Sat March 6 1999
   @author     Gabrielle Allen
   @desc 
   Decides if it is time to output a variable using the IO 2D output
   method
   @enddesc 
   @calledby   IOASCII_Output2DGH
   @history 
 
   @endhistory 
   @var     GH
   @vdesc   Pointer to CCTK GH
   @vtype   cGH
   @vio     in
   @vcomment 
   @endvar 
   @var     index
   @vdesc   index of variable
   @vtype   int
   @vio     in
   @vcomment 
   @endvar 
@@*/
int IOASCII_TimeFor2D (cGH *GH, int index)
{
  int return_type;
  asciiioGH *myGH;
  char *fullname;


  /* Default is do not do output */
  return_type = 0;

  /* Get the GH extensions for IOASCII */
  myGH = (asciiioGH *) GH->extensions [CCTK_GHExtensionHandle ("IOASCII")];

  CheckSteerableParameters (myGH);

  /* return if no output requested */
  if (myGH->out2D_every <= 0)
  {
    return (0);
  }

  /* Check if this variable should be output */
  if (myGH->do_out2D [index] && (GH->cctk_iteration % myGH->out2D_every == 0)) 
  {
    /* Check if variable wasn't already output this iteration */
    if (myGH->out2D_last[index] == GH->cctk_iteration)
    {
      fullname = CCTK_FullName (index);
      CCTK_VWarn (5, __LINE__, __FILE__, CCTK_THORNSTRING,
                  "Already done IOASCII 2D output for '%s' in current "
                  "iteration (probably via triggers)", fullname);
      free (fullname);
    }
    else
    {
      return_type = 1;
    }
  }

  return (return_type);
}


/*@@
   @routine    IOASCII_TriggerOutput2D
   @date       Sat March 6 1999
   @author     Gabrielle Allen
   @desc
   Triggers the output a variable using the IOASCII output method
   method
   @enddesc
   @calledby   CCTK scheduler
   @var     GH
   @vdesc   Pointer to CCTK GH
   @vtype   cGH
   @vio     in
   @vcomment
   @endvar
   @var     index
   @vdesc   index of variable to output
   @vtype   int
   @vio     in
   @vcomment
   @endvar
@@*/
int IOASCII_TriggerOutput2D (cGH *GH, int index)
{
  const char *var;
  asciiioGH *myGH;

  myGH = (asciiioGH *) GH->extensions [CCTK_GHExtensionHandle ("IOASCII")];
  
  var = CCTK_VarName (index);

  /* Do the 2D output */
#ifdef IO_DEBUG
  printf ("\nIn IOASCII TriggerOutput2D\n---------------------\n");
  printf ("  Index = %d\n", index);
  printf ("  Variable = -%s-\n", var);
#endif
  
  IOASCII_Write2D (GH, index, var);
  
  /* Register variables as having 2D output this iteration */
  myGH->out2D_last [index] = GH->cctk_iteration;
  
  return (0);
}


/**************************** local functions ******************************/
/* check if steerable parameters have changed */
static void CheckSteerableParameters (asciiioGH *myGH)
{
  DECLARE_CCTK_PARAMETERS
  int times_set;
  static int out2D_vars_lastset = -1;


  /* How often to output */
  myGH->out2D_every = out_every > 0 ? out_every : -1;
  if (out2D_every > 0)
  {
    myGH->out2D_every = out2D_every;
  }

  /* re-parse the 'out2D_vars' parameter if it was changed */
  times_set = CCTK_ParameterQueryTimesSet ("out2D_vars", CCTK_THORNSTRING);
  if (times_set != out2D_vars_lastset) 
  {
    memset (myGH->do_out2D, 0, CCTK_NumVars ());
    CCTK_TraverseString (out2D_vars, SetOutputFlag, myGH->do_out2D,
                         CCTK_GROUP_OR_VAR);

    /* Save the last setting of 'out2D_vars' parameter */
    out2D_vars_lastset = times_set;
  }

}


/* check if this variable can be output (static conditions) */
static int CheckOutputVar (int index)
{
  int groupindex;
  cGroup groupinfo;
  char *fullname;
  const char *errormsg;


  /* get the variable group indormation */
  groupindex = CCTK_GroupIndexFromVarI (index);
  CCTK_GroupData (groupindex, &groupinfo);

  /* check if variable is of dimension 3 and is a GF or array type */
  errormsg = NULL;
  if (groupinfo.dim != 3)
  {
    errormsg = "No IOASCII 2D output for '%s' (dim != 3)";
  }
  else if (groupinfo.grouptype != CCTK_GF && groupinfo.grouptype != CCTK_ARRAY)
  {
    errormsg = "No IOASCII 2D output for '%s' (not a grid function or array)";
  }

  if (errormsg)
  {
    fullname = CCTK_FullName (index);
    CCTK_VWarn (2, __LINE__, __FILE__, CCTK_THORNSTRING, errormsg, fullname);
    free (fullname);
  }

  return (errormsg != NULL);
}


/* callback for CCTK_TraverseString() to set the output flag 
   for the given variable */
static void SetOutputFlag (int index, const char *optstring, void *arg)
{
  char *flags = (char *) arg;


  /* Check the variable type */
  if (CheckOutputVar (index) == 0)
  {
    flags[index] = 1;
  }

  if (optstring)
  {
    CCTK_VWarn (5, __LINE__, __FILE__, CCTK_THORNSTRING,
                "Optional string '%s' in variable name ignored", optstring);
  }
}