File: ErrorMotifs.h

package info (click to toggle)
tvc 5.0.3%2Bgit20151221.80e144e%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 3,548 kB
  • sloc: cpp: 24,088; ansic: 3,933; python: 260; makefile: 16
file content (115 lines) | stat: -rw-r--r-- 2,855 bytes parent folder | download | duplicates (3)
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
/* Copyright (C) 2013 Ion Torrent Systems, Inc. All Rights Reserved */

//! @file     ErrorMotifs.h
//! @ingroup  VariantCaller
//! @brief    Handles motifs with high systematic error


#ifndef ERRORMOTIFS_H
#define ERRORMOTIFS_H

#include <iostream>
#include <fstream>
#include <string>
#include <stdlib.h>
#include <sstream>
#include <cstdlib> 
#include <map>
#include <vector>

using namespace std;

//-----------------------------------------------------

struct TMetaData
{
  unsigned short covered_sse_;
  unsigned short covered_nonsse_;

  TMetaData(unsigned short cov_sse,unsigned short  cov_nonsse) {
    covered_sse_    = cov_sse;
    covered_nonsse_ = cov_nonsse;
  }

  inline bool operator=(const TMetaData & md) {
    covered_sse_ = md.covered_sse_;
    covered_nonsse_ = md.covered_nonsse_;
    return false;
  }

  inline double calculate_probability() const {
    // Divide by zero protection
    double denominator = (double)(covered_sse_ + covered_nonsse_);
    if (denominator > 0)
      return (double)covered_sse_ / denominator;
    else
      return -1.0;
  }
};

//-----------------------------------------------------
class TIonMotif
{

  map <unsigned, TMetaData> motif_list;

  public:
    short left_size;
    short right_size;

    TIonMotif(){ left_size = 0; right_size = 0; };

    inline void add(unsigned hashKey, const TMetaData &mdata) {
      map <unsigned, TMetaData>::iterator itr;
      itr = motif_list.find(hashKey);

      if (itr == motif_list.end())
        motif_list.insert(make_pair(hashKey, mdata));
      else if (itr->second.calculate_probability() > mdata.calculate_probability())
        motif_list.at(hashKey) = mdata;
     };

     inline bool has_hashkey(unsigned hashKey, TMetaData &md) const {
       map <unsigned, TMetaData>::const_iterator itr;
       itr = motif_list.find(hashKey);

       if (itr == motif_list.end())
         return false;
       else {
         md = itr->second;
         return true;
       }
     };

     inline bool isEmpty() const {
       return (motif_list.size()==0);
     };
};

//-----------------------------------------------------
class TIonMotifSet
{
  unsigned MAX_HP_SIZE;
  vector<TIonMotif> motif_table;
  bool isempty;
  void add_my_motif(short key, string motif, const TMetaData &mdata, unsigned pos);

  public:
    TIonMotifSet() {
      MAX_HP_SIZE = 20;
      motif_table.resize(4*(MAX_HP_SIZE+1));
      isempty=true;
    };

    void     load_from_file(const char * fname);
    void     add_motif(string motif, const TMetaData &mdata);
    short    get_key(char hpChar, int hpSize) const;
    unsigned make_hashkey(string motif) const;

    float    get_sse_probability(string context, unsigned err_base_pos) const;

    inline bool isEmpty() { return isempty; };
};
//-----------------------------------------------------

#endif //ERRORMOTIFS_H