-
Notifications
You must be signed in to change notification settings - Fork 3
/
minigraph.h
183 lines (158 loc) · 4.6 KB
/
minigraph.h
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
#ifndef MINIGRAPH_H
#define MINIGRAPH_H
#include <stdint.h>
#include "gfa.h"
#define MG_VERSION "0.20-r559"
#define MC_VERSION "1.3"
#define MG_M_SPLICE 0x10
#define MG_M_SR 0x20
#define MG_M_FRAG_MODE 0x40
#define MG_M_FRAG_MERGE 0x80
#define MG_M_FOR_ONLY 0x100
#define MG_M_REV_ONLY 0x200
#define MG_M_HEAP_SORT 0x400
#define MG_M_VERTEX_COOR 0x800
#define MG_M_ALL_CHAINS 0x1000
#define MG_M_PRINT_2ND 0x2000
#define MG_M_CAL_COV 0x4000
#define MG_M_RMQ 0x8000
#define MG_M_COPY_COMMENT 0x10000
#define MG_M_INDEPEND_SEG 0x20000
#define MG_M_NO_QUAL 0x40000
#define MG_M_2_IO_THREADS 0x80000
#define MG_M_SHOW_UNMAP 0x100000
#define MG_M_NO_COMP_PATH 0x200000
#define MG_M_NO_DIAG 0x400000
#define MG_M_WRITE_LCHAIN 0x800000
#define MG_M_WRITE_MZ 0x1000000
#define MG_M_SKIP_GCHECK 0x2000000
#define MG_M_CIGAR 0x4000000
#define MG_G_NONE 0
#define MG_G_GGSIMPLE 1
#define MG_G_NO_QOVLP 0x1
#define MG_G_CAL_COV 0x2
#define MG_G_NO_INV 0x4
#define MG_G_CALL 0x8
typedef struct { uint64_t x, y; } mg128_t;
typedef struct { size_t n, m; mg128_t *a; } mg128_v;
typedef struct { int32_t n, m; uint32_t *a; } mg32_v;
typedef struct { int32_t n, m; uint64_t *a; } mg64_v;
typedef struct {
int w, k;
int bucket_bits;
} mg_idxopt_t;
typedef struct {
uint64_t flag;
int64_t mini_batch_size;
int seed;
int max_qlen;
int pe_ori;
int occ_max1, occ_max1_cap;
float occ_max1_frac;
int bw, bw_long;
int rmq_size_cap;
int rmq_rescue_size;
float rmq_rescue_ratio;
int max_gap_pre, max_gap, max_gap_ref, max_frag_len;
float div;
float chn_pen_gap, chn_pen_skip;
float l_chn_pen_gap, l_chn_pen_skip;
float tau_1, tau_2;
int max_lc_skip, max_lc_iter, max_gc_skip;
int min_lc_cnt, min_lc_score;
int min_gc_cnt, min_gc_score;
int gdp_max_ed, lc_max_trim, lc_max_occ;
float mask_level;
int sub_diff;
int best_n;
float pri_ratio;
int ref_bonus;
int64_t cap_kalloc;
int min_cov_mapq, min_cov_blen;
bool is_ggen;
bool is_hap;
int G;
int max_itr;
} mg_mapopt_t;
typedef struct {
uint64_t flag;
int algo;
int min_mapq;
int min_map_len, min_depth_len;
int min_var_len, match_pen;
// parameters specific to ggsimple/ggs
int ggs_shrink_pen;
int ggs_min_end_cnt;
float ggs_min_end_frac;
// scoring for SW check
float ggs_max_iden, ggs_min_inv_iden;
} mg_ggopt_t;
typedef struct {
const gfa_t *g;
gfa_edseq_t *es;
int32_t b, w, k, flag, n_seg;
struct mg_idx_bucket_s *B; // index (hidden)
} mg_idx_t;
typedef struct {
int32_t off, cnt:31, inner_pre:1;
uint32_t v;
int32_t rs, re, qs, qe;
int32_t score, dist_pre;
uint32_t hash_pre;
} mg_lchain_t;
typedef struct {
int32_t off, cnt;
uint32_t v;
int32_t score;
int32_t ed;
} mg_llchain_t;
typedef struct {
int32_t n_cigar, mlen, blen, aplen, ss, ee; // ss: start on the start vertex; ee: end on the end vertex
uint64_t cigar[];
} mg_cigar_t;
typedef struct {
int32_t id, parent;
int32_t off, cnt;
int32_t n_anchor, score;
int32_t qs, qe;
int32_t plen, ps, pe;
int32_t blen, mlen;
float div;
uint32_t hash;
int32_t subsc, n_sub;
uint32_t mapq:8, flt:1, dummy:23;
mg_cigar_t *p;
} mg_gchain_t;
typedef struct {
void *km;
int32_t n_gc, n_lc, n_a, rep_len;
mg_gchain_t *gc;
mg_llchain_t *lc;
mg128_t *a; // minimizer positions; see comments above mg_update_anchors() for details
} mg_gchains_t;
typedef struct mg_tbuf_s mg_tbuf_t;
extern int mg_verbose, mg_dbg_flag;
extern double mg_realtime0;
#ifdef __cplusplus
extern "C" {
#endif
// options
int mg_opt_set(const char *preset, mg_idxopt_t *io, mg_mapopt_t *mo, mg_ggopt_t *go);
int mg_opt_check(const mg_idxopt_t *io, const mg_mapopt_t *mo, const mg_ggopt_t *go);
void mg_opt_update(const mg_idx_t *gi, mg_mapopt_t *mo, mg_ggopt_t *go);
// index operations
mg_idx_t *mg_index(gfa_t *g, const mg_idxopt_t *io, int n_threads, mg_mapopt_t *mo); // combine mg_index_core() and mg_opt_update()
void mg_idx_destroy(mg_idx_t *gi);
// mapping
mg_tbuf_t *mg_tbuf_init(void);
void mg_tbuf_destroy(mg_tbuf_t *b);
mg_gchains_t *mg_map(const mg_idx_t *gi, int qlen, const char *seq, mg_tbuf_t *b, const mg_mapopt_t *opt, const char *qname);
void mg_map_frag(const mg_idx_t *gi, int n_segs, const int *qlens, const char **seqs, mg_gchains_t **gcs, mg_tbuf_t *b, const mg_mapopt_t *opt, const char *qname);
// high-level mapping APIs
int mg_map_files(gfa_t *g, int n_fn, const char **fn, const mg_idxopt_t *ipt, const mg_mapopt_t *opt0, int n_threads);
// graph generation
int mg_ggen(gfa_t *g, int32_t n_fn, const char **fn, const mg_idxopt_t *ipt, const mg_mapopt_t *opt0, const mg_ggopt_t *go, int n_threads);
#ifdef __cplusplus
}
#endif
#endif