m @Yc@sddlZddlZddlmZmZmZmZmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZmZddlmZmZddlmZmZddlmZdZdefd YZd efd YZd efd YZdefdYZdefdYZdefdYZdefdYZ defdYZ!defdYZ"defdYZ#defdYZ$defdYZ%d fd!YZ&dS("iN(toptimizet interpolatetsignaltstatstndimage(tProcessingUnitt Operation(t Parametersthildebrand_sekhon(tasarraytexp(t curve_fitiJxtParametersProccBs)eZdZdZdZdZRS(cCs8tj|d|_d|_d|_t|_dS(Ni(Rt__init__tNonetbuffert firstdatatimet profIndexRtdataOut(tself((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyR s     cCs|jj|j_|jj|j_|jj|j_|jj|j_|jj|j_|jjj |j_|jj j |j_ |jj |j_ |jj |j_ t jddg|j_|jj|j_|jj|j_|jj|j_|jj|j_|jj|j_|jj|j_|jj|j_|jj|j_|jj|j_|jj|j_|jj|j_ |jj|j_|jj|j_dS(Ntreals   N(t__name__t __module__RtnSecondsR R:RY(((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyR s "tPrecipitationProccBseZdZdZRS(su Funtion that uses Reflectivity factor (Z), and estimates rainfall Rate Input: self.dataOut.data_pre : SelfSpectra Output: self.dataOut.data_output : Reflectivity factor, rainfall Rate Parameters affected: Winds, height range, SNR cCs|jdj}|j}|j}tj|dddddf|d|dddddf|dg}|j}|jd}|jd}|jd} tj| |g} tj | |gtj } d} tj |d} tj | dk| tj } tj | |g} xt |D]~}| d|f|d|fd||dd| d|f<| d|f|d|fd||dd| d|feZdZdZddZdZdZdZRS(s Function that implements Full Spectral Analisys technique. Input: self.dataOut.data_pre : SelfSpectra and CrossSPectra data self.dataOut.groupList : Pairlist of channels self.dataOut.ChanDist : Physical distance between receivers Output: self.dataOut.data_output : Zonal wind, Meridional wind and Vertical wind Parameters affected: Winds, height range, SNR c Cs|jdj}|jdj}|jd}|jd}|jd}|j}|j}|j} tj||g} tj||g} tj||gdtj} tj||g} tj|}tj|}|j}|j }dtj |j }g}g}xt |D]}tj |tj|jdd|f|jdd|f|jdd|fg}tj |tj|d|f|d|f|d|fg}q$Wtjd|jdgtj}g}g}g}xSt |D]E}|j||||||| \}}}}t|dkrrt|dkrrtj ||}ntj |tj}t|dkrt|dkrtj ||}ntj |tj}t|dkrtj ||}ntj |tj}||d krtj||tj| d}?tj| d}@|<||=|d|7tj!|>tj tj|7}A|<||=|d|7tj!|?tj tj|7}B|<||=|d|7tj!|@tj tj|7}Ctj|A|B|Cg}Dtj|d|dd||g|d|dd||g|d|dd||gg}Etjj |E|D\}F}G}Htj|F|Hg|H|Ggg}Itj|< |= g}Jtjj |I|J\}K}L|L}M|K}Ntj |Md|Nd}Otj|N|M}P||+}QqMW|M|N|Q|,fS(NiiR#iRigh㈵>tp0g333333?sVerificador: Dentroig?tkeycst|S(N(Ry(R(teMinus1(sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pytsgHz>cst|dS(Ng?(Ry(R(tMaximun(sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyRsig?("R"RbR`RuRRetisfinitetNANRfRdRtarRtsqrtRytarctan2RRRvRwR Rtamaxt RuntimeErrorR RtminR_Rt linregresstpitlinalgtsolvetlog(RRRjRzRIR4RR0R5R}R~RRRRtxFrectE01tN01tE02tN02tE12tN12tztitzlinetFactNormt SmoothSPCRtcspcLinet chan_index0t chan_index1t CSPCFactortCSPCNormtyMeantyMean2Rt meanGaussRtpopttpcovtFitGausstHWpost HalfWidthtGCpostVposRtFijtRangpostRanget FrecRanget PhaseRangetslopet intercepttr_valuetp_valuetstd_errtcCt MijEijNijt MijResult0t MijResult1t MijResultstcFtcGtW01tW02tW12t WijResult0t WijResult1t WijResult2t WijResultst WijEijNijtcAtcBtcHtVxVyt VxVyResultstVxtVyRRtVmagtVangR((RRsQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyRx>s##,# !$$7F"*""  )"'  0'  /-!    !;;;c!(RZR[RrRYRRRRx(((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyRss  W   tSpectralMomentsc BsGeZdZdZddddddddddd ZdZRS(sd Function SpectralMoments() Calculates moments (power, mean, standard deviation) and SNR of the signal Type of dataIn: Spectra Configuration Parameters: dirCosx : Cosine director in X axis dirCosy : Cosine director in Y axis elevation : azimuth : Input: channelList : simple channel list to select e.g. [2,3,7] self.dataOut.data_pre : Spectral data self.dataOut.abscissaList : List of frequencies self.dataOut.noise : Noise level per channel Affected: self.dataOut.data_param : Parameters per channel self.dataOut.data_SNR : SNR per channel cCs|jd}|jd }|j}|jd}tj|d|jdf}x_t|D]Q}|j||ddddf|||||ddddf Function GetMoments() Input: self.dataOut.data_pre self.dataOut.abscissaList self.dataOut.noise self.dataOut.normFactor self.dataOut.data_SNR self.dataOut.groupList self.dataOut.nChannels Affected: self.dataOut.data_param cCs|jd}|jd}|jd}|jd}|jd}|jd}|j}|j} |j} |j} |j} xftt |D]R} || ddddf|| ddf|| ddddf|jdd}tj|dd}|jdd}tj|jdd}|dd|ddf}xZt|jdD]E} tj||| ddfjdd|| ddfeZdZeddddZdZdZdZRS(sg Function GetMoments() Input: Output: Variables modified: c/ Cs|dkrtjj|ntj||j_tj |}||j_ |j d}|j j }|j jj} d|j_|jjj|j } | |j_|j j} |j j} |j j} |j j}tj |j j}tjtj|j dd}tjtt|dd}|r|j|j}|j|j j }|j!|j j"|ddddf|||j_#nx[t$|D]M}||ddf}|j j"|ddddf| | }|j|j d|j d|j df}d}xk|D]c}tj ||||g}t%tj&tj'||kdddd||<|d7}q/W|j j(|ddddf| | }|d|}x$t$| D]}|dd|f}tj)|d|}d}x|D]}|d}|d}||dd|f} tj)| }!|!||| |d| || |d| f<|!||| |d| || |d| f<|d7}qWtj*j+|}"tj*j,|"}#|#j-}$tj.|$|}%|j j"|dd|f}&|dkrv|'ddkrv|jj|dd|df}(n$tj |jjj/|&| |}(ywt0j1|j2|(d |%|$| fd t3\})}*}+},}-tj4|+d dd| }.|jjj5|)| |$}'Wn*|(tj6})tj6}.|(tj6}'nX|jjdkrtj||(j| ftj6|j_tj||(jd| ftj6|j_7ntj8|.|'f|jj7|dd|f<|)|jj|dd|f-. 6-   44 &$9  (/.'cCs2|jjj||}tj||}||S(N(RRPt modelFunctionR"R`(RtpR~R}RRtfmtfmp((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyt__residFunctionSscCs3tj|dd}|j||}|j}|S(NR4i(R"taverageR_(RRR0tavgRO((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyt__getSNRZs c Cs`t|||\}}}tj||}tj||}tj||j||}|S(N(t setupLTdfmR"R`R_( RtchindexthindexR}RwRR~Rtchisq((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyt__chisqas N( RZR[RrRRRRYRcRZt_SpectralFitting__chisq(((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyRHs n  t WindProfilercBseZeZdZdZdZdZeZ dZ dZ dZ dZ dZdZdZdZdZddZdZd Zd Zd Zd Zd ZdZdZdZdZRS(cCstj|dS(N(RR (R((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyR yscCsd|tjd}|tjd}tjdtj|ddtj|d}tjdtj|d|d}tjtj|}tjtj|}tj||}tj||}||fS(NiZiii(R"RRtcosttantsigntsintcopysign(RtelevtazimtzentcosDirXtcosDirYtsignXtsignY((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyt__calculateCosDir|s5(c Cstjd|d|d}tj|}tj|||tjd}tj|tj|}tj|tj|}|||||fS(Niii(R"RtarccosRRRRR( Rttheta_xttheta_ytazimuthtdir_coswt zenith_arrt azimuth_arrtdir_cosutdir_cosv((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyt__calculateAngless !cCsk|rtj||f}ntj|||f}tj|}tjj|j||j}|S(N(R"tc_tasmatrixRR]t transpose(RRRRthorOnlytAtA1((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyt__calculateMatAs &cCs|j}|jt|}|jt|}tt|}|tj||} |tj||} | | dkj} t j | | } t j t|t| g} t j t|t| g} x|D]}|tj||}||ddf}t j ||dd}| }||}||ddf}t j ||dd}||}|| |ddf<|| |ddftj|d}tj|d}|j||\}}|d} |jd r|d } nt} |jd r|d } nd } |jd rj|d } t| d kr%t} ntj| } || ddddf}|| }|| }n|j||| \}}}}}|j||| ||\}}}|j |||| }|j ||}|||fS(s Function that implements Doppler Beam Swinging (DBS) technique. Input: Radial velocities, Direction cosines (x and y) of the Beam, Antenna azimuth, Direction correction (if necessary), Ranges and SNR Output: Winds estimation (Zonal, Meridional and Vertical) Parameters affected: Winds, height range, SNR RR!ROtdirCosxtdirCosyt elevationRtcorrectAzimuththorizontalOnlyt correctFactoriR iN( thas_keyR"R_t_WindProfiler__calculateCosDirR=RRRt_WindProfiler__calculateAnglest_WindProfiler__correctValuest_WindProfiler__calculateMatAt_WindProfiler__calculateVelUVW(Rtkwargst velRadial0RtSNR0RRRRRRRR t arrayChanneltparamRRRRRRRRRtwinds((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyt techniqueDBSs:          $%c Cst|}tj|}tj|}|dkr|tjd}|tj||tj|}| tj||tj|}n |}|}tj|}tj|} tj|} tj|} xt |D]} ||| d||| d|| <||| d||| d| |  )!     !!6cKs|d}|d}|d}|d}|d}|d}|d}|d} |j|||\} } | |} |j|||} | ddd f| ddd fd | | | ddd fjt| ddd fR=RRt_WindProfiler__isConfigtdatetimetutcfromtimestampRtreplacet total_secondst_WindProfiler__bufferRt_WindProfiler__firstdataR"Rt_WindProfiler__checkTimeRRRkRtfloorR1ReR_(RRt techniqueRRRR0R!RORiR RbRcRoR RR!RpRJRqRtlambR"Rtinitimet minuteAuxt minuteNewR((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyRY|s     #  !             =    '               :      B $ N(RZR[R=RrRRt_WindProfiler__lastdatatimet_WindProfiler__integrationtimeRwRRxRR RRRRRRRRRRRyRR1R$R#ReRY(((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyRis4  !  / *  & ? ? % Q  tEWDriftsEstimationcBs#eZdZdZdZRS(cCstj|dS(N(RR (R((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyR "scCs|j}|jt|}|jt|}tt|}|tj||} |tj||} | | dkj} t j | | } t j t|t| g} t j t|t| g} x|D]}|tj||}||ddf}t j ||dd}| }||}||ddf}t j ||dd}||}|| |ddf<|| |ddfRi(2RRpt SNRthreshR;t cohThreshtallDataRWRXR|ttSampRAR,tnLagsR@R|R+R<R!R,R(RR0RORtSNRmRRR~tcoh1RRXRYtccftcohtphaseAuxtvelRadt boolMetFintboolMet1tstructderPhasetboolMet2tcoordMetR>R?Ritacf1tacf2tspcWidthtboolMet3R^((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyRYhs         +#"  44k78@ ` -      5 @@@<     &&)cCstjj|\}}tj|}xtd|dD]}||k}|j} tj|\} } | j| j } | j| j } | dks| |ks| |kr;t ||(1RRthei_refttauindext phaseOffsetst cohDetectiontcohDet_timeStept cohDet_threshtnoise_timeSteptnoise_multipletmultDet_timeLimittmultDet_rangeLimitt phaseThresht SNRThreshRbRcRtchannelPositionst meteorOpst pairslist0t distancesRt voltsPShifttvoltsDCRtpowerNetR0tnoise1t signalThresht listMeteorstheiRanget rangeIntervalt rangeLimitt timeLimitt listMeteors1tthresht listMeteors2tlistMeteorsPowertlistMeteorsVoltst listMeteors3tradialStdThresht listMeteors4tdatetarrayParametersRItpairxtpairytjphRv((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyRYcs^   !>    0  B      cCst|d}t|d}|dddd||df}|jd|}d} d} tjt||f} xt|D]} | |7} tj|j|dd| | ddf|dg } tj | dd} | j | dd| f<| } qWd}| t j | d}tj |dd}|j|jdf}tjtj|||k|| |k}tj| |>c3 Cs!|jd}|jd} |d} |d} t|d} d| d} tj|}|dddf}t|}|jd|jdg}g}g}|dkrdd}n d d}tj||}x2tt|D]}tj d }||d}||d}||d}||d}|dd||d|f}|j |jd|jdd}|j ||\}}|dddd|f}|dd|f| } |dd|f}!|j ||ddf|||ddfR=t_SMPhaseCalibration__dataReadyRHt_SMPhaseCalibration__isConfigRsRtR(t_SMPhaseCalibration__initimeRuRvt_SMPhaseCalibration__bufferRRiRR"Rt_SMPhaseCalibration__checkTimeRAR?R/RqRRRRet_SMPhaseCalibration__getGammast_SMPhaseCalibration__getPhasesRXRRh(RRRbRcRRiRRtR|R"RRvRrRRRRRt boolErrorRRRt phasesOff((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyRY sN   = '          4   N(RZR[RRRR=RRRRRRRRY(((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyR s 6   7RcBsGeZdZdZdZdZdZdZdZRS(cCsdS(N((R((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyR scCs5|j}|d}|d} tjd} |dddf} |ddddf |} |j| ||| | |\|ddddf<|dddf<|dddf} |dddf} |dddf}|j| || || \|ddd f<|dddf<|S( Niiiii iiii(RR"Rt_SMOperations__getAOAt_SMOperations__getHeights(RtarrayParameters0RRvRIRRRRbRct AOAthreshRRtRangesR((sQ/home/erick/workspace/schain_ebocanegra/schainpy/model/proc/jroproc_parameters.pyR s    !JAcCstj|jddf}|j|||\}} |j| ||ddddfs>  (      w@bF