ó ‘Mc@s¶ddlZddlZddlmZddlmZddlmZejdƒriejdƒZndZdeZ d„Z d „Z d „Z d „Z d „Ze d „ZdS(iÿÿÿÿN(tGenome(t geneinfoDB(tenvirontCISTEMATIC_ROOTs /proj/genomes%s/D_rerio/drerio.genedbcCsÅg}d}tdd|ƒ}tj|ƒ}x‘|D]‰}td||fdƒ}|jƒ} xQ| dkr²g}d}| jƒd} |jƒ} xU| dkrí| ddkrí| jƒ} |t| ƒ7}|j| ƒ|jƒ} q™Wtj |dƒ} |d krMd | GH|j d| f| d t |ƒƒ|j | | d ƒn\d || f}tdt |fdƒ}|j| ƒ|jƒd|GH|j | |dƒ| } qbW|jƒq4WdS(NitdreriotdbFiles%s/%strtit>iÐsAdded contig %s to databaset chromosometdbs%s%s.bins%s%stws Added contig file %s to databasetfile(Rtostlistdirtopentreadlinetstriptlentappendtstringtjoint addSequencetstrtaddChromosomeEntrytcisRoottwritetclose(R t chromPatht chromOutPathtseqArraytseqLentdrGenometfilestfilenametinFiletheadertchromIDt currentLinetlineSeqtseqt outFileNametoutFile((sK/woldlab/castor/data00/home/georgi/erange-4.0a/cistematic/genomes/drerio.pytloadChromosome,s<       "    c CsŽg}g}tdd|ƒ}t|dƒ}tƒ}x5|D]-}|jdƒ}|d} yJ|jd| ƒ} t| ƒsŒd| GHw=n|j| ƒ} | d} Wn q=nX| dkrÅq=nt|d ƒ} t|d ƒ} |d }|d }|d krd}nd}d| f}||kr5d}nd}|j|ƒ|j||| | |d|fƒq=Wdt|ƒGH|j |ƒdS(NRRRs iscould not find %siRiiiit-tRtFt2t1tgenesAdding %d gene entries( RRRtsplitt getGeneIDRt getGeneInfotintRtaddGeneEntryBatch(R tgFilet geneEntriestseenGIDsR tgeneFiletidbtlinetcolstgidttempIDtgeneInfotstarttstoptsensetchromtgeneIDt gidVersion((sK/woldlab/castor/data00/home/georgi/erange-4.0a/cistematic/genomes/drerio.pytloadGeneEntriesQsD              &c CsTt|dƒ}tƒ}g}idd6dd6dd6}g}xÖ|D]Î}|jdƒ}t|dƒ} |d jd ƒ} |d jd ƒ} |d } ||d } t|dƒd}t|dƒd}|d}yJ|jd|ƒ}t|ƒs d|GHwFn|j|ƒ}|d}Wn qFnX|dkrDqFnd|f}||kred}nd}x¦t| ƒD]˜}t| |ƒd}t| |ƒd}||krã||krã|j||| ||| dfƒqx||kr/| dkrd}nd}|j||| ||| |fƒqx||kr{| dkrPd}nd}|j||| ||| |fƒqx||krù||krù| dkr¨d}nd}|j||| ||| dfƒ|j||| |d|| |fƒqx||krw||krw| dkr&d}nd}|j||| ||d| |fƒ|j||| ||| dfƒqx| dkr’d}d}n d}d}|j||| ||d| |fƒ|j||| ||| dfƒ|j||| |d|d| |fƒqxWqFW|j ƒt dd|ƒ}dt|ƒGH|j |ƒdS(NRR.t+R-R,t.s ii t,i iiiiiiRscould not find %sRR/R0tCDSt5UTRt3UTRRsAdding %d features( RRR2R5R3RR4trangeRRRtaddFeatureEntryBatch(R tgfileR:R;R9t senseArrayt insertArraytgeneLinet geneFieldstexonNumt exonStartst exonStopsRDRCtgstarttgstopR>R?R@RERFtindextestarttestoptfTypetfType1tfType2R ((sK/woldlab/castor/data00/home/georgi/erange-4.0a/cistematic/genomes/drerio.pytloadGeneFeatures~sŽ            %   %   %  ")  &%  &"2 cCs#tdd|ƒ}|j|ƒdS(NRR(Rt createGeneDB(R R ((sK/woldlab/castor/data00/home/georgi/erange-4.0a/cistematic/genomes/drerio.pyt createDBFileÛscCs tdd|ƒ}|jƒdS(NRR(Rt createIndices(R R ((sK/woldlab/castor/data00/home/georgi/erange-4.0a/cistematic/genomes/drerio.pytcreateDBindicesàscCsTdt}d}d|GHt|ƒdGHt|||ƒdGHt|ƒd|GHdS(s1 genes and annotations are from UCSC (dr3). s%s/download/dr3s /D_rerio/sCreating database %ssLoading chromosomessCreating IndicessFinished creating database %sN(RRbR+Rd(R t chromoPatht chromoOutPath((sK/woldlab/castor/data00/home/georgi/erange-4.0a/cistematic/genomes/drerio.pyt buildDrerioDBås    (RR tcistematic.genomesRtcistematic.core.geneinfoRRtgetRtgeneDBR+RGR`RbRdRg(((sK/woldlab/castor/data00/home/georgi/erange-4.0a/cistematic/genomes/drerio.pyts  % - ]