a <ÿùd%?ã@s¨ddlZddlZddlZddlmZejfdd„Zdd„Zdd„Z d d „Z dd„Zd d„Zd"dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zd d!„ZdS)#éN)Ú curve_fitcCs^g}g}tt|ƒƒD]<}|||kr0|krnq| ||¡| ||¡qt|||ƒS©N)ÚrangeÚlenÚappendr)ÚxÚyÚxminÚxmaxÚfuncÚxfÚyfÚi©rú%/home/danka/Public/Fajho2.0/FHeval.pyÚdo_fit srcCsˆ||_||_t|j|j||ƒ\}}| dt|ƒdt|ƒd¡| ||¡d|_|d|_ |d|_ t t |¡¡}|d|_dS)Nu ElÅ‘szakasz illesztÃ©se, x[ú; z], y = p0*x + p1Tré)Ú base_startÚbase_endrrrÚlogÚstrÚ fitreportÚbBaseÚbase_aÚbase_bÚnpÚsqrtÚdiagÚDbase_a©Úmeasr r ÚparamsÚcovsÚperrrrrÚfitbases" r%cCs|jstdƒdS||_||_g}g}tt|jƒƒD]B}||j|krR|kr4nq4| |j|¡| |j|¡q4d}d}d}t dd„|||||fd\} } | dt|ƒd t|ƒd ¡| | | ¡d|_ | d|_| d |_| d|_t t | ¡¡}|d |_dS)NúError: missing base fitçÍÌÌÌÌÌ@çü©ñÒMbP?écSs|t ||¡|Sr©rÚexp©ÚtÚaÚbÚcrrrÚ9ózfitexp..©Úp0u ExponenciÃ¡lis illesztÃ©se: x[rz], y = p0*exp(-p1 * x) + p2Trré)rÚprintÚ exp_startÚexp_endrrrrrrrrrÚbExpÚexp_AÚexp_bÚexp_crrrÚDexp_b)r!r r rr rÚa_guessÚb_guessÚc_guessr"r#r$rrrÚfitexp&s." rAcsvˆjstdƒdSˆjs$tdƒdS|dks6|dkrlg}g}ttˆjƒƒD]F}ˆjˆj|krnˆjkrLnqL| ˆj|¡| ˆj |¡qLd}d}ˆj }|dkrÜtdd „|||||fd \} } ˆ d¡tdƒnFt‡fd d „||||fd \} } ˆ dt ˆj ƒ¡tdt ˆj ƒƒˆ | | ¡dˆ_dˆ_| dˆ_| dˆ_|dkr`| dnˆj ˆ_n|dks€|dkrrg}g}ttˆjƒƒD]¤}ˆjdˆj|krÀˆjkrònn.| ˆj|¡| t ˆj|ˆjˆj¡¡ˆjˆj|krˆjdkr–nn| ˆj|¡| ˆj ¡q–ddlm} | ||ƒˆ_dˆ_dˆ_ˆ d¡tdƒdS)Nz"Error in dualfit: missing base fitz!Error in dualfit: missing exp fitr+Úer'r(ÚfreecSs|t ||¡|Srr*r,rrrr1cr2zdualfit..r3uU KettÅ‘s illesztÃ©s: exponenciÃ¡lis elÅ‘szakasz korrekciÃ³, y = p0*exp(-p1 * x) + cuTKettÅ‘s illesztÃ©s: exponenciÃ¡lis elÅ‘szakasz korrekciÃ³, y = p0*exp(-p1 * x) + ccs|t ||¡ˆjSr)rr+r<)r-r.r/©r!rrr1gr2ue KettÅ‘s illesztÃ©s: exponenciÃ¡lis elÅ‘szakasz korrekciÃ³ [fixÃ¡lt limesz], y = p0*exp(-p1 * x) + udKettÅ‘s illesztÃ©s: exponenciÃ¡lis elÅ‘szakasz korrekciÃ³ [fixÃ¡lt limesz], y = p0*exp(-p1 * x) + TFrrr5ÚsplineÚsr))ÚCubicSplineu? KettÅ‘s illesztÃ©s: elÅ‘szakasz korrekciÃ³ja kÃ¶bÃ¶s spline-alu>KettÅ‘s illesztÃ©s: elÅ‘szakasz korrekciÃ³ja kÃ¶bÃ¶s spline-al)rr6r9rrrrrrrr<rrrrÚbBaseExpÚbBaseSplineÚ baseexp_AÚ baseexp_bÚ baseexp_cÚFHutilÚflinrrr7Zscipy.interpolaterGÚ basespline)r!ÚmethodÚlimitrr rr>r?r@r"r#Z tandata_xZ tandata_yrGrrDrÚdualfitJsZ" ** rRc Cs*|jstdƒdS|js$tdƒdS|j}g|_tt|jƒƒD]à}|j||krd|j |j |¡q>|j|}g}g}tt|jƒƒD]B}||j|kr¢|kr„nq„| |j|¡| |j |¡q„t ||j|ƒ}tt|ƒƒD]}|||8<qä|j |j ||jt ||¡¡q>d|_dS)NúError: base fit not foundú Error: exponential fit not foundT)rr6r9rÚTsrrrrrÚ getTkValuer;rMÚtrapcalcÚbTs)r!Ú startpointrr-rr ÚjÚTkrrrÚgetTs‰s. &r\c CsÚ|jstdƒdS|js$tdƒdS|j}g|_d}tt|jƒƒD]Œ}|j||krh|j |j |¡qBt ||j|ƒ}t ||j|d|j |d||j||j ||¡}|j |j ||j |¡qBd|_dS)NrSrTrrT)rr6r9rrUrrrrrrVrMÚtrapaddr;rX)r!rYÚpresumrr[rrrÚgetTsOptimized¬s :r_ÚindepcCsþ|jstdƒdS|dkr”|jrZ||jkrs$ $?# 1(7B