diff --git a/CMakeLists.txt b/CMakeLists.txt
index 508a437..fb2c307 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -1,16 +1,16 @@
cmake_minimum_required(VERSION 3.0)
project(mylib VERSION 1.0)
add_executable(mylib_bench main_benchmark.cpp)
set(BENCHMARK_USE_BUNDLED_GTEST OFF)
set(BENCHMARK_ENABLE_GTEST_TESTS OFF)
set(BENCHMARK_ENABLE_TESTING OFF)
add_subdirectory(benchmark)
add_subdirectory(akantu_iterators)
find_package(TBB)
-target_link_libraries(mylib_bench benchmark::benchmark akantu_iterators)
+target_link_libraries(mylib_bench benchmark::benchmark akantu_iterators tbb)
set_target_properties(mylib_bench PROPERTIES CXX_STANDARD 17)
diff --git a/akantu_iterators/include/iterators/aka_zip_iterator.hh b/akantu_iterators/include/iterators/aka_zip_iterator.hh
index 30c3389..c8c268d 100644
--- a/akantu_iterators/include/iterators/aka_zip_iterator.hh
+++ b/akantu_iterators/include/iterators/aka_zip_iterator.hh
@@ -1,282 +1,290 @@
/**
* @file aka_zip_iterator.hh
*
* @author Nicolas Richart
*
* @date creation jeu déc 12 2019
*
* @brief A Documented file.
*
*
* Copyright (©) 2010-2011 EPFL (Ecole Polytechnique Fédérale de Lausanne)
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
*
* akantu-iterators is free software: you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or (at your
* option) any later version.
*
* akantu-iterators is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
* License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with akantu-iterators. If not, see .
*
*/
/* -------------------------------------------------------------------------- */
#include "aka_compatibilty_with_cpp_standard.hh"
#include "aka_iterator_tools.hh"
#include "aka_tuple_tools.hh"
/* -------------------------------------------------------------------------- */
#include
#include
/* -------------------------------------------------------------------------- */
#ifndef AKA_ZIP_ITERATOR_HH
#define AKA_ZIP_ITERATOR_HH
#ifndef AKANTU_ITERATORS_NAMESPACE
#define AKANTU_ITERATORS_NAMESPACE akantu
#endif
namespace AKANTU_ITERATORS_NAMESPACE {
/* -------------------------------------------------------------------------- */
namespace iterators {
/* ------------------------------------------------------------------------ */
template class Tuple, class... Iterators>
class ZipIterator_
: public details::CopyAssignmentEnabler<
aka::conjunction...,
std::is_copy_constructible...>::value>,
public details::MoveAssignmentEnabler<
aka::conjunction...,
std::is_move_constructible...>::value> {
private:
using tuple_t = Tuple;
public:
using value_type =
Tuple::value_type...>;
using difference_type = std::common_type_t<
typename std::iterator_traits::difference_type...>;
using pointer = Tuple::pointer...>;
using reference =
Tuple::reference...>;
using iterator_category = // std::input_iterator_tag;
std::common_type_t<
typename std::iterator_traits::iterator_category...>;
// using nb_iterators = sizeof...(Iterators);
public:
explicit ZipIterator_(tuple_t iterators)
: iterators(std::move(iterators)) {}
/* ---------------------------------------------------------------------- */
template ::value> * = nullptr>
ZipIterator_ & operator--() {
tuple::foreach ([](auto && it) { --it; }, iterators);
return *this;
}
template ::value> * = nullptr>
ZipIterator_ operator--(int) {
auto cpy = *this;
this->operator--();
return cpy;
}
// input iterator ++it
ZipIterator_ & operator++() {
tuple::foreach ([](auto && it) { ++it; }, iterators);
return *this;
}
// input iterator it++
ZipIterator_ operator++(int) {
auto cpy = *this;
this->operator++();
return cpy;
}
// input iterator it != other_it
bool operator!=(const ZipIterator_ & other) const {
// return tuple::are_not_equal(iterators, other.iterators);
return std::get<0>(iterators) !=
std::get<0>(other.iterators); // helps the compiler to optimize
}
// input iterator dereference *it
decltype(auto) operator*() {
return tuple::transform([](auto && it) -> decltype(auto) { return *it; },
iterators);
}
template ::value> * = nullptr>
- difference_type operator-(const ZipIterator_ & other) {
+ difference_type operator-(const ZipIterator_ & other) const {
return std::get<0>(this->iterators) - std::get<0>(other.iterators);
}
+ template ::value> * = nullptr>
+ bool operator<(const ZipIterator_ & other) const {
+ return std::get<0>(this->iterators) < std::get<0>(other.iterators);
+ }
+
// random iterator it[idx]
template ::value> * = nullptr>
decltype(auto) operator[](std::size_t idx) {
return tuple::transform(
[idx](auto && it) -> decltype(auto) { return it[idx]; }, iterators);
}
// random iterator it + n
template ::value> * = nullptr>
decltype(auto) operator+(std::size_t n) {
return ZipIterator_(std::forward(tuple::transform(
[n](auto && it) -> decltype(auto) { return it + n; }, iterators)));
}
// random iterator it - n
template ::value> * = nullptr>
decltype(auto) operator-(std::size_t n) {
return ZipIterator_(std::forward(tuple::transform(
[n](auto && it) -> decltype(auto) { return it - n; }, iterators)));
}
template <
class iterator_category_ = iterator_category,
std::enable_if_t::value> * = nullptr>
bool operator==(const ZipIterator_ & other) const {
return not tuple::are_not_equal(iterators, other.iterators);
}
private:
tuple_t iterators;
};
template
using ZipIterator = ZipIterator_;
template
using NamedZipIterator = ZipIterator_;
} // namespace iterators
/* -------------------------------------------------------------------------- */
template
decltype(auto) zip_iterator(std::tuple && iterators_tuple) {
auto zip = iterators::ZipIterator(
std::forward(iterators_tuple));
return zip;
}
template
decltype(auto)
zip_iterator(tuple::named_tuple && iterators_tuple) {
auto zip = iterators::NamedZipIterator(
std::forward(iterators_tuple));
return zip;
}
/* -------------------------------------------------------------------------- */
namespace containers {
template class Tuple, class... Containers>
class ZipContainer_ {
using containers_t = Tuple;
public:
explicit ZipContainer_(Containers &&... containers)
: containers(std::forward(containers)...) {}
decltype(auto) begin() const {
return zip_iterator(
tuple::transform([](auto && c) { return c.begin(); },
std::forward(containers)));
}
decltype(auto) end() const {
return zip_iterator(
tuple::transform([](auto && c) { return c.end(); },
std::forward(containers)));
}
decltype(auto) begin() {
return zip_iterator(
tuple::transform([](auto && c) { return c.begin(); },
std::forward(containers)));
}
decltype(auto) end() {
return zip_iterator(
tuple::transform([](auto && c) { return c.end(); },
std::forward(containers)));
}
private:
containers_t containers;
};
template
using ZipContainer = ZipContainer_;
template
using NamedZipContainer = ZipContainer_;
} // namespace containers
/* -------------------------------------------------------------------------- */
template decltype(auto) zip(Containers &&... conts) {
return containers::ZipContainer(
std::forward(conts)...);
}
template
decltype(auto) named_zip(NamedContainers &&... conts) {
return containers::NamedZipContainer(
std::forward(conts)...);
}
/* -------------------------------------------------------------------------- */
template
decltype(auto) make_zip_cat(zip_container_t &&... cont) {
return make_transform_adaptor(
zip(std::forward(cont)...),
[](auto && value) { return tuple::flatten(value); });
}
} // namespace AKANTU_ITERATORS_NAMESPACE
namespace std {
template class Tuple, typename... Its>
struct iterator_traits<
::AKANTU_ITERATORS_NAMESPACE::iterators::ZipIterator_> {
private:
using iterator_type =
typename ::AKANTU_ITERATORS_NAMESPACE::iterators::ZipIterator_;
public:
using iterator_category = typename iterator_type::iterator_category;
using value_type = typename iterator_type::value_type;
using difference_type = typename iterator_type::difference_type;
using pointer = typename iterator_type::pointer;
using reference = typename iterator_type::reference;
};
} // namespace std
#endif /* AKA_ZIP_ITERATOR_HH */
diff --git a/main_benchmark.cpp b/main_benchmark.cpp
index aefbe85..f951fd2 100644
--- a/main_benchmark.cpp
+++ b/main_benchmark.cpp
@@ -1,52 +1,51 @@
/* -------------------------------------------------------------------------- */
#include
/* -------------------------------------------------------------------------- */
#include
#include
-#include
#include
+#include
/* -------------------------------------------------------------------------- */
using namespace akantu;
class MyFixture : public benchmark::Fixture {
using stream_type = int;
+
public:
void SetUp(const ::benchmark::State &state) {
- auto && size = state.range(0) * 1024 / sizeof(stream_type);
+ auto &&size = state.range(0) * 1024 / sizeof(stream_type);
A.resize(size);
B.resize(size);
auto view = enumerate(A, B);
- std::for_each(std::execution::par_unseq, view.begin(), view.end(), [](auto &&data) {
- std::get<1>(data) = 2. * std::get<0>(data);
- std::get<2>(data) = 0.;
- });
+ std::for_each(std::execution::par_unseq, view.begin(), view.end(),
+ [](auto &&data) {
+ std::get<1>(data) = 2. * std::get<0>(data);
+ std::get<2>(data) = 0.;
+ });
}
void TearDown(const ::benchmark::State & /*state*/) {}
std::vector A;
std::vector B;
};
BENCHMARK_DEFINE_F(MyFixture, CopyZip)(benchmark::State &st) {
auto view = zip(A, B);
- std::for_each(view.begin(), view.end(),
+ std::for_each(std::execution::par_unseq, view.begin(), view.end(),
[](auto &&data) { std::get<1>(data) = std::get<0>(data); });
for (auto _ : st) {
- benchmark::DoNotOptimize(
- std::for_each(std::execution::par_unseq, view.begin(), view.end(), [](auto &&data) {
- std::get<1>(data) = std::get<0>(data);
- }));
+ std::for_each(std::execution::par_unseq, view.begin(), view.end(),
+ [](auto &&data) { std::get<1>(data) = std::get<0>(data); });
}
st.counters["size_kb"] =
A.size() * sizeof(typename decltype(A)::value_type) / 1024;
- st.SetBytesProcessed(int64_t(st.iterations()) * A.size() * sizeof(typename decltype(A)::value_type));
+ st.SetBytesProcessed(int64_t(st.iterations()) * A.size() *
+ sizeof(typename decltype(A)::value_type));
}
-BENCHMARK_REGISTER_F(MyFixture, CopyZip)
- ->RangeMultiplier(2)
- ->Range(2, 1 << 16);
+BENCHMARK_REGISTER_F(MyFixture, CopyZip)->RangeMultiplier(2)->Range(2, 1 << 16);
BENCHMARK_MAIN();