o Míh ã@sLddlZddlZddlmZddlmZmZddlmZGdd„de ƒZ dS)éN)ÚPath)Ú get_dir_pathÚchecker)Ú gen_scripc@sªeZdZdZdd„Zedd„ƒZejdd„ƒZedd„ƒZejd d„ƒZed d „ƒZ e jd d „ƒZ ed d„ƒZ e jdd„ƒZ edd„ƒZ e jdd„ƒZ dd„Z dd„Z dS)Úgenerate_gridszx generate_grids class handles all type of grid & SCRIP file generation. Returns lists of SCRIP filenames. cKsœ| dd¡|_| dd¡|_| dd¡|_| dd¡|_| dd¡|_| dd¡|_| dd¡|_| d d ¡|_| d d ¡|_ t   t t |jƒƒd ¡|_dS)NÚresÚiresÚfileÚinfileÚgridÚindÚÚoutÚxdimÚlonÚydimÚlatz /log.ggen)ÚgetÚ_resÚ_iresÚ_fileÚ_infileÚ_gridÚ_indirÚ_outdirrrÚloggingÚ getLoggerÚstrrÚlogger)ÚselfÚkwargs©r!úT/global/cfs/cdirs/e3sm/www/hass877/share/emis_processing/FlexEmis/src/ggen/ggrids.pyÚ__init__szgenerate_grids.__init__cCó|jS©N)r©rr!r!r"rózgenerate_grids.rescCóNdg|_dd„| d¡Dƒ}tt|ƒƒD] }|j ||¡q|j d¡dS)NrcSóg|]}| ¡‘qSr!©Ústrip©Ú.0Úxr!r!r"Ú !óz&generate_grids.res..ú,)rÚsplitÚrangeÚlenÚappendÚremove©rÚvalZressÚzzr!r!r"ró cCr$r%)rr&r!r!r"Úinres&r'zgenerate_grids.inrescCr()NrcSr)r!r*r,r!r!r"r/-r0z(generate_grids.inres..r1)rr2r3r4r5r6r7r!r!r"r;*r:cCr$r%©r r&r!r!r"r 2r'zgenerate_grids.gridcCsät|jƒt|ƒ}tt t|ƒ¡ƒdkrt t|ƒ¡|_dStt t|ƒ¡ƒdkr3t t|ƒ¡|_dS|jdkrpdg|_dd„t|ƒ d¡Dƒ}tt|ƒƒD]}t|jƒt||ƒ}tt|ƒƒ|j  |¡qN|j  d¡dSdS)NrcSr)r!r*r,r!r!r"r/?r0z'generate_grids.grid..r1) rrrr4Úglobrr2r3rr5r6©rr8ZdirvalÚfilesr9r r!r!r"r 6s  ùcCr$r%r<r&r!r!r"r Fr'zgenerate_grids.filecCsÖt|jƒt|ƒ}tt t|ƒ¡ƒdkrt t|ƒ¡|_dStt t|ƒ¡ƒdkr3t t|ƒ¡|_dSdg|_dd„t|ƒ d¡Dƒ}tt|ƒƒD]}t|jƒt||ƒ}tt|ƒƒ|j  |¡qI|j  d¡dS)NrcSr)r!r*r,r!r!r"r/Sr0z'generate_grids.file..r1) rrrr4r=rr2r3rr5r6r>r!r!r"r Js cCr$r%)r r&r!r!r"r Zr'zgenerate_grids.infilecCsštt|ƒƒ ¡st|jƒt|ƒ}tt|ƒƒt t|ƒ¡|_|jgkrKdg|_dd„| d¡Dƒ}t t |ƒƒD] }|j  ||¡q8|j  d¡dSdS)NrcSr)r!r*r,r!r!r"r/fr0z)generate_grids.infile..r1) rrÚis_filerrrr=rr2r3r4r5r6)rr8r?r9r!r!r"r ^s  ûc Csâg}|jdkr>|jD]1}|j dt|ƒ¡t||j|j|j|jd|j |j d  ¡}|j dt|ƒ¡|  t|ƒ¡q |S|j D]-}|j d¡t|j||j|j|jd|j |j d  ¡}|j dt|ƒ¡|  t|ƒ¡qA|S)a` Generate SCRIP files from input NetCDF files or resolution. Input resolution: For SE grids integer values (4, 30, 120 etc.) For RLL grids use latitudexlongitude (64x128, 180x360 etc.) Returns ------- in_file_list : List List of SCRIP file names for the input mesh. Nz Input Resolution: T©rr ÚpathZfdirr Úncrrú Generated zT Specifying input file suppresses resolution. (Recommended for SE to RLL conversion))rrÚinforrrrrrrrÚget_scrip_filer5rr)rZ in_file_listÚrÚfnameÚfr!r!r"Ú get_inp_scripks  ( û (zgenerate_grids.get_inp_scripc Csîg}|jdkrB|jD]5}|j d|¡|j d¡t|j|j|j|j|d|j|j d  ¡}|j dt |ƒ¡|  t |ƒ¡q |S|jD]/}|j d|¡t||j |j|j|jd|j|j d  ¡}|j dt |ƒ¡|  t |ƒ¡qE|S)aa Generate SCRIP files from output grid file or resolution. Output resolution: For SE grids integer values (4, 30, 120 etc.) For RLL grids use latitudexlongitude (64x128, 180x360 etc.) Returns ------- out_file_list : List List of SCRIP file names for the output mesh. Nz Using specified grid file: z, Specifying grid file suppresses resolution.TrArDz Output Resolution: )rrrErrrrrrrrFrr5r)rZ out_file_listÚgrHrGr!r!r"Ú get_out_scrip‡s   ( û(zgenerate_grids.get_out_scripN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r#ÚpropertyrÚsetterr;r r r rJrLr!r!r!r"rs2           r) r=rÚpathlibrÚsrc.ggen.utilsrrZsrc.ggen.gen_scrip_filerÚobjectrr!r!r!r"Ús