B ž3Rc$ã@s<dZddlmZGdd„deƒZeƒZdd„Zd d d „ZdS) zH functionality for drawing hierarchical catalogs on sping canvases é)Úpidc@s”eZdZdZdZdZe ddd¡Ze ddd¡Z e ddd¡Z e ddd¡Z ej Z e ddd¡Zd Zd Zd Zd Zejd ddZe ddd¡ZdZdS)ÚVisOptsé éég333333ã?gÍÌÌÌÌÌì?gš™™™™™é?gš™™™™™É?réééKéZ helvetica)ZfaceÚsizegð?gš™™™™™Ù?éN)Ú__name__Ú __module__Ú __qualname__ZcircRadÚ minCircRadÚ maxCircRadÚpiddleZColorÚ circColorZterminalEmptyColorZterminalOnColorZterminalOffColorZ transparentÚ outlineColorÚ lineColorÚ lineWidthÚ horizOffsetÚ vertOffsetÚ topMarginZFontÚ labelFontZhighlightColorZhighlightWidth©rrú=lib/python3.7/site-packages/rdkit/DataStructs/HierarchyVis.pyrs rcCs`d}x&| ¡D]}t||ƒ}t||ƒ}qWt| ¡ƒ}|tjdtjtj|tjf}|S)Néÿÿÿÿr )ÚkeysÚlenÚmaxÚvisOptsrrrr)ÚadjListÚ levelListZ maxAcrossÚkÚnHereZnLevsZminSizerrrÚGetMinCanvasSize(s  r&Nrç„×—Ac Csœ|dkr g}|dkri}| ¡}| ¡t||dƒ}t||dƒ}|j}i} x€t||ddƒD]j} | |} dtj| tjg} |  | g¡} t | ƒ}tj |_ |rbt |dƒ|}|tj8}t|dtjƒ}t|tjƒ}|dtj}|dd| d<|dr| d|d8<| d|dd|8<xš| D]’}|rL||kr8| |kr¨xP||D]D}|| kr`| |d}| | d| d|d|dtjtj¡q`Wt| ƒ|f| |<| d|7<q8WqbWxÄ|  ¡D]¸}| |\} }| d|| d|}}| d|| d|}}|  |tj¡}| ||||tjd|¡t|ƒ}| d| ¡d| d| |¡df}|j||d|dd d qÜW| S) z• Arguments: - adjList: adjacency list representation of the hierarchy to be drawn - levelList: dictionary mapping level -> list of ids Nrrrr g@gà?réZ)Zangle)rÚsortr Úminr Úranger!rrÚgetrrZ defaultFontÚfloatrrrZdrawLinerrÚtuplerZ drawEllipserÚstrZ fontHeightZ stringWidthZ drawString)r"r#ZcanvasZ entryColorsZbitIdsZminLevelZmaxLevelZ levelLengthsZdimsZdrawLocsZlevelLenZ nLevelsDownÚposZidsr%Z spacePerNodeZnodeRadZIDZneighborZp2Zx1Zy1Zx2Zy2Z drawColorZlabelZtxtLocrrrÚ DrawHierarchy3sZ         . * r1)NNrr') Ú__doc__ZspingrrÚobjectrr!r&r1rrrrÚs