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
|
// Copyright 2018-2019 Henry Schreiner and Hans Dembinski
//
// Distributed under the 3-Clause BSD License. See accompanying
// file LICENSE or https://github.com/scikit-hep/boost-histogram for details.
#pragma once
#include <bh_python/pybind11.hpp>
#include <bh_python/accumulators/ostream.hpp>
#include <bh_python/make_pickle.hpp>
#include <utility>
// This can be included here since we don't include Boost.Histogram's ostream
namespace boost {
namespace histogram {
namespace accumulators {
template <class CharT, class Traits, class W>
std::basic_ostream<CharT, Traits>& operator<<(std::basic_ostream<CharT, Traits>& os,
const sum<W>& x) {
if(os.width() == 0)
return os << x.large_part() << " + " << x.small_part();
return handle_nonzero_width(os, x);
}
} // namespace accumulators
} // namespace histogram
} // namespace boost
template <class A, class... Args>
py::class_<A> register_accumulator(py::module acc, Args&&... args) {
return py::class_<A>(std::move(acc), std::forward<Args>(args)...)
.def(py::init<>())
.def(py::self += py::self)
.def("__eq__",
[](const A& self, const py::object& other) {
try {
return self == py::cast<A>(other);
} catch(const py::cast_error&) {
return false;
}
})
.def("__ne__",
[](const A& self, const py::object& other) {
try {
return self != py::cast<A>(other);
} catch(const py::cast_error&) {
return true;
}
})
.def(py::self *= double())
.def("__add__",
[](const A& self, const A& other) {
A retval(self);
retval += other;
return retval;
})
// The c++ name is replaced with the Python name here
.def("__repr__",
[](const py::object& self) {
const A& item = py::cast<const A&>(self);
py::str const str = shift_to_string(item);
return py::str("{0.__class__.__name__}({1})").format(self, str);
})
.def("__copy__", [](const A& self) { return A(self); })
.def("__deepcopy__", [](const A& self, const py::object&) { return A(self); })
.def(make_pickle<A>());
}
|
