B 0d@sddlZddlmZddlZddlZddlZddlmZmZddl m Z m Z m Z m Z mZmZmZddl mZddl mZdZeeddZed d d gZed d ddddgZeddddddgZejereeeddZne Zded<e!dge!dge!dgd\Z"Z#de_$ddddgZ%d dd!gZ&d dd!d"d#d$gZ'd%d&e(e%e&e'DZ)d'\Z*Z+d(d&e,d)DZ-d*d+Z.d,d-Z/d.d/Z0d0d1Z1d2d3Z2d4d5Z3d6d7Z4dGd8d9Z5d:d;Z6dd?Z8d@dAZ9dBdCZ:dDdEZ;eBsr,)g& .>g& .>cCsg|]}t|ddttqS)r!r )rmom_atolmom_rtol)r)ir&r&r+r,Lsc Cs,ttdd}tjt|ddWdQRXdS)z3Writes the current res_dict to the target JSON filew)r)indentN)openresults_filepathjsondumpres_dict)fr&r&r+ write_dataQsr:cCs t|S)zConverts a namedtuple to a dict)dict_asdict)Z named_tupler&r&r+to_dictWsr=cCst|jt|jdS)z;Formats an MPResult namedtuple into a dict for JSON dumping)rr)r=rfloatr)rr&r&r+mp_res_to_dict\sr?cshtttfddfddfdd}t|dtgt tgdd d }|S) z7Straightforward implementation of studentized range CDFcs(t|t||t|dS)Nr)phiPhi)sz)rrr&r+innerjszcdf_mp..innercs&|dtt|||S)Nr)r@r)rBrC)rDnur&r+outermszcdf_mp..outercsBtdtdtdddd||S)Nr2r)rrr)rBrC)rrErFr&r+wholepszcdf_mp..wholerzgauss-legendrer )method maxdegree)r rr )rrrErGresr&)rDrrErFrr+cdf_mpfs rKcshtttfddfddfdd}t|dtgt tgdd d }|S) z7Straightforward implementation of studentized range PDFcs8t||t|t||t|dS)Nr2)r@rA)rBrC)rrr&r+rD}szpdf_mp..innercs"|tt|||S)N)r@r)rBrC)rDrEr&r+rFszpdf_mp..outercsJtdtddtdddd||S)Nr2r)rrr)rBrC)rrErFr&r+rGszpdf_mp..wholerzgauss-legendrer )rHrI)r rr )rrrErGrJr&)rDrrErFrr+pdf_mpys rLcsztttfddfddfddfdd}t|d tgd tgt tgd d d }|S) z.Implementation of the studentized range momentcs8t|||t|t|||t|dS)Nr2)r@rA)rrBrC)rr&r+rDszmoment_mp..innercs$|tt||||S)N)r@r)rrBrC)rDrEr&r+rFszmoment_mp..outercsLtdtddtdddd|||S)Nr2r)rrr)rrBrC)rrErFr&r+pdfszmoment_mp..pdfcs||||S)Nr&)rrBrC)rrMr&r+rGszmoment_mp..wholerzgauss-legendrer )rHrI)r rr )rrrErGrJr&)rDrrrErFrMr+ moment_mps  rNcsD|tkrt|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&)rJ) case_dictr&r+zresult_exists..r)r8 ValueErrorr=listfilterlen)set_keyr*Z existing_resr&)rOr+ result_existss  rWcCsLt}||}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_counterprintr?r)r* run_lambdarVindex total_casest_startrJr&r&r+runs(r`cCs$|\}}}t|||tdS)zSA callback for completed jobs. Inserts and writes a calculated result to file.N)r8insertr:)rJr]rVZ result_dictr&r&r+ write_results rbcsfddtD}ttt|dtdg}tt}x"|D]}||jt|tdqPW| | dS)z)Runs an array of cases and writes to filecs,g|]$\}}t|s||tfqSr&)rWrU)r)r]r*)casesr\rVr&r+r,szrun_cases..rXz? cases won't be calculated because their results already exist.)argscallbackN) enumerater[rUr num_poolsappendZ apply_asyncr`rbclosejoin)rcr\rVZjob_argjobspoolr*r&)rcr\rVr+ run_casess $ rmcCst|j|j|jS)N)rLrrr)r*r&r&r+run_pdfsrncCst|j|j|jS)N)rKrrr)r*r&r&r+run_cdfsrocCst|j|j|jS)N)rNrrr)r*r&r&r+ run_momentsrpcCst}dtttt}td|dtdttdtttdtdttdtttd td ttd ttt d td t|ddS)Nr2z 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.rzTest data generated in rB) rYrZrU cdf_pdf_cases moment_casesr[rmrnrorp)r_r^r&r&r+mains   rs__main__)rr)= itertools collectionsrr6rYosmultiprocessingrrZmpmathrrrrr r r r r@r rAr5maxrgrrrpathisfileloadr4r8r; setdefaultr'r(ZdpsZcp_qZcp_kZcp_nuproductrqr-r.rangerrr:r=r?rKrLrNrWr`rbrmrnrorprs__name__r&r&r&r+s` $