atm_init.c

Go to the documentation of this file.

/*
  This file is part of MPTRAC.
  
  MPTRAC is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.
  
  MPTRAC is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.
  
  You should have received a copy of the GNU General Public License
  along with MPTRAC. If not, see <http://www.gnu.org/licenses/>.
  
  Copyright (C) 2013-2026 Forschungszentrum Juelich GmbH
*/
 
#include "mptrac.h"
 
/* ------------------------------------------------------------
   Functions...
   ------------------------------------------------------------ */
 
void usage(
  void);
 
/* ------------------------------------------------------------
   Main...
   ------------------------------------------------------------ */
 
int main(
  int argc,
  char *argv[]) {
 
  atm_t *atm;
 
  ctl_t ctl;
 
  /* Allocate... */
  ALLOC(atm, atm_t, 1);
 
  /* Print usage information... */
  USAGE;
 
  /* Check arguments... */
  if (argc < 3)
    ERRMSG("Missing or invalid command-line arguments.\n\n"
           "Usage: atm_init <ctl> <atm_out> [KEY VALUE ...]\n\n"
           "Use -h for full help.");
 
  /* Read control parameters... */
  mptrac_read_ctl(argv[1], argc, argv, &ctl);
  const int ens =
    (int) scan_ctl(argv[1], argc, argv, "INIT_ENS", -1, "0", NULL);
  const double t0 = scan_ctl(argv[1], argc, argv, "INIT_T0", -1, "0", NULL);
  const double t1 = scan_ctl(argv[1], argc, argv, "INIT_T1", -1, "0", NULL);
  const double dt = scan_ctl(argv[1], argc, argv, "INIT_DT", -1, "1", NULL);
  const double z0 = scan_ctl(argv[1], argc, argv, "INIT_Z0", -1, "0", NULL);
  const double z1 = scan_ctl(argv[1], argc, argv, "INIT_Z1", -1, "0", NULL);
  const double dz = scan_ctl(argv[1], argc, argv, "INIT_DZ", -1, "1", NULL);
  const double lon0 =
    scan_ctl(argv[1], argc, argv, "INIT_LON0", -1, "0", NULL);
  const double lon1 =
    scan_ctl(argv[1], argc, argv, "INIT_LON1", -1, "0", NULL);
  const double dlon =
    scan_ctl(argv[1], argc, argv, "INIT_DLON", -1, "1", NULL);
  const double lat0 =
    scan_ctl(argv[1], argc, argv, "INIT_LAT0", -1, "0", NULL);
  const double lat1 =
    scan_ctl(argv[1], argc, argv, "INIT_LAT1", -1, "0", NULL);
  const double dlat =
    scan_ctl(argv[1], argc, argv, "INIT_DLAT", -1, "1", NULL);
  const double st = scan_ctl(argv[1], argc, argv, "INIT_ST", -1, "0", NULL);
  const double sz = scan_ctl(argv[1], argc, argv, "INIT_SZ", -1, "0", NULL);
  const double slon =
    scan_ctl(argv[1], argc, argv, "INIT_SLON", -1, "0", NULL);
  const double slat =
    scan_ctl(argv[1], argc, argv, "INIT_SLAT", -1, "0", NULL);
  const double sx = scan_ctl(argv[1], argc, argv, "INIT_SX", -1, "0", NULL);
  const double ut = scan_ctl(argv[1], argc, argv, "INIT_UT", -1, "0", NULL);
  const double uz = scan_ctl(argv[1], argc, argv, "INIT_UZ", -1, "0", NULL);
  const double ulon =
    scan_ctl(argv[1], argc, argv, "INIT_ULON", -1, "0", NULL);
  const double ulat =
    scan_ctl(argv[1], argc, argv, "INIT_ULAT", -1, "0", NULL);
  const int even =
    (int) scan_ctl(argv[1], argc, argv, "INIT_EVENLY", -1, "0", NULL);
  const int rep =
    (int) scan_ctl(argv[1], argc, argv, "INIT_REP", -1, "1", NULL);
  const double m = scan_ctl(argv[1], argc, argv, "INIT_MASS", -1, "0", NULL);
  const double vmr = scan_ctl(argv[1], argc, argv, "INIT_VMR", -1, "0", NULL);
  const double bellrad =
    scan_ctl(argv[1], argc, argv, "INIT_BELLRAD", -1, "0", NULL);
  const int idx_offset =
    (int) scan_ctl(argv[1], argc, argv, "INIT_IDX_OFFSET", -1, "0", NULL);
 
  /* Initialize random number generator... */
  gsl_rng_env_setup();
  gsl_rng *rng = gsl_rng_alloc(gsl_rng_default);
 
  /* Create grid... */
  for (double t = t0; t <= t1; t += dt)
    for (double z = z0; z <= z1; z += dz)
      for (double lon = lon0; lon <= lon1; lon += dlon)
        for (double lat = lat0; lat <= lat1; lat += dlat)
          for (int irep = 0; irep < rep; irep++) {
 
            /* Set position... */
            double rg = gsl_ran_gaussian_ziggurat(rng, st / 2.3548);
            double ru = ut * (gsl_rng_uniform(rng) - 0.5);
            atm->time[atm->np] = (t + rg + ru);
 
            rg = gsl_ran_gaussian_ziggurat(rng, sz / 2.3548);
            ru = uz * (gsl_rng_uniform(rng) - 0.5);
            atm->p[atm->np] = P(z + rg + ru);
 
            rg = gsl_ran_gaussian_ziggurat(rng, slon / 2.3548);
            double rx =
              gsl_ran_gaussian_ziggurat(rng, DX2DEG(sx, lat) / 2.3548);
            ru = ulon * (gsl_rng_uniform(rng) - 0.5);
            atm->lon[atm->np] = (lon + rg + rx + ru);
 
            if (ctl.qnt_ens >= 0)
              atm->q[ctl.qnt_ens][atm->np] = ens;
 
            do {
              rg = gsl_ran_gaussian_ziggurat(rng, slat / 2.3548);
              rx = gsl_ran_gaussian_ziggurat(rng, DY2DEG(sx) / 2.3548);
              ru = ulat * (gsl_rng_uniform(rng) - 0.5);
              atm->lat[atm->np] = (lat + rg + rx + ru);
            } while (even && gsl_rng_uniform(rng) >
                     fabs(cos(DEG2RAD(atm->lat[atm->np]))));
 
            /* Apply cosine bell (Williamson et al., 1992)... */
            if (bellrad > 0) {
              double x0[3], x1[3];
              geo2cart(0.0, 0.5 * (lon0 + lon1), 0.5 * (lat0 + lat1), x0);
              geo2cart(0.0, atm->lon[atm->np], atm->lat[atm->np], x1);
              const double rad =
                RE * acos(DOTP(x0, x1) / sqrt(DOTP(x0, x0)) /
                          sqrt(DOTP(x1, x1)));
              if (rad > bellrad)
                continue;
              if (ctl.qnt_m >= 0)
                atm->q[ctl.qnt_m][atm->np] =
                  0.5 * (1. + cos(M_PI * rad / bellrad));
              if (ctl.qnt_vmr >= 0)
                atm->q[ctl.qnt_vmr][atm->np] =
                  0.5 * (1. + cos(M_PI * rad / bellrad));
            }
 
            /* Set particle counter... */
            if ((++atm->np) > NP)
              ERRMSG("Too many particles!");
          }
 
  /* Check number of air parcels... */
  if (atm->np <= 0)
    ERRMSG("Did not create any air parcels!");
 
  /* Initialize mass... */
  if (ctl.qnt_m >= 0 && bellrad <= 0)
    for (int ip = 0; ip < atm->np; ip++)
      atm->q[ctl.qnt_m][ip] = m / atm->np;
 
  /* Initialize volume mixing ratio... */
  if (ctl.qnt_vmr >= 0 && bellrad <= 0)
    for (int ip = 0; ip < atm->np; ip++)
      atm->q[ctl.qnt_vmr][ip] = vmr;
 
  /* Initialize air parcel index... */
  if (ctl.qnt_idx >= 0)
    for (int ip = 0; ip < atm->np; ip++)
      atm->q[ctl.qnt_idx][ip] = idx_offset + ip;
 
  /* Initialize age of air... */
  if (ctl.qnt_aoa >= 0)
    for (int ip = 0; ip < atm->np; ip++)
      atm->q[ctl.qnt_aoa][ip] = atm->time[ip];
 
  /* Save data... */
  mptrac_write_atm(argv[2], &ctl, atm, 0);
 
  /* Free... */
  gsl_rng_free(rng);
  free(atm);
 
  return EXIT_SUCCESS;
}
 
/*****************************************************************************/
 
void usage(
  void) {
 
  printf("\nMPTRAC atm_init tool.\n\n");
  printf
    ("Create an atmospheric data file with initial air parcel positions.\n");
  printf("\n");
  printf("Usage:\n");
  printf("  atm_init <ctl> <atm_out> [KEY VALUE ...]\n");
  printf("\n");
  printf("Arguments:\n");
  printf("  <ctl>      Control file.\n");
  printf("  <atm_out>  Atmospheric output file.\n");
  printf("  [KEY VALUE]  Optional control parameters.\n");
  printf("\nFurther information:\n");
  printf("  Manual: https://slcs-jsc.github.io/mptrac/\n");
}