3 :>"^5@sdZddlZddlmZddlmZddlmZy@dDejddkoNdEknrbdd l m Z n dd l m Z Wn e k rdd l m Z YnXd d iZ y e jZWn:ek ree d sdd l m Zeje _ddZYnXxe jD]\ZZeeeqWejddkrdndZGdddeZddZddZedfddZddZddZe d fdd Zd!d"Zefd#d$Z d%d&Z!efd'd(Z"dFd)d*Z#d+d,Z$d-d.Z%d/d0Z&d1d2Z'd3d4Z(d5d6Z)Gd7d8d8e*Z+d9d:Z,d;d<Z-dGd>d?Z.d@dAZ/GdBdCdCe*Z0dS)HaPhyloXML reader/parser, writer, and associated functions. Instantiates tree elements from a parsed PhyloXML file, and constructs an XML file from a ``Bio.Phylo.PhyloXML`` object. About capitalization: - phyloXML means the file format specification - PhyloXML means the Biopython module ``Bio.Phylo.PhyloXML`` and its classes - Phyloxml means the top-level class used by ``PhyloXMLIO.read`` (but not ``Bio.Phylo.read``!), containing a list of Phylogenies (objects derived from ``BaseTree.Tree``) N) basestring)unicode)PhyloXML) ElementTree) cElementTreephyzhttp://www.phyloxml.org_namespace_mapcCs|tj|<dS)z"Set the namespace for ElementTree.N)rr )prefixurird/home/psgendb/BIRCHDEV/lib-linux-x86_64/python/lib64/python3.5/site-packages/Bio/Phylo/PhyloXMLIO.pyregister_namespace4srrzutf-8c@seZdZdZdS) PhyloXMLErrorzException raised when PhyloXML object construction cannot continue. XML syntax errors will be found and raised by the underlying ElementTree module; this exception is for valid XML that breaks the phyloXML specification. N)__name__ __module__ __qualname____doc__rrrrr@srcCs t|jS)zParse a phyloXML file or stream and build a tree of Biopython objects. The children of the root node are phylogenies and possibly other arbitrary (non-phyloXML) objects. :returns: a single ``Bio.Phylo.PhyloXML.Phyloxml`` object. )Parserread)filerrrrOs rcCs t|jS)aIterate over the phylogenetic trees in a phyloXML file. This ignores any additional data stored at the top level, but may be more memory-efficient than the ``read`` function. :returns: a generator of ``Bio.Phylo.PhyloXML.Phylogeny`` objects. )rparse)rrrrr[s rTcs~ddt|tjrnTt|tjjtjjfr:|j}n0t|drbtjifdd|Dd}ntdt |j |||dS) aRWrite a phyloXML file. :Parameters: obj an instance of ``Phyloxml``, ``Phylogeny`` or ``BaseTree.Tree``, or an iterable of either of the latter two. The object will be converted to a Phyloxml object before serialization. file either an open handle or a file name. cSsnt|tjr|St|tjr$|jSt|tjjr>tjj|St|tjjrbtjjtjj|dStddS)N)rootz)iterable must contain Tree or Clade types) isinstancePX PhylogenyCladeZ to_phylogenyBaseTreeTree from_tree ValueError)treerrr fix_singlets   zwrite..fix_single__iter__c3s|]}|VqdS)Nr).0t)r$rr szwrite..) phylogenieszXFirst argument must be a Phyloxml, Phylogeny, Tree, or iterable of Trees or Phylogenies.)encodingindent) rrPhyloxmlrr rZ to_phyloxmlhasattrr"Writerwrite)objrr*r+r)r$rr/gs   r/cCs&|ddkr"||jdddS|S)z;Extract the local tag from a namespaced tag name (PRIVATE).r{}rN)index)tagrrr_locals r5c Cs2y|ddjddStk r,d|fSXdS)z;Split a tag into namespace and local tag strings (PRIVATE).rNr2)splitr")r4rrr_split_namespacesr8cCs d||fS)z5Format an XML tag with the given namespace (PRIVATE).z{%s}%sr)r4 namespacerrr_nssr:cCs"|jt|}|dk r||SdS)z|Find a child node by tag, and pass it through a constructor (PRIVATE). Returns None if no matching child is found. N)findr:)parentr4 constructchildrrr _get_child_assr?cCs*|jt|}|dk r&|jr&||jSdS)z~Find a child node by tag; pass its text through a constructor (PRIVATE). Returns None if no matching child is found. N)r;r:text)r<r4r=r>rrr_get_child_textsrAcsfdd|jt|DS)zFind child nodes by tag; pass each through a constructor (PRIVATE). Returns an empty list if no matching child is found. csg|] }|qSrr)r&r>)r=rr sz$_get_children_as..)findallr:)r<r4r=r)r=r_get_children_assrDcsfdd|jt|DS)zFind child nodes by tag; pass each node's text through a constructor (PRIVATE). Returns an empty list if no matching child is found. csg|]}|jr|jqSr)r@)r&r>)r=rrrBsz&_get_children_text..)rCr:)r<r4r=r)r=r_get_children_textsrEcCsd|d}t|r|j s(|jj r2|d|_x8|D]0}t||d|j s^|jj r8|d|_q8W|j s|jj r||_n|r|j s|jj r||_dS)zAdd line breaks and indentation to ElementTree in-place (PRIVATE). Sources: - http://effbot.org/zone/element-lib.htm#prettyprint - http://infix.se/2007/02/06/gentlemen-indent-your-xml  z rN)lenr@strip_indenttail)elemlevelierrrrIs   rIcCs8|dks|dkrdS|dks$|dkr(dStd|dS) z$Convert string to boolean (PRIVATE).true1TZfalse0Fz*String could not be converted to boolean: N)r")r@rrr _str2bools rRcCs2|j}x$|D]}||krt||||<qW|S)zQReturn a new dictionary where string values are replaced with booleans (PRIVATE).)copyrR)dctkeysoutkeyrrr_dict_str2bools  rXc Cs*|dk r&yt|Stk r$dSXdS)z$Return text as an integer (PRIVATE).N)int Exception)r@rrr_ints r[c Cs*|dk r&yt|Stk r$dSXdS)z!Return text as a float (PRIVATE).N)floatrZ)r@rrr_floats r]cCs|dk rdj|jSdS)aReplace all spans of whitespace with a single space character (PRIVATE). Also remove leading and trailing whitespace. See "Collapse Whitespace Policy" in the phyloXML spec glossary: http://phyloxml.org/documentation/version_100/phyloxml.xsd.html#Glossary N )joinr7)r@rrr_collapse_wspace sr`cCs&x dD]}||kr|j|d}qW|S)zReplace tab, LF and CR characters with spaces, but don't collapse (PRIVATE). See "Replace Whitespace Policy" in the phyloXML spec glossary: http://phyloxml.org/documentation/version_100/phyloxml.xsd.html#Glossary  rF r^)rarFrb)replace)r@charrrr_replace_wspaces rec@s"eZdZdZddZddZddZdd Zd d d d gZdddddZ e ej e j j ddddgZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9d:Zd;d<Z d=d>Z!d?d@Z"dAdBZ#dCdDZ$dES)FraMethods for parsing all phyloXML nodes from an XML stream. To minimize memory use, the tree of ElementTree parsing events is cleared after completing each phylogeny, clade, and top-level 'other' element. Elements below the clade level are kept in memory until parsing of the current clade is finished -- this shouldn't be a problem because clade is the only recursive element, and non-clade nodes below this level are of bounded size. cCs.ttj|dd}t|\}}||_||_dS)zInitialize the class.startend)eventsN)rfrg)iterrZ iterparsenextrcontext)selfrrkeventrrrr__init__-s zParser.__init__c Cstjdd|jjD}d}x|jD]\}}t|j\}}|dkrx|tdkrZ|d7}q&|dkrx|j|}|j j ||dkr&|tdkr&|d8}|dkr&|j |||}|j j ||jj q&W|S) z7szParser.read..rrfr r phylogenyrg) rr,ritemsrkr8r4 NAMESPACES_parse_phylogenyr)appendotherclear) rlphyloxmlZ other_depthrmrKr9localtagrqotrrrrr5s$    z Parser.readccs>td}x0|jD]&\}}|dkr|j|kr|j|VqWdS)z@Parse the phyloXML file incrementally and return each phylogeny.rqrfN)r:rkr4rt)rlZphytagrmrKrrrrLsz Parser.parsecCs:tjft|jddg}ddddd}x |jD]\}}t|j\}}|dkrx|d krx|jd k rjtd |j ||_q0|d kr0|d kr|j P||krt |||j t |||q0|dkrt ||t |||q0|dkrt ||t|jq0|tdkr&|jj |j||||j q0td|q0W|S)aParse a single phylogeny within the phyloXML tree (PRIVATE). Recursively builds a phylogenetic tree with help from parse_clade, then clears the XML event history for the phylogeny element and returns control to the top-level parsing function. rooted rerootable confidences propertiesclade_relationssequence_relations) confidencepropertyclade_relationsequence_relationrfcladeNz)Phylogeny object should only have 1 cladergrqdateidname descriptionr zMisidentified tag: )rr)rr)rrrXattribrkr8r4rr" _parse_claderwgetattrrusetattrr`r@rsrvr)rlr<rqZ list_typesrmrKr9r4rrrrtUs8     zParser._parse_phylogenycolorrhbinary_charactersrr} distributions referencesr~)r distribution referencer branch_lengthrnode_idwidthcCstjf|j}|jdk r$t|j|_g}x|jD]\}}t|j\}}|dkr|dkrn|jj |j |q2|dkr|j j |j |q2|dkr|j j |j|q2||jkr|j ||dkr2|dkr|jP||dkrq2|j||jkrt||j|j t|||q2||jkr)rlrrrBsz Parser.other..)valuechildren)rZOtherrr@rH)rlrKr9ryr)rlrrvs z Parser.othercCstj|jj|jdS)zCreate accession object.source)rZ Accessionr@rHr)rlrKrrrrszParser.accessionc CsDtjftt|dt|d|jt|d|jt|d|jd|j S)zCreate annotation object.descrrr )rrr~r ) rZ Annotationr`rAr?rrDrr r)rlrKrrrrs    zParser.annotationc Csrdd}tj|jdt|jdt|jdt|jdt|jdt|d|t|d |t|d |t|d |d S) z Create binary characters object.cSs t|dS)z$Get binary characters from subnodes.bc)rE)rKrrr bc_gettersz+Parser.binary_characters..bc_gettertype gained_count lost_count present_count absent_countgainedlostpresentabsent) rrrrrrrrr)rZBinaryCharactersrr[r?)rlrKrrrrrs       zParser.binary_charactersc Cs6tj|jd|jd|jd|jdt|d|jdS)z!Create clade relationship object.rid_ref_0id_ref_1distancer)rr)rZ CladeRelationrr?r)rlrKrrrr&s zParser.clade_relationcs&fdddD\}}}tj|||S)zCreate branch color object.c3s|]}t|tVqdS)N)rArY)r&r)rKrrr(3szParser.color..redgreenblue)rrr)r BranchColor)rlrKrrrr)rKrr0sz Parser.colorcCstjt|j|jdS)zCreate confidence object.r)rZ Confidencer]r@r)rlrKrrrr7szParser.confidencec Cs<tj|jdtt|dt|dtt|dtt|dtdS)zCreate date object.unitrrminimummaximum)rrrrr)rDaterr`rAr\)rlrKrrrr;s    z Parser.datecCs.tjtt|dt|d|jt|d|jdS)z&Create geographic distribution object.rpointpolygon)rpointspolygons)rZ Distributionr`rArDrr)rlrKrrrrEs  zParser.distributioncCsBtj|jjt|jddt|jdt|jd|jddS)zCreate protein domain object.fromrtorr)rr)rZ ProteinDomainr@rHrYrr])rlrKrrrdomainMs   z Parser.domaincCs"tjt|jdt|d|jdS)z"Create domain architecture object.lengthr)rdomains)rZDomainArchitecturerYrrDr)rlrKrrrrWs zParser.domain_architecturec Cs<tjt|dt|dtt|dtt|dtt|d|jdS)zCreate events object.r duplications speciationslossesr)rrrrr)rZEventsrArYr?r)rlrKrrrrh^s    z Parser.eventscCs&|jdp|jd}tj|jj|S)zCreate identifier object.rr)rrrr@rH)rlrKrrrrrhsz Parser.idcCs.|jd}|dk rt|}tj|jj|dS)z!Create molecular sequence object. is_alignedN)r)rrRrZMolSeqr@rH)rlrKrrrrrms zParser.mol_seqcCs8tj|jdt|dtt|dtt|dt|jddS)z,Create point object, coordinates of a point.geodetic_datumlatlongaltalt_unit)rr)rZPointrrAr\)rlrKrrrrts    z Parser.pointcCstjt|d|jdS)z&Create polygon object, list of points.r)r)rZPolygonrDr)rlrKrrrr~szParser.polygoncCs:tj|jj|jd|jd|jd|jd|jddS)z*Create properties from external resources.ref applies_todatatyperid_ref)rr)rZPropertyr@rHr)rlrKrrrrszParser.propertycCstj|jdt|ddS)z#Create literature reference object.doir)rr)rZ ReferencerrA)rlrKrrrrszParser.referencec Cs:tj|jd|jd|jdt|jdt|d|jdS)zHCreate sequence relationship object, relationship between two sequences.rrrrr)rr)rZSequenceRelationrr]r?r)rlrKrrrrs  zParser.sequence_relationcCs&tj|jjt|jd|jddS)z(Create uri object, expected to be a url.rr)rr)rZUrir@rHr`r)rlrKrrrr s z Parser.uriN)%rrrrrnrrrtrrsetunionrUrrrrrvrrrrrrrrrrrhrrrrrrrr rrrrr"sH  .  9        rcCs4t|trt|jSt|tr,t|jSt|S)zMConvert a Python primitive to a phyloXML-compatible Unicode string (PRIVATE).)rr\rupperboollower)rrrr _serializes     rcCs4i}x*|D]"}t||}|dk r t|||<q W|S)zNCreate a dictionary from an object's specified, non-None attributes (PRIVATE).N)rr)r0attrsrVrWrorrr _clean_attribs   rFcs fdd}d|_|S)z2Handle to serialize nodes with subnodes (PRIVATE).cstjt|}xpD]h}t|trPt||dk r|jt||t||q|\}}x&t||D]}|jt|||qdWqWrt|j|_ |S)z$Wrap nodes and subnodes as elements.N) rElementrrrrrurrr@)rlr0rKsubnmethodZpluralitem)attribshas_textsubnodesr4rrwrappeds   z _handle_complex..wrappedz*Serialize a %s and its subnodes, in order.)r)r4rrrrr)rrrr4r_handle_complexs rcsfdd}d|_|S)z+Handle to serialize simple nodes (PRIVATE).cstj}t||_|S)zWrap node as element.)rrrr@)rlr0rK)r4rrrs  z_handle_simple..wrappedzSerialize a simple %s node.)r)r4rr)r4r_handle_simples  rc@s:eZdZdZddZedfddZddZd d Ze d dhdnZ e ddodxZ e d.dyfdd0Z e d1dzd|Z d6d7Ze dd}d~Ze d!fdZe ddfdd0Ze dddZe d)fdZdFdGZe dHddZe d'fdZe ddfdd0Ze dPdfdd0Ze d"dfdd0Ze dBddZe dDfdZe ddfdd0Ze d+ddZe d%ddZe dddZe d#ddZ e d5dfdd0Z!e"dVZ#e"dZ$e"dTZ%e"dUZ&e"dAZ'e"d@Z(e"d?Z)e"d Z*e"d=Z+e"dLZ,e"d<Z-e"dNZ.e"d;Z/e"dMZ0e"dfZ1e"d^Z2e"daZ3e"d4Z4e"dZ5e"d\Z6e"dZ7e"deZ8e"d_Z9e"d[Z:e"dcZ;e"dZrrrrvs  z Writer.otherrqr{r|branch_length_unitrrrrrrr}rrrrrrr~rv id_sourcerrrrrrrrrhrrrrrrrr)rrrevidencerr c Cs\tjdt|d }xDd D]<}tj|}x"t||D]}|j|j|q2W|j|qW|S) z4Serialize a binary_characters node and its subnodes.rrrrrrrrrr)rrrrr)rrrr)rrrrrur)rlr0rKrZsubelemtokenrrrr3s  zWriter.binary_charactersrrrrrrrrrrrrrrcCsftjdt|jdt|jd}|jdk r>|jdt|j|jdk rV|jd|jt|j |_ |S)zSerialize a domain node.rr)rrNrr) rrstrrfrgrrrrrr@)rlr0rKrrrrQs   z Writer.domainrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrN)r{r|rrrr}rrrrrr~rvrv) rrrrrrrrrr)rrr}rrrrrrrrrr~rrrvrv)rrrrrrrrrhrr rr r r r )r)rrrrrr~)rrrr )rrrr)r)rrr)r)r)rrrrrrrr)rrr)rrr)r)rrrrr)r)r)r)rr)rrrrr)r)rrrrr)r)r)rrrrrrvrv) rrrrrr rrr)rrrr)r)rrrrrrvrv) rrrrrrrr r)rr)=rrrrrnDEFAULT_ENCODINGr/rxrvrrqrrrrrrrrrrrrhrrrrrrrrrrr rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr.s            r.)rr)rr)r)F)1rsys Bio._py3krr Bio.Phylorr version_infoZ xml.etreerr ImportErrorrsrAttributeErrorr-ZET_pyr rrr r rrZrrrr/r5r8r:r?rArDrErIrRrXr[r]r`reobjectrrrrrr.rrrrs`   "     +