File: OutputList.hpp

package info (click to toggle)
pbseqlib 5.3.5%2Bdfsg-4
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 7,020 kB
  • sloc: cpp: 77,250; python: 331; sh: 103; makefile: 41
file content (59 lines) | stat: -rw-r--r-- 1,373 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
#ifndef _SIMULATOR_OUTPUT_LIST_HPP_
#define _SIMULATOR_OUTPUT_LIST_HPP_

#include <string>
#include <vector>

#include <alignment/simulator/CDFMap.hpp>

class Output
{
public:
    std::string output;
    int count;
};

class OutputList
{
public:
    int totalCount;
    std::vector<Output> outputs;
    CDFMap<std::vector<Output>::iterator> cdf;

    OutputList() { totalCount = 0; }

    int AddOutput(std::string str, int count)
    {
        outputs.resize(outputs.size() + 1);
        outputs[outputs.size() - 1].output = str;
        outputs[outputs.size() - 1].count = count;
        totalCount += count;
    }

    void StoreCumulativeCounts()
    {
        int outputIndex;
        int total = 0;
        for (outputIndex = 0; outputIndex < outputs.size(); outputIndex++) {
            total += outputs[outputIndex].count;
            cdf.cdf.push_back(total);
            cdf.data.push_back(outputs.begin() + outputIndex);
        }
        assert(total == totalCount);
    }

    int ReturnUniformRandomContextIt(std::vector<Output>::iterator &outputIt)
    {
        return cdf.SelectRandomValue(outputIt);
    }

    int SelectRandomContect(std::string &outputContext)
    {
        std::vector<Output>::iterator outIt;
        ReturnUniformRandomContextIt(outIt);
        outputContext = outIt->output;
        return outIt - outputs.begin();
    }
};

#endif