o Mh&@s@ddlZddlZddlmZddlmZddlmZddZdS)N)Path) gen_surf_emis) gen_elev_emisc+Cs>tjtjd}|jddddd|jddtd dd |jd d td dd |jdddddd|jddddd|jdddd|jdddd|jdddd|jdddd|jdd dd|jd!d"dd|jd#d$d%dd&|jd'd(dd|jd)d*dd|jd+dd,dd|jd-dd.dd|jd/dd0dd|jd1dd2dd|jd3dd4dd|jd5dd6dd|jd7dd8dd|jd9dd:dd|jd;dddd|jd?dd@dd|jdAddBdd|jdCddDdd|jdEdFdd|jdGdHgdId|jdJdKgdLd|jdMdNdd|jdOdPd%dd&|jdQdRdd|jdSdTdd|jdUdVdd|jdWdXdd|jdYdZdd|}|j}|j}|j}|j }|j }|j }|j }|j } |j} |j} |j} |j} |j}|j}|j}|j}|j}|j}|j}|j}|j}|j}|j}|j}|j}|j}|j }|j!}|j"}|j#}|j$} |j%}!|j&}"|j'}#|j(}$|j)}%t*+}&t,|t,|ksJd[|d\kr6t-||D]`\}'}(t.|||(/fid]|'/d^|d_|d`| da| db| dc| dd| de|df|dg|dh|di|dj|dk|dl|dm|dn|do|dp|dq|dr|ds|dt|du|"})|)0q|dvkr|#dkrBdw}#t-||D]\}'}(t.|||(/fid]|'/d^|d_|d`| da| db| dc| dd| de|df|dg|dh|di|dj|dk|dl|dm|dn|do|dp|dq|dr|ds|dt|du|"dx|#})|)1dy|)jvrt,t2t3|4dz|)j5d{|#d|kr|)6d}d~|)jvr|)6dd|)jvr|)6dqG|dkrUt-||D]i\}'}(t7|||(/fid]|'/d^|d_|d`| da| db| dc| dd| de|df|dg|dh|di|dj|dk|dl|dm|dn|dp|dq|dr|ds|dt|du|"d|d|!d|d| })|)0q|dkr |$dkrad}$t-||D]\}'}(t7|||(/fid]|'/d^|d_|d`| da| db| dc| dd| de|df|dg|dh|di|dj|dk|dl|dm|dn|dp|dq|dr|ds|dt|du|"d|d|!d|d| dx|$d|%})|)1dy|)jvrt,t2t3|4dz|)j5d{|$d|kr|)6d}d~|)jvr|)6dd|)jvr |)6dqft*+}*t8dt9|*|&d|ddS)N)Zformatter_classz-oz--output-optionszqS to produce 'Surface emissions data' E to produce 'Elevated emissions data'. Add P to show the parameter list.T)helprequiredz-syrz --start-yearz Starting year)typerrz-eyrz --end-yearz Ending yearz-fz --filename+zOriginal emission data filename)nargsrrz-vz=Variable names. Format: species_mode. ex: bc_a1, so4_a2 etc.)r rdefaultz-indirzRInput directory. Contains raw emissions, SCRIP, vertical distribution etc. files,)rr z-outdirzOOutput directory. All new emissions and intermediate data will be stored here.z-soutzXOutput surface emission file name. Ending string including the '.nc' for output NetCDF.z-eoutzYOutput elevated emission file name. Ending string including the '.nc' for output NetCDF.z-reszkOutput resolution. For SE-grid use 30 for ne30pg2 and so on. For lat/lon use 180x360 for 1-deg and so on.z-gridzQOutput grid mesh or SCRIP file. These files can be produced using the ggen tool.z-meanz6Produce climatological mean. The mean from syr to eyr.Z store_true)ractionr z-myrz5Climatological mean year. These are the model dates.z-profilezHProfile data. Vertical profile data used to produce vertical emissions.z-sectzList of sectorsz-syrszStarting year listz-eyrszEnding year listz-flistz Original emission filenames listz-plistzVertical profiles listz-diamzList of volume mean diametersz-mwzList of moelcular weightsz-rhozList of densitiesz-fraczList of fractionsz -sulfactorz~List of sulfactors. This is the ratio of sulfate (NH4)HSO4 mw and SO2 mw. Default value used for sulfate aerosols is 115/64.z-numstrzList of number stringsz-numfilezList of number filez-ncvarz#List of variable names in raw filesz -altitudez*Altitudes for the elevated emissions data.z -fractionsz>Vertical fractional distribution. Used in elevated emissions.) g?333333?r 皙?r r 皙?rrrrrrz -ind_fracz9Vertical fractional distribution for IND and ENE sectors.) gp= ף?gRQ?g)\(?rrrrrrrrrrz-paramsz5Parameter filename. Contains the default parameters.z-checkerzDVerify estimates. Compares original data with newly generated data.z-cgridz)Grid area data used when checker is True.z-sedatazAOutput in model native or SE-grid mesh. This is by default True.z-specieszAerosol species name (ex: bc, pom etc.). Used for species beyond default in place of -v option and in conjunction with -mode optionz-modezAerosol mode (ex: 1, 2, 4 etc.). Used for mode beond the default in place of -v option and in conjunction with -species optionz-prepzYPre-process data from different sources. Ex: 'GFED' for GFED4 biomass burning emissions.z8List of variables and files should have the same length!SPZvariableindiroutdirresgridmeanZmean_yrsectorsZsyr_listZeyr_list filenamesdiammwrhofrac sulfactornumstrnumfilencvarZ param_filecheckercgridZSEdataspeciesmodeSzE3SM_surf_emis.ncoutputZa4z num_a4_*_a4__Znum_a4Za1Znum_a1Za2Znum_a2ZEPprofileZprofiles fractionsind_fracEzE3SM_elev_emis.ncprepz Finished in z second(s)):argparseZArgumentParserZRawTextHelpFormatterZ add_argumentintZ parse_argsZoutput_optionsZ start_yearZend_yearfilenamevrrrrrmyrsectZsyrsZeyrsflistrrrrrrrrparamsrr sedatar!r'r(r)plistr"souteoutr+time perf_counterlenziprstripZ get_paramsZ prod_emislistrglobZytagZcombine_numvarsrprintround)+parserargsoutsefilevarrrrrrr0rZsyrListZeyrListr2rrrrrrrrr3rr r4r!r'r(r)r5r"r6r7r+ZtimerOnr/finitZtimerOffrJM/global/cfs/cdirs/e3sm/www/hass877/share/emis_processing/FlexEmis/src/main.pymainsF 4   46      4   46     rL) r8r,pathlibrZsrc.surfEmis_driverrZsrc.elevEmis_driverrrLrJrJrJrKs