aboutsummaryrefslogtreecommitdiff
path: root/kmer_utils.c
blob: e7373464e76874acda2f833e2fda3f8d89c1d5b8 (plain)
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
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

// This function takes a char array containing sequences and converts it
// into a kmer index (essentially a base 4 radix converstion to base 10, with
// some mumbo-jumbo of taking each of the characters we want (ACGT) and turning
// them into a radix-4 number we can use.
//
// kind of convoluted but it works.
//
// Arguemnts: 
//  char *str - a NULL terminated character array 
//  long kmer - how long of a index value you want to return
//  long error_pos - what index to return for a non ACGT character
//
inline long convert_kmer_to_index(const char *str, long kmer, long error_pos) {

  long i = 0;
  long out = 0;
  long multiply = 1;

  for(i = kmer - 1; i >= 0; i--){
    switch(str[i]) {
      case 'a':
      case 'A':
        break;
      case 'c':
      case 'C':
        out += 1 * multiply;
        break;
      case 'g':
      case 'G':
        out += 2 * multiply;
        break;
      case 't':
      case 'T':
        out += 3 * multiply;
        break;
      default:
        return error_pos; 
    }

    multiply = multiply << 2;
  }

  return out;
}