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
|
// -------------------------
// numeric ID
// -------------------------
// single value
PbiFilter filter{ PbiReadGroupFilter{ 2458765 } };
PbiFilterQuery query(filter);
for (const BamRecord& record : query) {
assert(record.ReadGroupNumericId() == 2458765);
}
// whitelist
vector<int32_t> whitelist = { 2458765, -32143 };
PbiFilter filter{ PbiReadGroupFilter{whitelist} };
PbiFilterQuery query(filter);
for (const BamRecord& record : query) {
assert(record.ReadGroupNumericId() == 2458765 ||
record.ReadGroupNumericId() == -32143);
}
// -------------------------
// printable ID
// -------------------------
// single value
PbiFilter filter{ PbiReadGroupFilter{ "12B33F00" } };
PbiFilterQuery query(filter);
for (const BamRecord& record : query) {
assert(record.ReadGroupId() == "12B33F00");
}
// whitelist
vector<string> whitelist = { "12B33F00", "123ABC77" };
PbiFilter filter{ PbiReadGroupFilter{whitelist} };
PbiFilterQuery query(filter);
for (const BamRecord& record : query) {
assert(record.ReadGroupId() == "12B33F00" ||
record.ReadGroupId() == "123ABC77");
}
// -------------------------
// read group
// -------------------------
BamFile file("foo.bam");
BamHeader header = file.Header();
assert(header.ReadGroups().size() > 1);
// single value
PbiFilter filter{ PbiReadGroupFilter{ header.ReadGroups()[0] } };
PbiFilterQuery query(filter);
for (const BamRecord& record : query) {
assert(record.ReadGroup() == header.ReadGroups()[0]);
}
// whitelist
vector<ReadGroupInfo> whitelist = { header.ReadGroups()[0], header.ReadGroups()[1] };
PbiFilter filter{ PbiReadGroupFilter{whitelist} };
PbiFilterQuery query(filter);
for (const BamRecord& record : query) {
assert(record.ReadGroup() == header.ReadGroups()[0] ||
record.ReadGroup() == header.ReadGroups()[1]);
}
|
