o 5ήc@sddlZddlmZddlZddlZddlZddlmZmZddl m Z m Z m Z m Z mZmZmZddl mZddl mZdZeeddZed d d gZed gd ZedgdZejerkeeeddZne Zded<e!dge!dge!dgd\Z"Z#de_$gdZ%gdZ&gdZ'dde(e%e&e'DZ)d\Z*Z+dde,d DZ-d!d"Z.d#d$Z/d%d&Z0d'd(Z1d)d*Z2d+d,Z3d-d.Z4d>d/d0Z5d1d2Z6d3d4Z7d5d6Z8d7d8Z9d9d:Z:d;d<Z;eAsr0)& .>r1cCsg|] }t|ddttqS)r%r$)rmom_atolmom_rtol)r-ir*r*r/r0KscCsBttdd}tjt|ddWddS1swYdS)z3Writes the current res_dict to the target JSON filewr)indentN)openresults_filepathjsondumpres_dict)fr*r*r/ write_dataQs"r?cCs t|S)zConverts a namedtuple to a dict)dict_asdict) named_tupler*r*r/to_dictWs rCcCst|jt|jdS)z;Formats an MPResult namedtuple into a dict for JSON dumping)rr)rCrfloatr)rr*r*r/mp_res_to_dict\srEchtttfddfddfdd}t|dtgt tgdd d }|S) z7Straightforward implementation of studentized range CDFcs(t|t||t|dSNrphiPhiszrrr*r/innerjs(zcdf_mp..innercs&|dtt|||SrGrIrrKrOnur*r/outerms&zcdf_mp..outercsBtdtdtdddd||SNr7rrrrrKrrRrSr*r/wholeps zcdf_mp..wholergauss-legendrer$method maxdegreer rr rrrRrWresr*rOrrRrSrr/cdf_mpfr`crF) z7Straightforward implementation of studentized range PDFcs8t||t|t||t|dSNr7rHrKrNr*r/rO}8zpdf_mp..innercs"|tt|||SNrPrKrQr*r/rSs"zpdf_mp..outercsJtdtddtdddd||SrTrUrKrVr*r/rWs "zpdf_mp..wholerrXr$rYr\r]r*r_r/pdf_mpyrarecsztttfddfddfddfdd}t|d tgd tgt tgd d d }|S) z.Implementation of the studentized range momentcs8t|||t|t|||t|dSrbrHrrLrM)rr*r/rOrczmoment_mp..innercs$|tt||||SrdrPrfrQr*r/rSs$zmoment_mp..outercsLtdtddtdddd|||SrTrUrfrVr*r/pdfs " zmoment_mp..pdfcs||||Srdr*rf)rrgr*r/rWszmoment_mp..wholerrXr$rYr\)rrrRrWr^r*)rOrrrRrSrgr/ moment_mps rhcsD|tvr t|dt|ttfddt|}t|dkS)znSearches the results dict for a result in the set that matches a case. Returns True if such a case exists.z not present in data structure!cs |dkS)Nrr*)r^ case_dictr*r/s zresult_exists..r)r= ValueErrorrClistfilterlen)set_keyr. existing_resr*rir/ result_existss  rrcCsLt}||}td|dd|dt|d||tt||fS)zARuns the single passed case, returning an mp dictionary and indexz Finished r/z in batch. (Took zs))time perf_counterprintrEr)r. run_lambdarpindex total_casest_startr^r*r*r/runs   r{cCs$|\}}}t|||tdS)zSA callback for completed jobs. Inserts and writes a calculated result to file.N)r=insertr?)r^rxrp result_dictr*r*r/ write_results  r~cs~fddtD}ttt|dtdg}tt}|D] }||jt|tdq'| | dS)z)Runs an array of cases and writes to filecs,g|]\}}t|s||tfqSr*)rrro)r-rxr.casesrwrpr*r/r0szrun_cases..rsz? cases won't be calculated because their results already exist.)argscallbackN) enumeratervror num_poolsappend apply_asyncr{r~closejoin)rrwrpjob_argjobspoolr.r*rr/ run_casess$ rcCt|j|j|jSrd)rerrrr.r*r*r/run_pdfrcCrrd)r`rrrrr*r*r/run_cdfrrcCrrd)rhrrrrr*r*r/ run_momentrrcCst}dtttt}td|dtdttdtttdtdttdtttd td ttd ttt d td t|ddS)Nr7z Processing z test caseszRunning 1st batch (z' PDF cases). These take about 30s each.rzRunning 2nd batch (z' CDF cases). These take about 30s each.rzRunning 3rd batch (zH moment cases). These take about anywhere from a few hours to days each.r zTest data generated in rL) rtruro cdf_pdf_cases moment_casesrvrrrr)rzryr*r*r/mains   r__main__)rr)= itertools collectionsrr;rtosmultiprocessingrrmpmathrrrrr r r r rIr rJr:maxrrrrpathisfileloadr9r=r@ setdefaultr+r,dpscp_qcp_kcp_nuproductrr2r3rangerr?rCrEr`rerhrrr{r~rrrrr__name__r*r*r*r/sl $