[4122] | 1 | #!/usr/bin/env python2.7 |
---|
| 2 | |
---|
| 3 | import os |
---|
| 4 | import shutil |
---|
| 5 | |
---|
| 6 | # Local imports |
---|
| 7 | import locator |
---|
| 8 | import nml |
---|
[4127] | 9 | import run |
---|
[4122] | 10 | |
---|
| 11 | def parse_args(): |
---|
| 12 | import argparse |
---|
| 13 | parser = argparse.ArgumentParser() |
---|
| 14 | parser.add_argument("date", metavar="DATE", |
---|
| 15 | help="Run date.") |
---|
| 16 | parser.add_argument("-w", "--work-dir", default=os.getcwd()) |
---|
| 17 | parser.add_argument("-f", "--forecast-types", default="forecast", |
---|
| 18 | help="Choice of forecast,persistence,climatology") |
---|
| 19 | parser.add_argument("-l", "--lead-times", default="12", |
---|
| 20 | help="Forecast lead times") |
---|
| 21 | parser.add_argument("-o", "--obs-types", default="profbfiles", |
---|
| 22 | help="Choice of namobs types.") |
---|
| 23 | parser.add_argument("--class4", dest="ln_cl4", action="store_true", |
---|
| 24 | help="Flag to choose class 4 file outputs") |
---|
[4136] | 25 | parser.add_argument("--dry-run", action="store_true", |
---|
| 26 | help="Flag to test namelist building without submitting.") |
---|
[4140] | 27 | parser.add_argument("--cmd", dest="command", default="./opa", |
---|
| 28 | help="Submit task to run.") |
---|
[4137] | 29 | parser.add_argument("-v", "--verbose", action="store_true", |
---|
| 30 | help="Prints difference between before and after namelists.") |
---|
[4122] | 31 | parser.add_argument("namelist", metavar="NAMELIST", |
---|
| 32 | help="NEMO namelist to edit.") |
---|
| 33 | args = parser.parse_args() |
---|
| 34 | args.forecast_types = args.forecast_types.split(',') |
---|
| 35 | args.obs_types = args.obs_types.split(',') |
---|
| 36 | args.lead_times = map(int, args.lead_times.split(',')) |
---|
| 37 | return args |
---|
| 38 | |
---|
[4137] | 39 | def printdiff(text1, text2): |
---|
| 40 | # Provides nice text difference summary of namelists |
---|
| 41 | import difflib |
---|
| 42 | lines1 = text1.splitlines() |
---|
| 43 | lines2 = text2.splitlines() |
---|
| 44 | d = difflib.Differ() |
---|
| 45 | result = list(d.compare(lines1, lines2)) |
---|
| 46 | text = '\n'.join(result) |
---|
| 47 | print text |
---|
| 48 | |
---|
[4122] | 49 | def main(): |
---|
| 50 | args = parse_args() |
---|
| 51 | date = args.date |
---|
[4136] | 52 | print "Processing", args.namelist, " for", args.date |
---|
[4122] | 53 | |
---|
| 54 | # Move to working directory |
---|
| 55 | if not os.path.exists(args.work_dir): |
---|
| 56 | os.makedirs(args.work_dir) |
---|
| 57 | os.chdir(args.work_dir) |
---|
| 58 | |
---|
| 59 | # Collect forecast files |
---|
| 60 | types = args.forecast_types |
---|
| 61 | lead_times = args.lead_times |
---|
[4132] | 62 | namooo, namcl4 = locator.forecasts(date=date, |
---|
[4122] | 63 | types=types, |
---|
| 64 | lead_times=lead_times) |
---|
| 65 | |
---|
| 66 | # Process NEMO namelist |
---|
| 67 | text = nml.reader(args.namelist) |
---|
| 68 | sublists = nml.namelists(text) |
---|
| 69 | |
---|
[4137] | 70 | # Verbose save original text |
---|
| 71 | if args.verbose: |
---|
| 72 | original_text = text |
---|
| 73 | |
---|
[4132] | 74 | # namooo |
---|
| 75 | if "namooo" not in sublists: |
---|
[4122] | 76 | # Attach boilerplate |
---|
[4132] | 77 | text += nml.new("namooo") |
---|
| 78 | text = nml.update("namooo", text, data=namooo) |
---|
[4122] | 79 | |
---|
| 80 | # namcl4 |
---|
| 81 | if "namcl4" not in sublists: |
---|
| 82 | # Attach boilerplate |
---|
| 83 | text += nml.new("namcl4") |
---|
| 84 | namcl4["cl4_leadtime"] = lead_times |
---|
[4133] | 85 | namcl4["cl4_date"] = nml.quote(date) |
---|
[4122] | 86 | namcl4["cl4_match_len"] = len(namcl4["cl4_vars"]) |
---|
| 87 | namcl4["cl4_fcst_len"] = len(namcl4["cl4_leadtime"]) |
---|
[4130] | 88 | # Add naming convention |
---|
| 89 | namcl4["cl4_sys"] = "FOAM" |
---|
| 90 | namcl4["cl4_cfg"] = "orca025" |
---|
[4132] | 91 | namcl4["cl4_vn"] = "'1.0'" |
---|
[4130] | 92 | namcl4["cl4_prefix"] = "class4" |
---|
| 93 | namcl4["cl4_contact"] = "example@example.com" |
---|
| 94 | namcl4["cl4_inst"] = "institute" |
---|
[4122] | 95 | text = nml.update("namcl4", text, data=namcl4) |
---|
| 96 | |
---|
| 97 | # namrun |
---|
[4138] | 98 | namrun = {"nn_date0": nml.quote(date)} |
---|
[4122] | 99 | text = nml.update("namrun", text, data=namrun) |
---|
| 100 | |
---|
| 101 | # namobs |
---|
| 102 | namobs = locator.observations(date=date, |
---|
| 103 | types=args.obs_types) |
---|
| 104 | namobs["ln_cl4"] = args.ln_cl4 |
---|
| 105 | text = nml.update("namobs", text, data=namobs) |
---|
| 106 | |
---|
[4137] | 107 | # Verbose print namelist differences |
---|
| 108 | if args.verbose: |
---|
| 109 | printdiff(original_text, text) |
---|
| 110 | |
---|
[4122] | 111 | # pipe text to file |
---|
| 112 | tmp = args.namelist+".tmp" |
---|
| 113 | nml.writer(tmp, text) |
---|
| 114 | shutil.move(tmp, args.namelist) |
---|
| 115 | |
---|
| 116 | # Run job |
---|
[4136] | 117 | if not args.dry_run: |
---|
[4140] | 118 | run.submit(command=args.command) |
---|
[4122] | 119 | |
---|
| 120 | |
---|
| 121 | |
---|
| 122 | if __name__ == '__main__': |
---|
| 123 | main() |
---|
| 124 | |
---|