/***************************************************************************

* Copyright (c) 2016, Johan Mabille, Sylvain Corlay and Wolf Vollprecht *

* *

* Distributed under the terms of the BSD 3-Clause License. *

* *

* The full license is in the file LICENSE, distributed with this software. *

****************************************************************************/

#ifndef BENCHMARK_SHAPE_HPP

#define BENCHMARK_SHAPE_HPP

#include <benchmark/benchmark.h>

#include "xtensor/containers/xstorage.hpp"

#include "xtensor/core/xshape.hpp"

namespace xt

{

namespace benchmark_xshape

{

template <class T>

void xshape_initializer(benchmark::State& state)

{

for (auto _ : state)

{

T sv({2, 3, 1});

benchmark::DoNotOptimize(sv.data());

}

}

template <class T>

void xshape_initializer_long(benchmark::State& state)

{

for (auto _ : state)

{

T sv({2, 3, 1, 2, 6, 1, 2, 3, 45, 6, 12, 3, 5, 45, 5, 6});

benchmark::DoNotOptimize(sv.data());

}

}

template <class T>

void xshape_access(benchmark::State& state)

{

T a({3, 2, 1, 3});

for (auto _ : state)

{

a[0] = a[1] * a[2] + a[3];

a[3] = a[1];

a[1] = a[2] + a[3];

a[2] = a[3];

benchmark::DoNotOptimize(a.data());

}

}

BENCHMARK_TEMPLATE(xshape_initializer, std::vector<std::size_t>);

BENCHMARK_TEMPLATE(xshape_initializer, xt::svector<std::size_t, 4>);

BENCHMARK_TEMPLATE(xshape_initializer, std::array<std::size_t, 4>);

BENCHMARK_TEMPLATE(xshape_initializer_long, xt::svector<std::size_t, 4>);

BENCHMARK_TEMPLATE(xshape_initializer_long, xt::uvector<std::size_t>);

BENCHMARK_TEMPLATE(xshape_initializer_long, std::vector<std::size_t>);

BENCHMARK_TEMPLATE(xshape_access, xt::uvector<double>);

BENCHMARK_TEMPLATE(xshape_access, std::vector<double>);

BENCHMARK_TEMPLATE(xshape_access, xt::svector<std::size_t, 4>);

BENCHMARK_TEMPLATE(xshape_access, std::array<std::size_t, 4>);

}

}

#endif