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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
// -----------------------------------------------------------------------------------------------------
// Copyright (c) 2006-2020, Knut Reinert & Freie Universität Berlin
// Copyright (c) 2016-2020, Knut Reinert & MPI für molekulare Genetik
// This file may be used, modified and/or redistributed under the terms of the 3-clause BSD-License
// shipped with this file and also available at: https://github.com/seqan/seqan3/blob/master/LICENSE.md
// -----------------------------------------------------------------------------------------------------
#include <benchmark/benchmark.h>
#include <fstream>
#include <iostream>
#include <iterator>
#include <seqan3/alphabet/adaptation/char.hpp>
#include <seqan3/io/stream/iterator.hpp>
#include <seqan3/test/performance/sequence_generator.hpp>
#include <seqan3/test/seqan2.hpp>
#include <seqan3/test/tmp_filename.hpp>
#if SEQAN3_HAS_SEQAN2
#include <seqan/stream.h>
#endif
enum class tag
{
std_stream_it,
std_streambuf_it,
seqan3_streambuf_it,
seqan3_streambuf_it_write_range,
seqan2_stream_it,
seqan2_stream_it_write_range
};
template <tag id>
void write_all(benchmark::State & state)
{
/* prepare file for writing */
seqan3::test::tmp_filename filename{"foo"};
std::ofstream os{filename.get_path(), std::ios::binary};
// sequence to write:
std::vector<char> sequence = seqan3::test::generate_sequence<char>(10'000, 0, 0);
#ifdef SEQAN3_HAS_SEQAN2
auto seqan2_sequence = seqan3::test::generate_sequence_seqan2<char>(10'000, 0, 0);
#endif
/* start benchmark */
for (auto _ : state)
{
std::ofstream os{filename.get_path(), std::ios::binary};
auto it = [&os]()
{
if constexpr (id == tag::std_stream_it)
{
return std::ostream_iterator<char>{os};
}
else if constexpr (id == tag::std_streambuf_it)
{
return std::ostreambuf_iterator<char>{os};
}
else if constexpr (id == tag::seqan3_streambuf_it || id == tag::seqan3_streambuf_it_write_range)
{
return seqan3::detail::fast_ostreambuf_iterator<char>{*os.rdbuf()};
}
#ifdef SEQAN3_HAS_SEQAN2
else if constexpr (id == tag::seqan2_stream_it || id == tag::seqan2_stream_it_write_range)
{
return seqan::Iter<std::ofstream, seqan::StreamIterator<seqan::Output>>{os};
}
#endif
}();
if constexpr (id == tag::seqan3_streambuf_it_write_range)
{
it.write_range(sequence);
}
#ifdef SEQAN3_HAS_SEQAN2
else if constexpr (id == tag::seqan2_stream_it_write_range)
{
seqan::write(it, seqan2_sequence);
}
#endif
else
{
for (auto chr : sequence)
*it = chr;
}
}
}
BENCHMARK_TEMPLATE(write_all, tag::std_stream_it);
BENCHMARK_TEMPLATE(write_all, tag::std_streambuf_it);
BENCHMARK_TEMPLATE(write_all, tag::seqan3_streambuf_it);
BENCHMARK_TEMPLATE(write_all, tag::seqan3_streambuf_it_write_range);
#ifdef SEQAN3_HAS_SEQAN2
BENCHMARK_TEMPLATE(write_all, tag::seqan2_stream_it);
BENCHMARK_TEMPLATE(write_all, tag::seqan2_stream_it_write_range);
#endif
BENCHMARK_MAIN();
|
