AfrtProcess.h

Go to the documentation of this file.

/*
 * AfrtProcess.h
 *
 *  Created on: May, 2018
 *      Author: Sam Anderson
 */
 
#ifndef INCLUDE_AfrtProcess_H_
#define INCLUDE_AfrtProcess_H_
 
#include <math.h>
#include <netcdf>
#include <string>
#include <AfrtConstants.h>
 
#ifdef WITH_MPI
#include <mpi.h>
#endif
 
using namespace std;
using namespace netCDF;
using namespace netCDF::exceptions;
 
static const int NUM_PACE_BANDS = 61;
extern const double pace_wavelengths[NUM_PACE_BANDS];
 
static const double EQUAL_VOLUME = 1.0;
static const double EQUAL_AREA = 0.5;
static const int RADIUS_MAXIMUM = 2.0;
 
static const int SPHEROID = -1;
static const int CYLINDER = -2;
static const int CHEBYSHEV = 1;
 
static const int JUNGE = 1;
static const int DEIRMENDJIAN = 2;
static const int LOGNORMAL = 3;
static const int BIMODAL = 4;
static const int TRIMODAL = 5;
 
static const int nw = 250;
static const int ntau = 10;
static const int nrh = 10;
static const int nsdt = 100;
static const int nwnd = 8;
static const int ntf = 1801;
static const int nmd = 3;
static const int nstk = 4;
static const int nlyr = 200;
static const int nozn = 101;
static const int ndlvl = 161;
static const int nsig = 10;
static const int nsolz = 25;
static const int nthe = 25;
static const int nsza = 25;
static const int nphi = 46;
static const int ntrx = 16;
static const int npti = 8;
 
/* AFPHS
 *
 */
struct phs_in {
    string infile;
    string indir;
    int inw;
    int insd;
    int ilam1;
    int ilam2;
    int isd1;
    int isd2;
    int ifunc[nsdt];
    int jnorm[nsdt];
    int jrgm[nsdt];
    int mfunc[nsdt];
    int irh[nsdt];
    int iset[nsdt];
    char xtitle[nsdt][80];
    double ww[nw][nsdt];
    double refr[nw][nsdt][nmd];
    double refi[nw][nsdt][nmd];
    double rg[nw][nsdt][nmd];
    double sig[nw][nsdt][nmd];
    double npar[nw][nsdt][nmd];
    double dx[nw][nsdt][nmd];
    double epar[nw][nsdt][nmd];
};
 
struct phs_out {
    string outfile;
    string outdir;
    //    double angl[ntf];
    int_1darray ifunc;
    int_1darray mfunc;
    double_1darray angl;
    double_1darray ww;
    double_3darray refr;
    double_3darray refi;
    double_3darray rg;
    double_3darray sig;
    double_3darray npar;
    double_3darray dx;
    double_3darray epar;
    double_3darray rmin;
    double_3darray rmax;
    double_3darray deltar;
    double_3darray deltax;
    double_2darray r11;
    double_2darray r22;
    double_2darray r33;
    double_2darray r44;
    double_2darray reff;
    double_2darray veff;
    double_2darray ccnsml;
    double_2darray bsr;
    double_2darray salb;
    double_2darray asf;
    double_2darray qscat;
    double_2darray qext;
    double_3darray phfu;
    double_3darray pol;
    double_3darray wt;
    double_3darray rbar;
    double_3darray dnzp;
    double_3darray dnp;
    double_3darray dvp;
    double_3darray sumnp;
    double_3darray thd;
    double_4darray t;
};
 
/* AFOCN
 *
 */
struct ocn_in {
    string infile;
    string indir;
    int    inw;
    int    iopt;
    int    isig;
    int    isolz;
    int    insza;
    int    inthe;
    int    inphi;
    int    insig;
    int    ilam1;
    int    ilam2;
    int    iprob;
    double dlth;
    double dlph;
    double xr;
    double xi;
    double v;
    double thcel;
    double phcel;
    double wav[nw];
    double tha[nthe];
    double sig_hg[nsig];
    double solz[nsolz];
};
 
struct ocn_out {
    string outfile;
    string outdir;
    double_1darray thp;
    double_1darray phcnd;
    double_1darray thcnd;
    double_6darray txx;
    double_6darray pti;
    double_5darray xinlpi;
    double_5darray xi_hgpi;
    double_5darray xin2pi;
    double_5darray xin2top;
};
 
/* AFRT1
 *
 */
struct rt1_in {
    string infile;
    string phsdir;
    string rt1dir;
    string ozone_lut;
    string atm_lut;
    string coeff_lut;
    int nsd;
    int iair[nmd];
    int iprin;
    int ipsudo;
    int ioptn;
    int idust;
    int iaprof;
    int iblyr;
    int icld;
    int inch;
    int int55;
    int inrh;
    int ikset;
    int ilam1;
    int ilam2;
    int isd1;
    int isd2;
    int itau1;
    int itau2;
    int kznum;
    int krhum[nrh];
    int nortau;
    int ioznab;
    int nrmww;
    double deltau;
    double psrfc;
    double ht_dust;
    double sigma_dust;
    double taum55[ntau];
    double pcldtop;
    double pcldbtm;
    double xlamb[nw];
    double c0[nw];
    double c1[nw];
    double c2[nw];
    double beta[nw];
    double rho[nw];
    double ppo[nozn];
    double x101[nozn];
    double temp101[nozn];
    double htdd[nlyr];
    double ppodd[nlyr];
    double dxdd[nlyr];
    double znpdd[nlyr];
    double qst[nw][nsdt][ntau];
    double qtt[nw][nsdt][ntau];
    double qst5[nw][nsdt][ntau];
    double qtt5[nw][nsdt][ntau];
};
 
struct rt1_out {
    string outfile;
    string rt1dir;
    int_3darray nolyr;
    int_3darray ifc;
    int_3darray nmodl;
    double_4darray htlvl;
    double_4darray pplvl;
    double_4darray fr;
    double_4darray fn;
    double_4darray fa;
    double_4darray ftot;
    double_4darray dtrr;
    double_4darray dtmm;
    double_4darray dtaa;
    double_4darray dtot;
    double_4darray salb;
    double_4darray turbhl;
    double_3darray trp;
    double_3darray tmp;
    double_3darray tap;
    double_3darray tautot;
    double_4darray ppo;
    double_4darray dx;
    double_4darray x;
    double_4darray tozn101;
    double_4darray htdv;
    double_4darray pdv;
    double_4darray taur;
    double_4darray taum;
    double_4darray taua;
    double_4darray tau;
    double_3darray wvlth;
    double_3darray psrfc;
    double_3darray rho;
    double_3darray xozn;
    double_3darray dtau1;
    double_3darray tcar;
    double_3darray twat;
    double_3darray tozn;
};
 
/* AFRT1
 *
 */
struct rt2_in {
    string infile;
    string rt1dir;
    string phsdir;
    string rt2dir;
    string ocndir;
    string outdir;
    int    iref;
    int    itrans;
    int    iset;
    int    ilm1;
    int    ilm2;
    int    isd1;
    int    isd2;
    int    irh1;
    int    irh2;
    int    iwnd1;
    int    iwnd2;
    int    itau1;
    int    itau2;
    int    ithe01;
    int    ithe02;
    int    inx;
    int    inthe0;
    int    inwl;
    int    inrh;
    int    insd;
    int    iww[nsig];
    int    krhum[nrh];
    int    iprin;
    int    ipol;
    int    inpass1;
    int    inpass2;
    int    icrft;
    int    iactflx;
    int    isurf;
    int    iglint;
    int    iocn;
    int    ifoam;
    int    iwatr;
    int    iconc;
    double dtheta;
    double dphi;
    double hcrft;
    double rx[nthe];
    double the0in[nsza];
    double phi[nphi];
    double albwat[nwnd];
    double wwl[nw];
// phs
    int    ifunc;
    int    mfunc;
    double refr[nmd];
    double refi[nmd];
    double rmin[nmd];
    double rmax[nmd];
    double deltar[nmd];
    double deltax[nmd];
    double rg[nmd];
    double sig[nmd];
    double npar[nmd];
    double r11;
    double r22;
    double r33;
    double r44;
    double reff;
    double veff;
    double ccnsml;
    double bsr;
    double salb;
    double asf;
    double qscat;
    double qext;
    double t[nstk][ntf];
    double thd[ntf];
// ocn
    double xr;
    double xi;
    double v;
    double thcel;
    double phcel;
    double txx[nsza][nthe][nphi][ntrx];
    double pti[nsza][nthe][nphi][ntrx];
// rt1
    int    ifc;
    int    nolyr;
    int    nmodl;
    double wvlth;
    double psrfc;
    double rho;
    double xozn;
    double tautot;
    double deltau;
    double trp;
    double tmp;
    double tap;
    double tcar;
    double twat;
    double tozn;
    double htlvl[nlyr];
    double pplvl[nlyr];
    double dtrr[nlyr];
    double dtmm[nlyr];
    double dtaa[nlyr];
    double dtot[nlyr];
    double htdv[nlyr];
    double pdv[nlyr];
    double taur[nlyr];
    double taum[nlyr];
    double taua[nlyr];
    float  bfr[nw][nsdt][ntau][nsig][nlyr];
};
 
struct rt2_out {
    string outfile;
    string outdir;
    int    sd;
    double wl;
    double tau;
    double wnd;
    double_5darray tma;
    double_5darray tmb;
    double_5darray tmc;
    double_5darray tmfd;
    double_5darray tmfu;
    double_5darray tms;
    double_5darray tmg;
    double_5darray tmh;
    double_5darray tmp;
    double_5darray tmq;
    double_5darray tmt;
    double_5darray tmpp;
    double_5darray tmqq;
    double_5darray tmrr;
    double_5darray tmss;
    double_5darray transm;
    double_5darray tmcfd;
    double_5darray tmcfu;
    double_6darray tmf1;
    double_5darray tmf2;
    double_8darray xzeroz;
    double_8darray xzerod;
    double_8darray tupz;
    double_8darray tdwnz;
    double_7darray crfttup;
    double_7darray crftzu;
    double_7darray crfttdn;
    double_7darray crftzd;
    double_7darray surfzu;
    double_7darray radsky;
    double_7darray radocn;
    double_7darray raddir;
};
 
 
class AfrtProcess
{
public:
    string  output_filepath_;
 
    // output file attributes
 
    string title_ ;
    string prod_name_ ;
 
    // output file global attributes
 
    string sensor_ ;
    string platform_;
    string processing_version_;
    string conventions_ ;
    string institution_;
    string license_ ;
    string naming_authority_;
    string date_created_ ;
    string keywords_vocabulary_;
    string stdname_vocabulary_ ;
    string creator_name_;
    string creator_email_;
    string creator_url_;
    string project_;
    string publisher_name_;
    string publisher_url_;
    string publisher_email_;
    string processing_level_;
    string cdm_data_type_;
    string source_files_;
    string time_coverage_start_;
    string time_coverage_end_;
    string pge_start_time_;
    string pge_end_time_;
    string versionid_;
 
    AfrtProcess();
 
    virtual ~AfrtProcess ();
 
    virtual int initialize();
 
    virtual int read_af();
 
    virtual int process();
 
    NcFile* open_nc4(string ofile);
 
    virtual int write_nc4(NcFile* ncout);
 
    int write_global_attributes( NcFile* nc_output );
 
    string history_;
 
    void setHistory(std::string history) {
        history_ = history;
    }
    std::string getHistory() {
        return history_;
    }
 
protected:
 
    int num_bands_;
    STAGE_ENUM stage_;
};
 
class AfrtProcessPhs : public AfrtProcess
{
public:
 
    AfrtProcessPhs();
 
    ~AfrtProcessPhs ();
 
    int initialize();
    int initialize(int l1, int l2, int s1, int s2);
 
    int read_af();
 
    int read_nc4(std::string filepath, int l1, int l2, int s1, int s2);
 
    int process();
 
    int write_nc4(NcFile* ncout);
 
    bool bNC4_;
    phs_in* in_;
    phs_out* out_;
};
 
class AfrtProcessOcn : public AfrtProcess
{
public:
 
    AfrtProcessOcn();
 
    ~AfrtProcessOcn();
 
    int initialize();
 
    int read_af();
 
    ocn_in* in_;
    ocn_out* out_;
};
 
class AfrtProcessRt1 : public AfrtProcess
{
public:
    constexpr static int klam=250;
    constexpr static int ktau=10;
    constexpr static int klyr=200;
    constexpr static int kair=3;
    constexpr static int kmac=100;
    constexpr static int kozn=100;
    constexpr static int koznp=101;
    constexpr static int kdlyr=160;
    constexpr static int kdlvl=161;
    constexpr static int kmacp=101;
    constexpr static int nsc=1000;
    constexpr static int krh=10;
    constexpr static int ksd=100;
    constexpr static int knm=85;
 
    AfrtProcessRt1();
 
    ~AfrtProcessRt1();
 
    int initialize();
 
    int read_af();
 
    rt1_in* in_;
    rt1_out* out_;
};
 
class AfrtProcessRt2 : public AfrtProcess
{
public:
 
    AfrtProcessRt2();
 
    ~AfrtProcessRt2();
 
    virtual int initialize();
 
    int read_af();
 
    rt2_in* in_;
    rt2_out* out_;
};
 
class AfrtProcessNC4 : public AfrtProcessRt2
{
public:
 
  int mpi_id=-1;
  int n_mpi_nodes;
 
    AfrtProcessNC4();
 
    ~AfrtProcessNC4();
 
    int initialize();
 
    int process();
 
    int process_mpi();
 
 
    int initialize_nc4(string ofile);
 
    using AfrtProcessRt2::write_nc4;
    int write_nc4(size_t l1, size_t s1, size_t t1, size_t w1);
 
protected:
    string ofilepath_;
    NcFile* nc_out_;
    double_5darray F_;
 
    AfrtProcessPhs* phs_;
    AfrtProcessOcn* ocn_;
    AfrtProcessRt1* rt1_;
 
};
 
#endif /* INCLUDE_AfrtProcess_H_ */