o
    Mhl                     @   sP   d dl Zd dlZd dlZd dlmZ d dlZd dlZd dl	m
Z
mZ dd ZdS )    N)datetime)
sec_in_monget_dir_pathc           !         s   d}g d}	 t }dd}t||}	t|	}
|
|
d | k | jd }t| t	|| d d d	d
f}d}t
|}t||d D ]|}tt|d t| d  t|d t| d }t|d}tdD ]Q}d||  d }|| d d  }t	d}tdD ] }d||  d ||  }|| d d  }||| ||  7 }qt||d }|| d }|||d|  < qk|d7 }qEtjt|d t|d dd}tdddd}tdddd}ddg|f| g d |jd!d"ifi}d#gtj|jtjd$d%d#d&fd'gtj|jtjd$d(d'd&fd)}d*d+t d,d-d.d/}tj|||d0}d1dhi|_td d2  fd3d4|jD } |  fd5d4|jD  t
|}|  j!|t| d6 t| d7 t| d8  | d9 t| d6 t| d7 t| d8 S ):N)Z01Z0203Z04Z05Z06Z07Z08Z091011Z12)ZSAVAZBORFZTEMFZDEFOZPEATZAGRI/)z..datazGFED4_Emission_Factors.csvZspcr             z
/GFED4.1s_z.hdf5rz/emissions/z/DM)r   r      z/partitioning/DM_g     @@z-01-01z-12-31M)freqg     xV@g     Vg      п   g     |fg     f@g      ?time)r   latitude	longitudeunitszkg/m2/sr   )dtypedegrees_north)r   	long_namer   degrees_east)r   r   z6GFED data produced for the E3SM emission pre-processorz&Taufiq Hassan (taufiq.hassan@pnnl.gov)z%Y-%m-%d %H:%M:%Sz#0.25x0.25 degree latitudexlongitudez25 km)commentcontactcreation_dategridnominal_resolution)	data_varscoordsattrsunlimited_dims)
_FillValuec                       i | ]}| qS  r&   ).0varcompr&   S/global/cfs/cdirs/e3sm/www/hass877/share/emis_processing/FlexEmis/src/prep_GFED4.py
<dictcomp>M       zgen_GFED4.<locals>.<dictcomp>c                    r%   r&   r&   )r'   coordr)   r&   r+   r,   N   r-   Z_biomass_burning_emis_GFED_-z.nc)encoding)"__name__joinpkg_resourcesresource_streampdread_csvvaluesprintnpzerosr   rangestrh5pyFiler   
date_rangearangeroundr	   arrayfloat32r   todaystrftimexrDatasetr0   dictr    updater!   load	to_netcdf)!Zaerindiroutdir
start_yearend_yearmonthssourcesresource_packageresource_pathZEFdataZEFsZEF_COZbb_yearidir_pathyearstringfmonthZDM_emissionsZCO_emissionssourceZcontributionfactZ	conv_valsr   Zlat_biomZlon_biomr    r!   r"   ds	encodingsr&   r)   r+   	gen_GFED4   sf   



 

6$r^   )numpyr9   pandasr5   xarrayrF   r   r3   r=   	src.utilsr   r   r^   r&   r&   r&   r+   <module>   s   