write_model_state.c

来自「超强的大尺度水文模拟工具」· C语言 代码 · 共 221 行

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

static char vcid[] = "$Id: write_model_state.c,v 4.2.2.4 2004/05/06 19:57:37 tbohn Exp $";

#if SAVE_STATE

void write_model_state(dist_prcp_struct    *prcp,
		       global_param_struct *gp,
		       int                  Nveg,
		       int                  cellnum,
		       outfiles_struct     *outfiles,
		       soil_con_struct     *soil_con,
		       char                 STILL_STORM,
		       int                  DRY_TIME) 
/*********************************************************************
  write_model_state      Keith Cherkauer           April 14, 2000

  This subroutine saves the model state at hour 0 of the date 
  defined in the global control file using STATEDAY, STATEMONTH,
  and STATEYEAR.  The saved files can then be used to initialize 
  the model to the same state as when the files were created.

  Soil moisture, soil thermal, and snowpack variables  are stored 
  for each vegetation type and snow band.  However moisture variables
  from the distributed precipitation model are averaged so that the
  model is restarted with mu = 1.

  Modifications:
  04-10-03 Rewritten to handle updates to vicNl_def.h and to write
           the file as binary to minimize write time and differences
           with simulations started with the state file.         KAC
  04-10-03 Model is now restarted with the correct values for mu
           and LAST_STORM
  06-03-03 Modified to create ASCII as well as BINARY state file.  KAC
  09-05-2003 Modified to print space before dz_node for ASCII state
             file, this corrects a problem with state files created
             for models using the Cherkauer and Lettenmaier (1999) heat 
             flux formulation.                                   KAC
  09-Oct-03 Added "\n" after mu in ASCII file, to jive with
	    read_initial_model_state.                            TJB

*********************************************************************/
{
  extern option_struct options;

  double tmpval;
  double Nsum;
  int    veg;
  int    band;
  int    lidx;
  int    nidx;
  int    dist;
  int    Ndist;
  int    Nbands;
  int    byte, Nbytes;

  cell_data_struct     ***cell;
  snow_data_struct      **snow;
  energy_bal_struct     **energy;
  veg_var_struct       ***veg_var;

  if(options.DIST_PRCP) 
    Ndist = 2;
  else 
    Ndist = 1;
  Nbands = options.SNOW_BAND;

  cell    = prcp->cell;
  veg_var = prcp->veg_var;
  snow    = prcp->snow;
  energy  = prcp->energy;
  
  /* write cell information */
  if ( options.BINARY_STATE_FILE ) {
    fwrite( &cellnum, 1, sizeof(int), outfiles->statefile );
    fwrite( &Nveg, 1, sizeof(int), outfiles->statefile );
    fwrite( &Nbands, 1, sizeof(int), outfiles->statefile );
  }
  else {
    fprintf( outfiles->statefile, "%i %i %i", cellnum, Nveg, Nbands );
  }
  // This stores the number of bytes from after this value to the end 
  // of the line.  DO NOT CHANGE unless you have changed the values
  // written to the state file.
  // IF YOU EDIT THIS FILE: UPDATE THIS VALUE!
  if ( options.BINARY_STATE_FILE ) {
    Nbytes = ( options.Nnode * sizeof(double) // dz_node
	       + sizeof(char) // STILL_STORM
	       + sizeof(int) // DRY_TIME
	       + Nveg * sizeof(double) // mu
	       + Nveg * Nbands * 2 * sizeof(int) // veg & band
	       + Nveg * Nbands * Ndist * options.Nlayer * sizeof(double) // soil moisture
	       + Nveg * Nbands * Ndist * options.Nlayer * sizeof(double) // soil ice
	       + (Nveg-1) * Nbands * sizeof(double) // dew
	       + Nveg * Nbands * sizeof(int) // last_snow
	       + Nveg * Nbands * sizeof(char) // MELTING
	       + Nveg * Nbands * sizeof(double) * 9 // other snow parameters
	       + Nveg * Nbands * options.Nnode * sizeof(double) ); // soil temperatures
    fwrite( &Nbytes, 1, sizeof(int), outfiles->statefile );
  }

  /* Write soil thermal node depths */
  Nsum = 0;
  for ( nidx = 0; nidx < options.Nnode; nidx++ ) {
    if ( options.BINARY_STATE_FILE )
      fwrite( &soil_con->dz_node[nidx], 1, sizeof(double), 
	      outfiles->statefile );
    else
      fprintf( outfiles->statefile, " %f", soil_con->dz_node[nidx] );
  }    
  if ( !options.BINARY_STATE_FILE )
    fprintf( outfiles->statefile, "\n" );

  // Store distributed precipitation variables
  if ( options.BINARY_STATE_FILE ) {
    fwrite( &STILL_STORM, 1, sizeof(char), outfiles->statefile );
    fwrite( &DRY_TIME, 1, sizeof(int), outfiles->statefile );
  }
  else {
    fprintf( outfiles->statefile, "%i %i\n", STILL_STORM, DRY_TIME );
  }

  /* Output for all vegetation types */
  for ( veg = 0; veg <= Nveg; veg++ ) {

    // Store distributed precipitation fraction
    if ( options.BINARY_STATE_FILE )
      fwrite( &prcp->mu[veg], 1, sizeof(double), outfiles->statefile );
    else
      fprintf( outfiles->statefile, "%f\n", prcp->mu[veg] );

    /* Output for all snow bands */
    for ( band = 0; band < Nbands; band++ ) {
      
      /* Write cell identification information */
      if ( options.BINARY_STATE_FILE ) {
	fwrite( &veg, 1, sizeof(int), outfiles->statefile );
	fwrite( &band, 1, sizeof(int), outfiles->statefile );
      }
      else {
	fprintf( outfiles->statefile, "%i %i", veg, band );
      }
      
      for ( dist = 0; dist < Ndist; dist ++ ) {
	// Store both wet and dry fractions if using distributed precipitation

	/* Write total soil moisture */
	for ( lidx = 0; lidx < options.Nlayer; lidx++ ) {
	  tmpval = cell[dist][veg][band].layer[lidx].moist;
	  if ( options.BINARY_STATE_FILE )
	    fwrite( &tmpval, 1, sizeof(double), outfiles->statefile );
	  else
	    fprintf( outfiles->statefile, " %f", tmpval );
	}

	/* Write ice content */
	for ( lidx = 0; lidx < options.Nlayer; lidx++ ) {
	  tmpval = cell[dist][veg][band].layer[lidx].ice;
	  if ( options.BINARY_STATE_FILE )
	    fwrite( &tmpval, 1, sizeof(double), outfiles->statefile );
	  else
	    fprintf( outfiles->statefile, " %f", tmpval );
	}
      
	/* Write dew storage */
	if ( veg < Nveg ) {
	  tmpval = veg_var[dist][veg][band].Wdew;
	  if ( options.BINARY_STATE_FILE )
	    fwrite( &tmpval, 1, sizeof(double), outfiles->statefile );
	  else
	    fprintf( outfiles->statefile, " %f", tmpval );
	}
      }
      
      /* Write snow data */
      if ( options.BINARY_STATE_FILE ) {
	fwrite( &snow[veg][band].last_snow, 1, sizeof(int), outfiles->statefile );
	fwrite( &snow[veg][band].MELTING, 1, sizeof(char), outfiles->statefile );
	fwrite( &snow[veg][band].coverage, 1, sizeof(double), outfiles->statefile );
	fwrite( &snow[veg][band].swq, 1, sizeof(double), outfiles->statefile );
	fwrite( &snow[veg][band].surf_temp, 1, sizeof(double), outfiles->statefile );
	fwrite( &snow[veg][band].surf_water, 1, sizeof(double), outfiles->statefile );
	fwrite( &snow[veg][band].pack_temp, 1, sizeof(double), outfiles->statefile );
	fwrite( &snow[veg][band].pack_water, 1, sizeof(double), outfiles->statefile );
	fwrite( &snow[veg][band].density, 1, sizeof(double), outfiles->statefile );
	fwrite( &snow[veg][band].coldcontent, 1, sizeof(double), outfiles->statefile );
	fwrite( &snow[veg][band].snow_canopy, 1, sizeof(double), outfiles->statefile );
      }
      else {
	fprintf( outfiles->statefile, " %i %i %f %f %f %f %f %f %f %f %f", 
		 snow[veg][band].last_snow, (int)snow[veg][band].MELTING, 
		 snow[veg][band].coverage, snow[veg][band].swq, 
		 snow[veg][band].surf_temp, snow[veg][band].surf_water, 
		 snow[veg][band].pack_temp, snow[veg][band].pack_water, 
		 snow[veg][band].density, snow[veg][band].coldcontent, 
		 snow[veg][band].snow_canopy );
      }
      
      /* Write soil thermal node temperatures */
      for ( nidx = 0; nidx < options.Nnode; nidx++ ) 
	if ( options.BINARY_STATE_FILE )
	  fwrite( &energy[veg][band].T[nidx], 1, sizeof(double), 
		  outfiles->statefile );
	else
	  fprintf( outfiles->statefile, " %f", energy[veg][band].T[nidx] );

      if ( !options.BINARY_STATE_FILE ) fprintf( outfiles->statefile, "\n" );
      
    } 
  }

  /* Force file to be written */
  fflush(outfiles->statefile);

}

#endif