forked from BenLangmead/bowtie2
-
Notifications
You must be signed in to change notification settings - Fork 0
/
alphabet.h
127 lines (111 loc) · 3.68 KB
/
alphabet.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
/*
* Copyright 2011, Ben Langmead <[email protected]>
*
* This file is part of Bowtie 2.
*
* Bowtie 2 is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Bowtie 2 is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Bowtie 2. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef ALPHABETS_H_
#define ALPHABETS_H_
#include <stdexcept>
#include <string>
#include <sstream>
#include <stdint.h>
#include "assert_helpers.h"
using namespace std;
/// Convert an ascii char to a DNA category. Categories are:
/// 0 -> invalid
/// 1 -> unambiguous a, c, g or t
/// 2 -> ambiguous
/// 3 -> unmatchable
extern uint8_t asc2dnacat[];
/// Convert masks to ambiguous nucleotides
extern char mask2dna[];
/// Convert ambiguous ASCII nuceleotide to mask
extern uint8_t asc2dnamask[];
/// Convert mask to # of alternative in the mask
extern int mask2popcnt[];
/// Convert an ascii char to a 2-bit base: 0=A, 1=C, 2=G, 3=T, 4=N
extern uint8_t asc2dna[];
/// Convert an ascii char to a 2-bit base: 0=A, 1=C, 2=G, 3=T, 4=N
extern uint8_t asc2col[];
/// Convert a pair of 2-bit (and 4=N) encoded DNA bases to a color
extern uint8_t dinuc2color[5][5];
/// Convert a 4-bit mask into an IUPAC code
extern signed char mask2iupac[16];
/// Convert bit encoded DNA char to its complement
extern int dnacomp[5];
/// String of all DNA and IUPAC characters
extern const char *iupacs;
/// Map from masks to their reverse-complement masks
extern int maskcomp[16];
/**
* Return true iff c is an unambiguous DNA character.
*/
static inline bool isUnambigNuc(char c) {
return asc2dnacat[(int)c] == 1;
}
/**
* Return the DNA complement of the given ASCII char.
*/
static inline char comp(char c) {
switch(c) {
case 'a': return 't';
case 'A': return 'T';
case 'c': return 'g';
case 'C': return 'G';
case 'g': return 'c';
case 'G': return 'C';
case 't': return 'a';
case 'T': return 'A';
default: return c;
}
}
/**
* Return the reverse complement of a bit-encoded nucleotide.
*/
static inline int compDna(int c) {
assert_leq(c, 4);
return dnacomp[c];
}
/// Convert a pair of 2-bit (and 4=N) encoded DNA bases to a color
extern uint8_t dinuc2color[5][5];
/**
* Decode a not-necessarily-ambiguous nucleotide.
*/
static inline void decodeNuc(char c , int& num, int *alts) {
switch(c) {
case 'A': alts[0] = 0; num = 1; break;
case 'C': alts[0] = 1; num = 1; break;
case 'G': alts[0] = 2; num = 1; break;
case 'T': alts[0] = 3; num = 1; break;
case 'M': alts[0] = 0; alts[1] = 1; num = 2; break;
case 'R': alts[0] = 0; alts[1] = 2; num = 2; break;
case 'W': alts[0] = 0; alts[1] = 3; num = 2; break;
case 'S': alts[0] = 1; alts[1] = 2; num = 2; break;
case 'Y': alts[0] = 1; alts[1] = 3; num = 2; break;
case 'K': alts[0] = 2; alts[1] = 3; num = 2; break;
case 'V': alts[0] = 0; alts[1] = 1; alts[2] = 2; num = 3; break;
case 'H': alts[0] = 0; alts[1] = 1; alts[2] = 3; num = 3; break;
case 'D': alts[0] = 0; alts[1] = 2; alts[2] = 3; num = 3; break;
case 'B': alts[0] = 1; alts[1] = 2; alts[2] = 3; num = 3; break;
case 'N': alts[0] = 0; alts[1] = 1; alts[2] = 2; alts[3] = 3; num = 4; break;
default: {
std::cerr << "Bad IUPAC code: " << c << ", (int: " << (int)c << ")" << std::endl;
throw std::runtime_error("");
}
}
}
extern void setIupacsCat(uint8_t cat);
#endif /*ALPHABETS_H_*/