&cSc@sdZddlZddlmZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z mZddlmZmZmZddlmZmZddlmZddlmZdd lmZdd lmZd Zd Zd Z dZ!dZ"dZ#dZ$dZ%dZ&de j'efdYZ(de)fdYZ*e*Z+dZ,dZ-dZ.dZ/dZ0de j1fdYZ2de)fd YZ3d!e)fd"YZ4d#e4fd$YZ5d%e j1fd&YZ6d'e3fd(YZ7d)e2fd*YZ8d+e j1fd,YZ9d-fd.YZ:d/e j;fd0YZ<e=d1kre j>ej?Z@e jAe jBd2d2d2jCe jBjDe(ZEeEjFd3d4d5eEjGejHe@jIndS(6s3 This module is a graph and map visualizing tool. iN(togr(tQttQtCoretQtGuitQtOpenGLtQwt5(tarraytsqrttaverage(trandinttuniform(tUi_Form(tcm(t MethodType(tSimpleXMLRPCServertqtcCs|j}|tjjkr1|jddnL|tjjkrV|jddn'|tjjkr{|jddn|tjjkr|jddn|tjjkr|j dn|tjj kr|j ddn|tjj ks |tjj krjxn|j jD]D}t|tr|jdtjddtjdqqWntjj||dS(Niiig333333?iiji,(tkeyRRtKey_Upt translatetKey_DowntKey_Leftt Key_RighttKey_Plust scaleViewt Key_Minust Key_Spacet Key_Entertscenetitemst isinstancetPolygontsetPostqrandRt QGraphicsViewt keyPressEvent(tselfteventRtitem((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyR"s$ $6cCs(|jtjd|j ddS(Ng@gn@(Rtmathtpowtdelta(R#R$((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt wheelEvent/scCsM|jj||jtjddddj}|j||dS(Nii(tmatrixtscaletmapRectRtQRectFtwidth(R#t scaleFactortfactor((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyR2s9cCssx|jD]}|jq Wt}x&|jD]}|jr.t}q.q.W|so|j|jd|_ndS(Ni(tnodestcalculateForcestFalsetadvancetTruet killTimerttimerId(R#R$tnodet itemsMoved((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt timerEvent8s  cCs&|js"|jdd|_ndS(Nii(R7t startTimer(R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt itemMovedGs cCs2d}x|D]}||d7}q Wtj|S(Nii(R&R(tatacctvalue((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt array_magKs cCsKt|}|j}x,tt|D]}|||||ssstateChanged(int)N(tnodeListLayouttsortRt QCheckBoxt scrollAreatsetTextRRRkRlRntselectt setToolTiptstrtgeocodet addWidget(R#tpltlotntcb((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pytfill_nodel_lists  "tallc Cs|jj}|jdgt|j}|dkrzd}xK|jjD]"\}}t|j|| Plots a time series curve to the plot window iN(RCRSRRRR{RRRRRdtnextRRyRRgtpolyDictR(R#tgcRRRRRR((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt addTsCurves c Cstd|d|d|d||_|jj|jj}}|jj|jj}}||d}||d} ||d} ||d} tj|j|j_ |jj j |||| |jj |||| |jj |jj |jj |jj j|jj| | |jjtjj|jjtjj|jjtjj|jjtjjx'|jjD]} |jj j| qW|j|jj|jjdd|jj|} |jj| | |j j!r |j"dnd S( s> Draws the map stored in the shapefile fname. tfnametdisplayt namefieldt geocfieldg?g@i,RN(#tMapRgtxmintymintxmaxtymaxRtQGraphicsSceneRhRt setSceneRectt fitInViewtsetScenetupdateSceneRectt sceneRecttcenterOnt setCacheModeR!tCacheBackgroundt setRenderHinttQPaintert AntialiasingtsetTransformationAnchortAnchorUnderMousetsetResizeAnchortAnchorViewCentertpolyListtaddItemRtsetMinimumSizeR.R+RLt isCheckedR(R#tfilenameRRRRRRtxxstyxstxctycRt scale_factor((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pytdrawMaps2$c CsW|jjdt||_d|j_tt|j|j_tt|j|j_t j |j|j_ g|D]}|d|d f^qz}t |j dd\}}t |jdd\}}x|D]}t|j|d|d} | j|d|d ft||||tjt|d| _|jj j| |jj| qW|jj|j_x|D]} t|jjt| d|jjt| d} t||||tjt|d| _|jj j| |jj| qW|||_|_|||_|_|j d S( s Draws a graph in the scene nlist: is a lis of nodes in the format(x,y,geocode,name) elist: is a list of edges described by tuples of indices to the first list. s Network Viewiiitaxisig?g?N(!tlabelRtGraphRgRhR7R R<R:RRRRtmintmaxtNodeRR&RRCtsizeRt insertNodeR1tEdgetintt arrowSizet insertEdgeRRRRRt( R#tnlisttelistRtnposRRRRR8teted((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt drawGraphs0 ( 1 71c Css|j|j}}|j|j}}||d}||d}||d}||d}|jjjtjj |jjj |||||jj |||||jj |jj|jj |jjj|jj|||jj|} |jj| | |jjtjj|jjtjj|jjtjj|jjtjjdS(sJ Centers the scene and fits the specified rectangle to it g?g@N(RRRRRhRtsetItemIndexMethodRRtNoIndexRRRRRRR.R+RR!RRRRRRRR( R#RRRRRRRRR((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRts$cCs1t|jdkr-t|j}nd}x|jD]\}}y||:}Wn(tk r}|G|G||G|GHnX|j|}t|}|jjj|r@t j t|ddt|ddt|ddt|dd|jj|_ |jj|j q@q@WdS(s Paint the polygons with the data from data dict datadict is a dictionary of the form {geocode:value,...} ig?iiiiN( RRRt TypeErrorRKRRgRthas_keyRtQColort fillColortupdate(R#tdatadicttnormwRtvalRtcol((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt paintPols3s ]cst|j|jfd}tjjtjd|jtjjtjd|jtjj|j tjd|j dS(s? Replay the time series from beggining to end. csjdS(N(tquit((trw(sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt stop_replayOstdrawSteptflashs released()N( t ReplayWorkerRSRTRRkRlRnRt flashBorderst pushButton_2trender(R#R((RsL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRrJs """cCsG||_|j||jj||jj|||j|t|dD],}|||fr7|j||fq7q7Wq W|S(s: Extract edges from the adjacency matrix. ii(tshapeRBR*(R#R*tsizteltctl((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pytgetEdgeFromMatrix?s cCsdS(N((R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt centerViewMscCs-|j|j}tjddd}xt|jD]}x1|jD]&}|j|jrE|jqEqEWx|j D]}|j s|j r|j |j n|j r|j|j n|j|n|j||j7}|tt|j |_qyWx|jD]}|jqWq5WdS(sP Perform one Iteration of the system by advancing one timestep. iN(R4R6RtvectortrangeR.tcalcSpringForcetdampingtcalcDampingForceR1tfixedR7tcalcGravityForceR8tcalcViscosityForcetcalcNewLocationt clearForceR0tfloatRCtcenterR(R#t microstepRYRFR,R8((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyR4Ps&        cCsdS(s>Process the drag and drop interaction from the mouse. N((R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt dispatchDnDiscCs|jtj|jdS(spPerform one step. This is a convenience method. It actually calls dispatchDnD() and advance(). N(R4RxtsleepR4(R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRfos cCsx |jqdS(s[Start the mainloop, which means that step() is called in an infinite loop. N(Rf(R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pytmainloopwsN(R R RRORNRRBRDRRFRLRMR4R[RfR](((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRs         tBaseMapcBs)eZdddZdZdZRS(t NOME_ZONASt ZONA_TRAFEcCs||_||_g|_i|_i|_g|_g|_i|_tj j |rj|j |ntj j d|}tj j |r|j |nIt jjd|jd|jdj||tj|jddS(Ns..s Map not founds"Neither {} nor {} were found in {}s&OK(RRt centroidstcentdicttgeomdictRRRtostpathtexiststReadertjoinRt QMessageBoxt informationROttrUtf8tformattgetcwd(R#RRRtupfname((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRNs"          !cCsYtj|}|jd}d}g}|j}x|dk r|jj}|j}|jdkr|jd} |j |j } |j |j } | |j | <| jdkr| j} |jj|| j| j| jf|j| Returns the bounding rectangle for the graph iiii(R1R0RRR(R#R((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pytgetRectIs ???C(R R RNR(((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyR>s RcBsheZdZdZdZdZdZdZdZdZ dZ d Z d Z RS( sO Physical model and visual representation of a node as a mass using Qt cCswtj||g|_||_||_tjddd|_d|_|j t |jd|t |_ dS(s# Construct a mass. iiis - N( RRNR"RRRRRRRRR3tselected(R#tgraphwRR((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRNYs    cCs`|jrt|_|jjjr|jjj|jjj|jjj|j}tj t |ddt |ddt |ddt |dd|_ ntj ddd|_ t j j|_|jj|jjjjn_t|_tj ddd|_ t j j|_|jjj|j|j|jjjj|jdS(Niiiii(RR3R!RRSRKRfRRRRRRRR]RRR RyRR5RRRR(R#R$R((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRhs  2V  cCs|jdS(N(R(R#R$((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRscCstjS(N(RR((R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyR)scCs|jj||jdS(N(R"R*R+(R#R,((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyR-scCs|jS(N(R"(R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyR.scCs,d}tjd|d|d|d|S(Ng@ii(RR-(R#R+((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRscCs,tj}|jdd|j|j|S(Ni(RRt addEllipseR(R#Re((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRGs cCsd|jtjj|jtjj|jdd|j|jtj ddd}|j tj j @r|j dd|jdd|jdtj|jjd|jdtj|jjdjdn/|jd|j|jd|jjd|jtj||jtjtjjd|jd d |j|jdS( Niii iiixiii(RRRtNoPenRtdarkGrayt drawEllipseRRtQRadialGradienttstatetQStylet State_Sunkent setCentert setFocalPointt setColorAtRRtlighttdarkertQBrushRR](R#RRRtgradient((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRs%1cCs+|tjjkrntjj|||S(N(RR tItemPositionChanget itemChange(R#tchangeR?((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRs( R R RRNRRR)R-R.RRGRR(((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRUs         RcBs|eZejZdeZejjdZ dZ dZ dZ dZ dZdZdZd Zd ZRS( g@icCstjj|d|_tj|_tj|_|jtj j ||_ ||_ |j j ||j jj|j |j j ||j jj|j |jdS(Ng$@(RR RNRRR#t sourcePointt destPointtsetAcceptedMouseButtonsRtNoButtontsourcetdestR-R&R*R+(R#t sourceNodetdestNode((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRNs   cCstjS(N(RR((R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyR)scCs|jS(N(R(R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRscCs||_|jdS(N(RR+(R#R8((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt setSourceNodes cCs|jS(N(R(R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRscCs||_|jdS(N(RR+(R#R8((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt setDestNodes cCs|j s|j rdStj|j|jdd|j|jdd}|j}|dkrjdStj|jd||jd|}|j |j ||_ |j ||_ dS(Nigi (RRRtQLineFt mapFromItemtlengthR#tdxtdytprepareGeometryChangetp1Rtp2R(R#tlineRt edgeOffset((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyR+s6  . cCs|j s|j rtjSd}||jd}tj|jtj|jj|jj|jj |jj j j | | ||S(Nig@( RRRR-RRtQSizeFRRRt normalizedtadjusted(R#tpenWidthtextra((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRs  c Cs|j s|j rdStj|j|j}|jdkrFdS|jtj tj j dtj j tj j tj j|j|tj|j|j}|jdkrtj|}n|jtjtj|tjd|jtj|tjd|j}|jtjtj|tjtjd|jtj|tjtjd|j}|jtjtj|tjd|jtj|tjd|j}|jtjtj|tjtjd|jtj|tjtjd|j} |jtj j |jtj|j||g|jtj|j|| gdS(Ngiii( RRRRRRRRRRRR]t SolidLinetRoundCapt RoundJointdrawLineR&tacosRRRtTwoPiR#tsintPiRtcosRRRRR( R#RRRRtanglet sourceArrowP1t sourceArrowP2t destArrowP1t destArrowP2((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRs*: *%1,*%1,%(R R R&tpiRRRR R3R(RNR)RRRRR+RR(((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRs          t MapServercBs#eZdZddZdZRS(s xmlrpc server iPcCs@td|f|_d|_d|_tjdd|_dS(NRiRKi2(RtserverROR9RfR RRRK(R#tporta((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRNs  cCs|jjdS(N(Rt serve_forever(R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pytstarts(R R RRNR(((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRs RcBs2eZdddZdZdZdZRS(g?cCsPtjj||tj|_tj|_||_||_||_ dS(N( RtQThreadRNtQMutextmutextQWaitConditiont conditionRSRTtperiod(R#ttstarrRtparent((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRN s   cCs5|jj|jj|jj|jdS(N(RtlockRtwakeOnetunlocktwait(R#((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyt__del__'s   cCs tj|j}|jdS(N(Rt QMutexLockerRR(R#tlocker((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyR,scCsxtt|jD]}|jj|jtjd||j||jj|j j |r|jtjd||j |nt j |j qW|jj|jj|j|jjdS(NRR(RBRCRSRRtemitRRnRRTRRxR\RRR(R#R((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pytrun/s # & N(R R RORNRRR(((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyRs  t__main__isriozonas_LatLong.shpR_R`(JRR&tosgeoRt threadingRURxRdtsystPyQt4RRRRRR{tnumpyRRRt numpy.randomR R t Ui_displayR t matplotlibR ttypesR RRR"R)RR:R<R@RGRIRMRJtobjectRt BaseCylinderRRRRtrhoR RRR^RRRRRRRRR t QApplicationtargvtapptqsrandtQTimetsecsTot currentTimeRRRtexittexec_(((sL/home/fccoelho/Documentos/Projects_Software/epigrass-code/Epigrass/dgraph.pyts^   $(        6    AQX^^ .