File: PulseFile.cpp

package info (click to toggle)
pbseqlib 0~20161219-1
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 5,924 kB
  • ctags: 5,123
  • sloc: cpp: 82,727; makefile: 305; python: 239; sh: 8
file content (81 lines) | stat: -rw-r--r-- 2,978 bytes parent folder | download | duplicates (2)
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
// Author: Mark Chaisson

#include "PulseFile.hpp"

void PulseFile::CopySignal(HalfWord *signalData, // either a vector or matrix
                           int signalNDims,
                           DSLength pulseStartPos, // 0 if baseToPulseIndex maps to abs position
                           int *baseToPulseIndex,
                           Nucleotide *readSeq,
                           DNALength readLength,
                           HalfWord *readData) {
    // if baseToPulseIndex maps bases to absolute pulse postions
    // pulseStartPos must be 0; 
    // otherwise, pulseStartPos is pulseStartPositions[holeIndex]

    std::map<char, size_t> baseMap = GetBaseMap();
    if (signalNDims == 1) {
        for (DNALength i = 0; i < readLength; i++) {
            readData[i] = signalData[pulseStartPos + baseToPulseIndex[i] ];
        }
    }
    else {
        for (DNALength i = 0; i < readLength; i++) {
            readData[i] = signalData[baseToPulseIndex[i]*4 + baseMap[readSeq[i]]];
        }
    }
}

void PulseFile::CopyReadAt(uint32_t plsReadIndex, int *baseToPulseIndex, SMRTSequence &read) {
    DSLength pulseStartPos = pulseStartPositions[plsReadIndex];
    bool OK = true;
    if (midSignal.size() > 0) {
    assert(midSignal.size() > pulseStartPos);
    OK = Realloc(read.midSignal, read.length);
    CopySignal(&midSignal[0], 
                midSignalNDims, 
                pulseStartPos,
                baseToPulseIndex,
                read.seq, read.length,
                read.midSignal);
    }

    if (maxSignal.size() > 0) {
        assert(maxSignal.size() > pulseStartPos); 
        OK = OK and Realloc(read.maxSignal, read.length);
        CopySignal(&maxSignal[0], 
                    maxSignalNDims, 
                    pulseStartPos,
                    baseToPulseIndex,
                    read.seq, read.length,
                    read.maxSignal);
    }

    if (meanSignal.size() > 0) {
        assert(meanSignal.size() > pulseStartPos); 
        OK = OK and Realloc(read.meanSignal, read.length);
        CopySignal(&meanSignal[0], 
                    meanSignalNDims, 
                    pulseStartPos,
                    baseToPulseIndex,
                    read.seq, read.length,
                    read.meanSignal);
    }
    if (plsWidthInFrames.size() > 0) {
        OK = OK and Realloc(read.widthInFrames, read.length);
        StoreField(plsWidthInFrames, baseToPulseIndex, read.widthInFrames, read.length);
    }
    if (classifierQV.size() > 0) {
        OK = OK and Realloc(read.classifierQV, read.length);
        StoreField(classifierQV, baseToPulseIndex, read.classifierQV, read.length);
    }
    if (startFrame.size() > 0) {
        OK = OK and Realloc(read.startFrame, read.length);
        StoreField(startFrame, baseToPulseIndex, read.startFrame, read.length);
    }

    if (not OK) {
        std::cout << "ERROR, failed to CopyReadAt(" << plsReadIndex << ")" << std::endl;
        exit(1);
    }
}