U
    þp*gð  ã                   @   sF   d dl Z d dlZd dlZG dd„ dƒZG dd„ dƒZG dd„ dƒZdS )é    Nc                   @   s   e Zd Zdd„ ZdS )Úread_GEOS_CF_SO2c                 C   sÄ   || _ | j | }t |d¡}t |jd d d … ¡| _t |jd d d … ¡| _t |jd d d … ¡| _t |jd d d … ¡| _	t |jd d d … ¡| _
t |jd d d … ¡| _| ¡  d S )NÚrZPSÚPHISÚlatÚlonÚSO2ÚZPBL)ÚpathÚnetCDF4ÚDatasetÚnpÚarrayÚ	variablesZPsfcr   ÚLatÚLonr   r   Úclose)ÚselfÚfilenamer	   ÚinfileÚf© r   úP/run/cephfs/ACPS_Scratch/cli3/TEMPO/Analysis/sfc_wind_2024/../readers/read_CF.pyÚ__init__   s    
zread_GEOS_CF_SO2.__init__N©Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r      s   r   c                   @   s   e Zd Zdd„ ZdS )Úread_GEOS_CF_METc                 C   sÎ   t  |d¡}t |jd d d … ¡| _t |jd d d … ¡| _t |jd d d … ¡| _t |jd d d … ¡| _t |jd d d … ¡| _	t |jd d d … ¡| _
t |jd d d … ¡| _| ¡  d S )	Nr   ZZLÚUÚVr   r   ÚSLPÚDELP)r
   r   r   r   r   Zmid_layer_heightsr   r   r   r   r    r!   r   ©r   r   r   r   r   r   r      s    zread_GEOS_CF_MET.__init__Nr   r   r   r   r   r      s   r   c                   @   s   e Zd Zdd„ ZdS )Úread_GEOS_CF_sfc_windc                 C   s`   t  |d¡}|d d d … | _|d d d … | _|d d d … | _|d d d … | _| ¡  d S )Nr   z/latz/lonz/Uz/V)Úh5pyZFiler   r   r   r   r   r"   r   r   r   r   '   s    zread_GEOS_CF_sfc_wind.__init__Nr   r   r   r   r   r#   &   s   r#   )r
   Únumpyr   r$   r   r   r#   r   r   r   r   Ú<module>   s
   