NWPC业务系统笔记：构建数值预报业务系统

suite downscl 
  clock hybrid    
  repeat day 1    
  
  family t639_downscale 
    limit job_num 14
    inlimit job_num

    loop HH ( 00 06 12 18 ) do
      set fam $HH
      family $HH
        edit HH $fam
        case fam
          in( 00 ) do time 05:15 ;endin
          in( 06 ) do time 12:45 ;endin
          in( 12 ) do time 17:15 ;endin
          in( 18 ) do time 00:15 ;endin
        endcase

        set list "000 003 006 009 012 015 018 021 024"
        family downscale
          loop h ( ${list} ) do
            task downscale_$h
              edit FTIMEH $h
          endloop
        endfamily #downscale
      endfamily #HH
    endloop   #HH
  endfamily  #t639_downscale

  family archive 
    time 00:40
    trigger ./t639_downscale==complete
    edit HH NONE
    edit ID NONE
    task archiving 
  endfamily #archive 

  family housekeep
    time 01:00
    trigger ./archive==complete
    edit HH NONE
    edit ID NONE
    task housekeeping
  endfamily #housekeep
endsuite

forecast_hour_list = ["{hour:03}".format(hour=hour) for hour in range(0, 25, 3)]

class GrapesMeso3kmPost(object):
    def __init__(self, name, root_node_path):
        pass

    def build_one_cycle(self):
        """
        Family: one start hour
        """
        with Family(self.name) as fm_one_cycle:
            fm_one_cycle.add_variable('HH', self.name)
            fm_one_cycle.add_variable(self.attrs)
            fm_one_cycle.add_variable(common.lljob_s("serial_op"))

            with self.initial() as tk_initial:
                fm_one_cycle.add_task(tk_initial)

            with self.togrib2() as fm_togrib2:
                fm_one_cycle.add_family(fm_togrib2)

            with self.tograph() as fm_tograph:
               fm_one_cycle.add_family(fm_tograph)

            with self.upload() as fm_upload:
                fm_one_cycle.add_family(fm_upload)

            with self.housekeep_cycle() as fm_housekeep_cycle:
                fm_one_cycle.add_family(fm_housekeep_cycle)

            return fm_one_cycle

    def initial(self):
        """
        Task: initial
        """
        with Task("initial") as tk_initial:
            for a_forecast_hour in self.config['forecast_hour_list']:
                tk_initial.add_event(
                    'grapes_grib2_{hour}'.format(hour=a_forecast_hour))
            return tk_initial

    def togrib2(self):
        """
        Family: togrib2
        """
        with Family("togrib2") as fm_togrib2:
            # ...skip...
            return fm_togrib2

    def tograph(self):
        """
        Family: tograph
        """
        with Family("tograph") as fm_tograph:
            # ...skip...
            return fm_tograph

    def upload(self):
        with Family('upload') as fm_upload:
            # ...skip...
            return fm_upload
    
    def housekeep_cycle(self):
        with Family("housekeep_cycle") as fm_housekeep_cycle:
            # ...skip...
            return fm_housekeep_cycle

class GrapesMeso3kmPostSystem(object):
    def __init__(self, suite_name, suite_attrs, start_hour_config, attrs, config):
        pass

    def build_suite(self):
        suite = Suite(self.suite_name)

        for HH_name in sorted(self.start_hour_config.keys()):
            gmf_grapes_gfs_post_one_time = GrapesMeso3kmPost(
                HH_name, "/{suite_name}".format(suite_name=self.suite_name))

            with gmf_grapes_gfs_post_one_time.build_one_cycle() as fm_HH:
                suite.add_family(fm_HH)

        with self.housekeep_final() as tk_housekeep_final:
            suite.add_task(tk_housekeep_final)
        
        return suite

    def housekeep_final(self):
        with Task('housekeep_final') as tk_housekeep_final:
            # ...skip...
            return tk_housekeep_final

def create_suite():
    # start hour settings
    start_hour_config = {
        "00": {},
        "06": {},
        "12": {},
        "18": {}
    }

    # varaibles set in suite level
    suite_attrs = {}

    # variables set in cycle level
    common_attrs = {}

    # config for each component.
    config = {
        'initial': {},
        'togrib2': {},
        'tograph': {},
        'upload': {},
        'housekeep_final': {}
    }

    grapes_meso_3km_post_system = GrapesMeso3kmPostSystem(
        sys_name, suite_attrs, start_hour_config, common_attrs, config)
    st_sys = grapes_meso_3km_post_system.build_suite()

    return {
        'suite_name': sys_name,
        'suite': st_sys
    }

def main():
    suite_result = create_suite()

    defs=Defs()
    defs.add_suite(suite_result['suite'])

    defs_filename = suite_result['suite_name'] + ".def"
    defs_file = defs.save_as_defs(defs_filename)

run_date=int(commands.getoutput("date +%Y%m%d"))
suite.add_repeat(RepeatDate("ECF_DATE", run_date, 20280101))

for a_forecast_hour in self.config['forecast_hour_list']:
    with fm_plot.add_task(
            'cn_grapes_meso_APWC_sep_{hour}'.
            format(hour=a_forecast_hour)) as tk_plot:
        tk_plot.add_variable("ECF_SCRIPT_CMD", "cat {ecf_files}/cn_grapes_meso_APWC_sep.ecf".format(
            ecf_files=self.config['meso_chartos']['attrs']['ECF_FILES']
        ))