nQc@sdZddlZddlmZmZddlmZmZmZddl m Z m Z dZ dZ defd YZd efd YZd efd YZdefdYZdS(sIModel GFF3 records generically, and genes, mRNAs, and exons specifically.iN(t StringTypestDictionaryType(t SeqRecordtSeqtAlphabet(t SeqFeaturetFeatureLocationiccCs$||krtS||kr tStS(N(tFalsetTrue(tstart0tend0tstart1tend1((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pytintervals_overlap s   t GFF3RecordcBseZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0d/Z1d0Z2d1Z3RS(2sGeneric GFF3 recordc Cse||_||_||_t||_t||_||_||_||_|j | dS(N( t_seqIDt_sourcet_typetintt_startt_endt_scoret_strandt_phasetset_attributes( tselftseqIDtsourcetso_typetstarttendtscoretstrandtphaset attributes((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt__init__s      cCs|jS(N(R(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt get_seqID"scCst||_dS(N(tstrR(Rtscaffold((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt set_seqID%scCs|jS(N(R(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt get_score(scCs |jS(N(R((R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pytgetScore+scCs|jS(N(R(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt get_strand.scCs |jS(N(R*(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt getStrand1scCs ||_dS(N(R(Rtvalue((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt set_score4scCs|j|dS(N(R-(RR,((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pytsetScore7scCs ||_dS(N(R(RR,((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt set_strand:scCs|j|dS(N(R/(RR,((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt setStrand=scCs9gt|jjD]}d|^q}dj|S(Ns%s=%st;(tsortedt _attributestitemstjoin(Rtpairt attr_list((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pytget_attr_string@s,c CsU|j|j|jt|jt|jt|j|jt|j|j g S(N( RRRR%RRRRRR8(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pytget_field_listDscCsdj|jS(Ns (R5R9(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt__str__GscCsd|jS(NsGFF3Record(%s)(R:(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt__repr__JscCstdt|j|jS(Ni(RtabsRR(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt__len__MscCs.tjd}t|j|jjdS(Ns\D*(\d+)i(tretcompileRtmatchRtgroup(Rtpat((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt chrom_numPscCs|jS(N(R(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt get_startTscCs|jS(N(R(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pytget_endWscCst||_dS(N(RR(RR((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyt set_startZscCst||_dS(N(RR(RR((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pytset_end]scCs|jS(N(R3(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pytget_attributes`scCsi|_t|trqx|jdjdD]9}y&|jd\}}||j|s 5  2cCs&tg|jD]}t|^q S(N(tsumRR(RR((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyRQscCs|j|j|jS(N(RRR(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyRTscCsN|j}|j}|dkr9|j||| !S|j||SdS(Ni(RRR(RRtutr5tutr3((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pytget_coding_sequenceWs    c Cs|j|}t|j|j}t|j|j}td||d}||j||d!j}|dkr||j|||!jn tjd}|jdkrt |j |t |j St ||t |SdS(NiitRw( RRRRRR$RRRR%R( RRt pad_lengthtcdstlefttrightt left_starttleft_padt right_pad((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pytget_padded_coding_sequence_s9$cCs|j}|d=t|S(Ni(R9R(RR((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyRls ($RRRR#RuRRRRRRRRRRRRRRR/R:RtRRRRRRRRRRt PAD_LENGTHRRR(((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyRAs@                           tGFF3ExoncBs\eZdZdZdZdZdZedZdZ dZ dZ RS( scThe Parent of an exon is a set of mRNAs, implemented here as a comma-separated list of strings c Cs2tt|j||d|||||| dS(NR(RRR#( RRRRRRR R!R"((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyR#wsc CsLt|j|jt|jt|jt|j|j|j|j S(N( RRRR%RRRRRR8(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyRuzscCs#|jr|jjdSdS(Nt,(RiRLR(R((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyR}s cCsT|j}|r1||kr:|j|q:n |g}|jdj|dS(NR(RRRkR5(RtparentRj((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyRs    cCs|j}|d=t|S(Ni(R9R(RR((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyRs cCs&|j}|r"||kr"tStS(N(RRR(Rt transcriptIDRj((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pytisInTranscripts cCsC|j}|r?x*|jD]}|j|rtSqWntS(N(RRRRR(RRRjR((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyRs   cCsRt|j|jd}t|j|j}||j||!jS(Ni(RRDRERR$R(RRRR((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyRs( RRRR#RuRRRRRRR(((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyRts      (RR>ttypesRRtBioRRRtBio.SeqFeatureRRRR tobjectRRRR(((sT/woldlab/castor/data00/home/georgi/programs/RNA-Seq-Simulator.v1.0/lib/gff3Record.pyts  _4