-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathbntseq.c
514 lines (473 loc) · 16 KB
/
bntseq.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <zlib.h>
#include <unistd.h>
#include <errno.h>
#include "bntseq.h"
#include "utils.h"
#include "bwt.h"
#include "kseq.h"
KSEQ_DECLARE(gzFile)
#include "khash.h"
KHASH_MAP_INIT_STR(str, int)
#ifdef USE_MALLOC_WRAPPERS
# include "malloc_wrap.h"
#endif
unsigned char nst_nt4_table[256] = {
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 5 /*'-'*/, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 0, 4, 1, 4, 4, 4, 2, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 3, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 0, 4, 1, 4, 4, 4, 2, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 3, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4
};
unsigned char aa_encode_hash[256] = {
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x00
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x08
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x10
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x18
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x20
0x15, 0x15, 0x14, 0x15, 0x15, 0x16, 0x15, 0x15, // 0x28 *-
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x30
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x38
0x15, 0x00, 0x02, 0x01, 0x02, 0x03, 0x04, 0x05, // 0x40 A(B)CDEFG
0x06, 0x07, 0x15, 0x08, 0x09, 0x0A, 0x0B, 0x15, // 0x48 HI[J]KLMN[O]
0x0C, 0x0D, 0x0E, 0x0F, 0x10, 0x15, 0x11, 0x12, // 0x50 PQRST[U]VW
0x15, 0x13, 0x0D, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x58 [X]Y(Z)
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x60
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x68
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x70
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x78
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x80
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x88
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x90
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0x98
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xA0
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xA8
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xB0
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xB8
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xC0
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xC8
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xD0
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xD8
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xE0
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xE8
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xF0
0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, 0x15, // 0xF8
};
unsigned char aa_ascii_hash[256] = {
'A', 'C', 'D', 'E', 'F', 'G', 'H', 'I', // 0x00
'K', 'L', 'M', 'N', 'P', 'Q', 'R', 'S', // 0x08
'T', 'V', 'W', 'Y', 'X', 'X', 'X', 'X', // 0x10
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x18
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x20
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x28 -
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x30
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x38
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x40
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x48
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x50
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x58
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x60
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x68
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x70
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x78
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x80
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x88
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x90
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0x98
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xA0
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xA8
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xB0
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xB8
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xC0
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xC8
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xD0
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xD8
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xE0
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xE8
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xF0
'X', 'X', 'X', 'X', 'X', 'X', 'X', 'X', // 0xF8
};
void bns_dump(const bntseq_t *bns, const char *prefix)
{
char str[1024];
FILE *fp;
int i;
{ // dump .ann
strcpy(str, prefix); strcat(str, ".ann");
fp = xopen(str, "w");
err_fprintf(fp, "%lld %d %u\n", (long long)bns->l_pac, bns->n_seqs, bns->seed);
for (i = 0; i != bns->n_seqs; ++i) {
bntann1_t *p = bns->anns + i;
err_fprintf(fp, "%d %s", p->gi, p->name);
if (p->anno[0]) err_fprintf(fp, " %s\n", p->anno);
else err_fprintf(fp, "\n");
err_fprintf(fp, "%lld %d %d\n", (long long)p->offset, p->len, p->n_ambs);
}
err_fflush(fp);
err_fclose(fp);
}
{ // dump .amb
strcpy(str, prefix); strcat(str, ".amb");
fp = xopen(str, "w");
err_fprintf(fp, "%lld %d %u\n", (long long)bns->l_pac, bns->n_seqs, bns->n_holes);
for (i = 0; i != bns->n_holes; ++i) {
bntamb1_t *p = bns->ambs + i;
err_fprintf(fp, "%lld %d %c\n", (long long)p->offset, p->len, p->amb);
}
err_fflush(fp);
err_fclose(fp);
}
}
bntseq_t *bns_restore_core(const char *ann_filename, const char* amb_filename, const char* pac_filename)
{
char str[8192];
FILE *fp;
const char *fname;
bntseq_t *bns;
long long xx;
int i;
int scanres;
bns = (bntseq_t*)calloc(1, sizeof(bntseq_t));
{ // read .ann
fp = xopen(fname = ann_filename, "r");
scanres = fscanf(fp, "%lld%d%u", &xx, &bns->n_seqs, &bns->seed);
if (scanres != 3) goto badread;
bns->l_pac = xx;
bns->anns = (bntann1_t*)calloc(bns->n_seqs, sizeof(bntann1_t));
for (i = 0; i < bns->n_seqs; ++i) {
bntann1_t *p = bns->anns + i;
char *q = str;
int c;
// read gi and sequence name
scanres = fscanf(fp, "%u%s", &p->gi, str);
if (scanres != 2) goto badread;
p->name = strdup(str);
// read fasta comments
while (q - str < sizeof(str) - 1 && (c = fgetc(fp)) != '\n' && c != EOF) *q++ = c;
while (c != '\n' && c != EOF) c = fgetc(fp);
if (c == EOF) {
scanres = EOF;
goto badread;
}
*q = 0;
if (q - str > 1 && strcmp(str, " (null)") != 0) p->anno = strdup(str + 1); // skip leading space
else p->anno = strdup("");
// read the rest
scanres = fscanf(fp, "%lld%d%d", &xx, &p->len, &p->n_ambs);
if (scanres != 3) goto badread;
p->offset = xx;
}
err_fclose(fp);
}
{ // read .amb
int64_t l_pac;
int32_t n_seqs;
fp = xopen(fname = amb_filename, "r");
scanres = fscanf(fp, "%lld%d%d", &xx, &n_seqs, &bns->n_holes);
if (scanres != 3) goto badread;
l_pac = xx;
xassert(l_pac == bns->l_pac && n_seqs == bns->n_seqs, "inconsistent .ann and .amb files.");
bns->ambs = bns->n_holes? (bntamb1_t*)calloc(bns->n_holes, sizeof(bntamb1_t)) : 0;
for (i = 0; i < bns->n_holes; ++i) {
bntamb1_t *p = bns->ambs + i;
scanres = fscanf(fp, "%lld%d%s", &xx, &p->len, str);
if (scanres != 3) goto badread;
p->offset = xx;
p->amb = str[0];
}
err_fclose(fp);
}
{ // open .pac
bns->fp_pac = xopen(pac_filename, "rb");
}
return bns;
badread:
if (EOF == scanres) {
err_fatal(__func__, "Error reading %s : %s\n", fname, ferror(fp) ? strerror(errno) : "Unexpected end of file");
}
err_fatal(__func__, "Parse error reading %s\n", fname);
}
bntseq_t *bns_restore(const char *prefix)
{
char ann_filename[1024], amb_filename[1024], pac_filename[1024], alt_filename[1024];
FILE *fp;
bntseq_t *bns;
strcat(strcpy(ann_filename, prefix), ".ann");
strcat(strcpy(amb_filename, prefix), ".amb");
strcat(strcpy(pac_filename, prefix), ".pac");
bns = bns_restore_core(ann_filename, amb_filename, pac_filename);
if (bns == 0) return 0;
if ((fp = fopen(strcat(strcpy(alt_filename, prefix), ".alt"), "r")) != 0) { // read .alt file if present
char str[1024];
khash_t(str) *h;
int c, i, absent;
khint_t k;
h = kh_init(str);
for (i = 0; i < bns->n_seqs; ++i) {
k = kh_put(str, h, bns->anns[i].name, &absent);
kh_val(h, k) = i;
}
i = 0;
while ((c = fgetc(fp)) != EOF) {
if (c == '\t' || c == '\n' || c == '\r') {
str[i] = 0;
if (str[0] != '@') {
k = kh_get(str, h, str);
if (k != kh_end(h))
bns->anns[kh_val(h, k)].is_alt = 1;
}
while (c != '\n' && c != EOF) c = fgetc(fp);
i = 0;
} else str[i++] = c; // FIXME: potential segfault here
}
kh_destroy(str, h);
fclose(fp);
}
return bns;
}
void bns_destroy(bntseq_t *bns)
{
if (bns == 0) return;
else {
int i;
if (bns->fp_pac) err_fclose(bns->fp_pac);
free(bns->ambs);
for (i = 0; i < bns->n_seqs; ++i) {
free(bns->anns[i].name);
free(bns->anns[i].anno);
}
free(bns->anns);
free(bns);
}
}
//#define _set_pac(pac, l, c) ((pac)[(l)>>2] |= (c)<<((~(l)&3)<<1))
//#define _get_pac(pac, l) ((pac)[(l)>>2]>>((~(l)&3)<<1)&3)
uint8_t *add1(const kseq_t *seq, bntseq_t *bns, uint8_t *pac, int64_t *m_pac, int *m_seqs, int *m_holes, bntamb1_t **q) {
bntann1_t *p;
int i, lasts;
if (bns->n_seqs == *m_seqs) {
*m_seqs <<= 1;
bns->anns = (bntann1_t*)realloc(bns->anns, *m_seqs * sizeof(bntann1_t));
}
p = bns->anns + bns->n_seqs;
p->name = strdup((char*)seq->name.s);
p->anno = seq->comment.l > 0? strdup((char*)seq->comment.s) : strdup("(null)");
p->gi = 0; p->len = seq->seq.l;
p->offset = (bns->n_seqs == 0)? 0 : (p-1)->offset + (p-1)->len;
p->n_ambs = 0;
// Iterate through each AA in the sequence
for (i = lasts = 0; i < seq->seq.l; ++i) {
// Hash encoded value
int c = (int) aa_encode_hash[(int)(seq->seq.s[i])];
// Process unknowns
if (c >= VALUE_DEFINED) {
// Contiguous unknowns
if (lasts == seq->seq.s[i]) {
++(*q)->len;
}
else {
if (bns->n_holes == *m_holes) {
(*m_holes) <<= 1;
bns->ambs = (bntamb1_t*)realloc(bns->ambs, (*m_holes) * sizeof(bntamb1_t));
}
*q = bns->ambs + bns->n_holes;
(*q)->len = 1;
(*q)->offset = p->offset + i;
(*q)->amb = seq->seq.s[i];
++p->n_ambs;
++bns->n_holes;
}
}
// Store value to check for contiguous runs
lasts = seq->seq.s[i];
// Expand buffer if necessary
if (bns->l_pac == *m_pac) { // double the pac size
*m_pac <<= 1;
pac = realloc(pac, *m_pac);
memset(pac + bns->l_pac, 0, (*m_pac - bns->l_pac));
}
// Pack and set value
if (c >= VALUE_DEFINED) c = lrand48() % VALUE_DEFINED;
pac[bns->l_pac] = c;
++bns->l_pac;
}
++bns->n_seqs;
return pac;
}
int64_t bns_fasta2bntseq(gzFile fp_fa, const char * prefix, int for_only) {
kseq_t *seq;
char name[1024];
bntseq_t *bns;
uint8_t *pac = 0;
int32_t m_seqs, m_holes;
int64_t ret = -1, m_pac, l;
bntamb1_t *q;
FILE *fp;
// Initialization of sequence related structures
seq = kseq_init(fp_fa);
bns = (bntseq_t*)calloc(1, sizeof(bntseq_t));
bns->seed = 11;
srand48(bns->seed);
m_seqs = m_holes = 8; m_pac = 0x10000;
bns->anns = (bntann1_t*)calloc(m_seqs, sizeof(bntann1_t));
bns->ambs = (bntamb1_t*)calloc(m_holes, sizeof(bntamb1_t));
q = bns->ambs;
// Allocate packed buffer to size of sequence (for now - to be converted to 40-bit word)
pac = calloc(m_pac, 1);
// Open packed file for writing
strcpy(name, prefix); strcat(name, ".pac");
fp = xopen(name, "wb");
// Read sequences into pac buffer
while (kseq_read(seq) >= 0) pac = add1(seq, bns, pac, &m_pac, &m_seqs, &m_holes, &q);
// Add reverse complement if not forward only
if (!for_only) {
m_pac = bns->l_pac * 2;
pac = realloc(pac, m_pac);
memset(pac + bns->l_pac, 0, m_pac - bns->l_pac);
for (l = bns->l_pac - 1; l >= 0; --l, ++bns->l_pac) {
pac[bns->l_pac] = VALUE_DEFINED - 1 - pac[l];
}
}
ret = bns->l_pac;
// Write pac file
ubyte_t ct;
err_fwrite(pac, 1, bns->l_pac, fp);
// Pad to nearest 40-bit word
ct = 0;
err_fwrite(&ct, 1, 1, fp);
ct = 0;
err_fwrite(&ct, 1, 1, fp);
// Close pac file
err_fflush(fp);
err_fclose(fp);
bns_dump(bns, prefix);
bns_destroy(bns);
kseq_destroy(seq);
free(pac);
return ret;}
int bwa_fa2pac(int argc, char *argv[])
{
int c, for_only = 0;
gzFile fp;
while ((c = getopt(argc, argv, "f")) >= 0) {
switch (c) {
case 'f': for_only = 1; break;
}
}
if (argc == optind) {
fprintf(stderr, "Usage: bwa fa2pac [-f] <in.fasta> [<out.prefix>]\n");
return 1;
}
fp = xzopen(argv[optind], "r");
bns_fasta2bntseq(fp, (optind+1 < argc)? argv[optind+1] : argv[optind], for_only);
err_gzclose(fp);
return 0;
}
int bns_pos2rid(const bntseq_t *bns, int64_t pos_f)
{
int left, mid, right;
if (pos_f >= bns->l_pac) return -1;
left = 0; mid = 0; right = bns->n_seqs;
while (left < right) { // binary search
mid = (left + right) >> 1;
if (pos_f >= bns->anns[mid].offset) {
if (mid == bns->n_seqs - 1) break;
if (pos_f < bns->anns[mid+1].offset) break; // bracketed
left = mid + 1;
} else right = mid;
}
return mid;
}
int bns_intv2rid(const bntseq_t *bns, int64_t rb, int64_t re)
{
int is_rev, rid_b, rid_e;
if (rb < bns->l_pac && re > bns->l_pac) return -2;
assert(rb <= re);
rid_b = bns_pos2rid(bns, bns_depos(bns, rb, &is_rev));
rid_e = rb < re? bns_pos2rid(bns, bns_depos(bns, re - 1, &is_rev)) : rid_b;
return rid_b == rid_e? rid_b : -1;
}
int bns_cnt_ambi(const bntseq_t *bns, int64_t pos_f, int len, int *ref_id)
{
int left, mid, right, nn;
if (ref_id) *ref_id = bns_pos2rid(bns, pos_f);
left = 0; right = bns->n_holes; nn = 0;
while (left < right) {
mid = (left + right) >> 1;
if (pos_f >= bns->ambs[mid].offset + bns->ambs[mid].len) left = mid + 1;
else if (pos_f + len <= bns->ambs[mid].offset) right = mid;
else { // overlap
if (pos_f >= bns->ambs[mid].offset) {
nn += bns->ambs[mid].offset + bns->ambs[mid].len < pos_f + len?
bns->ambs[mid].offset + bns->ambs[mid].len - pos_f : len;
} else {
nn += bns->ambs[mid].offset + bns->ambs[mid].len < pos_f + len?
bns->ambs[mid].len : len - (bns->ambs[mid].offset - pos_f);
}
break;
}
}
return nn;
}
uint8_t *bns_get_seq(int64_t l_pac, const uint8_t *pac, int64_t beg, int64_t end, int64_t *len)
{
uint8_t *seq = 0;
if (end < beg) end ^= beg, beg ^= end, end ^= beg; // if end is smaller, swap
if (end > l_pac<<1) end = l_pac<<1;
if (beg < 0) beg = 0;
if (beg >= l_pac || end <= l_pac) {
int64_t k, l = 0;
*len = end - beg;
seq = malloc(end - beg);
if (beg >= l_pac) { // reverse strand
int64_t beg_f = (l_pac<<1) - 1 - end;
int64_t end_f = (l_pac<<1) - 1 - beg;
for (k = end_f; k > beg_f; --k)
seq[l++] = VALUE_DEFINED - 1 - pac[k];
//seq[l++] = 3 - _get_pac(pac, k);
} else { // forward strand
for (k = beg; k < end; ++k)
seq[l++] = pac[k];
//seq[l++] = _get_pac(pac, k);
}
} else *len = 0; // if bridging the forward-reverse boundary, return nothing
return seq;
}
uint8_t *bns_fetch_seq(const bntseq_t *bns, const uint8_t *pac, int64_t *beg, int64_t mid, int64_t *end, int *rid)
{
int64_t far_beg, far_end, len;
int is_rev;
uint8_t *seq;
if (*end < *beg) *end ^= *beg, *beg ^= *end, *end ^= *beg; // if end is smaller, swap
assert(*beg <= mid && mid < *end);
*rid = bns_pos2rid(bns, bns_depos(bns, mid, &is_rev));
far_beg = bns->anns[*rid].offset;
far_end = far_beg + bns->anns[*rid].len;
if (is_rev) { // flip to the reverse strand
int64_t tmp = far_beg;
far_beg = (bns->l_pac<<1) - far_end;
far_end = (bns->l_pac<<1) - tmp;
}
*beg = *beg > far_beg? *beg : far_beg;
*end = *end < far_end? *end : far_end;
seq = bns_get_seq(bns->l_pac, pac, *beg, *end, &len);
if (seq == 0 || *end - *beg != len) {
fprintf(stderr, "[E::%s] begin=%ld, mid=%ld, end=%ld, len=%ld, seq=%p, rid=%d, far_beg=%ld, far_end=%ld\n",
__func__, (long)*beg, (long)mid, (long)*end, (long)len, seq, *rid, (long)far_beg, (long)far_end);
}
assert(seq && *end - *beg == len); // assertion failure should never happen
return seq;
}