void testScanEventSet(SpectrumListPtr sl)
{
if (os_) *os_ << "testScanEventSet:\n";
IntegerSet scanEventSet;
scanEventSet.insert(0,0);
scanEventSet.insert(2,3);
SpectrumList_Filter filter(sl, SpectrumList_FilterPredicate_ScanEventSet(scanEventSet));
if (os_)
{
printSpectrumList(filter, *os_);
*os_ << endl;
}
unit_assert(filter.size() == 7);
unit_assert(filter.spectrumIdentity(0).id == "scan=100");
unit_assert(filter.spectrumIdentity(1).id == "scan=102");
unit_assert(filter.spectrumIdentity(2).id == "scan=103");
unit_assert(filter.spectrumIdentity(3).id == "scan=104");
unit_assert(filter.spectrumIdentity(4).id == "scan=106");
unit_assert(filter.spectrumIdentity(5).id == "scan=107");
unit_assert(filter.spectrumIdentity(6).id == "scan=108");
}
Duration RandomInsertTestImpl(IntegerSet set, int n) {
std::mt19937 random(5757);
std::uniform_int_distribution<> distribution(0, 16384);
auto start = Now();
for (int i = 0; i < n; ++i) {
set.insert(distribution(random));
}
return Now() - start;
}
void testScanNumberSet(SpectrumListPtr sl)
{
if (os_) *os_ << "testScanNumberSet:\n";
IntegerSet scanNumberSet;
scanNumberSet.insert(102,104);
scanNumberSet.insert(107);
SpectrumList_Filter filter(sl, SpectrumList_FilterPredicate_ScanNumberSet(scanNumberSet));
if (os_)
{
printSpectrumList(filter, *os_);
*os_ << endl;
}
unit_assert(filter.size() == 4);
unit_assert(filter.spectrumIdentity(0).id == "scan=102");
unit_assert(filter.spectrumIdentity(1).id == "scan=103");
unit_assert(filter.spectrumIdentity(2).id == "scan=104");
unit_assert(filter.spectrumIdentity(3).id == "scan=107");
}
void testIndexSet(SpectrumListPtr sl)
{
if (os_) *os_ << "testIndexSet:\n";
IntegerSet indexSet;
indexSet.insert(3,5);
indexSet.insert(7);
indexSet.insert(9);
SpectrumList_Filter filter(sl, SpectrumList_FilterPredicate_IndexSet(indexSet));
if (os_)
{
printSpectrumList(filter, *os_);
*os_ << endl;
}
unit_assert(filter.size() == 5);
unit_assert(filter.spectrumIdentity(0).id == "scan=103");
unit_assert(filter.spectrumIdentity(1).id == "scan=104");
unit_assert(filter.spectrumIdentity(2).id == "scan=105");
unit_assert(filter.spectrumIdentity(3).id == "scan=107");
unit_assert(filter.spectrumIdentity(4).id == "scan=109");
}
void test()
{
// instantiate IntegerSet
IntegerSet a;
unit_assert(a.empty());
a.insert(1);
unit_assert(!a.empty());
a.insert(2);
a.insert(IntegerSet::Interval(0,2));
a.insert(0,2);
a.insert(4);
// verify virtual container contents: 0, 1, 2, 4
if (os_)
{
copy(a.begin(), a.end(), ostream_iterator<int>(*os_," "));
*os_ << endl;
}
vector<int> b;
copy(a.begin(), a.end(), back_inserter(b));
unit_assert(b.size() == 4);
unit_assert(b[0] == 0);
unit_assert(b[1] == 1);
unit_assert(b[2] == 2);
unit_assert(b[3] == 4);
// insert [2,4], and verify contents: 0, 1, 2, 3, 4
a.insert(2,4);
if (os_)
{
copy(a.begin(), a.end(), ostream_iterator<int>(*os_," "));
*os_ << endl;
}
b.clear();
copy(a.begin(), a.end(), back_inserter(b));
unit_assert(b.size() == 5);
for (int i=0; i<5; i++)
unit_assert(i == b[i]);
}
void testMSLevelSet(SpectrumListPtr sl)
{
if (os_) *os_ << "testMSLevelSet:\n";
IntegerSet msLevelSet;
msLevelSet.insert(1);
SpectrumList_Filter filter(sl, SpectrumList_FilterPredicate_MSLevelSet(msLevelSet));
if (os_)
{
printSpectrumList(filter, *os_);
*os_ << endl;
}
unit_assert(filter.size() == 4);
unit_assert(filter.spectrumIdentity(0).id == "scan=100");
unit_assert(filter.spectrumIdentity(1).id == "scan=103");
unit_assert(filter.spectrumIdentity(2).id == "scan=106");
unit_assert(filter.spectrumIdentity(3).id == "scan=109");
IntegerSet msLevelSet2;
msLevelSet2.insert(2);
SpectrumList_Filter filter2(sl, SpectrumList_FilterPredicate_MSLevelSet(msLevelSet2));
if (os_)
{
printSpectrumList(filter2, *os_);
*os_ << endl;
}
unit_assert(filter2.size() == 6);
unit_assert(filter2.spectrumIdentity(0).id == "scan=101");
unit_assert(filter2.spectrumIdentity(1).id == "scan=102");
unit_assert(filter2.spectrumIdentity(2).id == "scan=104");
unit_assert(filter2.spectrumIdentity(3).id == "scan=105");
unit_assert(filter2.spectrumIdentity(4).id == "scan=107");
unit_assert(filter2.spectrumIdentity(5).id == "scan=108");
}
