ó ?îec@svdZddlZddlZddlZd„ZejddƒZd„Zd„Z d„Z d „Z d „Z dS( sw General tools for dealing with ATAC-Seq data using Python. @author: Alicia Schep, Greenleaf Lab, Stanford University iÿÿÿÿNcCs_tj|ƒ}|j|j|j|jƒ}|jdkrKt|ƒ}n|jƒ|j ƒS(s©obtain sequence for an interval chunk: chunk object for which sequenceuence is to be fetched fastafile: filename for fasta file with sequenceuence t-( tpysamt FastaFiletfetchtchromtstarttendtstrandtreverse_complementtclosetupper(tchunkt fastafilethandletsequence((s0/tmp/pip-install-bGcd2k/NucleoATAC/pyatac/seq.pyt get_sequence s  tACGTtTGCAcCs |jtƒS(s#Get complement of DNA sequenceuence(t translatetDNA_Translation(R((s0/tmp/pip-install-bGcd2k/NucleoATAC/pyatac/seq.pyt complementscCst|ddd…ƒS(s+Get reverse complement of DNA sequenceuenceNiÿÿÿÿ(R(R((s0/tmp/pip-install-bGcd2k/NucleoATAC/pyatac/seq.pyR sc sæt|dƒ‰t‡fd†tt|ƒDƒƒsDtdƒ‚ntjt|ƒt|ƒˆdfƒ}xrtt|ƒƒD]^}tjttgtt|ƒˆdƒD]!}|||ˆ!||k^q­ƒƒ||(ssiUsage Error! Nucleotides must all be of same length! No mixing single nucleotides with dinucleotides, etci( tlentalltmapt Exceptiontnptzerostrangetarraytint(Rt nucleotidestmattitj((Rs0/tmp/pip-install-bGcd2k/NucleoATAC/pyatac/seq.pyt seq_to_mat%s%)\cCs«tjt|ƒƒ}d}t|dƒ}xl|D]d}|ddkr1|jdƒjƒ}|g|D]}|j|ƒ^qf7}|t|ƒ7}q1q1W|jƒ||S(s%Get genomewide nucleotide frequenciesgtrit>s (RRRtopentrstripR tcountR (tfastaR!touttntftlineRR#((s0/tmp/pip-install-bGcd2k/NucleoATAC/pyatac/seq.pyt getNucFreqs/s & c Cs­tjt|ƒƒ}d}tj|ƒ}xn|D]f}|j|j|j|jƒ}|j ƒ}|g|D]}|j |ƒ^qk7}|t|ƒ7}q1W|j ƒ||S(s2Get nucleotide frequences within regions of genomeg( RRRRRRRRRR R*R ( tchunksR+R!R,R-R R RR#((s0/tmp/pip-install-bGcd2k/NucleoATAC/pyatac/seq.pytgetNucFreqsFromChunkList=s  & ( t__doc__tstringtnumpyRRRt maketransRRRR%R0R2(((s0/tmp/pip-install-bGcd2k/NucleoATAC/pyatac/seq.pyts