File: PoaGraph.hpp

package info (click to toggle)
consensuscore 1.1.1%2Bdfsg-4
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 3,404 kB
  • sloc: cpp: 38,940; python: 2,083; ansic: 542; sh: 184; makefile: 82; cs: 10
file content (79 lines) | stat: -rw-r--r-- 1,915 bytes parent folder | download | duplicates (4)
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
// Author: David Alexander

#pragma once

#include <boost/tuple/tuple.hpp>
#include <climits>
#include <string>
#include <utility>
#include <vector>

#include <ConsensusCore/Mutation.hpp>
#include <ConsensusCore/Types.hpp>

namespace ConsensusCore {
struct PoaConsensus;

namespace detail {
class PoaGraphImpl;
}

class PoaAlignmentMatrix
{
public:
    virtual ~PoaAlignmentMatrix(){};
    virtual float Score() const = 0;
};

/// \brief An object representing a Poa (partial-order alignment) graph
class PoaGraph
{
public:
    typedef size_t Vertex;
    typedef size_t ReadId;

public:  // Flags enums for specifying GraphViz output features
    enum
    {
        COLOR_NODES = 0x1,
        VERBOSE_NODES = 0x2
    };

public:
    PoaGraph();
    PoaGraph(const PoaGraph& other);
    PoaGraph(const detail::PoaGraphImpl& o);  // NB: this performs a copy
    ~PoaGraph();

    //
    // Easy API
    //
    void AddRead(const std::string& sequence, const AlignConfig& config,
                 detail::SdpRangeFinder* rangeFinder = NULL,
                 std::vector<Vertex>* readPathOutput = NULL);

    //
    // API for more control
    //
    void AddFirstRead(const std::string& sequence, std::vector<Vertex>* readPathOutput = NULL);

    PoaAlignmentMatrix* TryAddRead(const std::string& sequence, const AlignConfig& config,
                                   detail::SdpRangeFinder* rangeFinder = NULL) const;

    void CommitAdd(PoaAlignmentMatrix* mat, std::vector<Vertex>* readPathOutput = NULL);

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

    size_t NumReads() const;

    std::string ToGraphViz(int flags = 0, const PoaConsensus* pc = NULL) const;

    void WriteGraphVizFile(std::string filename, int flags = 0,
                           const PoaConsensus* pc = NULL) const;

    const PoaConsensus* FindConsensus(const AlignConfig& config, int minCoverage = -INT_MAX) const;

private:
    detail::PoaGraphImpl* impl;
};
}