mirror of
https://github.com/triqs/dft_tools
synced 2024-12-25 05:43:40 +01:00
47cb8a03f7
- Simplify group_indices - Only for C ordered, remove complex compile time. - Could be generalized to non C ordered, but no need. - Fix slice for custom orders. - Generalize the group_indices for the custom order. - Add c_ordered_transposed_view (useful ?) - Improve slice, special for ellipsis (quicker). - Simplify TraversalOrder - Assignement. Specialize one case for speed. - use FORCEINLINE in foreach, according to speed test for clang - add one speed test - Modify iterators for better speed. - along the lines decided for the foreach - update doc.
78 lines
2.7 KiB
C++
78 lines
2.7 KiB
C++
/*******************************************************************************
|
|
*
|
|
* TRIQS: a Toolbox for Research in Interacting Quantum Systems
|
|
*
|
|
* Copyright (C) 2011 by O. Parcollet
|
|
*
|
|
* TRIQS is free software: you can redistribute it and/or modify it under the
|
|
* terms of the GNU General Public License as published by the Free Software
|
|
* Foundation, either version 3 of the License, or (at your option) any later
|
|
* version.
|
|
*
|
|
* TRIQS 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 General Public License for more
|
|
* details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with
|
|
* TRIQS. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
******************************************************************************/
|
|
#pragma once
|
|
#include <triqs/utility/first_include.hpp>
|
|
namespace triqs {
|
|
namespace arrays {
|
|
|
|
/// ------------ Pretty Printing : specializing the default behaviour for d=1,2 -------------------------
|
|
namespace PrettyPrint_details {
|
|
template <int R, typename A> struct print_impl {
|
|
std::ostream &out;
|
|
A const &a;
|
|
print_impl(std::ostream &out_, A const &a_) : out(out_), a(a_) {}
|
|
template <typename A0, typename... Args> void operator()(A0 const &a0, Args const &... args) const {
|
|
out << a(a0, args...) << " ";
|
|
}
|
|
void print() const {
|
|
out << "[";
|
|
// indexmaps::cuboid::foreach<permutations::identity(A::domain_type::rank)>(a.domain(),
|
|
// std::cref(*this)); // foreach(a, std::cref(*this));
|
|
indexmaps::cuboid::foreach_impl(_traversal_fortran{}, a.domain(), FORTRAN_LAYOUT, *this);
|
|
// foreach(a, *this);
|
|
out << "]";
|
|
}
|
|
};
|
|
|
|
template <typename A> struct print_impl<1, A> {
|
|
std::ostream &out;
|
|
A const &a;
|
|
print_impl(std::ostream &out_, A const &a_) : out(out_), a(a_) {}
|
|
void print() const {
|
|
auto d = a.indexmap().domain();
|
|
out << "[";
|
|
for (size_t i = 0; i < d.lengths()[0]; ++i) out << (i > 0 ? "," : "") << a(i);
|
|
out << "]";
|
|
}
|
|
};
|
|
|
|
template <typename A> struct print_impl<2, A> {
|
|
std::ostream &out;
|
|
A const &a;
|
|
print_impl(std::ostream &out_, A const &a_) : out(out_), a(a_) {}
|
|
void print() const {
|
|
auto d = a.indexmap().domain();
|
|
out << "\n[";
|
|
for (size_t i = 0; i < d.lengths()[0]; ++i) {
|
|
out << (i == 0 ? "[" : " [");
|
|
for (size_t j = 0; j < d.lengths()[1]; ++j) out << (j > 0 ? "," : "") << a(i, j);
|
|
out << "]" << (i == d.lengths()[0] - 1 ? "" : "\n");
|
|
}
|
|
out << "]";
|
|
}
|
|
};
|
|
}
|
|
template <typename A> void pretty_print(std::ostream &out, A const &a) {
|
|
PrettyPrint_details::print_impl<A::domain_type::rank, A>(out, a).print();
|
|
}
|
|
}
|
|
}
|