wtdbg.h 65 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291
  1. /*
  2. *
  3. * Copyright (c) 2011, Jue Ruan <ruanjue@gmail.com>
  4. *
  5. *
  6. * This program is free software: you can redistribute it and/or modify
  7. * it under the terms of the GNU General Public License as published by
  8. * the Free Software Foundation, either version 3 of the License, or
  9. * (at your option) any later version.
  10. *
  11. * This program is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. * GNU General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU General Public License
  17. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  18. */
  19. #ifndef __WTDBG_H_RJ
  20. #define __WTDBG_H_RJ
  21. #include "kbm.h"
  22. #include "kbmpoa.h"
  23. #include "filewriter.h"
  24. #include "pgzf.h"
  25. #include <getopt.h>
  26. #include <regex.h>
  27. #define WT_MAX_RD 0x3FFFFFFF // 1 G
  28. #define WT_MAX_RDLEN 0x00FFFFFF // 16 Mb
  29. #define WT_MAX_RDBIN 0x0000FFFF // 64 K bins
  30. #define WT_MAX_NODE 0x000000FFFFFFFFFFLLU
  31. #define WT_MAX_EDGE 0x000000FFFFFFFFFFLLU
  32. #define WT_MAX_NODE_EDGES 0xFFFF
  33. #define WT_MAX_EDGE_COV 0xFFF
  34. typedef struct {
  35. u8i rid:30, dir:1, beg:16, end:16, closed:1;
  36. } rd_frg_t;
  37. define_list(rdfrgv, rd_frg_t);
  38. typedef struct {
  39. u8i node;
  40. u8i rid:30, dir:1, beg:16, end:16, closed:1;
  41. } rd_reg_t;
  42. define_list(rdregv, rd_reg_t);
  43. typedef struct {
  44. u8i rid:30, dir:1, beg:16, end:16, closed:1;
  45. u8i read_link;
  46. } rd_rep_t;
  47. define_list(rdrepv, rd_rep_t);
  48. typedef struct {
  49. u8i node;
  50. u8i rid:30, dir:1, beg:16, end:16, closed:1;
  51. u8i read_link;
  52. } reg_t;
  53. define_list(regv, reg_t);
  54. typedef struct { uint32_t x, y; } xy_t;
  55. define_list(xyv, xy_t);
  56. typedef struct { uint64_t idx:46, cnt:18; } vec_ref_t;
  57. typedef struct { uint64_t idx:46, cnt:18; } ptr_ref_t;
  58. typedef struct { uint64_t idx:46, cnt:18, fix:1, rank:45, score:18; } rnk_ref_t;
  59. static const vec_ref_t VEC_REF_NULL = (vec_ref_t){0, 0};
  60. static const ptr_ref_t PTR_REF_NULL = (ptr_ref_t){0, 0};
  61. static const rnk_ref_t RNK_REF_NULL = (rnk_ref_t){0, 0, 0, 0, 0};
  62. define_list(vecrefv, vec_ref_t);
  63. define_list(ptrrefv, ptr_ref_t);
  64. define_list(rnkrefv, rnk_ref_t);
  65. #define ptrref_hashcode(E) u64hashcode((E).idx)
  66. #define ptrref_hashequals(E1, E2) ((E1).idx == (E2).idx)
  67. define_hashset(ptrrefhash, ptr_ref_t, ptrref_hashcode, ptrref_hashequals);
  68. #define WT_EDGE_CLOSED_NULL 0
  69. #define WT_EDGE_CLOSED_MASK 1
  70. #define WT_EDGE_CLOSED_LESS 2
  71. #define WT_EDGE_CLOSED_HARD 3
  72. typedef struct {
  73. uint64_t node1:45, dir1:1, dir2:1, status:1, closed:2, flag:2, cov:12;
  74. uint64_t node2:45; int64_t off:19;
  75. } edge_t;
  76. define_list(edgev, edge_t);
  77. static inline uint64_t _edge_hashcode(edge_t e){
  78. const uint64_t m = 0xc6a4a7935bd1e995LLU;
  79. const int r = 47;
  80. uint64_t h = 1023 ^ (16 * m);
  81. uint64_t k = (e.node1 << 1) | e.dir1;
  82. k *= m;
  83. k ^= k >> r;
  84. k *= m;
  85. h ^= k;
  86. h *= m;
  87. k = (e.node2 << 1) | e.dir2;
  88. k *= m;
  89. k ^= k >> r;
  90. k *= m;
  91. h ^= k;
  92. h *= m;
  93. h ^= h >> r;
  94. h *= m;
  95. h ^= h >> r;
  96. return h;
  97. }
  98. #define EDGEHASH(idx) ((edgev *)set->userdata)->buffer[idx]
  99. #define edge_hashcode(E) _edge_hashcode(EDGEHASH(E))
  100. #define edge_hashequals(E1, E2) (EDGEHASH(E1).node1 == EDGEHASH(E2).node1 && EDGEHASH(E1).node2 == EDGEHASH(E2).node2 \
  101. && EDGEHASH(E1).dir1 == EDGEHASH(E2).dir1 && EDGEHASH(E1).dir2 == EDGEHASH(E2).dir2)
  102. define_hashset(edgehash, uint64_t, edge_hashcode, edge_hashequals);
  103. typedef struct { uint64_t idx:63, flg:1; uint64_t next; } edge_ref_t;
  104. static const edge_ref_t EDGE_REF_NULL = (edge_ref_t){0x7FFFFFFFFFFFFFFLLU, 1, 0};
  105. define_list(edgerefv, edge_ref_t);
  106. #define MAX_REP_IDX MAX_U4
  107. typedef struct {
  108. u8i rep_idx:32;
  109. u4i unvisit:16, cov:16;
  110. u8i closed:1, single_in:1, bt_visit:45, rep_dir:1, bt_idx:16, init_end:1;
  111. vec_ref_t regs;
  112. ptr_ref_t edges[2];
  113. } node_t;
  114. define_list(nodev, node_t);
  115. typedef struct {
  116. u8i idx:39, flg:1, cnt:24;
  117. } hit_lnk_t;
  118. define_list(hitlnkv, hit_lnk_t);
  119. typedef struct {
  120. rd_frg_t frgs[2];
  121. hit_lnk_t lnks[2]; // idx+flg: link, cnt[0]: mat, cnt[1]:cglen
  122. } rd_hit_t;
  123. define_list(rdhitv, rd_hit_t);
  124. typedef struct {
  125. u8i visit:63, flag:1;
  126. hit_lnk_t hits; // point to the g->rdhits
  127. int clps[2];
  128. ptr_ref_t regs;
  129. //u2i corr_bincnt;
  130. } read_t;
  131. define_list(readv, read_t);
  132. typedef struct {
  133. u8i node;
  134. //node_t *n;
  135. edge_ref_t edges[2];
  136. u4i dir:2, cov:30;
  137. int off;
  138. } trace_t;
  139. define_list(tracev, trace_t);
  140. #define WT_TRACE_MSG_ZERO 0
  141. #define WT_TRACE_MSG_ONE 1
  142. #define WT_TRACE_MSG_MORE 2
  143. #define WT_TRACE_MSG_VISITED 3
  144. #define WT_TRACE_MSG_UNDEF 4
  145. typedef struct {
  146. union {
  147. struct {
  148. u4i frg1:31, dir1:1;
  149. u4i frg2:31, dir2:1;
  150. };
  151. u8i key;
  152. };
  153. u4i cov:13, flag:2, tidx1:8, tidx2:8, weak:1, closed:2;
  154. b4i off;
  155. } lnk_t;
  156. define_list(lnkv, lnk_t);
  157. #define lnk_hashcode(E) (E).key
  158. #define lnk_hashequals(E1, E2) (E1).key == (E2).key
  159. define_hashset(lnkhash, lnk_t, lnk_hashcode, lnk_hashequals);
  160. typedef struct {
  161. u8i toff:46, tcnt:18; // extended traces and core traces
  162. u4i tx, ty; // core traces
  163. ptr_ref_t lnks[2];
  164. u4i len, length;
  165. u8i rep_idx:48, unvisit:16;
  166. u8i closed:1, single_in:1, bt_visit:46, rep_dir:1, bt_idx:16;
  167. } frg_t;
  168. define_list(frgv, frg_t);
  169. typedef struct {
  170. edge_ref_t lnks[2];
  171. u4i frg;
  172. u4i dir:2, cov:30;
  173. int off;
  174. u4i tx, ty;
  175. } path_t;
  176. define_list(pathv, path_t);
  177. #define WT_PATH_MSG_ZERO 0
  178. #define WT_PATH_MSG_ONE 1
  179. #define WT_PATH_MSG_MORE 2
  180. #define WT_PATH_MSG_VISITED 3
  181. #define WT_PATH_MSG_UNDEF 4
  182. typedef struct {
  183. u8i node1:63, dir1:1;
  184. u8i node2:63, dir2:1;
  185. b8i off:42, len:22;
  186. } seqlet_t;
  187. define_list(seqletv, seqlet_t);
  188. typedef struct {
  189. KBM *kbm;
  190. KBMPar *par, *rpar;
  191. regv *regs;
  192. readv *reads;
  193. rdhitv *rdhits;
  194. BitsVec *cigars;
  195. nodev *nodes;
  196. edgev *edges;
  197. edgehash *ehash;
  198. edgerefv *erefs;
  199. frgv *frgs;
  200. lnkv *lnks;
  201. edgerefv *lrefs;
  202. tracev *traces;
  203. u8i genome_size;
  204. u4i num_index;
  205. //u4i corr_mode, corr_min, corr_max;
  206. //u4i corr_bsize, corr_bstep;
  207. //float corr_cov, corr_gcov;
  208. int node_order, mem_stingy;
  209. u4i n_fix, only_fix; // first n sequences are accurate contigs; only_fix means whether to include other pacbio sequenes
  210. u4i reglen, regovl, bestn;
  211. int min_node_mats;
  212. int max_overhang, chainning_hits, uniq_hit;
  213. float node_max_conflict; // 0.25
  214. float node_merge_cutoff;
  215. uint32_t max_node_cov, min_node_cov, exp_node_cov, min_edge_cov, max_edge_span;
  216. u4i max_node_cov_sg, max_sg_end;
  217. int cut_tip;
  218. int store_low_cov_edge;
  219. int rep_filter, rep_detach;
  220. uint32_t bub_step, tip_step, rep_step;
  221. int min_ctg_len, min_ctg_nds, minimal_output;
  222. vplist *utgs;
  223. u4i major_nctg;
  224. vplist *ctgs;
  225. } Graph;
  226. static const char *colors[2][2] = {{"blue", "green"}, {"red", "gray"}};
  227. static inline Graph* init_graph(KBM *kbm){
  228. Graph *g;
  229. u4i rid;
  230. g = malloc(sizeof(Graph));
  231. g->kbm = kbm;
  232. g->par = kbm->par;
  233. g->rpar = NULL;
  234. g->regs = init_regv(32);
  235. g->reads = init_readv(kbm->reads->size);
  236. g->reads->size = kbm->reads->size;
  237. for(rid=0;rid<g->reads->size;rid++){
  238. g->reads->buffer[rid].clps[0] = 0;
  239. g->reads->buffer[rid].clps[1] = g->kbm->reads->buffer[rid].bincnt;
  240. }
  241. g->nodes = init_nodev(32);
  242. g->rdhits = init_rdhitv(1024);
  243. g->cigars = init_bitsvec(1024, 3);
  244. g->edges = init_edgev(32);
  245. g->ehash = init_edgehash(1023);
  246. set_userdata_edgehash(g->ehash, g->edges);
  247. g->erefs = init_edgerefv(32);
  248. g->frgs = init_frgv(32);
  249. g->lnks = init_lnkv(32);
  250. g->lrefs = init_edgerefv(32);
  251. g->traces = init_tracev(32);
  252. g->genome_size = 1024 * 1024 * 1024LLU;
  253. g->num_index = 1;
  254. //g->corr_mode = 0;
  255. //g->corr_min = 5;
  256. //g->corr_max = 10;
  257. //g->corr_cov = 0.75;
  258. //g->corr_gcov = 5.0;
  259. //g->corr_bsize = 2048;
  260. //g->corr_bstep = 2048 - 512;
  261. g->node_order = 0;
  262. g->n_fix = 0;
  263. g->only_fix = 0;
  264. g->reglen = 1024;
  265. g->regovl = 256;
  266. g->node_max_conflict = 0.25;
  267. g->node_merge_cutoff = 0.8;
  268. g->max_overhang = -1;
  269. g->bestn = 0;
  270. g->min_node_mats = 1;
  271. g->chainning_hits = 0;
  272. g->uniq_hit = 0;
  273. g->min_node_cov = 4;
  274. g->max_node_cov = 60;
  275. g->exp_node_cov = 40; // expected node cov
  276. g->min_edge_cov = 4;
  277. g->max_edge_span = 1024; // 1024 * 256 = 256 kb
  278. g->max_node_cov_sg = 2;
  279. g->max_sg_end = 5;
  280. g->store_low_cov_edge = 1;
  281. g->rep_filter = 1;
  282. g->rep_detach = 1;
  283. g->bub_step = 10;
  284. g->tip_step = 5;
  285. g->rep_step = 20;
  286. g->cut_tip = 1;
  287. g->min_ctg_len = 10000;
  288. g->min_ctg_nds = 5;
  289. g->minimal_output = 0;
  290. g->utgs = init_vplist(32);
  291. g->ctgs = init_vplist(32);
  292. g->major_nctg = 0;
  293. return g;
  294. }
  295. static inline void free_graph(Graph *g){
  296. u8i i;
  297. free_regv(g->regs);
  298. free_readv(g->reads);
  299. free_nodev(g->nodes);
  300. free_rdhitv(g->rdhits);
  301. free_bitsvec(g->cigars);
  302. free_edgev(g->edges);
  303. free_edgehash(g->ehash);
  304. free_edgerefv(g->erefs);
  305. free_frgv(g->frgs);
  306. free_lnkv(g->lnks);
  307. free_edgerefv(g->lrefs);
  308. free_tracev(g->traces);
  309. for(i=0;i<g->utgs->size;i++) free_tracev(g->utgs->buffer[i]);
  310. free_vplist(g->utgs);
  311. for(i=0;i<g->ctgs->size;i++) free_tracev(g->ctgs->buffer[i]);
  312. free_vplist(g->ctgs);
  313. free(g);
  314. }
  315. #define KBM_MAP2RDHIT_QUICK
  316. static inline int map2rdhits_graph(Graph *g, kbm_map_t *hit){
  317. u4i k, f, d, p, n, add;
  318. rd_hit_t *rh, *hp, *hn;
  319. read_t *rd;
  320. if(hit->mat == 0) return 0;
  321. rh = next_ref_rdhitv(g->rdhits);
  322. rh->frgs[0] = (rd_frg_t){hit->qidx, hit->qdir, hit->qb, hit->qe, 0};
  323. rh->frgs[1] = (rd_frg_t){hit->tidx, hit->tdir, hit->tb, hit->te, 0};
  324. rh->lnks[0].cnt = hit->mat;
  325. rh->lnks[1].cnt = hit->cglen;
  326. add = 0;
  327. for(k=0;k<2;k++){
  328. rd = ref_readv(g->reads, rh->frgs[k].rid);
  329. #ifdef KBM_MAP2RDHIT_QUICK
  330. { // Just add it
  331. rh->lnks[k].idx = rd->hits.idx;
  332. rh->lnks[k].flg = rd->hits.flg;
  333. rd->hits.idx = offset_rdhitv(g->rdhits, rh);
  334. rd->hits.flg = k;
  335. rd->hits.cnt ++;
  336. add ++;
  337. continue;
  338. }
  339. #endif
  340. hn = ref_rdhitv(g->rdhits, rd->hits.idx);
  341. f = rd->hits.flg;
  342. hp = NULL;
  343. p = 0;
  344. n = 1;
  345. while(1){
  346. if(hn->lnks[0].cnt <= rh->lnks[0].cnt){ // hit->mat
  347. break;
  348. }
  349. p = f;
  350. hp = hn;
  351. if(hn->lnks[f].idx == 0) break;
  352. f = hn->lnks[p].flg;
  353. hn = ref_rdhitv(g->rdhits, hn->lnks[p].idx);
  354. n ++;
  355. }
  356. if(g->bestn && n > g->bestn){
  357. continue;
  358. }
  359. if(hp == NULL){
  360. rh->lnks[k].idx = rd->hits.idx;
  361. rh->lnks[k].flg = rd->hits.flg;
  362. rd->hits.idx = offset_rdhitv(g->rdhits, rh);
  363. rd->hits.flg = k;
  364. } else {
  365. rh->lnks[k].idx = hp->lnks[p].idx;
  366. rh->lnks[k].flg = hp->lnks[p].flg;
  367. hp->lnks[p].idx = offset_rdhitv(g->rdhits, rh);
  368. hp->lnks[p].flg = k;
  369. }
  370. rd->hits.cnt ++;
  371. if(g->bestn && rd->hits.cnt > g->bestn){
  372. hn = rh;
  373. p = k;
  374. while(n < g->bestn){
  375. f = hn->lnks[p].flg;
  376. hn = ref_rdhitv(g->rdhits, hn->lnks[p].idx);
  377. p = f;
  378. n ++;
  379. }
  380. hp = hn;
  381. f = p;
  382. while(hn->lnks[f].idx){
  383. d = hn->lnks[f].flg;
  384. hn = ref_rdhitv(g->rdhits, hn->lnks[f].idx);
  385. hn->frgs[f].closed = 1;
  386. f = d;
  387. }
  388. hp->lnks[p].idx = 0;
  389. hp->lnks[p].flg = 0;
  390. rd->hits.cnt = n;
  391. }
  392. add ++;
  393. }
  394. if(add == 0){
  395. trunc_rdhitv(g->rdhits, 1);
  396. g->cigars->size -= hit->cglen;
  397. }
  398. return add;
  399. }
  400. static inline int is_dovetail_overlap(Graph *g, kbm_map_t *hit){
  401. read_t *q, *t;
  402. int overhangs[2][2];
  403. q = ref_readv(g->reads, hit->qidx);
  404. t = ref_readv(g->reads, hit->tidx);
  405. overhangs[1][0] = hit->tb - t->clps[0];
  406. overhangs[1][1] = (t->clps[1] - t->clps[0]) - hit->te;
  407. if(hit->qdir){
  408. overhangs[0][0] = (q->clps[1] - q->clps[0]) - hit->qe;
  409. overhangs[0][1] = hit->qb;
  410. } else {
  411. overhangs[0][0] = hit->qb;
  412. overhangs[0][1] = (q->clps[1] - q->clps[0]) - hit->qe;
  413. }
  414. if(overhangs[0][0] > g->max_overhang && overhangs[1][0] > g->max_overhang){
  415. return 0;
  416. }
  417. if(overhangs[0][1] > g->max_overhang && overhangs[1][1] > g->max_overhang){
  418. return 0;
  419. }
  420. return 1;
  421. }
  422. // qlen is measured by KBM_BIN_SIZE
  423. static inline int hit2rdregs_graph(Graph *g, rdregv *regs, int qlen, kbm_map_t *hit, BitsVec *cigars, u4v *maps[3]){
  424. KBM *kbm;
  425. u8i ndoff;
  426. u4i bpos[2][2], npos[2][2], clen, ndbeg, qn, j, qbincnt;
  427. int tmp, bt, tlen, x, y, mat, beg, end, min_node_len, max_node_len;
  428. int mask, closed;
  429. kbm = g->kbm;
  430. mask = 0;
  431. if(g->max_overhang >= 0){
  432. if(!is_dovetail_overlap(g, hit)){
  433. mask = 1;
  434. }
  435. }
  436. qn = g->reglen;
  437. min_node_len = (qn - 1);
  438. max_node_len = (qn + 1);
  439. // translate into reverse sequence order
  440. if(qlen == 0){
  441. qbincnt = kbm->reads->buffer[hit->qidx].bincnt;
  442. qlen = qbincnt;
  443. } else if(qlen > kbm->reads->buffer[hit->qidx].bincnt){
  444. fprintf(stderr, " -- something wrong in %s -- %s:%d --\n", __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  445. abort();
  446. } else {
  447. qbincnt = qlen;
  448. }
  449. tlen = kbm->reads->buffer[hit->tidx].bincnt;
  450. if(hit->qdir){
  451. tmp = qlen - hit->qb;
  452. hit->qb = qlen - hit->qe;
  453. hit->qe = tmp;
  454. }
  455. {
  456. clear_u4v(maps[0]);
  457. clear_u4v(maps[1]);
  458. clear_u4v(maps[2]);
  459. clen = hit->cglen;
  460. x = -1; y = -1;
  461. while(clen){
  462. bt = get_bitsvec(cigars, hit->cgoff + clen - 1);
  463. push_u4v(maps[2], !(bt >> 2));
  464. bt = bt & 0x03;
  465. switch(bt){
  466. case 0: x ++; y ++; break;
  467. case 1: x ++; break;
  468. case 2: y ++; break;
  469. }
  470. push_u4v(maps[0], (x < 0)? 0 : x);
  471. push_u4v(maps[1], (y < 0)? 0 : y);
  472. clen --;
  473. }
  474. push_u4v(maps[0], x + 1);
  475. push_u4v(maps[1], y + 1);
  476. push_u4v(maps[2], 0);
  477. #if DEBUG
  478. if(x + 1 + hit->qb != hit->qe || y + 1 + hit->tb != hit->te){
  479. fprintf(stderr, " -- something wrong in %s -- %s:%d --\n", __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  480. print_hit_kbm(g->kbm, g->kbm->reads->buffer[hit->qidx].tag, g->kbm->reads->buffer[hit->qidx].bincnt * KBM_BIN_SIZE, hit, cigars, NULL, stderr);
  481. abort();
  482. }
  483. #endif
  484. }
  485. bpos[0][0] = hit->qb;
  486. bpos[0][1] = hit->qe;
  487. bpos[1][0] = hit->tb;
  488. bpos[1][1] = hit->te;
  489. #if 0
  490. fprintf(stdout, "BPOS\t%d\t%d\t%d\t%d\n", bpos[0][0], bpos[0][1], bpos[1][0], bpos[1][1]);
  491. for(j=0;j<maps[0]->size;j++){
  492. fprintf(stdout, "%d,%d\t", maps[0]->buffer[j], maps[1]->buffer[j]);
  493. if((j % 10) == 9) fprintf(stdout, "\n");
  494. }
  495. fprintf(stdout, "\n");
  496. #endif
  497. {
  498. ndoff = kbm->reads->buffer[hit->qidx].binoff;
  499. if(hit->qdir){
  500. ndbeg = qbincnt - bpos[0][0];
  501. ndbeg = ndbeg % qn;
  502. ndoff = ndoff + ((qbincnt - (ndbeg + bpos[0][0])) / qn) - 1;
  503. } else {
  504. ndbeg = (bpos[0][0] % qn)? qn - (bpos[0][0] % qn) : 0;
  505. ndoff = ndoff + ((ndbeg + bpos[0][0]) / qn);
  506. }
  507. x = 0;
  508. for(j=ndbeg+bpos[0][0];j+qn<=bpos[0][1];j+=qn){
  509. mat = 0;
  510. while(maps[0]->buffer[x] < j - bpos[0][0]){
  511. x ++;
  512. }
  513. npos[0][0] = maps[1]->buffer[x];
  514. while(maps[0]->buffer[x] == j - bpos[0][0]){
  515. if(maps[2]->buffer[x]) mat ++;
  516. x ++;
  517. }
  518. npos[0][1] = maps[1]->buffer[x];
  519. while(maps[0]->buffer[x] < j + qn - 1 - bpos[0][0]){
  520. if(maps[2]->buffer[x]) mat ++;
  521. x ++;
  522. }
  523. npos[1][0] = maps[1]->buffer[x];
  524. while(maps[0]->buffer[x + 1] == j + qn - 1 - bpos[0][0]){
  525. if(maps[2]->buffer[x]) mat ++;
  526. x ++;
  527. }
  528. npos[1][1] = maps[1]->buffer[x];
  529. // TODO: arbitrarily use outer boundary as matched region, need to test its effect
  530. // TODO: whether to remove regs outside clps
  531. if(mat >= g->min_node_mats){
  532. beg = (npos[0][0] + bpos[1][0]);
  533. end = (npos[1][1] + bpos[1][0] + 1);
  534. if(end - beg >= min_node_len && end - beg <= max_node_len){
  535. closed = 0;
  536. } else {
  537. closed = 1;
  538. }
  539. #if DEBUG
  540. if(beg >= end || beg < 0){
  541. fprintf(stderr, " -- something wrong in %s -- %s:%d --\n", __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  542. abort();
  543. }
  544. #endif
  545. push_rdregv(regs, (rd_reg_t){ndoff, hit->tidx, hit->qdir, beg, end, mask | closed});
  546. }
  547. if(hit->qdir){
  548. ndoff --;
  549. } else {
  550. ndoff ++;
  551. }
  552. #if 0
  553. rd_reg_t *rg = ref_rdregv(regs, regs->size - 1);
  554. fprintf(stdout, "NPOS\tX=%d,%d\tY=%d,%d\t%d\t%d\t%d\t%d\n", j, j - bpos[0][0], j + qn - 1, j + qn - 1 - bpos[0][0], npos[0][0], npos[0][1], npos[1][0], npos[1][1]);
  555. fprintf(stdout, "REG\t%llu\t%s\t%c\t%d\t%d\t%d\n", rg->node, kbm->reads->buffer[rg->rid].tag, "+-"[rg->dir], rg->beg, rg->end, rg->end - rg->beg);
  556. #endif
  557. }
  558. }
  559. //if(!g->corr_mode){
  560. {
  561. ndoff = kbm->reads->buffer[hit->tidx].binoff;
  562. ndbeg = (bpos[1][0] % qn)? qn - (bpos[1][0] % qn): 0;
  563. ndoff = ndoff + ((ndbeg + bpos[1][0]) / qn);
  564. x = 0;
  565. for(j=ndbeg+bpos[1][0];j+qn<=bpos[1][1];j+=qn){
  566. mat = 0;
  567. while(maps[1]->buffer[x] < j - bpos[1][0]){
  568. x ++;
  569. }
  570. npos[0][0] = maps[0]->buffer[x];
  571. while(maps[1]->buffer[x] == j - bpos[1][0]){
  572. if(maps[2]->buffer[x]) mat ++;
  573. x ++;
  574. }
  575. npos[0][1] = maps[0]->buffer[x];
  576. while(maps[1]->buffer[x] < j + qn - 1 - bpos[1][0]){
  577. if(maps[2]->buffer[x]) mat ++;
  578. x ++;
  579. }
  580. npos[1][0] = maps[0]->buffer[x];
  581. while(maps[1]->buffer[x + 1] == j + qn - 1 - bpos[1][0]){
  582. if(maps[2]->buffer[x]) mat ++;
  583. x ++;
  584. }
  585. npos[1][1] = maps[0]->buffer[x];
  586. // TODO: arbitrarily use outer boundary as matched region, need to test its effect
  587. if(mat >= g->min_node_mats){
  588. if(hit->qdir){
  589. beg = qlen - (npos[1][1] + bpos[0][0] + 1);
  590. end = qlen - (npos[0][0] + bpos[0][0]);
  591. } else {
  592. beg = (npos[0][0] + bpos[0][0]);
  593. end = (npos[1][1] + bpos[0][0] + 1);
  594. }
  595. if(end - beg >= min_node_len && end - beg <= max_node_len){
  596. closed = 0;
  597. } else {
  598. closed = 1;
  599. }
  600. #if DEBUG
  601. if(beg >= end || beg < 0){
  602. fprintf(stderr, " -- something wrong in %s -- %s:%d --\n", __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  603. abort();
  604. }
  605. #endif
  606. push_rdregv(regs, (rd_reg_t){ndoff, hit->qidx, hit->qdir, beg, end, mask | closed});
  607. #if 0
  608. fprintf(stdout, "NPOS\tX=%d,%d\tY=%d,%d\t%d\t%d\t%d\t%d\n", j, j - bpos[1][0], j + qn - 1, j + qn - 1 - bpos[1][0], npos[0][0], npos[0][1], npos[1][0], npos[1][1]);
  609. rd_reg_t *rg = ref_rdregv(regs, regs->size - 1);
  610. fprintf(stdout, "REG\t%llu\t%s\t%c\t%d\t%d\t%d\n", rg->node, kbm->reads->buffer[rg->rid].tag, "+-"[rg->dir], rg->beg, rg->end, rg->end - rg->beg);
  611. #endif
  612. }
  613. ndoff ++;
  614. }
  615. }
  616. if(hit->qdir){
  617. tmp = qlen - hit->qb;
  618. hit->qb = qlen - hit->qe;
  619. hit->qe = tmp;
  620. }
  621. return !mask;
  622. }
  623. thread_beg_def(mdbg);
  624. Graph *g;
  625. KBMAux *aux, *raux;
  626. CTGCNS *cc;
  627. reg_t reg;
  628. rdregv *regs;
  629. BitVec *rdflags;
  630. u4i beg, end;
  631. int raw;
  632. FILE *alno;
  633. int task;
  634. thread_end_def(mdbg);
  635. thread_beg_func(mdbg);
  636. Graph *g;
  637. KBM *kbm;
  638. KBMAux *aux, *raux;
  639. rdregv *regs;
  640. BitVec *rdflags;
  641. u4v *maps[3], *tidxs;
  642. volatile reg_t *reg;
  643. g = mdbg->g;
  644. kbm = g->kbm;
  645. reg = (reg_t*)&mdbg->reg;
  646. aux = mdbg->aux;
  647. raux = mdbg->raux;
  648. regs = mdbg->regs;
  649. rdflags = mdbg->rdflags;
  650. maps[0] = init_u4v(32);
  651. maps[1] = init_u4v(32);
  652. maps[2] = init_u4v(32);
  653. tidxs = init_u4v(16);
  654. thread_beg_loop(mdbg);
  655. if(mdbg->task == 1){
  656. if(reg->closed) continue;
  657. //if(g->corr_mode){
  658. if(0){
  659. } else {
  660. query_index_kbm(aux, NULL, reg->rid, kbm->rdseqs, (kbm->reads->buffer[reg->rid].seqoff + UInt(reg->beg)) * KBM_BIN_SIZE, UInt(reg->end - reg->beg) * KBM_BIN_SIZE);
  661. map_kbm(aux);
  662. }
  663. if(raux && aux->hits->size){ // refine
  664. kbm_read_t *rd;
  665. u4i i, j, tidx;
  666. clear_kbm(raux->kbm);
  667. bitpush_kbm(raux->kbm, NULL, 0, kbm->rdseqs->bits, (kbm->reads->buffer[reg->rid].seqoff + UInt(reg->beg)) * KBM_BIN_SIZE, UInt(reg->end - reg->beg) * KBM_BIN_SIZE);
  668. ready_kbm(raux->kbm);
  669. simple_index_kbm(raux->kbm, 0, raux->kbm->bins->size);
  670. clear_u4v(tidxs);
  671. sort_array(aux->hits->buffer, aux->hits->size, kbm_map_t, num_cmpgt(a.tidx, b.tidx));
  672. for(i=0;i<aux->hits->size;i++){
  673. if(i && tidxs->buffer[tidxs->size - 1] == aux->hits->buffer[i].tidx) continue;
  674. push_u4v(tidxs, aux->hits->buffer[i].tidx);
  675. }
  676. clear_kbmmapv(aux->hits);
  677. clear_bitsvec(aux->cigars);
  678. for(i=0;i<tidxs->size;i++){
  679. tidx = get_u4v(tidxs, i);
  680. rd = ref_kbmreadv(aux->kbm->reads, tidx);
  681. query_index_kbm(raux, rd->tag, tidx, aux->kbm->rdseqs, rd->seqoff * KBM_BIN_SIZE, rd->bincnt * KBM_BIN_SIZE);
  682. map_kbm(raux);
  683. for(j=0;j<raux->hits->size;j++){
  684. flip_hit_kbmaux(aux, raux, j);
  685. }
  686. }
  687. }
  688. sort_array(aux->hits->buffer, aux->hits->size, kbm_map_t, num_cmpgt(b.mat, a.mat));
  689. }
  690. thread_end_loop(mdbg);
  691. free_u4v(maps[0]);
  692. free_u4v(maps[1]);
  693. free_u4v(maps[2]);
  694. free_u4v(tidxs);
  695. thread_end_func(mdbg);
  696. typedef struct {
  697. int pos:19;
  698. u4i dir:1, spur:1, dep:11;
  699. } rd_clp_t;
  700. define_list(rdclpv, rd_clp_t);
  701. static inline void clip_read_algo(int clps[2], rdclpv *brks, rdclpv *chis, int avg_dep, int min_dep){
  702. rd_clp_t *c;
  703. u4i i;
  704. int dep, x, y, max, mx, my;
  705. sort_array(brks->buffer, brks->size, rd_clp_t, num_cmpgt((a.pos << 1) | a.dir, (b.pos << 1) | b.dir));
  706. x = y = 0; max = 0; dep = 0; mx = my = 0;
  707. for(i=0;i<brks->size;i++){
  708. c = ref_rdclpv(brks, i);
  709. if(dep >= min_dep){
  710. y = c->pos;
  711. if(y - x > max){
  712. max = y - x;
  713. mx = x;
  714. my = y;
  715. }
  716. }
  717. if(c->dir){ // end of overlap
  718. c->dep = dep;
  719. dep --;
  720. } else {
  721. dep ++;
  722. c->dep = dep;
  723. if(dep == min_dep){
  724. x = c->pos;
  725. }
  726. }
  727. if(c->spur){
  728. push_rdclpv(chis, (rd_clp_t){c->pos - KBM_BIN_SIZE, 0, 0, c->dep});
  729. push_rdclpv(chis, (rd_clp_t){c->pos - 1, 1, 0, c->dep});
  730. push_rdclpv(chis, (rd_clp_t){c->pos, 0, 1, c->dep});
  731. push_rdclpv(chis, (rd_clp_t){c->pos + KBM_BIN_SIZE, 1, 0, c->dep});
  732. }
  733. }
  734. clps[0] = mx / KBM_BIN_SIZE;
  735. clps[1] = my / KBM_BIN_SIZE;
  736. if((int)chis->size < avg_dep){
  737. return;
  738. }
  739. sort_array(chis->buffer, chis->size, rd_clp_t, num_cmpgtx(a.pos, b.pos, b.dir, a.dir));
  740. dep = 0; max = 0; mx = 0;
  741. for(i=0;i<chis->size;i++){
  742. c = ref_rdclpv(chis, i);
  743. if(c->dir){
  744. if(c->spur){
  745. if(dep >= max){
  746. max = dep;
  747. mx = i;
  748. }
  749. }
  750. dep --;
  751. } else {
  752. dep ++;
  753. if(c->spur){
  754. if(dep >= max){
  755. max = dep;
  756. mx = i;
  757. }
  758. }
  759. }
  760. }
  761. if(max * 2 < avg_dep){
  762. return;
  763. }
  764. c = ref_rdclpv(chis, mx);
  765. if(c->dep >= avg_dep){
  766. return;
  767. }
  768. if(2 * c->dep > max + 1){
  769. return;
  770. }
  771. if(c->pos <= clps[0] * KBM_BIN_SIZE || c->pos >= clps[1] * KBM_BIN_SIZE){
  772. return;
  773. }
  774. if(c->pos - clps[0] > clps[1] - c->pos){
  775. clps[1] = (c->pos + 1) / KBM_BIN_SIZE;
  776. } else {
  777. clps[0] = (c->pos + 1) / KBM_BIN_SIZE;
  778. }
  779. }
  780. static inline void clip_read_core(Graph *g, u4i rid, hitlnkv *lnks, rdclpv *brks, rdclpv *chis){
  781. read_t *rd;
  782. hit_lnk_t *lnk;
  783. rd_hit_t *hit;
  784. rd_frg_t *f1, *f2;
  785. u4i i;
  786. int rlen, dlen, margin, min_dep, dep, avg, x, y;
  787. rd = ref_readv(g->reads, rid);
  788. rlen = g->kbm->reads->buffer[rid].bincnt;
  789. if(rlen == 0){
  790. return;
  791. }
  792. lnk = &rd->hits;
  793. clear_rdclpv(brks);
  794. margin = g->max_overhang > 0? g->max_overhang : 4;
  795. min_dep = 2;
  796. dep = 0;
  797. clear_hitlnkv(lnks);
  798. while(lnk->idx){
  799. hit = ref_rdhitv(g->rdhits, lnk->idx);
  800. f1 = hit->frgs + lnk->flg;
  801. if(f1->closed == 0) push_hitlnkv(lnks, *lnk);
  802. lnk = hit->lnks + lnk->flg;
  803. }
  804. for(i=0;i<lnks->size;i++){
  805. lnk = ref_hitlnkv(lnks, i);
  806. hit = ref_rdhitv(g->rdhits, lnk->idx);
  807. f1 = hit->frgs + lnk->flg;
  808. if(f1->closed == 0){
  809. // whether be contained
  810. if(f1->beg <= margin && f1->end + margin >= rlen){
  811. rd->clps[0] = 0;
  812. rd->clps[1] = rlen;
  813. return;
  814. }
  815. f2 = hit->frgs + !lnk->flg;
  816. dlen = g->kbm->reads->buffer[f2->rid].bincnt;
  817. if(f1->dir ^ f2->dir){
  818. x = (f1->beg > margin && f2->end + margin < dlen);
  819. y = (f1->end + margin < rlen && f2->beg > margin);
  820. } else {
  821. x = (f1->beg > margin && f2->beg > margin);
  822. y = (f1->end + margin < rlen && f2->end + margin < dlen);
  823. }
  824. if(x && y){
  825. //push_rdclpv(brks, (rd_clp_t){f1->beg, 0, 1, 0});
  826. //push_rdclpv(brks, (rd_clp_t){f1->end, 1, 1, 0});
  827. } else if(x){
  828. push_rdclpv(brks, (rd_clp_t){f1->beg * KBM_BIN_SIZE, 0, 1, 0});
  829. push_rdclpv(brks, (rd_clp_t){f1->end * KBM_BIN_SIZE, 1, 0, 0});
  830. } else if(y){
  831. push_rdclpv(brks, (rd_clp_t){f1->beg * KBM_BIN_SIZE, 0, 0, 0});
  832. push_rdclpv(brks, (rd_clp_t){f1->end * KBM_BIN_SIZE, 1, 1, 0});
  833. } else {
  834. dep += f1->end - f1->beg;
  835. push_rdclpv(brks, (rd_clp_t){f1->beg * KBM_BIN_SIZE, 0, 0, 0});
  836. push_rdclpv(brks, (rd_clp_t){f1->end * KBM_BIN_SIZE, 1, 0, 0});
  837. }
  838. }
  839. }
  840. clear_rdclpv(chis);
  841. avg = (dep + rlen - 1) / rlen;
  842. clip_read_algo(rd->clps, brks, chis, avg, min_dep);
  843. }
  844. thread_beg_def(mclp);
  845. Graph *g;
  846. int task;
  847. thread_end_def(mclp);
  848. thread_beg_func(mclp);
  849. Graph *g;
  850. rdclpv *brks, *chis;
  851. hitlnkv *lnks;
  852. u4i rid;
  853. g = mclp->g;
  854. brks = init_rdclpv(32);
  855. chis = init_rdclpv(32);
  856. lnks = init_hitlnkv(32);
  857. thread_beg_loop(mclp);
  858. if(mclp->task == 1){
  859. for(rid=mclp->t_idx;rid<mclp->g->reads->size;rid+=mclp->n_cpu){
  860. clip_read_core(mclp->g, rid, lnks, brks, chis);
  861. }
  862. } else if(mclp->task == 2){
  863. hitlnkv *lnks, *chks;
  864. read_t *rd;
  865. hit_lnk_t *lnk;
  866. u4i i;
  867. lnks = init_hitlnkv(1024);
  868. chks = init_hitlnkv(1024);
  869. for(rid=mclp->t_idx;rid<mclp->g->reads->size;rid+=mclp->n_cpu){
  870. rd = ref_readv(g->reads, rid);
  871. if(rd->hits.idx == 0) continue;
  872. clear_hitlnkv(lnks);
  873. lnk = &rd->hits;
  874. while(1){
  875. if(lnk->idx == 0) break;
  876. push_hitlnkv(lnks, *lnk);
  877. lnk = g->rdhits->buffer[lnk->idx].lnks + lnk->flg;
  878. }
  879. if(lnks->size < 2) continue;
  880. sort_array(lnks->buffer, lnks->size, hit_lnk_t, num_cmpgt(g->rdhits->buffer[b.idx].lnks[0].cnt, g->rdhits->buffer[a.idx].lnks[0].cnt));
  881. lnk = &rd->hits;
  882. for(i=0;i<lnks->size;i++){
  883. lnk->idx = lnks->buffer[i].idx;
  884. lnk->flg = lnks->buffer[i].flg;
  885. lnk = g->rdhits->buffer[lnks->buffer[i].idx].lnks + lnks->buffer[i].flg;
  886. }
  887. lnk->idx = 0;
  888. lnk->flg = 0;
  889. {
  890. clear_hitlnkv(chks);
  891. lnk = &rd->hits;
  892. while(1){
  893. if(lnk->idx == 0) break;
  894. push_hitlnkv(chks, *lnk);
  895. lnk = g->rdhits->buffer[lnk->idx].lnks + lnk->flg;
  896. }
  897. if(lnks->size != chks->size){
  898. fprintf(stderr, " -- something wrong in %s -- %s:%d --\n", __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  899. abort();
  900. }
  901. for(i=0;i<lnks->size;i++){
  902. if(lnks->buffer[i].idx != chks->buffer[i].idx || lnks->buffer[i].flg != chks->buffer[i].flg){
  903. fprintf(stderr, " -- something wrong in %s -- %s:%d --\n", __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  904. abort();
  905. }
  906. }
  907. }
  908. if(g->bestn && lnks->size > mclp->g->bestn){
  909. lnk = lnks->buffer + g->bestn - 1;
  910. lnk = g->rdhits->buffer[lnk->idx].lnks + lnk->flg;
  911. lnk->idx = 0;
  912. lnk->flg = 0;
  913. for(i=g->bestn;i<lnks->size;i++){
  914. lnk = lnks->buffer + i;
  915. g->rdhits->buffer[lnk->idx].frgs[lnk->flg].closed = 1;
  916. }
  917. rd->hits.cnt = g->bestn;
  918. }
  919. }
  920. free_hitlnkv(lnks);
  921. free_hitlnkv(chks);
  922. }
  923. thread_end_loop(mclp);
  924. free_rdclpv(brks);
  925. free_rdclpv(chis);
  926. free_hitlnkv(lnks);
  927. thread_end_func(mclp);
  928. static inline u8i check_read_reg_conflict_core(Graph *g, rd_reg_t *hit, int *conflict){
  929. read_t *rd;
  930. reg_t *reg;
  931. u8i idx, pre;
  932. int x, y;
  933. rd = ref_readv(g->reads, hit->rid);
  934. idx = rd->regs.idx;
  935. pre = 0;
  936. *conflict = 0;
  937. while(idx){
  938. reg = ref_regv(g->regs, idx);
  939. if(hit->beg >= reg->beg) pre = idx;
  940. idx = reg->read_link;
  941. if(hit->end <= reg->beg) break;
  942. if(hit->beg >= reg->end) continue;
  943. if(reg->closed) continue;
  944. if(hit->beg <= reg->beg){
  945. x = reg->beg;
  946. if(hit->end >= reg->end){
  947. y = reg->end;
  948. } else y = hit->end;
  949. } else {
  950. x = hit->beg;
  951. if(hit->end <= reg->end){
  952. y = hit->end;
  953. } else y = reg->end;
  954. }
  955. if(x < y){
  956. if(x + (int)g->regovl < y) *conflict = 1;
  957. }
  958. }
  959. return pre;
  960. }
  961. thread_beg_def(mupd);
  962. Graph *g;
  963. rdregv *regs;
  964. rnk_ref_t *nd;
  965. u8v *ins;
  966. u8i vt;
  967. thread_end_def(mupd);
  968. thread_beg_func(mupd);
  969. Graph *g;
  970. rdregv *regs;
  971. rd_reg_t *hit;
  972. u8i pre;
  973. u4i i;
  974. int conflict;
  975. g = mupd->g;
  976. regs = mupd->regs;
  977. thread_beg_loop(mupd);
  978. if(mupd->nd == NULL) continue;
  979. clear_u8v(mupd->ins);
  980. if(1){
  981. for(i=1;i<mupd->nd->cnt;i++){
  982. if(regs->buffer[mupd->nd->idx + i].rid == regs->buffer[mupd->nd->idx + i - 1].rid){
  983. regs->buffer[mupd->nd->idx + i].closed = 1;
  984. regs->buffer[mupd->nd->idx + i - 1].closed = 1;
  985. }
  986. }
  987. }
  988. for(i=0;i<mupd->nd->cnt;i++){
  989. hit = ref_rdregv(regs, mupd->nd->idx + i);
  990. conflict = 0;
  991. pre = check_read_reg_conflict_core(g, hit, &conflict);
  992. if(conflict){
  993. hit->closed = 1;
  994. }
  995. push_u8v(mupd->ins, pre);
  996. }
  997. thread_end_loop(mupd);
  998. thread_end_func(mupd);
  999. //TODO: reg_t to store regs, sort them by rank, and sort them by rid in another u4i/u8i array
  1000. //TODO: fast generate nodes and read_link, then select important intervals
  1001. static inline void mul_update_regs_graph(Graph *g, rdregv *regs, rnkrefv *nds, u4i ncpu, u8i upds[3]){
  1002. u8i idx, vt;
  1003. u4i i, j, pass;
  1004. read_t *rd;
  1005. node_t *n;
  1006. reg_t *reg, *r;
  1007. rd_reg_t *hit;
  1008. int conflict, closed;
  1009. thread_prepare(mupd);
  1010. for(i=0;i<g->reads->size;i++){
  1011. g->reads->buffer[i].visit = 0; // in which round did the read be touched
  1012. }
  1013. encap_regv(g->regs, regs->size + 1); // make sure next_ref_regv in this function is thread-safe
  1014. thread_beg_init(mupd, ncpu);
  1015. mupd->g = g;
  1016. mupd->regs = regs;
  1017. mupd->nd = NULL;
  1018. mupd->ins = init_u8v(64);
  1019. mupd->vt = 0;
  1020. thread_end_init(mupd);
  1021. vt = 1;
  1022. upds[0] = upds[1] = upds[2] = 0;
  1023. for(idx=0;idx<nds->size+ncpu;idx++){
  1024. thread_wait_next(mupd);
  1025. if(mupd->nd){
  1026. pass = 0;
  1027. for(j=0;j<mupd->nd->cnt;j++){
  1028. hit = ref_rdregv(regs, mupd->nd->idx + j);
  1029. rd = ref_readv(g->reads, hit->rid);
  1030. if(rd->visit >= mupd->vt){ // previous node had changed the layout of this read, need to check conflict again
  1031. mupd->ins->buffer[j] = check_read_reg_conflict_core(g, hit, &conflict);
  1032. if(conflict){
  1033. hit->closed = 1;
  1034. }
  1035. }
  1036. rd->visit = idx;
  1037. if(hit->closed == 0) pass ++;
  1038. }
  1039. do {
  1040. closed = 0;
  1041. if(0 && mupd->nd->cnt > g->max_node_cov){
  1042. // Repetitive nodes should be in graph to avoid mis-assembling
  1043. upds[1] ++;
  1044. closed = 1;
  1045. continue;
  1046. }
  1047. if(mupd->nd->fix){
  1048. if(pass == 0){
  1049. upds[1] ++;
  1050. closed = 1;
  1051. continue;
  1052. } else {
  1053. upds[0] ++;
  1054. }
  1055. } else {
  1056. if(pass < g->min_node_cov || pass < (u4i)(mupd->nd->cnt * (1 - g->node_max_conflict))){
  1057. upds[1] ++;
  1058. closed = 1;
  1059. continue;
  1060. } else {
  1061. upds[0] ++;
  1062. }
  1063. }
  1064. n = next_ref_nodev(g->nodes);
  1065. n->rep_idx = MAX_REP_IDX;
  1066. n->unvisit = 0;
  1067. n->closed = 0;
  1068. n->single_in = 0;
  1069. n->bt_visit = 0;
  1070. n->bt_idx = 0;
  1071. n->init_end = 0;
  1072. n->regs.idx = g->regs->size;
  1073. n->regs.cnt = 0;
  1074. n->cov = 0;
  1075. n->edges[0] = PTR_REF_NULL;
  1076. n->edges[1] = PTR_REF_NULL;
  1077. for(j=0;j<mupd->nd->cnt;j++){
  1078. hit = ref_rdregv(regs, mupd->nd->idx + j);
  1079. rd = ref_readv(g->reads, hit->rid);
  1080. n->regs.cnt ++;
  1081. reg = next_ref_regv(g->regs); // Now, it is thread-safe
  1082. reg->node = g->nodes->size - 1;
  1083. reg->rid = hit->rid;
  1084. reg->dir = hit->dir;
  1085. reg->beg = hit->beg;
  1086. reg->end = hit->end;
  1087. reg->closed = hit->closed | closed;
  1088. if(reg->closed == 0) n->cov ++;
  1089. reg->read_link = 0;
  1090. if(mupd->ins->buffer[j]){
  1091. r = ref_regv(g->regs, mupd->ins->buffer[j]);
  1092. reg->read_link = r->read_link;
  1093. r->read_link = g->regs->size - 1;
  1094. } else {
  1095. reg->read_link = rd->regs.idx;
  1096. rd->regs.idx = g->regs->size - 1;
  1097. }
  1098. rd->regs.cnt ++;
  1099. }
  1100. } while(0);
  1101. }
  1102. if(idx < nds->size){
  1103. mupd->nd = ref_rnkrefv(nds, idx);
  1104. mupd->vt = idx;
  1105. thread_wake(mupd);
  1106. } else {
  1107. mupd->nd = NULL;
  1108. }
  1109. }
  1110. thread_beg_close(mupd);
  1111. free_u8v(mupd->ins);
  1112. thread_end_close(mupd);
  1113. }
  1114. static inline u8i chainning_hits_core(kbmmapv *hits, BitsVec *cigars, int uniq_hit, float aln_var){
  1115. kbm_map_t HIT;
  1116. u8i idx, bst, ite, lst, lsx, nrm;
  1117. u4i state;
  1118. sort_array(hits->buffer, hits->size, kbm_map_t, num_cmpgtxx(a.qidx, b.qidx, a.tidx, b.tidx, a.qdir, b.qdir));
  1119. nrm = 0;
  1120. for(idx=lst=lsx=0;idx<=hits->size;idx++){
  1121. state = 0;
  1122. if(idx == hits->size){
  1123. state = 1;
  1124. } else if(hits->buffer[lst].qidx == hits->buffer[idx].qidx && hits->buffer[lst].tidx == hits->buffer[idx].tidx){
  1125. if(hits->buffer[lst].qdir == hits->buffer[idx].qdir){
  1126. state = 0;
  1127. } else {
  1128. state = 2;
  1129. }
  1130. } else {
  1131. state = 1;
  1132. }
  1133. if(state){
  1134. if(idx > lst + 1){
  1135. if(simple_chain_all_maps_kbm(hits->buffer + lst, idx - lst, cigars, &HIT, cigars, aln_var)){
  1136. hits->buffer[lst++] = HIT;
  1137. while(lst < idx){
  1138. hits->buffer[lst++].mat = 0; // closed = 1
  1139. nrm ++;
  1140. }
  1141. }
  1142. }
  1143. if(state == 1){
  1144. // Choose the best hit
  1145. if(uniq_hit && idx > lsx + 1){
  1146. bst = lsx;
  1147. for(bst=lsx;bst<idx;bst++){
  1148. if(hits->buffer[bst].mat) break;
  1149. }
  1150. for(ite=bst+1;ite<idx;ite++){
  1151. if(hits->buffer[ite].mat == 0) continue;
  1152. if((hits->buffer[ite].aln > hits->buffer[bst].aln) || (hits->buffer[ite].aln == hits->buffer[bst].aln && hits->buffer[ite].mat > hits->buffer[bst].mat)){
  1153. bst = ite;
  1154. }
  1155. }
  1156. for(ite=lsx;ite<bst;ite++){
  1157. hits->buffer[ite].mat = 0;
  1158. }
  1159. for(ite=bst+1;ite<idx;ite++){
  1160. hits->buffer[ite].mat = 0;
  1161. }
  1162. nrm += idx - lsx - 1;
  1163. }
  1164. lsx = idx;
  1165. }
  1166. lst = idx;
  1167. }
  1168. }
  1169. return nrm;
  1170. }
  1171. static inline u8i load_alignments_core(Graph *g, FileReader *pws, int raw, rdregv *regs, u4v *maps[3]){
  1172. kbmmapv *hits;
  1173. BitsVec *cigars;
  1174. u4v *cgs;
  1175. kbm_map_t *hit, HIT, *h;
  1176. cuhash_t *cu;
  1177. u8i nhit;
  1178. u4i i, rid;
  1179. int qlen, val, flg, nwarn, mwarn, ncol;
  1180. char *cgstr, *qtag;
  1181. mwarn = 20;
  1182. nwarn = 0;
  1183. cgs = init_u4v(4);
  1184. hits = init_kbmmapv(32);
  1185. cigars = g->chainning_hits? init_bitsvec(1024, 3) : g->cigars;
  1186. memset(&HIT, 0, sizeof(kbm_map_t));
  1187. hit = &HIT;
  1188. nhit = 0;
  1189. while((ncol = readtable_filereader(pws)) != -1){
  1190. if((pws->n_line % 100000) == 0){
  1191. fprintf(KBM_LOGF, "\r%llu", pws->n_line); fflush(KBM_LOGF);
  1192. }
  1193. if(pws->line->buffer[0] == '#'){
  1194. //if(strncasecmp(pws->line->buffer, "#corr_mode=1", 12) == 0){
  1195. //g->corr_mode = 1;
  1196. //}
  1197. continue;
  1198. }
  1199. if(ncol < 15) continue;
  1200. if((cu = get_cuhash(g->kbm->tag2idx, get_col_str(pws, 0))) == NULL){
  1201. if(nwarn < mwarn){
  1202. fprintf(stderr, " -- Cannot find read \"%s\" in LINE:%llu in %s -- %s:%d --\n", get_col_str(pws, 0), pws->n_line, __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  1203. nwarn ++;
  1204. }
  1205. continue;
  1206. } else rid = cu->val;
  1207. hit->qidx = rid;
  1208. qtag = get_col_str(pws, 0);
  1209. hit->qdir = (get_col_str(pws, 1)[0] == '-');
  1210. qlen = atoi(get_col_str(pws, 2));
  1211. //if(g->corr_mode){
  1212. if(0){
  1213. //g->reads->buffer[hit->qidx].corr_bincnt = qlen / KBM_BIN_SIZE;
  1214. } else if(qlen != (int)g->kbm->reads->buffer[hit->qidx].bincnt * KBM_BIN_SIZE){
  1215. if(nwarn < mwarn){
  1216. fprintf(stderr, " -- inconsisitent read length \"%s\" %d != %d in %s -- %s:%d --\n", qtag, qlen, g->kbm->reads->buffer[hit->qidx].bincnt * KBM_BIN_SIZE, __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  1217. nwarn ++;
  1218. }
  1219. }
  1220. hit->qb = atoi(get_col_str(pws, 3)) / KBM_BIN_SIZE;
  1221. hit->qe = atoi(get_col_str(pws, 4)) / KBM_BIN_SIZE;
  1222. if((cu = get_cuhash(g->kbm->tag2idx, get_col_str(pws, 5))) == NULL){
  1223. if(nwarn < mwarn){
  1224. fprintf(stderr, " -- Cannot find read \"%s\" in LINE:%llu in %s -- %s:%d --\n", get_col_str(pws, 5), pws->n_line, __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  1225. nwarn ++;
  1226. }
  1227. continue;
  1228. } else rid = cu->val;
  1229. if(hit->qidx >= rid) continue;
  1230. hit->tidx = rid;
  1231. hit->tdir = 0; // col 6 always eq '+'
  1232. // skil col 7
  1233. hit->tb = atoi(get_col_str(pws, 8)) / KBM_BIN_SIZE;
  1234. hit->te = atoi(get_col_str(pws, 9)) / KBM_BIN_SIZE;
  1235. // skip col 10-13
  1236. hit->mat = atoi(get_col_str(pws, 10));
  1237. if(hit->mat < g->par->min_mat) continue;
  1238. hit->aln = atoi(get_col_str(pws, 11)) / KBM_BIN_SIZE;
  1239. if(hit->aln < g->par->min_aln) continue;
  1240. if(hit->mat < hit->aln * KBM_BIN_SIZE * g->par->min_sim) continue;
  1241. if(num_diff(hit->qe - hit->qb, hit->te - hit->tb) > (int)num_max(g->par->aln_var * hit->aln, 1.0)) continue;
  1242. hit->cnt = atoi(get_col_str(pws, 12));
  1243. hit->gap = atoi(get_col_str(pws, 13));
  1244. if(g->chainning_hits){
  1245. if(hits->size && peer_kbmmapv(hits)->qidx != hit->qidx){
  1246. chainning_hits_core(hits, cigars, g->uniq_hit, g->kbm->par->aln_var);
  1247. for(i=0;i<hits->size;i++){
  1248. h = ref_kbmmapv(hits, i);
  1249. if(h->mat == 0) continue;
  1250. append_bitsvec(g->cigars, cigars, h->cgoff, h->cglen);
  1251. h->cgoff = g->cigars->size - h->cglen;
  1252. nhit ++;
  1253. map2rdhits_graph(g, h);
  1254. }
  1255. clear_kbmmapv(hits);
  1256. clear_bitsvec(cigars);
  1257. }
  1258. }
  1259. hit->cgoff = cigars->size;
  1260. clear_u4v(cgs);
  1261. cgstr = get_col_str(pws, 14);
  1262. if(cgstr[0] == '*'){ // no cigar
  1263. continue;
  1264. }
  1265. val = 0;
  1266. while(cgstr[0]){
  1267. if(cgstr[0] >= '0' && cgstr[0] <= '9'){
  1268. val = val * 10 + (cgstr[0] - '0');
  1269. } else {
  1270. flg = -1;
  1271. switch(cgstr[0]){
  1272. case 'M': flg = 0; break;
  1273. case 'I': flg = 1; break;
  1274. case 'D': flg = 2; break;
  1275. case 'm': flg = 4; break;
  1276. case 'i': flg = 5; break;
  1277. case 'd': flg = 6; break;
  1278. default:
  1279. fprintf(stderr, " -- Bad cigar '%c' \"%s\" in LINE:%llu in %s -- %s:%d --\n", cgstr[0], get_col_str(pws, 14), pws->n_line, __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  1280. exit(1);
  1281. }
  1282. if(val == 0) val = 1;
  1283. push_u4v(cgs, (val << 8) | flg);
  1284. val = 0;
  1285. }
  1286. cgstr ++;
  1287. }
  1288. for(i=cgs->size;i>0;i--){
  1289. val = cgs->buffer[i - 1] >> 8;
  1290. flg = cgs->buffer[i - 1] & 0xFF;
  1291. pushs_bitsvec(cigars, flg, val);
  1292. }
  1293. hit->cglen = cigars->size - hit->cgoff;
  1294. if(raw){
  1295. hit2rdregs_graph(g, regs, qlen / KBM_BIN_SIZE, hit, cigars, maps);
  1296. clear_bitsvec(cigars);
  1297. } else if(g->chainning_hits){
  1298. push_kbmmapv(hits, *hit);
  1299. } else {
  1300. map2rdhits_graph(g, hit);
  1301. }
  1302. }
  1303. if(g->chainning_hits){
  1304. if(hits->size){
  1305. chainning_hits_core(hits, cigars, g->uniq_hit, g->kbm->par->aln_var);
  1306. for(i=0;i<hits->size;i++){
  1307. h = ref_kbmmapv(hits, i);
  1308. if(h->mat == 0) continue;
  1309. append_bitsvec(g->cigars, cigars, h->cgoff, h->cglen);
  1310. h->cgoff = g->cigars->size - h->cglen;
  1311. nhit ++;
  1312. map2rdhits_graph(g, h);
  1313. }
  1314. clear_kbmmapv(hits);
  1315. clear_bitsvec(cigars);
  1316. }
  1317. }
  1318. fprintf(KBM_LOGF, "\r%llu lines, %llu hits\n", pws->n_line, nhit);
  1319. print_proc_stat_info(g_mpi_comm_rank); //每个分布节点比对完成后消耗时间 ningwenmo@outlook.com
  1320. free_kbmmapv(hits);
  1321. if(g->chainning_hits) free_bitsvec(cigars);
  1322. free_u4v(cgs);
  1323. return nhit;
  1324. }
  1325. static inline u8i proc_alignments_core(Graph *g, int ncpu, int raw, rdregv *regs, u4v *maps[3], char *prefix, char *dump_kbm){
  1326. kbm_map_t *hit;
  1327. kbm_read_t *pb;
  1328. BitVec *rdflags;
  1329. BufferedWriter *bw;
  1330. FILE *alno, *kmlog;
  1331. u8i nbp, mbp, nhit;
  1332. u8i i, ib, ie, ic;
  1333. u4i rid, qb, qe, ii, in;
  1334. int reset_kbm, n_cpu;
  1335. thread_prepare(mdbg);
  1336. if(KBM_LOG) n_cpu = 1;
  1337. else n_cpu = ncpu;
  1338. ic = (g->kbm->bins->size + g->num_index - 1) / g->num_index;
  1339. ie = 0;
  1340. UNUSED(mbp);
  1341. #if 0
  1342. if(g->corr_mode){
  1343. mbp = g->genome_size * g->corr_gcov;
  1344. qb = qe = g->kbm->reads->size / 2;
  1345. nbp = g->kbm->reads->buffer[qb].bincnt * KBM_BIN_SIZE;
  1346. while(nbp < mbp && qb && qe + 1 < g->kbm->reads->size){
  1347. qb --;
  1348. qe ++;
  1349. nbp += g->kbm->reads->buffer[qb].bincnt * KBM_BIN_SIZE;
  1350. nbp += g->kbm->reads->buffer[qe].bincnt * KBM_BIN_SIZE;
  1351. }
  1352. if(qe < g->kbm->reads->size) qe ++;
  1353. fprintf(KBM_LOGF, "[%s] turn correct-mode on, reads[%u ~ %u = %u] (%llu bp), genome-size=%llu, corr-gcov=%0.2f, corr-dep=[%d,%d,%0.2f]\n", date(), qb, qe, qe - qb, nbp, g->genome_size, g->corr_gcov, g->corr_min, g->corr_max, g->corr_cov); fflush(KBM_LOGF);
  1354. } else {
  1355. #else
  1356. {
  1357. #endif
  1358. qb = 0;
  1359. qe = g->reads->size;
  1360. }
  1361. // alno = open_file_for_write(prefix, ".alignments.gz", 1);
  1362. // 每个分布节点输出相应计算的alignments ningwenmo@outlook.com
  1363. char* alignments_file_name[64];
  1364. sprintf(alignments_file_name, ".%u_%u.alignments", g_mpi_comm_rank, g_mpi_comm_size);
  1365. alno = open_file_for_write(prefix, alignments_file_name, 1);
  1366. // bw = zopen_bufferedwriter(alno, 1024 * 1024, ncpu, 0);
  1367. #if 0
  1368. if(g->corr_mode){
  1369. beg_bufferedwriter(bw);
  1370. fprintf(bw->out, "#corr_mode=1\n");
  1371. end_bufferedwriter(bw);
  1372. }
  1373. rdflags = (!g->corr_mode && g->par->skip_contained)? init_bitvec(g->kbm->reads->size) : NULL;
  1374. in = g->corr_mode? 1 : g->num_index;
  1375. #else
  1376. rdflags = (g->par->skip_contained)? init_bitvec(g->kbm->reads->size) : NULL;
  1377. in = g->num_index;
  1378. #endif
  1379. if(g->kbm->seeds->size){
  1380. reset_kbm = 0;
  1381. if(in > 1){
  1382. fprintf(KBM_LOGF, " ** WARNNING: change number of kbm index to 1 **\n"); fflush(KBM_LOGF);
  1383. in = 1;
  1384. }
  1385. } else {
  1386. reset_kbm = 1;
  1387. }
  1388. //fix_node = 0;
  1389. nhit = 0;
  1390. for(ii=0;ii<in;ii++){
  1391. ib = ie;
  1392. ie = num_min(ib + ic, g->kbm->bins->size);
  1393. while(ie > 0 && ie < g->kbm->bins->size && g->kbm->bins->buffer[ie - 1].ridx == g->kbm->bins->buffer[ie].ridx) ie ++;
  1394. qb = 0;
  1395. qe = ie ? g->kbm->bins->buffer[ie - 1].ridx : 0;
  1396. nbp = ((u8i)(ie - ib)) * KBM_BIN_SIZE;
  1397. //每个分布节点计算分别计算相应数据 ningwenmo@outlook.com
  1398. if (g_mpi_comm_size >= 2 && ii != (g_mpi_comm_rank - 1))
  1399. continue;
  1400. if(reset_kbm){
  1401. reset_index_kbm(g->kbm);
  1402. fprintf(KBM_LOGF, "[%s] indexing bins[(%llu,%llu)/%llu] (%llu/%llu bp), %d threads\n", date(), ib, ie, (u8i)g->kbm->bins->size, nbp, (u8i)g->kbm->rdseqs->size, ncpu); fflush(KBM_LOGF);
  1403. kmlog = (in > 1)? NULL : open_file_for_write(prefix, ".kmerdep", 1);
  1404. index_kbm(g->kbm, ib, ie, ncpu, kmlog);
  1405. if(kmlog){
  1406. fclose(kmlog);
  1407. kmlog = NULL;
  1408. }
  1409. fprintf(KBM_LOGF, "[%s] Done\n", date()); fflush(KBM_LOGF);
  1410. if(in == 1 && dump_kbm){
  1411. FILE *dump;
  1412. fprintf(KBM_LOGF, "[%s] dump kbm-index to %s ...", date(), dump_kbm); fflush(KBM_LOGF);
  1413. dump = open_file_for_write(dump_kbm, NULL, 1);
  1414. mem_dump_obj_file(g->kbm, 1, &kbm_obj_desc, 1, 0, dump);
  1415. fclose(dump);
  1416. fprintf(KBM_LOGF, " Done\n"); fflush(KBM_LOGF);
  1417. }
  1418. if(in == 1){
  1419. u4i *deps;
  1420. u8i hidx;
  1421. kmlog = open_file_for_write(prefix, ".binkmer", 1);
  1422. deps = calloc(KBM_BIN_SIZE + 1, 4);
  1423. for(hidx=0;hidx<g->kbm->bins->size;hidx++){
  1424. deps[g->kbm->bins->buffer[hidx].degree] ++;
  1425. }
  1426. for(hidx=0;hidx<KBM_BIN_SIZE;hidx++){
  1427. fprintf(kmlog, "%u\n", deps[hidx]);
  1428. }
  1429. fclose(kmlog);
  1430. free(deps);
  1431. if(!g->minimal_output && 0){
  1432. kbm_bin_t *bn;
  1433. kmlog = open_file_for_write(prefix, ".closed_bins", 1);
  1434. for(hidx=0;hidx<g->kbm->bins->size;hidx++){
  1435. bn = ref_kbmbinv(g->kbm->bins, hidx);
  1436. if(bn->closed){
  1437. fprintf(kmlog, "%s_F_%d_%d\t%d\n", g->kbm->reads->buffer[bn->ridx].tag, bn->off * KBM_BIN_SIZE, KBM_BIN_SIZE, bn->degree);
  1438. }
  1439. }
  1440. fclose(kmlog);
  1441. }
  1442. }
  1443. }
  1444. {
  1445. {
  1446. thread_beg_init(mdbg, n_cpu);
  1447. mdbg->g = g;
  1448. memset((void*)&mdbg->reg, 0, sizeof(reg_t));
  1449. mdbg->reg.closed = 1;
  1450. mdbg->aux = init_kbmaux(g->kbm);
  1451. if(g->rpar){
  1452. mdbg->raux = init_kbmaux(init_kbm(g->rpar));
  1453. } else {
  1454. mdbg->raux = NULL;
  1455. }
  1456. #if 0
  1457. if(g->corr_mode){
  1458. KBMBlock *kb;
  1459. POGPar par;
  1460. kb = init_kbmblock(g->corr_bsize, g->corr_bstep);
  1461. //mdbg->cc = init_ctgcns(kb, iter_kbmblock, info_kbmblock, 1, 1, 1, g->corr_max, 200, 100, 1, 96, 2, -5, -2, -4, -1, 16, 3, 0.5, g->corr_bsize - g->corr_bstep + KBM_BIN_SIZE);
  1462. par = DEFAULT_POG_PAR;
  1463. par.refmode = 1;
  1464. mdbg->cc = init_ctgcns(kb, iter_kbmblock, info_kbmblock, 1, 1, g->corr_max, 200, 100, 1, g->corr_bsize - g->corr_bstep + KBM_BIN_SIZE, &par);
  1465. } else {
  1466. #else
  1467. {
  1468. #endif
  1469. mdbg->cc = NULL;
  1470. }
  1471. mdbg->aux->par = (KBMPar*)malloc(sizeof(KBMPar));
  1472. memcpy(mdbg->aux->par, g->par, sizeof(KBMPar));
  1473. mdbg->regs = regs;
  1474. mdbg->rdflags = rdflags;
  1475. mdbg->beg = 0;
  1476. mdbg->end = 0;
  1477. mdbg->raw = raw;
  1478. mdbg->alno = alno;
  1479. thread_end_init(mdbg);
  1480. }
  1481. thread_beg_iter(mdbg);
  1482. mdbg->task = 1;
  1483. thread_end_iter(mdbg);
  1484. for(rid=qb;rid<=qe+ncpu;rid++){
  1485. if(rid < qe){
  1486. if(!KBM_LOG && ((rid - qb) % 2000) == 0){ fprintf(KBM_LOGF, "\r%u|%llu", rid - qb, nhit); fflush(KBM_LOGF); }
  1487. thread_wait_one(mdbg);
  1488. } else {
  1489. thread_wait_next(mdbg);
  1490. pb = NULL;
  1491. }
  1492. if(mdbg->reg.closed == 0){
  1493. KBMAux *aux = mdbg->aux;
  1494. //if(g->corr_mode && mdbg->cc->cns->size){
  1495. //g->reads->buffer[mdbg->reg.rid].corr_bincnt = mdbg->cc->cns->size / KBM_BIN_SIZE;
  1496. //}
  1497. if(alno){
  1498. // beg_bufferedwriter(bw);
  1499. //if(g->corr_mode && mdbg->aux->cns->size){
  1500. //fprintf(bw->out, "#corrected\t%s\t%u\t", mdbg->cc->tag->string, (u4i)mdbg->cc->cns->size);
  1501. //println_fwdseq_basebank(mdbg->cc->cns, 0, mdbg->cc->cns->size, bw->out);
  1502. //}
  1503. // for(i=0;i<mdbg->aux->hits->size;i++){
  1504. // hit = ref_kbmmapv(mdbg->aux->hits, i);
  1505. // fprint_hit_kbm(mdbg->aux, i, bw->out);
  1506. // }
  1507. // end_bufferedwriter(bw);
  1508. for (i = 0; i < mdbg->aux->hits->size; i++)
  1509. {
  1510. hit = ref_kbmmapv(mdbg->aux->hits, i);
  1511. fprint_hit_kbm(mdbg->aux, i, alno);
  1512. }
  1513. }
  1514. for(i=0;i<mdbg->aux->hits->size;i++){
  1515. hit = ref_kbmmapv(mdbg->aux->hits, i);
  1516. if(hit->mat == 0) continue;
  1517. if(rdflags
  1518. && g->kbm->reads->buffer[hit->tidx].bincnt < g->kbm->reads->buffer[hit->qidx].bincnt
  1519. && (hit->tb <= 1 && hit->te + 1 >= (int)(g->kbm->reads->buffer[hit->tidx].bincnt))
  1520. && (hit->qb > 1 || hit->qe + 1 < (int)(g->kbm->reads->buffer[hit->qidx].bincnt))
  1521. ){
  1522. one_bitvec(rdflags, hit->tidx);
  1523. }
  1524. }
  1525. if(g->chainning_hits){
  1526. chainning_hits_core(aux->hits, aux->cigars, g->uniq_hit, g->kbm->par->aln_var);
  1527. }
  1528. for(i=0;i<aux->hits->size;i++){
  1529. hit = ref_kbmmapv(aux->hits, i);
  1530. if(hit->mat == 0) continue;
  1531. //hit->qb /= KBM_BIN_SIZE;
  1532. //hit->qe /= KBM_BIN_SIZE;
  1533. //hit->tb /= KBM_BIN_SIZE;
  1534. //hit->te /= KBM_BIN_SIZE;
  1535. //hit->aln /= KBM_BIN_SIZE;
  1536. nhit ++;
  1537. append_bitsvec(g->cigars, aux->cigars, hit->cgoff, hit->cglen);
  1538. hit->cgoff = g->cigars->size - hit->cglen;
  1539. if(raw){
  1540. //hit2rdregs_graph(g, regs, g->corr_mode? mdbg->cc->cns->size / KBM_BIN_SIZE : 0, hit, mdbg->aux->cigars, maps);
  1541. hit2rdregs_graph(g, regs, 0, hit, mdbg->aux->cigars, maps);
  1542. } else {
  1543. map2rdhits_graph(g, hit);
  1544. }
  1545. }
  1546. if(KBM_LOG){
  1547. fprintf(KBM_LOGF, "QUERY: %s\t+\t%d\t%d\n", g->kbm->reads->buffer[mdbg->reg.rid].tag, mdbg->reg.beg, mdbg->reg.end);
  1548. for(i=0;i<mdbg->aux->hits->size;i++){
  1549. hit = ref_kbmmapv(mdbg->aux->hits, i);
  1550. fprintf(KBM_LOGF, "\t%s\t%c\t%d\t%d\t%d\t%d\t%d\n", g->kbm->reads->buffer[hit->tidx].tag, "+-"[hit->qdir], g->kbm->reads->buffer[hit->tidx].bincnt * KBM_BIN_SIZE, hit->tb * KBM_BIN_SIZE, hit->te * KBM_BIN_SIZE, hit->aln * KBM_BIN_SIZE, hit->mat);
  1551. }
  1552. }
  1553. mdbg->reg.closed = 1;
  1554. }
  1555. if(rid < qe && (rdflags == NULL || get_bitvec(rdflags, rid) == 0)){
  1556. pb = ref_kbmreadv(g->kbm->reads, rid);
  1557. mdbg->reg = (reg_t){0, rid, 0, 0, pb->bincnt, 0, 0};
  1558. thread_wake(mdbg);
  1559. }
  1560. }
  1561. {
  1562. thread_beg_close(mdbg);
  1563. free(mdbg->aux->par);
  1564. free_kbmaux(mdbg->aux);
  1565. //if(g->corr_mode){
  1566. //free_kbmblock((KBMBlock*)mdbg->cc->obj);
  1567. //free_ctgcns(mdbg->cc);
  1568. //}
  1569. if(mdbg->raux){
  1570. free_kbm(mdbg->raux->kbm);
  1571. free_kbmaux(mdbg->raux);
  1572. }
  1573. thread_end_close(mdbg);
  1574. }
  1575. }
  1576. if(!KBM_LOG) fprintf(KBM_LOGF, "\r%u reads|total hits %llu\n", qe - qb, nhit);
  1577. if(reset_kbm){
  1578. reset_index_kbm(g->kbm);
  1579. }
  1580. print_proc_stat_info(g_mpi_comm_rank);
  1581. }
  1582. // if(bw) close_bufferedwriter(bw);
  1583. if(alno) fclose(alno);
  1584. if(rdflags) free_bitvec(rdflags);
  1585. return nhit;
  1586. }
  1587. static inline void build_nodes_graph(Graph *g, u8i maxbp, int ncpu, FileReader *pws, int rdclip, char *prefix, char *dump_kbm){
  1588. kbm_map_t *hit, HIT;
  1589. BitVec *rks;
  1590. u4v *maps[3];
  1591. FILE *clplog;
  1592. u8i idx, rank, kcnts[256], upds[3], fix_node, nhit;
  1593. u4i rid, i, dep;
  1594. int raw;
  1595. rdregv *regs;
  1596. rnkrefv *nds;
  1597. rnk_ref_t *nd;
  1598. thread_preprocess(mclp);
  1599. regs = init_rdregv(1024);
  1600. nds = init_rnkrefv(1024);
  1601. renew_rdhitv(g->rdhits, 1024);
  1602. maps[0] = init_u4v(4);
  1603. maps[1] = init_u4v(4);
  1604. maps[2] = init_u4v(4);
  1605. clear_regv(g->regs);
  1606. next_ref_regv(g->regs);
  1607. clear_rdhitv(g->rdhits);
  1608. ZEROS(next_ref_rdhitv(g->rdhits));
  1609. //clear_kbmmapv(g->pwalns);
  1610. clear_bitsvec(g->cigars);
  1611. raw = !((g->bestn > 0) || rdclip);
  1612. //if(g->corr_mode){
  1613. //raw = 1;
  1614. //}
  1615. fix_node = 0; // TODO: new code hasn't coped with contigs+longreads mode
  1616. if(pws){
  1617. nhit = load_alignments_core(g, pws, raw, regs, maps);
  1618. } else {
  1619. UNUSED(maxbp);
  1620. nhit = proc_alignments_core(g, ncpu, raw, regs, maps, prefix, dump_kbm);
  1621. }
  1622. // 每个节点分布计算完成后返回 ningwenmo@outlook.com
  1623. if (g_mpi_comm_size >= 2 && g_mpi_comm_rank != 0){
  1624. // int message = 0;
  1625. // MPI_Send(&message, 1, MPI_INT, 0, 1, MPI_COMM_WORLD);
  1626. // fprintf(KBM_LOGF, "[%s] send %u/%u\n", date(), g_mpi_comm_rank, g_mpi_comm_size);
  1627. // fflush(KBM_LOGF);
  1628. return;
  1629. }
  1630. // print_proc_stat_info(g_mpi_comm_rank);
  1631. if(raw){
  1632. fprintf(KBM_LOGF, "[%s] generated %llu regs\n", date(), (u8i)regs->size);
  1633. } else {
  1634. #ifdef KBM_MAP2RDHIT_QUICK
  1635. // sort rdhits and pick bestn
  1636. {
  1637. fprintf(KBM_LOGF, "[%s] sorting rdhits ... ", date()); fflush(KBM_LOGF);
  1638. thread_beg_init(mclp, ncpu);
  1639. mclp->g = g;
  1640. mclp->task = 2;
  1641. thread_end_init(mclp);
  1642. thread_wake_all(mclp);
  1643. thread_beg_close(mclp);
  1644. thread_end_close(mclp);
  1645. fprintf(KBM_LOGF, "Done\n"); fflush(KBM_LOGF);
  1646. }
  1647. #endif
  1648. // clip reads
  1649. if(rdclip){
  1650. fprintf(KBM_LOGF, "[%s] clipping ... ", date()); fflush(KBM_LOGF);
  1651. clplog = open_file_for_write(prefix, ".clps", 1);
  1652. thread_beg_init(mclp, ncpu);
  1653. mclp->g = g;
  1654. mclp->task = 1;
  1655. thread_end_init(mclp);
  1656. thread_wake_all(mclp);
  1657. thread_beg_close(mclp);
  1658. thread_end_close(mclp);
  1659. u8i tot, clp;
  1660. tot = clp = 0;
  1661. for(rid=0;rid<g->reads->size;rid++){
  1662. tot += g->kbm->reads->buffer[rid].bincnt * KBM_BIN_SIZE;
  1663. clp += (g->reads->buffer[rid].clps[1] - g->reads->buffer[rid].clps[0]) * KBM_BIN_SIZE;
  1664. fprintf(clplog, "%s\t%d\t%d\t%d\n", g->kbm->reads->buffer[rid].tag, g->kbm->reads->buffer[rid].bincnt * KBM_BIN_SIZE, g->reads->buffer[rid].clps[0] * KBM_BIN_SIZE, g->reads->buffer[rid].clps[1] * KBM_BIN_SIZE);
  1665. }
  1666. fclose(clplog);
  1667. fprintf(KBM_LOGF, "%.2f%% bases\n", ((tot - clp) * 100.0) / tot); fflush(KBM_LOGF);
  1668. }
  1669. fprintf(KBM_LOGF, "[%s] generating regs ... ", date()); fflush(KBM_LOGF);
  1670. rd_hit_t *rh;
  1671. u8i cgoff;
  1672. cgoff = 0;
  1673. hit = &HIT;
  1674. hit->cnt = 0;
  1675. hit->gap = 0;
  1676. for(idx=0;idx<g->rdhits->size;idx++){
  1677. rh = ref_rdhitv(g->rdhits, idx);
  1678. hit->cgoff = cgoff;
  1679. hit->cglen = rh->lnks[1].cnt;
  1680. cgoff += hit->cglen;
  1681. if(rh->frgs[0].closed && rh->frgs[1].closed){
  1682. continue;
  1683. }
  1684. hit->qidx = rh->frgs[0].rid;
  1685. hit->tidx = rh->frgs[1].rid;
  1686. hit->qdir = rh->frgs[0].dir;
  1687. hit->tdir = rh->frgs[1].dir;
  1688. hit->qb = rh->frgs[0].beg;
  1689. hit->qe = rh->frgs[0].end;
  1690. hit->tb = rh->frgs[1].beg;
  1691. hit->te = rh->frgs[1].end;
  1692. hit->mat = rh->lnks[0].cnt;
  1693. hit->aln = num_min(hit->qe - hit->qb, hit->te - hit->tb);
  1694. //hit2rdregs_graph(g, regs, g->reads->buffer[hit->qidx].corr_bincnt, hit, g->cigars, maps);
  1695. hit2rdregs_graph(g, regs, g->kbm->reads->buffer[hit->qidx].bincnt, hit, g->cigars, maps);
  1696. }
  1697. free_bitsvec(g->cigars); g->cigars = init_bitsvec(1024, 3);
  1698. fprintf(KBM_LOGF, "%llu\n", (u8i)regs->size); fflush(KBM_LOGF);
  1699. }
  1700. free_u4v(maps[0]);
  1701. free_u4v(maps[1]);
  1702. free_u4v(maps[2]);
  1703. // add node itself
  1704. //if(!g->corr_mode){
  1705. {
  1706. rks = init_bitvec(g->kbm->bins->size);
  1707. for(idx=0;idx<regs->size;idx++){
  1708. one_bitvec(rks, regs->buffer[idx].node);
  1709. }
  1710. for(idx=0;idx<g->kbm->bins->size;idx++){
  1711. if(get_bitvec(rks, idx)){
  1712. rid = g->kbm->bins->buffer[idx].ridx;
  1713. kbm_read_t *rd = ref_kbmreadv(g->kbm->reads, rid);
  1714. push_rdregv(regs, (rd_reg_t){idx, rid, 0, (idx - rd->binoff) * g->reglen, (idx + 1 - rd->binoff) * g->reglen, 0});
  1715. }
  1716. }
  1717. free_bitvec(rks);
  1718. }
  1719. // generating nodes
  1720. fprintf(KBM_LOGF, "[%s] sorting regs ... ", date()); fflush(KBM_LOGF);
  1721. psort_array(regs->buffer, regs->size, rd_reg_t, ncpu, num_cmpgtxx((a.node << 30) | a.rid, (b.node << 30) | b.rid, a.beg, b.beg, b.end, a.end));
  1722. fprintf(KBM_LOGF, " Done\n"); fflush(KBM_LOGF);
  1723. u4v *kbcnts;
  1724. kbcnts = init_u4v(1024);
  1725. rank = 0xFFFFFFFFFFFFFFFFLLU;
  1726. nd = NULL;
  1727. fprintf(KBM_LOGF, "[%s] generating intervals ... ", date()); fflush(KBM_LOGF);
  1728. for(idx=0;idx<regs->size;idx++){
  1729. if(0){
  1730. kbm_read_t *rd1, *rd2;
  1731. rd_reg_t *r;
  1732. r = ref_rdregv(regs, idx);
  1733. rd1 = ref_kbmreadv(g->kbm->reads, g->kbm->bins->buffer[r->node].ridx);
  1734. rd2 = ref_kbmreadv(g->kbm->reads, r->rid);
  1735. fprintf(stdout, "%s\t%c\t%d\t%d\t%s\t%c\t%d\t%d\n", rd1->tag, '+', (int)((r->node) - rd1->binoff) * g->reglen,
  1736. (int)(r->node - rd1->binoff + 1) * g->reglen,
  1737. rd2->tag, "+-"[r->dir], r->beg, r->end);
  1738. }
  1739. if(regs->buffer[idx].node != rank){
  1740. if(nd){
  1741. while(nd->cnt >= kbcnts->size) push_u4v(kbcnts, 0);
  1742. kbcnts->buffer[nd->cnt] ++;
  1743. }
  1744. nd = next_ref_rnkrefv(nds);
  1745. nd->idx = idx;
  1746. nd->rank = regs->buffer[idx].node;
  1747. nd->fix = regs->buffer[idx].node < fix_node;
  1748. nd->cnt = 1;
  1749. //nd->score = (regs->buffer[idx].beg >= g->reads->buffer[regs->buffer[idx].rid].clps[0] && regs->buffer[idx].end <= g->reads->buffer[regs->buffer[idx].rid].clps[1]);
  1750. rank = regs->buffer[idx].node;
  1751. } else {
  1752. nd->cnt ++;
  1753. //nd->score += (regs->buffer[idx].beg >= g->reads->buffer[regs->buffer[idx].rid].clps[0] && regs->buffer[idx].end <= g->reads->buffer[regs->buffer[idx].rid].clps[1]);
  1754. }
  1755. }
  1756. if(nd){
  1757. while(nd->cnt >= kbcnts->size) push_u4v(kbcnts, 0);
  1758. kbcnts->buffer[nd->cnt] ++;
  1759. }
  1760. // find medean k-bin depth
  1761. u4i gcov;
  1762. for(gcov=0,rank=0;gcov<kbcnts->size&&rank<nds->size;gcov++){
  1763. rank += kbcnts->buffer[gcov];
  1764. }
  1765. //psort_array(nds->buffer, nds->size, rnk_ref_t, ncpu, num_cmpgtx(b.cnt, a.cnt, a.rank, b.rank));
  1766. psort_array(nds->buffer, nds->size, rnk_ref_t, ncpu, num_cmpgtx(num_diff(a.cnt, gcov), num_diff(b.cnt, gcov), a.rank, b.rank));
  1767. fprintf(KBM_LOGF, " %llu intervals\n", (u8i)nds->size); fflush(KBM_LOGF);
  1768. //psort_array(nds->buffer, nds->size, rnk_ref_t, ncpu, num_cmpgtx(b.score, a.score, a.rank, b.rank));
  1769. if(0){
  1770. memset(kcnts, 0, sizeof(uint64_t) * 256);
  1771. for(idx=0;idx<nds->size;idx++){
  1772. dep = num_min(nds->buffer[idx].cnt, 255);
  1773. kcnts[dep] ++;
  1774. }
  1775. for(i=1;i<51;i++){
  1776. fprintf(KBM_LOGF, "%10llu", (long long unsigned int)kcnts[i]);
  1777. if(((i - 1) % 10) == 9) fprintf(KBM_LOGF, "\n");
  1778. }
  1779. if(((i - 1) % 10) != 0) fprintf(KBM_LOGF, "\n");
  1780. }
  1781. fprintf(KBM_LOGF, "[%s] selecting important intervals from %llu intervals\n", date(), (u8i)nds->size);
  1782. mul_update_regs_graph(g, regs, nds, ncpu, upds);
  1783. free_rdregv(regs);
  1784. free_rnkrefv(nds);
  1785. encap_regv(g->regs, 1);
  1786. memset(g->regs->buffer + g->regs->size, 0xFF, sizeof(reg_t));
  1787. if(!KBM_LOG) fprintf(KBM_LOGF, "[%s] Intervals: kept %llu, discarded %llu\n", date(), upds[0], upds[1]);
  1788. print_proc_stat_info(g_mpi_comm_rank);
  1789. }
  1790. uint32_t estimate_genome_size(Graph *g, unsigned long long tot_bp, FILE *out){
  1791. uint64_t kcnts[256];
  1792. node_t *n;
  1793. uint64_t nid, sum, ncnt, pmax;
  1794. uint32_t i, dep, peak, mid;
  1795. float avg;
  1796. ncnt = g->nodes->size;
  1797. memset(kcnts, 0, sizeof(uint64_t) * 256);
  1798. sum = 0;
  1799. for(nid=0;nid<ncnt;nid++){
  1800. n = ref_nodev(g->nodes, nid);
  1801. dep = num_min(n->regs.cnt, 255);
  1802. sum += n->regs.cnt;
  1803. kcnts[dep] ++;
  1804. }
  1805. mid = pmax = 0;
  1806. while(mid < 255){
  1807. pmax += kcnts[mid];
  1808. if(pmax >= ncnt / 2) break;
  1809. mid ++;
  1810. }
  1811. avg = 1.0 * sum / (ncnt + 1);
  1812. fprintf(out, "[%s] median node depth = %d\n", date(), mid);
  1813. return mid;
  1814. //TODO: calculate the genome coverage
  1815. for(i=1;i<51;i++){
  1816. fprintf(out, "%10llu", (long long unsigned int)kcnts[i]);
  1817. if(((i - 1) % 10) == 9) fprintf(out, "\n");
  1818. }
  1819. if(((i - 1) % 10) != 0) fprintf(out, "\n");
  1820. return avg;
  1821. pmax = 0; peak = avg;
  1822. for(i=g->min_node_cov+1;i<256;i++){
  1823. if(kcnts[i] > pmax){ peak = i; pmax = kcnts[i]; }
  1824. else if(i > avg && kcnts[i] < 0.8 * pmax) break;
  1825. }
  1826. fprintf(out, "[%s] depth peak = %d\n", date(), peak);
  1827. fprintf(out, "[%s] genome size = %llu\n", date(), tot_bp / peak);
  1828. return peak;
  1829. }
  1830. // MUST be called before build_edges
  1831. static u8i mask_nodes_by_cov_graph(Graph *g, FILE *out){
  1832. node_t *n;
  1833. u8i ret, i;
  1834. ret = 0;
  1835. for(i=0;i<g->nodes->size;i++){
  1836. n = ref_nodev(g->nodes, i);
  1837. if(n->regs.cnt > g->max_node_cov || n->regs.cnt < g->min_node_cov){
  1838. n->closed = 1;
  1839. ret ++;
  1840. if(out) fprintf(out, "MASK_COV\tN%llu\t%u\t%u\n", (u8i)i, (u4i)n->regs.cnt, n->cov);
  1841. }
  1842. }
  1843. return ret;
  1844. }
  1845. static inline void remove_all_edges_graph(Graph *g){
  1846. node_t *n;
  1847. uint64_t nid;
  1848. free_edgev(g->edges);
  1849. g->edges = init_edgev(32);
  1850. free_edgehash(g->ehash);
  1851. g->ehash = init_edgehash(1023);
  1852. set_userdata_edgehash(g->ehash, g->edges);
  1853. free_edgerefv(g->erefs);
  1854. g->erefs = init_edgerefv(32);
  1855. for(nid=0;nid<g->nodes->size;nid++){
  1856. n = ref_nodev(g->nodes, nid);
  1857. n->edges[0] = n->edges[1] = PTR_REF_NULL;
  1858. }
  1859. }
  1860. typedef struct {
  1861. uint64_t idx:46; int off:18;
  1862. } edge_off_t;
  1863. define_list(edgeoffv, edge_off_t);
  1864. static inline int estimate_edge_length(edge_off_t *ps, uint32_t size, uint32_t idxs[2]){
  1865. int64_t tot, var;
  1866. uint32_t i, b, e, mi;
  1867. int v, max, len, avg, std;
  1868. idxs[0] = 0; idxs[1] = size;
  1869. if(size == 0){ return 0; }
  1870. if(size <= 2){ return ps[size/2].off; }
  1871. b = 0; e = size;
  1872. for(i=b,tot=0;i<e;i++) tot += ps[i].off;
  1873. len = tot / (e - b);
  1874. //fprintf(stdout, "b=%d\te=%d\tlen=%d\n", b, e, len);
  1875. while(b + 2 < e){
  1876. max = 0; mi = 0;
  1877. for(i=b+1;i<e;i++){
  1878. if(ps[i].off - ps[i-1].off > max){
  1879. max = ps[i].off - ps[i-1].off;
  1880. mi = i;
  1881. }
  1882. }
  1883. if(max < len * 0.5) break;
  1884. else if(max < 100) break;
  1885. if(mi - b > e - mi) e = mi;
  1886. else b = mi;
  1887. for(i=b,tot=0;i<e;i++) tot += ps[i].off; avg = tot / (e - b);
  1888. if(num_diff(avg, len) < num_max(avg * 0.2, 50)) break;
  1889. len = avg;
  1890. }
  1891. //fprintf(stdout, "b=%d\te=%d\tlen=%d\n", b, e, len);
  1892. if(0){
  1893. if(b + 1 < e){
  1894. for(i=b,var=0;i<e;i++){
  1895. v = ((int)ps[i].off) - ((int)len);
  1896. var += v * v;
  1897. }
  1898. std = sqrt(var / (e - b));
  1899. //fprintf(stdout, "std=%d\n", std);
  1900. b = 0; e = size;
  1901. while(b < e && num_diff(ps[b].off, len) > 3 * std) b ++;
  1902. while(b < e && num_diff(ps[e - 1].off, len) > 3 * std) e --;
  1903. }
  1904. idxs[0] = b;
  1905. idxs[1] = e;
  1906. }
  1907. return len;
  1908. }
  1909. static inline int calculate_edge_cov_off_graph(Graph *g, edge_t *e, edgeoffv *offs){
  1910. node_t *n1, *n2;
  1911. reg_t *r1, *r2;
  1912. uint32_t dir1, dir2, cov, idxs[2];
  1913. int off;
  1914. n1 = ref_nodev(g->nodes, e->node1);
  1915. n2 = ref_nodev(g->nodes, e->node2);
  1916. r1 = ref_regv(g->regs, n1->regs.idx);
  1917. r2 = ref_regv(g->regs, n2->regs.idx);
  1918. cov = e->cov;
  1919. clear_edgeoffv(offs);
  1920. while(1){
  1921. if(r1->rid > r2->rid){
  1922. r2 ++;
  1923. if(r2->node != e->node2) break;
  1924. } else if(r1->rid < r2->rid){
  1925. r1 ++;
  1926. if(r1->node != e->node1) break;
  1927. } else {
  1928. if(r1->beg < r2->beg){
  1929. if(r1->node < r2->node){
  1930. dir1 = r1->dir;
  1931. dir2 = r2->dir;
  1932. } else {
  1933. dir1 = !r2->dir;
  1934. dir2 = !r1->dir;
  1935. }
  1936. off = r2->beg - r1->end;
  1937. } else {
  1938. if(r2->node < r1->node){
  1939. dir1 = r2->dir;
  1940. dir2 = r1->dir;
  1941. } else {
  1942. dir1 = !r1->dir;
  1943. dir2 = !r2->dir;
  1944. }
  1945. off = ((int)r1->beg) - r2->end;
  1946. }
  1947. if(dir1 == e->dir1 && dir2 == e->dir2){
  1948. push_edgeoffv(offs, (edge_off_t){e - g->edges->buffer, off});
  1949. }
  1950. r1 ++;
  1951. if(r1->node != e->node1) break;
  1952. r2 ++;
  1953. if(r2->node != e->node2) break;
  1954. }
  1955. }
  1956. e->off = estimate_edge_length(offs->buffer, offs->size, idxs);
  1957. e->cov = idxs[1] - idxs[0];
  1958. if(cov != e->cov) return 1;
  1959. else return 0;
  1960. }
  1961. static inline void build_edges_graph(Graph *g, int ncpu, FILE *log){
  1962. read_t *rd;
  1963. node_t *n;
  1964. edge_t *E, *e;
  1965. vplist *regs;
  1966. reg_t *r1, *r2;
  1967. edge_ref_t f1, f2;
  1968. edgeoffv *offs;
  1969. uint64_t idx, lst, cnt, x, *u;
  1970. uint32_t rid, i, idxs[2];
  1971. int exists;
  1972. UNUSED(log);
  1973. clear_edgev(g->edges);
  1974. E = next_ref_edgev(g->edges);
  1975. memset(E, 0, sizeof(edge_t));
  1976. E->closed = WT_EDGE_CLOSED_MASK;
  1977. E->cov = 1;
  1978. //E->status = 0;
  1979. clear_edgehash(g->ehash);
  1980. offs = init_edgeoffv(32);
  1981. regs = init_vplist(32);
  1982. for(rid=0;rid<g->kbm->reads->size;rid++){
  1983. rd = ref_readv(g->reads, rid);
  1984. if(rd->regs.cnt < 2) continue;
  1985. clear_vplist(regs);
  1986. idx = rd->regs.idx;
  1987. while(idx){
  1988. r2 = ref_regv(g->regs, idx);
  1989. idx = r2->read_link;
  1990. if(g->nodes->buffer[r2->node].closed) continue;
  1991. if(r2->closed) continue;
  1992. if(!(r2->beg >= rd->clps[0] && r2->end <= rd->clps[1])) continue;
  1993. for(i=regs->size;i>0;i--){
  1994. r1 = (reg_t*)get_vplist(regs, i - 1);
  1995. if(r1->end + g->max_edge_span < r2->beg) break;
  1996. E = ref_edgev(g->edges, 0);
  1997. if(r1->node < r2->node){
  1998. E->node1 = r1->node;
  1999. E->node2 = r2->node;
  2000. E->dir1 = r1->dir;
  2001. E->dir2 = r2->dir;
  2002. } else {
  2003. E->node1 = r2->node;
  2004. E->node2 = r1->node;
  2005. E->dir1 = !r2->dir;
  2006. E->dir2 = !r1->dir;
  2007. }
  2008. E->off = ((int)r2->beg) - r1->end;
  2009. u = prepare_edgehash(g->ehash, 0, &exists);
  2010. if(exists){
  2011. //if(g->edges->buffer[*u].cov < WT_MAX_EDGE_COV) g->edges->buffer[*u].cov ++;
  2012. } else {
  2013. *u = g->edges->size;
  2014. push_edgev(g->edges, *E);
  2015. }
  2016. if(i == regs->size){
  2017. e = ref_edgev(g->edges, *u);
  2018. //if(e->cov < WT_MAX_EDGE_COV) e->cov ++;
  2019. //if(e->cov >= 1){
  2020. e->closed = 0;
  2021. //}
  2022. }
  2023. push_edgeoffv(offs, (edge_off_t){*u, ((int)r2->beg) - r1->end});
  2024. }
  2025. push_vplist(regs, r2);
  2026. }
  2027. }
  2028. free_vplist(regs);
  2029. if(g->edges->size == 0){
  2030. free_edgeoffv(offs);
  2031. return;
  2032. }
  2033. psort_array(offs->buffer, offs->size, edge_off_t, ncpu, num_cmpgtx(a.idx, b.idx, a.off, b.off));
  2034. lst = 0;
  2035. for(idx=1;idx<=offs->size;idx++){
  2036. if(idx < offs->size && offs->buffer[idx].idx == offs->buffer[lst].idx) continue;
  2037. if(1){
  2038. g->edges->buffer[offs->buffer[lst].idx].off = offs->buffer[(lst+idx)/2].off;
  2039. g->edges->buffer[offs->buffer[lst].idx].cov = idx - lst;
  2040. } else {
  2041. g->edges->buffer[offs->buffer[lst].idx].off = estimate_edge_length(offs->buffer + lst, idx - lst, idxs);
  2042. g->edges->buffer[offs->buffer[lst].idx].cov = idxs[1] - idxs[0];
  2043. }
  2044. if(0){
  2045. uint64_t m;
  2046. for(m=lst;m<idx;m++) fprintf(stdout, "%u\t", offs->buffer[m].off);
  2047. fprintf(stdout, "\n");
  2048. for(m=lst+idxs[0];m<lst+idxs[1];m++) fprintf(stdout, "%u\t", offs->buffer[m].off);
  2049. fprintf(stdout, "\n");
  2050. }
  2051. lst = idx;
  2052. }
  2053. free_edgeoffv(offs);
  2054. clear_edgerefv(g->erefs);
  2055. push_edgerefv(g->erefs, EDGE_REF_NULL);
  2056. for(idx=1;idx<g->edges->size;idx++){
  2057. if(g->edges->buffer[idx].cov < g->min_edge_cov){
  2058. if(g->store_low_cov_edge) g->edges->buffer[idx].closed = WT_EDGE_CLOSED_LESS;
  2059. else continue;
  2060. }
  2061. if(g->nodes->buffer[g->edges->buffer[idx].node1].closed || g->nodes->buffer[g->edges->buffer[idx].node2].closed){
  2062. g->edges->buffer[idx].closed = WT_EDGE_CLOSED_HARD;
  2063. } else if(g->edges->buffer[idx].closed == WT_EDGE_CLOSED_MASK){
  2064. g->edges->buffer[idx].closed = WT_EDGE_CLOSED_LESS;
  2065. //} else {
  2066. //g->edges->buffer[idx].closed = WT_EDGE_CLOSED_NULL;
  2067. }
  2068. push_edgerefv(g->erefs, (edge_ref_t){idx, 0, 0});
  2069. push_edgerefv(g->erefs, (edge_ref_t){idx, 1, 0});
  2070. }
  2071. psort_array(g->erefs->buffer + 1, g->erefs->size - 1, edge_ref_t, ncpu, num_cmpgtx(
  2072. (a.flg? ((g->edges->buffer[a.idx].node2 << 1) | !g->edges->buffer[a.idx].dir2) : ((g->edges->buffer[a.idx].node1 << 1) | g->edges->buffer[a.idx].dir1)),
  2073. (b.flg? ((g->edges->buffer[b.idx].node2 << 1) | !g->edges->buffer[b.idx].dir2) : ((g->edges->buffer[b.idx].node1 << 1) | g->edges->buffer[b.idx].dir1)),
  2074. g->edges->buffer[a.idx].off, g->edges->buffer[b.idx].off));
  2075. f1.idx = g->nodes->size; f1.flg = 0;
  2076. cnt = 0;
  2077. for(lst=idx=1;idx<g->erefs->size;idx++){
  2078. if(g->erefs->buffer[idx].flg){
  2079. f2.idx = g->edges->buffer[g->erefs->buffer[idx].idx].node2;
  2080. f2.flg = !g->edges->buffer[g->erefs->buffer[idx].idx].dir2;
  2081. } else {
  2082. f2.idx = g->edges->buffer[g->erefs->buffer[idx].idx].node1;
  2083. f2.flg = g->edges->buffer[g->erefs->buffer[idx].idx].dir1;
  2084. }
  2085. if(f1.idx == f2.idx && f1.flg == f2.flg) continue;
  2086. if(lst < idx){
  2087. n = ref_nodev(g->nodes, f1.idx);
  2088. n->edges[f1.flg].idx = lst;
  2089. n->edges[f1.flg].cnt = 0;
  2090. for(x=lst;x+1<idx;x++){
  2091. g->erefs->buffer[x].next = x + 1;
  2092. if(g->edges->buffer[g->erefs->buffer[x].idx].closed == WT_EDGE_CLOSED_NULL) n->edges[f1.flg].cnt ++;
  2093. }
  2094. if(g->edges->buffer[g->erefs->buffer[x].idx].closed == WT_EDGE_CLOSED_NULL) n->edges[f1.flg].cnt ++;
  2095. }
  2096. lst = idx;
  2097. f1 = f2;
  2098. }
  2099. if(lst < idx){
  2100. n = ref_nodev(g->nodes, f1.idx);
  2101. n->edges[f1.flg].idx = lst;
  2102. n->edges[f1.flg].cnt = 0;
  2103. for(x=lst;x+1<idx;x++){
  2104. g->erefs->buffer[x].next = x + 1;
  2105. if(g->edges->buffer[g->erefs->buffer[x].idx].closed == WT_EDGE_CLOSED_NULL) n->edges[f1.flg].cnt ++;
  2106. }
  2107. if(g->edges->buffer[g->erefs->buffer[x].idx].closed == WT_EDGE_CLOSED_NULL) n->edges[f1.flg].cnt ++;
  2108. }
  2109. }
  2110. static inline void load_nodes_graph(Graph *g, FileReader *clp, FileReader *nds){
  2111. read_t *rd;
  2112. node_t *n;
  2113. reg_t *reg, *r;
  2114. uint64_t nid;
  2115. uint32_t i, nreg, rid;
  2116. char *str, *tok;
  2117. int ncol, closed, nwarn;
  2118. clear_nodev(g->nodes);
  2119. clear_regv(g->regs);
  2120. next_ref_regv(g->regs);
  2121. nwarn = 0;
  2122. while((ncol = readtable_filereader(clp)) != -1){
  2123. if(clp->line->string[0] == '#') continue;
  2124. if(ncol < 4) continue;
  2125. if((rid = getval_cuhash(g->kbm->tag2idx, get_col_str(clp, 0))) == MAX_U4){
  2126. if(nwarn < 10){
  2127. fprintf(stderr, " -- Cannot find read \"%s\" in LINE:%llu in %s -- %s:%d --\n", get_col_str(clp, 0), clp->n_line, __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  2128. nwarn ++;
  2129. }
  2130. continue;
  2131. }
  2132. rd = ref_readv(g->reads, rid);
  2133. rd->clps[0] = atoi(get_col_str(clp, 2)) / KBM_BIN_SIZE;
  2134. rd->clps[1] = atoi(get_col_str(clp, 3)) / KBM_BIN_SIZE;
  2135. }
  2136. nwarn = 0;
  2137. while((ncol = readtable_filereader(nds)) != -1){
  2138. if(nds->line->string[0] == '#') continue;
  2139. if(ncol < 2) continue;
  2140. nreg = atoi(get_col_str(nds, 1));
  2141. //if(nreg == 0) continue;
  2142. //if(nreg < g->min_node_cov) continue;
  2143. //node_id ~ N(\d+)(\*?)
  2144. nid = atol(get_col_str(nds, 0) + 1);
  2145. if(get_col_str(nds, 0)[get_col_len(nds, 0) - 1] == '*'){ // assert get_col_len(nds, 0) > 0
  2146. closed = 1;
  2147. } else {
  2148. closed = 0;
  2149. }
  2150. while(g->nodes->size < nid){
  2151. n = next_ref_nodev(g->nodes);
  2152. ZEROS(n);
  2153. n->closed = 1;
  2154. }
  2155. n = next_ref_nodev(g->nodes);
  2156. ZEROS(n);
  2157. n->closed = closed;
  2158. n->regs.idx = g->regs->size;
  2159. n->regs.cnt = nreg;
  2160. for(i=0;i<nreg;i++){
  2161. reg = next_ref_regv(g->regs);
  2162. reg->node = nid;
  2163. reg->closed = 0;
  2164. reg->read_link = 0;
  2165. str = get_col_str(nds, 2 + i);
  2166. if(0){
  2167. if(str[get_col_len(nds, 2 + i)-1] == '*'){
  2168. reg->closed = 1;
  2169. }
  2170. tok = index(str, '_'); *tok = '\0';
  2171. reg->rid = getval_cuhash(g->kbm->tag2idx, str);
  2172. str = tok + 1; *tok = '_'; tok = index(str, '_'); *tok = '\0';
  2173. reg->dir = (str[0] == 'R');
  2174. str = tok + 1; *tok = '_'; tok = index(str, '_'); *tok = '\0';
  2175. reg->beg = atoi(str);
  2176. str = tok + 1;
  2177. reg->end = atoi(str);
  2178. reg->end += reg->beg;
  2179. } else {
  2180. if(str[get_col_len(nds, 2 + i)-1] == '*'){
  2181. reg->closed = 1;
  2182. }
  2183. tok = rindex(str, '_'); *tok = '\0'; tok ++;
  2184. reg->end = atoi(tok);
  2185. tok = rindex(str, '_'); *tok = '\0'; tok ++;
  2186. reg->beg = atoi(tok);
  2187. reg->end += reg->beg;
  2188. tok = rindex(str, '_'); *tok = '\0'; tok ++;
  2189. reg->dir = (tok[0] == 'R');
  2190. if((reg->rid = getval_cuhash(g->kbm->tag2idx, str)) == MAX_U4){
  2191. g->regs->size --;
  2192. if(nwarn < 10){
  2193. fprintf(stderr, " -- Cannot find read \"%s\" in LINE:%llu in %s -- %s:%d --\n", str, nds->n_line, __FUNCTION__, __FILE__, __LINE__); fflush(stderr);
  2194. nwarn ++;
  2195. }
  2196. continue;
  2197. }
  2198. }
  2199. rd = ref_readv(g->reads, reg->rid);
  2200. if(rd->regs.idx){
  2201. r = ref_regv(g->regs, rd->regs.idx);
  2202. if(r->beg > reg->beg){
  2203. reg->read_link = rd->regs.idx;
  2204. rd->regs.idx = g->regs->size - 1;
  2205. } else {
  2206. while(1){
  2207. if(r->read_link == 0) break;
  2208. if(g->regs->buffer[r->read_link].beg > reg->beg) break;
  2209. r = ref_regv(g->regs, r->read_link);
  2210. }
  2211. reg->read_link = r->read_link;
  2212. r->read_link = g->regs->size - 1;
  2213. }
  2214. } else {
  2215. rd->regs.idx = g->regs->size - 1;
  2216. }
  2217. rd->regs.cnt ++;
  2218. }
  2219. }
  2220. encap_regv(g->regs, 1);
  2221. g->regs->buffer[g->regs->size].node = WT_MAX_NODE;
  2222. }
  2223. #endif