o [Yb@sddlmZmZmZddlmZddlZddlZddl Z ddl m Z ddl m Z ddlmZmZmZmZddlZddlmZdZdZd Zd Zd Zd Zd ZeeZeeZdeZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)ddZ*ddZ+ddZ,ddZ-ddZ.dd Z/d!d"Z0d#d$Z1d%d&Z2dId)d*Z3dId+d,Z4dJd-d.Z5dJd/d0Z6d1d2Z7Gd3d4d4e8Z9Gd5d6d6e8Z:Gd7d8d8e8Z;Gd9d:d:e8Zd>e8Z>Gd?d@d@e8Z?GdAdBdBe8Z@GdCdDdDe8ZAdEdFZBeCdGkreDdHdSdS)K)print_functiondivisionabsolute_import)rangeN)interp1d)read_nc)MLUTLUTIdxread_mlut_hdf5gkHc2@g-C<@goG@ggK҅Dg:#@g; @g^t0;?geWѧ?gn#@g7̉Zwk?gP#@linearg[@cCsB|}|}|dr|}|}|ds ||dS)N#)tellreadline startswithseek)fpostmprf/Users/jing/Downloads/radiative_transfer_demo/ARTDECO/artdeco/pyartdeco/f2py_utils/pyartdeco_runlib.py skipcomment;s  rcCs t|}t||}|S)N)npasarrayabsargmin)arrayvalueidxrrr find_nearestEs r cCs(tddtt|}||||SNrdryeps)Z d_kgperkgtemppaphumrmoistrrrget_densLr*cCs(tddtt|}||||Sr!r#)Zwcr&r'r(r)rrr get_massratioTr+r,cCs|dvrdtd|ddd|dddd}n|dkr;dtd|ddd|dddd}|dd|dd|dd}||d }|S Nliquidliq cn@QE1@fffffq@皙m@iceQk6@fffffq@Y@rexp)ptuairuh2ophaseppsppvrhrrrget_rh_s42$ rCcCs|dvrdtd|ddd|dddd}n|dkr;dtd|ddd|dddd}||d }|dd||dd}|Sr-r9)r;r<r=rBr?r@rAurrr get_u_from_rhis42 rEcCs|dd|dd|dd}||d}|dvr2t|dddt|dd}|S|dkrJt|ddd t|dd}|S) Nr8r.r1r4r2r3r5r7r6)rlog)r;rBr=r>r?rAr@r<rrrget_t_from_rh_uss$ ((rGcCs|dd|SNr ư>rppmvh2o_ppmv_moistrrrmoist_to_dry_ppmvrMcCs|dd|SrHrrJrrrdry_to_moist_ppmvrNrOFTcCsJ|r |dd|}n|}dtt|}d|d}|s#|d|}|S)Nr rI.Ar"mairmh2o)Zh2o_Z h2o_moistZ hum_moisth2oar(rrrget_hums  rVcCsJ|r |ddt|}n|}|dd|tt}|s#|d|}|S)Nr rIrP)get_h2o_ppmv_moistmo3rR)Zo3_r(Zo3_moistZ ozo_moisto3ozorrrget_ozos r[cCs4d|d|tt}|dd|}|r|S|SNrPr rIrQ)r(dryZ h2o_ppmv_dryrLrrrrW rWcCs4d|d|tt}|dd|}|r|S|Sr\)rRrX)rZr(rLr]Z ozo_ppmv_dryZozo_ppmv_moistrrrget_o3_ppmv_moistr^r_c CsRdtd}tdtdt}d|dtd|td}td|td} ttd|t|tjdd|t|tjdd} | d } td|t|tjdd| dt|tjdt|tjdd} | d | } t |}t |}t |}t |}t |}dt d t }|dd||dd|dd<|||t |d }t |}t t |d D]}||||d ||<q|t |d ||t |d <t |d }|| |||t|||||<| d|d| |d||d | }|d kr1| t|d||<nd}| d||<t t |dddD]]}|| ||d ||t||d ||||<| dd||d d| d||d d||d | }|d kr| t|d||<qCd}| d||<qC|d|d fS)Nr g@@@@g1@g,@g @f@g@@gMb`?r8r"MbP?g0.++)flattomegaeqradgraversqrtsinpicos zeros_likergcrRlenrrF)latr&r'psurf elevationqZdflatfacbetaetaZrearthZrlhZgravlZgravh_rZ int_dmairdmairZppwZztempZhgplcZ dmair_surfdpZilevrrr calc_hgplsF % FX      $   ,,  <D r|c@s*eZdZdZddgdfddZddZdS) solar_irradiance noneasciic Cs||_dd}|jdkr<|dkr>d|_|dkrtj|rt|d}t||}t | d|_ t||}t | d|_ t|j d |_d d g} t|j |_t|j |_t|t|j D]} |}t | d|j| <t | d |j| <qj|d |_t|dkrt||_ |j|d |_|j|d |_|j|d |_nstd td|tnf|dkr%d |_t|d} t| dd|_t| dd|_| ddjd|_| t|j|_ t|j d |_d|_ t|dkr%t||_ |j|d |_|j|d |_|j|d |_||jdds@tdtdtdSdSdSdS)NcSt|dd|ddkSNrfr"rallrUrrr z+solar_irradiance.__init__..kdisrFrrrr"wlrTz(solar_spectrum) ERRORz Missing file:h5_kdisdef central_wvlsolradZsolspecrfz solrad ERRORz8 wavelengths must be sorted in increasing order) solar_specmodeZ kdis_solradospathisfileopenrrfloatsplitZkdis_solrad_solcsteintkdis_solrad_nwvlrarangekdis_solrad_channelszeroskdis_solrad_wvlkdis_solrad_F0rcloserqprintexith5pyFilecopyattrs) self artdeco_in file_path file_format channel_listr is_sortedrrnamesiwvlffrrr__init__sl          9zsolar_irradiance.__init__c Csvd|vr9g}t|D]\}}|tt||jkq t|}t||_|j||_|j ||_ |j ||_ dS)Nrf) enumerateappendrsqueezeargwhererrrqrrrrZ list_chanZind_chanZichanchanrrrselect_chan_solradas     z#solar_irradiance.select_chan_solradN)__name__ __module__ __qualname____doc__rrrrrrr}s Fr}c@seZdZdddZdS)ptcle_optical_propertiesrFTc CsLt|dkrdSt|dkrtdtdtdtg|_g|_tt|D]B} || d|jvrI|j|| d|j|| dq)|j|j|| d|| dkrktdtd td tq)t|j|_||_ t ||_ ||_ ||_d g|j|_d g|j|_d g|j|_||_d g|j|_d g|j|_d g|j|_t|jD]+} |rtd |j| d |j| dd dvrtdtq|j| } tj| stdtdtd| tt| d} |j| | vrtdtdtd|j| d| t| |j| } d} gd}|rM| ddjddkrGtdtdtdtdt|gd}d| dvr[|dd | dvri|d d!| dvrw|d!d"| dvrd#} |d"|d$|r| ddjd|j| <| t!| ||j| d%d&}|j"d't |j"d'k#rtdtdtd(| tt$|j"d'|d}t$|j"d'|d)}|j"d'||dkr|dkr|d)}|j"d'||d)kr|t|j"d'd)kr|d)}d't%||d)i}|js=|j |j"d'vr=tdtdtd*| td+tn|j |j"d'vrPd't&|j d%d,i}nd't&|j i}|r|j |j"d-vrxtdtdtd.| td/tn |'d-t&|j d%d,i|d0(||j| <|rtd |j| )d|vr|d(||j| <d |vr|d (||j| <d!|vr|d!(||j| <|(||j| <|rtd |j| )qt|jD]>} | |krqt *|j|j"d1|j| j"d1s"|r td2|j| d |j| (d1t&|j|j"d1i|j| <qdS)3a ptcle_type_list : list of dictionnary e.g. ptcle = [{"name":, "file_path":} {"name":, "file_path":}] wl : is a wavelength interval all optical properties found in the given file will be loaded for wavelengths falling in this interval e.g [0.70, 0.80] wlref : the reference wavelength for the opacity rNrez(optical_properties) ERRORz1 wl is a wavelength interval z2 It must be [wvl_min, wvl_max]namerz= Same ptcle_name with different opt_path z rfz# (optical_propertie) read opt for  .)h5Zhdf5z'(optical_propertie) format must be HDF5z(optical_propertie) ERROR Missing file:rz Missing particle z in file )Cextsingle_scattering_albedop11_phase_functionp44_phase_functionp21_phase_functionp34_phase_functiondata alpha1_betalZ truncationdmK pyARTDECO only suited for DeltaM truncated optical prop. library)r alpha2_betal alpha3_betal alpha4_betal beta1_betal beta2_betalrecomposed_p11_phase_functionrecomposed_p22_phase_functionrecomposed_p33_phase_functionrecomposed_p44_phase_functionrecomposed_p21_phase_functionrecomposed_p34_phase_functiontruncation_coefficientintegrate_p11_recompZextinction_coefficient_over_wcZ"extinction_coefficient_over_dry_acZ#extinction_coefficient_over_rh80_acp22_phase_functionp33_phase_functionT)groupZlazy wavelengthszG Wavelength axe should be sorted in increasing order in file r"z@ Required reference wavelength value is not sampled in ' and no interpolation allowed roundnbetalz4 Required -nbetal- value is not sampled in rmuz0 (optical_propertie) angular interpolation for )+rqrr type_namerrrindexntypewlrefrsortrr opt_interp trunc_method cext_reflambopt read_betalZSext_o_wc_reflambZSext_o_dry_ac_reflambZSext_o_rh80_ac_reflambrrrrrrkeysrgetdecoderr axesrr slicer updatesubZ print_info array_equal)rZptcle_type_listrrrZ ind_ref_angrverboserifilenamerrnelemZ ls_datasetsmZiwvl1Ziwvl2 sub_lut_dicsub_reflamb_dicrrrrss  &              "0      &*z!ptcle_optical_properties.__init__N)rFFT)rrrrrrrrrqsrc@seZdZddZdddZdS) particlecst|dkr d|_dSt||_dd|D|_tdd|D|_dg|j|_j|_j|_dd|D|_ t |j|_ t |j|_ t |jD]p}|j |dvrmtd td td |j |td t|j |d vr~||d|j |<nd|j |<|j |dvrd||vrtd td tdtd||dtd t||d|j |<qMd|j |<qMd|j vr0d}|D]}|ddkrt|d|krt|d}qtj|jdd|_t|j|f|_t|j|f|_t |jD]0}||ddkr/t||d|j|<||d|j|ddf<||d|j|ddf<qg|_t |jD] }|j|jvrWtd td td|j|ttt|j|tjkjjd}t|}ttfddt |D} tj|tjjdtd} td|f} td t| |f} i} i}gd!}|D]}|jd"jvrc|||vrtd td td#|d$||dt|js|||jj|vrtd td td%|d&td't|||tjj|tq|||jj|vrI| |t!|||d(d)i| |t!|||d(d)iq| |t!|||i| |t!|||iqd*djj"D}t|dkrj#$|j"}nj#j"}t| dkrgjd+$| j"| dddf<jd,$| j"| d-ddf<d.| d/ddf<jd"$| j"| dddddf<jd0$| j"| dddddf<jd1$| j"| d2ddddf<jd3$| j"| d4ddddf<d5|vr$| j"d|j'd-ddf|d2ddddf<jd?$| j"d|j'd-ddf|d4ddddf<jd@$| j"|dddddf<jdA$| j"|d-ddddf<jdB$| j"|d/ddddf<jdC$| j"|dddddf<jdD$| j"|d2ddddf<jdE$| j"|d4ddddf<jdF$| j"|dd<jdG$| j"|dd<n&jd:j"d|j'd-ddf|dddddf<jd;j"d|j'd-ddf|d-ddddf<jd<j"d|j'd-ddf|d/ddddf<jd=j"d|j'd-ddf|dddddf<jd>j"d|j'd-ddf|d2ddddf<jd?j"d|j'd-ddf|d4ddddf<jd@j"|dddddf<jdAj"|d-ddddf<jdBj"|d/ddddf<jdCj"|dddddf<jdDj"|d2ddddf<jdEj"|d4ddddf<jdFj"|dd<jdGj"|dd<|j)|||| | | | d(|j'|||||dHn|j)|||| | | | ddItjjd|_*jjd|_+q8dS)Ja* wlref is the reference wavelength for the opacity ptcle is a list of dictionnary. One dictionnary per particle to use. e.g. [ {"file":string, "name":string, "tau":float, "alt_distrib":string, "z":float, "dz":float, "user_vdist":np.array, "user_alt":np.array, "reff":float, "veff":float, "humidity":float} ] If alt_distrib is "layer" z (z top) must be given is "homogeneous" z (z top) and dz must be given is "gauss" z (z middle) and dz must be given is "sh" z (scale height) must be given is "user" "user_vdist" and "user_alt" must be given optical_prop : optical properties class rNcSg|]}|dqS)rr.0xrrr ez%particle.__init__..cSr)taurrrrrrgrFcSr) alt_distribrrrrrrqr)layer homogeneousshgaussuserz(particle) ERRORzF -alt_distrib- must be layer, homogeneous, sh, gauss or userz required:r)rrrr z)rr dzzR dz must be provided for -homogeneous- and -gauss- vertical distributionz for particle rr rZuser_altrdtype user_vdistz, Missing optical properties for :rcsg|] }jjdqS)r)rr)rZiiiii optical_proprrrsrr)humidityZreffZveffr&Zlogiwcrrz9 should be provided in the description (dictionnary) for z Required -z- value is not sampledrTrcSr)rrrrrrrrrrr"rerrrrrrrrzF Required -nbetal- value is not sampled in optical propertiesrrrrrrrrrrrrrr)rropt_wvl nang_phasematr u_phasematphasemat betal_flagrrbetal trunc_coefftrunc_phasemattrunc_normphasemat)rrrrrrrr),rqnptcletyperrZ tau_reflambZhgrrZ vdist_typeroZh_vdistZsd_gauss_vdistrrrrrZuser_vdist_naltZuser_vdist_altroptical_propertiesrrrrr transposefullrmaxrrr rrrrrnstreamsrrrr)rrptclerrZnaltmaxrrZnopt_wvlrrrrrrZ prop_listpropdatasetsrrrrrr rrrrrJs       "$ &  " ""(((( (*(*"""" "$((   >>>>>>(((((("888888""""""  zparticle.__init__F/tmp/c Cst|}t|jd|j|f}t|jd|f}tj|j|fdtd} |dkrOt|jd|d|f} t|j|f} t|jd|j|f} t|j|f} nd} d} d} d} t|jD][}t|j |dt d|j |j |d ggd d }t|j |d t d |j |d gddgd }|dkr|j |dr||j |dkr||j |dkrd| |ddf<t|j |dt dt |j |dd|j |d ggdd }t|j |d|j |d gdgd }t|j |dt d|j |j |d ggd d }t|j |d|j |d gdgd }t|D]}|j s?|||j |d vr?t dt dtn|||j |d vrVdt||ddi}ndt||i}||j||dd|f<||j||dddd|f<| ||fr||j| |dddd|f<||j| ||f<||j| |dddd|f<||j| ||f<q!q\|rt|jD]}t|d|j|dd}|d|d |d!|d"|d#|d"|d$|d%|d&|d'|d(|d)t|D]0}d*||d+|jd*||d|fd*||d|fd*||d,|fd-}||q|d.t|D]f}|d/||d-|d0t|jD]M}d1|j |d1||d||fd1||d||fd1||d,||fd1||d||fd1||d2||fd1||d3||fd-}||qaqJ|q||j || | | | | fS)4NrrFrrr"rg@r)imatrwvl)rrrg@Zioptr-rrrTr)r,lr-rrr z (particle.get_it) ERRORzN No wavelength interpolation allowed in the class definitionrZopt_.datwz,# Optical properties to be used by ARTDECO z# z,# Used model to obtain that properties is: z unknown z# Used material is: z%# Number of phase matrix elements : z 6 z# number of wavelengths z%iz zI# lambda(microns) nteta Cext (microns^2) SSA g z %.9ez %ire z# Phase matrix z# lambda = %.4fzu# u F11 F22 F33 F44 F21 F34 z %.17err)rqrrr!rr%boolrr r#rrrrrr rrrr"writer)rrrrZ ascii_saveZdir_ascii_savenwlrrrrZ trunccoeffrr ZiptcleZ lut_phasematZlut_optZ lut_betalZlut_trunccoeffZlut_trunc_phasematZlut_trunc_normphasematiwlrZfoptrsZiangrrrget_opt]sB,                P    zparticle.get_optN)Fr+)rrrrr7rrrrrHs rc@s eZdZgfddZddZdS) kdis_coeffcCs|jdkrdS|j|_dd}|dkr|d|jd}t|d}t|dd |_t|d  }t ||_ d |_ g|_ g|_d |_g|_g|_t|D]W\} } d t|d | j vrp|d | jd rmd } nd} nd} | s|j d|_ |j| |j |d | jdqP|jd|_|j| |j|d | jdqPt|j |_ t|j|_t |dd|_td|jf|_t|dd|jd ddf<t|dd|jdddf<t|dd|jdddf<t|jd|_t|dd|_t|dd|_t |j|_t |j|_|jd krUt|dd|_|ddjd|_ t |j|_!||jsTt"dt"dt#nd|_ ||jd ddfsot"dt"dt#||jst"dt"dt#||jst"dt"dt#t |d krEt ||_|jdd|df|_|j|d|_|j d krvtj|j |jfd d!|_$t|j |j|j|j|jf|_%t|j |j|jf|_&t|j |jf|_'t|jD]\} } t|dD]{\} } t|d | d"| |j$| | f<t|d | d#| d |j$| | fddddf|j%| | d |j$| | fddddf<t|d | d$| d |j$| | ff|j&| | d |j$| | ff<t|d | d%| |j'| | f<qq|jd krCtj|j|jfd d!|_(t|j|j|j|j|j|j!f|_)t|j|j|jf|_*t|j|jf|_+t|jD]\} } t|dD]\} } t|d | d"| |j(| | f<t|d | d#| d |j(| | fddddddf|j)| | d |j(| | fddddddf<t|d | d$| d |j(| | ff|j*| | d |j(| | ff<t|d | d%| |j+| | f<qqnt|j d krtj|j |jfd d!|_$t|j |j|j|j|jf|_%t|j |j|jf|_&t|j |jf|_'t|jD]{\} } t|d | d"|j$| ddf<t,|j$| ddf}t|d | d#ddd |ddddf|j%| ddd |ddddf<t|d | d$ddd |f|j&| ddd |f<t|d | d%|j'| ddf<q|jd krtj|j|jfd d!|_(t|j|j|j|j|j|j!f|_)t|j|j|jf|_*t|j|jf|_+t|jD]{\} } t|d | d"|j(| ddf<t,|j(| ddf}t|d | d#ddd |ddddf|j)| ddd |ddddf<t|d | d$ddd |f|j*| ddd |f<t|d | d%|j+| ddf<q=|-t|jD]\} } |j| .|j| <qt|jD]\} } |j| .|j| <qdS|d&k r|d|jd'}t/j01|s t"d(t"d)|t#t2|d}t3||4}t5|6d |_t3||4}t5|6d |_ d |_ g|_ g|_d |_g|_g|_t3|t7|j D]W}|4}t5|6dd kr{|j d|_ |j|6d |j t8|6dqLt5|6ddkr|jd|_|j|6d |jt8|6dqLt|j |_ t|j|_t3||4}t5|6d |_td|jf|_t|jd|_t3|t7|jD]O}|4}t8|6d|jd |f<t8|6d|jd|f<t8|6d|jd|f<|d kr,|jd |f|jd |dfkr,t"dt"d*t#qt3||4}t3|t5|6d |_t|j|_t7|jD]/}|4}t8|6d |j|<|d kr}|j||j|dkr}t"dt"dt#qOt3||4}t3|t5|6d |_t|j|_t7|jD]/}|4}t8|6d |j|<|d kr|j||j|dkrt"dt"dt#q|jd kr't3||4}t3|t5|6d |_!t|j!|_t7|j!D]/}|4}t8|6d |j|<|d kr%|j||j|dkr%t"dt"dt#q|-|j d krctj|j |jfd d!|_$t|j |j|j|j|jf|_%t|j |j|jf|_&t|j |jf|_'|jd krtj|j|jfd d!|_(t|j|j|j|j|j|j!f|_)t|j|j|jf|_*t|j|jf|_+t7|j D]} |d|jd+|j| d,}t/j01|st"d(t"d)|t#t2|d}t3|t7|jD]!} |4}t5|6d|j$| | f<t8|6d|j'| | f<qt7|jD]c} |j$| | fdkr^t3||4}t7|j$| | fD]}t8|6||j&| | |f<qt7|jD]-}t7|jD]$}|4}t7|j$| | fD]}t8|6||j%| | |||f<qFq7q0q|-q|jd krpd-|_ nd|_ t7|jD]} |d|jd+|j| d,}t/j01|st"d(t"d)|t#t2|d}t3|t7|jD]!} |4}t5|6d|j(| | f<t8|6d|j+| | f<qt7|jD]m} |j(| | fdk r>t3||4}t7|j(| | fD]}t8|6||j*| | |f<qt7|j!D]7}t7|jD].}t7|jD]%}|4}t7|j(| | fD]}t8|6||j)| | ||||f< q# q q qq|-qxt |d k r|j|d|_|jdd|df|_|j d k r|j$dd|df|_$|j%dd|dddddddf|_%|j&dd|dddf|_&|j'dd|df|_'|jd k r|j(dd|df|_(|j)dd|dddddddddf|_)|j*dd|dddf|_*|j+dd|df|_+|jj9d|_dS).NrcSrrrrrrrrrz%kdis_coeff.__init__..rZkdis_z.h5rrZmaxnaicoeffrrho_depTFr"Z add_continuumrrZmin_wvlZmax_wvlrepressureZ temperaturerhodescz kdis_coeff ERRORz< concentration must be sorted in increasing orderrzF (h5 format) wavelengths must be sorted in increasing orderz7 pressure must be sorted in increasing orderz: temperature must be sorted in increasing orderrrnaikiaixsectrz_def.datz(kdis_coef) ERRORrz: wavelengths must be sorted in increasing order_r/density):r kdis_modelmodelrrrrZnmaxailistrrqZnsp_totZnspZfcontspeciesnsp_cZfcont_c species_crrrrnwvlrwvlbandrchannelsr;r<ntrzrZc_descncrrr>r?r@rAnai_cki_cai_cxsect_cnanmaxrlowerrrrrrrrrrrshape)rrdir_dataformatrrrrZ species_totispZspecier:rZ iwvl_fileZnai_tmpZfdefrrZiaiitipicrrrrsX                     $^F( $$jF(  $R:( $$R:(;      "       $    $      &  * 0zkdis_coeff.__init__c CsRd|vrg}t|D]\}}|tt||jkq t|}t||_|j dd|f|_ |j||_|j dd|f|_ |j dd|ddddddf|_ |j dd|ddf|_ |j dd|f|_ |jdkr|jdd|f|_|jdd|ddddddddf|_|jdd|ddf|_|jdd|f|_dS)Nrfr)rrrrrrLrrqrJrKr>r?r@rArHrOrPrQrRrrrrselect_chan_kdiss$   & ,zkdis_coeff.select_chan_kdisN)rrrrr\rrrrr8s   Xr8c@s8eZdZ         dd d Zd d ZddZdS)surfaceFrcNTrarfffff&A@rfr cCs||_||_||_||_||_||_||_|js7|dkr7tdtdtdtdtdtdt|rX|dkrXtdtdtdtdtdtdt||_ |j dvrktdtdt|j d kr||_ |t ||_ |t ||_d S||_ |d vrtdtd t|d kr||_||_||_||_||_||_||_||_d S| t ||_| t ||_| t ||_|t ||_ |r| t ||_| t ||_|t ||_d Sd S)a sdistr : 3 = isotropic surface : 2 = Anisotropic Gaussian distribution : 1 = Anisotropic Gaussian distribution with Gram Charlier series correction azw : phi_sun - phi_wind (for sdistr = 2 or 1) _6s_glitter : True = 6SV glitter False = M. I. Mishchenko and L. D. Travis, J. Geophys. Res. 102, 16989-17013 (1997). Rsw is an irradiance reflectance rrzERROR : zN 6S glitter must be used (_6s_glitter = True) if anisotropic surface requiredz3 Shadow possible only if isotropic ocean surface )brdflambertERRORz, Surface family must be -brdf- or -lambert-r`)oceanZlandz* Surface family must be -ocean- or -land-rbN)rr&interpbpdf test_glitterocean_whitecaps _6s_glitterrrfamilykindrargsortralbwdspdsdistrazwshadowxsalpclRswwvl_Rswisovolgeocvnrni)rrrhrircrdr&rrkrtrurvrwrxryrerfrgrorlrmrnrprqrsrrrrrr8sn    zsurface.__init__cCs`t|jts t|jtrt||j}|Sd|jvr t|d}t|j|jtdtj d|}|S)NrfF)ri bounds_error fill_value) isinstancerrrrr full_likerrs surf_interpnan)rrrrrrrget_Rsws zsurface.get_RswcCs~|jdkr tddS|jdkrCt|}t|}t|}t|}|rB|js3tdtdtt|}t|}t|} n t|}t|} |jst |D]m} || |j vrdtdtqS|jdkr|j |j || k|| <|j |j || k|| <|j |j || k|| <|r|j|j || k|| <|j|j || k|| <|j|j || k| | <qS|j|j || k| | <qSne|jdkrt|j |j td} t|j |j td} t|j |j td}| |}| |}||}|rt|j |jtd}t|j |jtd}t|j |jtd}||}||}||} n t|j |jtd}||} |jdkr/| S|r:|||||| fS|||fS) Nrbz%surface.get_it() is useless for oceanr_z(surface.get_it) ERRORz0 BPDF was not defined in surfacezL(surface.get_it) : No wavelength interpolation allowed in surface definitionrir`)rirrhrqrrrdrrcrrrtrurvrwrxryrkrr~)rrrdr4rtrurvrwrxryrkr5ZfisoZfvolZfgeoZfcvZfnrZfniZfalbrrrget_itsr              zsurface.get_it)FFrcNNNNNNNNFTFFrarrcr^rcrfr )rrrrrrrrrrr]6s V r]c@s8eZdZddddidddfddZd dd Zd d ZdS) ptqo_atmosphereTgF@Nr0Fc! Cs"| |_|t||_|t||_||_||_||_|dvr$d|_n|dvr+d|_|dkr3d|_ n||_ | |_ |j dkrd|vrOt dt d t|D]}||jvrp||jvrpt d t d |d t d |jtqQd|vsyd|vrd|vr|dd|vrd|_nt||_|jdkr||_d}nd|_t|}t| ||| | |\}}t|}t |d|_t ||_t |d|_t||_t|D]<}|dkr |j|d|j|kr t|dD]}t d|d|j|d|j|qt dt dtq||d}|d|}|tt|t t|_|jd|_|jdkrt|j|f|_!t|jD]}|j|dkro|ttd|j!|ddf<t"|j|j|j|j!|ddf|j|_#q@|j|dkr||dt$td|j!|ddf<q@|j|t%| &vrt d t dt d|j|t dt|jd | |j|t'|jd!d ||j!|ddf<q@nd|_!|s|dur|rt|dd"d"D]>}d#|j|d|j|d|j|d$|j|}t|jD]}|d$|j!||f|j|d}q t |qd%}t|jD] }|d&|j|}q-t |t d'tj(||jdd( dtj(|j|jdd( }t)|jD]!\}}t d)|tj(|j!|ddf|jdd( |dfq_|rt d*t d+t d*|durt*|d,} t|D]C}d-|j|d.|j|d|j|d$|j|}t|jD]}|d$|j!||f}qt ||dur| +|d/qd0}t|jD] }|d1|j|}qt ||dur| +|d/|dur| ,dS)2ap p in Pa t in K hum in kg/kg over moist air o3 in kg/kg over moist air gas list of gas to be accounted for in absorption ppmv ppmv profile over dry air dictionnary of interp1d( np.log10(p (hPa), ppmv_over_dry_air ) for several gases surfalt surface altitude in m psurf surface pressure in Pa wvl_depol Rayleigh depol coeff wavelengths depol Rayleigh depol coeff interp whether we can interpolate depol r0r/r0r5r5Nrr(atmopshere) ERROR. kdis mode without absorbing gas !(atmosphere) ERROR Required gas  for the atmsophere$ is not defined in KDIS o2n2rTrrdr8r"rez %iz %.2f(atmospher) ERROR 1 Altitude must be strictly decreasing r`rPrYz, you must provide a ppmv profile for gas rIr rfz %.3fz %.5ez) alt p T uair z %sz column of WV (kg/m**2) = rz ppmv of %5s is %18.9frZ densitiesr0z %8.3fz %15.6fr1z2 alt p T uair z %s)-rcrrjwldepoldepolZhum_origZo3_origtau_rayrh_phasenewgridrgasabsrrrrGrIrErngasrqgasrWr|alt_origt_origp_origr u_air_origrrSnarRu_origrCrh_origrXrFrlog10trapzrrr3r)!rrr;r<r(rYrrrrKZsurfaltrsrcrrrrrrrZ save_asciigrLZaltituderynaltialtZialt1Zwv_gpercubicmeterZddairigasr6Z column_airrrrrrs            *    ,*B8(  <   8   zptqo_atmosphere.__init__c Cs|jdkr,t|j}t|j}t|j}t|j}|jdkr)t|j}nd}nt |j}t |ddd}t |j|jt dd}t |jt |jt dd}t |jt |jt dd} ||}t||}t| |}|jdkrt|jt|f}t|jD]5} t |jt |j| ddft dd} t| ||| ddf<d|| ddft|| ddf<qnd}|||||fS)W Set up (altitude) grid specified in format 'start[step1]stop1[step2]stop' or similar. NrrfFriZ assume_sortedrc)rrrrrrrrr parseGridSpecrr atm_interprFr:rrqrisnan) rdebugaltr<r;u_airrDftfpfuairrfurrrregrids2        &(zptqo_atmosphere.regridcCt|}t|}|js8t|D]%}|||jvr)tdtdtdtq|j|j||k||<q|St |j|jt d}||}|SNrzS(atmo.get_depol) : No wavelength interpolation allowed in depolarization definitionr rqrrrcrrrrrr depol_interprrr4rr5Zfdepolrrr get_depol  zptqo_atmosphere.get_depolF)rrrrrrrrrrrs  ) *rc @s|eZdZddddddidddf ddZdd Zd d Zd d ZddZddZdddZ ddZ ddZ ddZ ddZ dS) atmosphereTNrfr0Fc. Cs||_||_| dkrd|_n| |_| |_|t||_|t||_||_|dvr/d|_ n|dvr6d|_ d|_ |j dkrUd|vrRt dt dt d t nd |_ d gt||_|dd|jdd<|j d krd |vr|t dt dt |D]}|| jvr|| jvrt dt d|dt d| jt q~d|vsd|vrd|vr|d|dd |vrd}d|_n t|}t||_||jkrt dt |jdkr||_d}||_d}nd |_|j rd|_t|jdkr t dt dt dt dt dt |dkrJt d||||}tj|s/t dt d|t t|d }t||}t|d}t||}t|d}t|t ||_!t ||_"t ||_#t$t|j!tj%|_&t |}t |d!|f}|}t'|D] }t(||||<qt|t'|D]a}|}t(|d|j!|<t(|d!|j#|<t(|d|j"|<|dkr|j!||j!|d!krt dt d"t d t t'|D]}t(|d#|||d!|f<qq|)|jddkrd}nt*||jd}|jd!dkr%d}nt*||jd!}g|_|||d!|_+|dd||d!f|_,|j!|_-|d$kr}t d%t d&t t d||t.||d'|_,t.||d(|_-t.||d)d*|_+n|d+krt/0||d }t1|d,d-|_!t1|d,d.|_"t1|d,d/|_#t$t|j!tj%|_&|j!|_-t1|d0d1} |d0d1d|d0d1d!krt dt dt d2||t d3t dt |jddkrt*| |d0d4}n t*| d5|jd}|jd!dkrt*| |d0d6}n t*| d5|jd!}t ||d!t|j!d!f|_,t2|d73}!d8|jvrE|!|_|jD]}||!vrct dt d|dt d9||t qH|!D]-}"|"|jvr|j,ddddf|d7|"||d!ddf|j,ddddf<qg|)g|_d5| ||d!ddd|_+t4|j,dddddf|_,n|d:krN||}tj|st dt d|t t|d }t||}t|d}t||}t|d}#g}$t||#dkr t'|#D]}%|}|$|dqn|}t|t ||_!t ||_"t ||_#t ||_&i}&t'|#D] }'t ||&|$|'<qFt'|D]h}|}t(|d|j!|<|dkr|j!|d!|j!|krt d;t d<t t(|d!|j#|<t(|d|j"|<t(|d#|j&|<t'|#D]}'t(|d=|'|&|$|'|<qqW|)|jdkrJt |j|f|_5t'|jD]p}(|j|(|$vr|&|j|(|j5|(ddf<|j|(dkrt dt d>|j|(t d?t d@t6t q|j|(dkr6t dt dAt dB|j|(t dCt d@t6t |j&dD|j|(|j5|(ddf<qnd|_5nt dEg|_7d |_8d|_9d|_:t'|jD]H}(|j|(dFkrt;tj<|j5|(ddf|j!dGdH})|)dI|_9|j|(dkrt;tj<|j5|(ddf|j!dGdH})|)t=t>j?|_:qcdF|jvr| dkrt'|jD])}(|j|(dFkr| |j9|_@| |_9|j5|(ddf|j@|j5|(ddf<nqd|jvrZ|dkrZt'|jD]c}(|j|(dkrX|dJkrdJ|_:dJ|j5|(ddf<nDtA|j#|j"|j&|j5|(ddf|j }*||j:}+||_:|j5|(ddf|+|j5|(ddf<|jrVtB|j#|*|j&|j5|(ddf|j |_"nqd|jvrt'|jD]k}(|j|(dkrtA|j#|j"|j&|j5|(ddf|j |_C|jrt|jC|jCtDkdkrt dKt dLtDt't|j!D]'}dM|dN|j!|dN|j#|dN|j"|dO|jC|},t |,qnqedP|_E| |_F| dPkr-|j rt dQt dR| tG|j#}-|j#|-|_#|j&|-|_&|jdkr-t'|jD]%}(|j|(dFkr+|j|(dkr+|j5|(ddf|-|j5|(ddf<q|j!d|j!d!krEt dQt dSt dSdS)TNrr0rr5Flblz(atmosphere) ERROR z% If artdeco mode is lbl zZ you must provide an atm file with lbl format: -lbl_forum- or -lbl_liradtran-TrrrrrrrrrrrrTrfrz(atmosphere) nppmv != ngasrez(atmosphere) zZ(atmosphere) in lbl mode two boundary wavelengths must be provided in -artdeco_in.wavel- Zlbl_iprtz (atmosphere) read rrr"z@ with -lbl_iprt- fmt, alt must be sorted decreasing rZlbl_libradtranz`This must evolve to include, P,T,alt in the file without modifiyng the libRatran format basis...z? file mustthen include at the / : wvl, z, tau, wvlmin, wvlmaxrr r-rdZ lbl_forumZAtmosZAltitudeZ TemperaturePressureZSpectral_InformationsZ Wavenumberz(atmosphere) In z>(atmosphere) Wavenumber should be sorted in increasing orderZWavmaxg@ZWavminZMol_Optical_Depthrz is not defined in rrrrz! ppmv must be -1 for z2 cause it is already defined in file :rz- you must provide a ppmv not = -1rz, cause it is not defined in filerIz9(atmosphere) ERROR : atmosphere file format not supportedrYrj@gPZWCrc(atmosphere) WARNING 7 Relative humidity superior to %.2f percent%3i %10.3f %10.2fr (atm regird) ERROR/ No P0 scale is allowed if lbl modez8 alt_orig must be sorted in decreasing order)Hwarnrrrcrrjrrscale_t_rh_csterlbl_atmrrrrqrrGrIrErrrrKwavelrrrrrrrrrrrrr%rrrrrr wvl_tauabsgasZ tauabsgasZ alt_tauabsgasrrrrrFrr$rZatm_path rh100_boundsrh100_wvc_csteDU_o3waterrrrScsteN_Ascale_o3rCrGrlimit_rh cloudy_waterP0amax).rrrVZatm_filerrKZfmt_atmrrrcrrrrrrrrrrZnppmvrrrrrJrZ dtauabsgasrriminimaxZ wn_tauabsgasZgas_listZgas_nameZnspeciesrGrZ u_speciesrXrcolumn rh_before scale_h2or6factrrrrsJ                    $  ""     H               $   &( ( & & $&  "(" @       $ zatmosphere.__init__cCs||_|dkrS|jrtdtd|t|j}|j||_|j||_|jdkrSt|jD]"}|j |dkrR|j |dkrR|j |ddf||j |ddf<q0dS)Nr rrrrYrT) rrrrrrrrrrr)rrrrrrrset_P0ys   $zatmosphere.set_P0cCs(tt|j|jddd}|t|SNFrrzr{)rrrFrr)rr;rrrrget_altszatmosphere.get_altcCst|j|jddd}||Sr)rrrrrrrrrget_tszatmosphere.get_tcCs*t|jt|jdtjd}t||S)NFr)rrrrFrrr:rrrr get_pressszatmosphere.get_presscCrrrrrrrrrzatmosphere.get_depolc# Cs|jdkr/t|j}t|jdkr.|j}|j}|j}|j dkr,|j }d|j vr+|j }nd}n|j r=tdtdtt|j}t|ddd}t|jdkrt|j|jtdd}t|jt|jtdd} t|jt|jtdd} ||}t| |}t| |}|j dkrt|j t|f}t|j D]K} t|jt|j | ddftdd} t| ||| ddf<d || ddft|| ddf<|j | dkrt||||| ddf|j}qnd}t|jdkrtt|jD]!} t||j| }t||j| d }t||j| d }qtt|ddd}||t|jk}||t|jk}t|j|jtdd}t|jt|jtdd} t|jt|jtdd} ||}t| |}t| |}|j dkrt|j t|f}t|j D]M} t|jt|j | ddftdd} t| ||| ddf<d || ddft|| ddf<|j | dkrt||||| ddf|j}qtt|d}t|j D]}|j |dkrt|}t||ddf}tt|jD]>} d }t ||krt |}|||t|j| k|t|j| k@<d ||t|j| k|t|j| k@<qt!|||||j}|j"r2t ||d k}t|}d ||dk<t#tj$||d dt%t&j'}t|}d ||d k||kB<t#tj$||d dt%t&j'}t|}d ||d k||kB<t#tj$||d dt%t&j'}t|}d ||dk<t#tj$||d dt%t&j'}||}|d krtdtdt||krd}||||}nd }d|dd<|j(||}||}||}d |dd<||d k||d k<||d k||d k<||d k||d k<|||ddf<t|||||ddf|j}t#tj$||ddf|d d} | t%t&j'|_)|rtdtt|D]5}!d|!d||!d||!d||!d||!d||!d||!d||!}"t|"qptd|j)td|j(q|j*rt||t+kdkrtdtdt+tt|D]#}!d|!d||!d||!d||!d||!}"t|"q|||||fS)rNrrTrz- No regrid is allowed if lbl moderfFrrch㈵>r8Trrz delta_wv_in < 0z(atmosphere regrid) set rh100 rrrz WVC =z WVC before =z(atmosphere regrid) WARNING z> Relative humidity superior to %.2f percent),rrrrrqrrrrrrrrrrrrrrrrFr:rrrrCrruniqueminr&r%rSrErrrrSrrrrrr)#rrrr<r;rrDrBrrrrribZ flag_cloudyrrZ u_h2o_beforeZrh_cloudZu_h2oZ max_alt_cloudZu_inZwater_in_cloudZu_out_upZwater_out_up_cloudZ u_out_lowZwater_out_low_cloudZ u_in_beforeZwater_in_before_cloudZ delta_wv_inZch_upscalerrr6rrrrs      &&  &&  ,0 " " " "    $\   ( 8 zatmosphere.regridcCsrd|jvr7|dkr7t|jD](}|j|dkr6||j|_||_|j|ddf|j|j|ddf<dSqdS)NrYrf)rrrrrr)rrrrrrapply_scale_O3 s &zatmosphere.apply_scale_O3c Cstd|jvr|dkrt|jD]}|j|dkr|dkr*d|_d|j|ddf<dSt|j|j|j|j|ddf|j }||j}||_|j|ddf||j|ddf<|j rmt |j||j|j|ddf|j |_|j rt |j|jtkdkrtdtdttt |jD]&}d|d|j|d|j|d|j|d |j|}t|qdSqdS) NrTrfrcrrrrrr)rrrrrrCrrrrrrGrrqrrrr)rrrrrrr6rrrapply_scale_h2o s0& $& @ zatmosphere.apply_scale_h2ocCst|jdkrtdttt|t|f}t|jdt|jddd}t |D]\}}t ||}||t ||dd|f<q/|S)Nrz>No tau_ray dict was set when initializing the atmosphere classr-rT)rz) rqrrrrrrrrFrr:r&)rZp_profZwvlarrZtau_odZfrayrr-Ztau_tmprrr get_ray_od s  zatmosphere.get_ray_odc Cst|tstdtdtdtdtdtd|jvr[g|_tt|D].}t||tsJtdtdtdtdtdt|jt t ||q*dStdtdtddS)NrzF(set_h2o_in_cloud) alt_bounds should be a list of altitude boundaries zW e.g. alt_bounds = [ [alt_min1, alt_max_1], [[alt_min2, alt_max_2]] ]rTz5(set_h2o_in_cloud) No water vapour in the atmosphere ) r|rFrrrrrrqrrrr)rZ alt_boundsrrrrset_rh100_limits s.   zatmosphere.set_rh100_limitsr)rrrrrrrrrrrrrrrrrrrs"+  % rc@seZdZdZddZdS)geometryr~cCs||_t|t|ddds tdtdtdt||_t|t|s;tdtdtdt||_d|_dS)Nrfrz3(geometry) : vza must be sorted in decreasing orderz3(geometry) : vaa must be sorted in increasing orderr) szarrrrrvzavaaZlon_lat_day_time)rrrrrrrr( s zgeometry.__init__Nrrrrrrrrrr% s rc@sReZdZdZdgddgddgddddd d d d d dd d ddddd d dd fddZdS)rr~rz kdis totorfrdoadTFr"rcri g$@g&@cCsFd|vr dg|_n||_|ddvrtdtdt|d|_|jdkr2|d|_d|vr:d|_ntdtdt||_|jdkr^t|d vr^tdtd t|jd kr~t |t |swtdtd tt ||_ nd |vrt ||_ n||_ |dvrtdtdt||_ |dkr|dks|dkrtdtdt||_ ||_|dkr|dks|dks|dks|dkrtdtdt||krtdtdt||_||_t||_|dkrt||_| |_t| |_t|jd|_nN|dkr1|dkr|j|_n||_||_||_||_| |_||_n,|dkrQ|r;d}| rMtdtdtdtd|_n tdtd|t||_||_|jdkrw| rwtdtd t| |_| dkr|jd!vrtdtd"t| |_ |dkr|j|_!dS||_!dS)#Nrr)monorrz(artdeco_in) ERRORz1 mode must be -mono-, -lbl- or -kdis-rr"z' filter use not implemented)r"rez> In kdis mode, you must provide 1 or 2 wavelenghtsrzI In mono mode, wavelengths must be sorted in increasing orderr )rrdfitpotterzB Trunc_method must be -none-, -dfir-, -potter- or -dm-rrbrcz5 Angular cut for dfit must be >0 and <180rzN theta_min and theta_max for Potter truncation must be >0 and <180zK theta_min must be smaller than theta_max for Potter truncationrreZdisortrfsinscaFz3 Thermal emission is not accounted for z) with the -sinsca- RTE solverz/ rt_model not implemented for f2py:zA To account for thermal IR, rt model must be -disort-)rrzM To account for polarization, rt model must be -doad- or -sinsca-)"keywordsrrrrrDfiltersrqrrrrr dfit_thetac dfit_fitallpotter_theta_minpotter_theta_maxrr'doad_ncoef_min_brdfdoad_eps doad_nfoumxZ doad_nmugod_nstr od_deltam od_corint od_do_secscaod_accur od_use_fisotrt_modelcorintthermalnmat corint_brdf)rrrrrrr'r r r r rrrrrrrrr rrrrrrrrrB s                zartdeco_in.__init__Nrrrrrr? s6rc Csg}g}tt|D]}||dkr||q ||dkr$||q q t|t|kr4tdtg}g}g}tt|D]}|dkrO||dd}nd}|t|dkr`||d}nt|}z|t||||Wn tytdtwz|t|||d||Wn tytdtwz|t|||d|Wq@tytdtwg} tt|D]4}ttt ||||||} |t|dk} | r| d7} | t t j ||||| | d qt | S) r[]zhcannot parse grid specification number of opening and closing braces differs! Use format start[step]stoprr"z:cannot parse grid start specification string not a number!z9cannot parse grid step specification string not a number!z9cannot parse grid stop specification string not a number!)endpoint)rrqrr SystemExitr ValueErrorrrrextendrFrlinspacer) ZgridSpeclpZrprZ gridStartZgridStopZgridStepstartstopZnewGridnrrrrr sN  & " $ & r__main__zpyartdeco_runlib is a library)FTr)E __future__rrrbuiltinsrnumpyrstringrZscipy.interpolaterZncutilsrZlutsrr r r rZscipy.constants constantsrrSrRrXrZgravityrpZkbr$r%Z scale_epsrhrirgrjZphasemat_interpr~rrrrr r*r,rCrErGrMrOrVr[rWr_r|objectr}rrr8r]rrrrrrrrrrrs             bYXzx,p[0