32#ifndef MADNESS_MRA_MRA_H__INCLUDED
33#define MADNESS_MRA_MRA_H__INCLUDED
48#define FUNCTION_INSTANTIATE_1
49#define FUNCTION_INSTANTIATE_2
50#define FUNCTION_INSTANTIATE_3
51#if !defined(HAVE_IBMBGP) || !defined(HAVE_IBMBGQ)
52#define FUNCTION_INSTANTIATE_4
53#define FUNCTION_INSTANTIATE_5
54#define FUNCTION_INSTANTIATE_6
96 template<
typename T, std::
size_t NDIM>
99 template<
typename T, std::
size_t NDIM>
102 template<
typename T, std::
size_t NDIM>
105 template<
typename T, std::
size_t NDIM>
106 class FunctionFactory;
108 template<
typename T, std::
size_t NDIM>
109 class FunctionFunctorInterface;
111 template<
typename T, std::
size_t NDIM>
114 template<
typename T, std::
size_t NDIM>
117 template<
typename T, std::
size_t NDIM>
120 template<
typename T, std::
size_t NDIM, std::
size_t LDIM,
typename opT>
123 template<
typename T, std::
size_t NDIM,
typename opT>
126 template<
typename T, std::
size_t NDIM>
138 template <
typename T, std::
size_t NDIM>
146 std::shared_ptr< FunctionImpl<T,NDIM> >
impl;
193 if (
this != &
f)
impl =
f.impl;
201 template <
typename R, std::
size_t MDIM>
213 const double eps=1
e-15;
220 for (std::size_t
d=0;
d<
NDIM; ++
d) {
221 if (
xsim[
d] < -eps) {
224 else if (
xsim[
d] < eps) {
228 if (
xsim[
d] > 1.0+eps) {
231 else if (
xsim[
d] > 1.0-eps) {
246 const double eps=1
e-15;
253 for (std::size_t
d=0;
d<
NDIM; ++
d) {
254 if (
xsim[
d] < -eps) {
257 else if (
xsim[
d] < eps) {
261 if (
xsim[
d] > 1.0+eps) {
264 else if (
xsim[
d] > 1.0-eps) {
286 std::vector<std::pair<bool,T>>& results)
const {
287 const double eps=1
e-15;
290 thread_local std::vector<coordT>
xsim;
297 for (std::size_t
d=0;
d<
NDIM; ++
d) {
301 else if (
xs[
d] < eps) {
305 if (
xs[
d] > 1.0+eps) {
308 else if (
xs[
d] > 1.0-eps) {
314 results.resize(
xuser.size());
321 std::vector<std::pair<bool,T>> results;
334 const double eps=1
e-15;
341 for (std::size_t
d=0;
d<
NDIM; ++
d) {
342 if (
xsim[
d] < -eps) {
345 else if (
xsim[
d] < eps) {
349 if (
xsim[
d] > 1.0+eps) {
352 else if (
xsim[
d] > 1.0-eps) {
371 const double eps=1
e-15;
378 for (std::size_t
d=0;
d<
NDIM; ++
d) {
379 if (
xsim[
d] < -eps) {
382 else if (
xsim[
d] < eps) {
386 if (
xsim[
d] > 1.0+eps) {
389 else if (
xsim[
d] > 1.0-eps) {
411 const std::vector<long>& npt,
415 const double eps=1
e-14;
419 for (std::size_t
d=0;
d<
NDIM; ++
d) {
428 for (std::size_t
d=0;
d<
NDIM; ++
d) {
457 impl->world.gop.broadcast(result);
465 T operator()(
double x,
double y=0,
double z=0,
double xx=0,
double yy=0,
double zz=0)
const {
468 if (
NDIM>=2) r[1] = y;
469 if (
NDIM>=3) r[2] = z;
470 if (
NDIM>=4) r[3] = xx;
471 if (
NDIM>=5) r[4] = yy;
472 if (
NDIM>=6) r[5] = zz;
492 impl->world.gop.broadcast(result);
504 template <
typename funcT>
519 template <
typename funcT>
528 impl->world.gop.fence();
545 return impl->is_compressed();
556 return impl->is_reconstructed();
566 return impl ?
impl->is_nonstandard() :
false;
574 return impl ?
impl->is_redundant() :
false;
582 return impl ?
impl->is_redundant_after_merge() :
false;
589 return impl->tree_size();
595 print(
"function",
name,
"not assigned yet");
605 return impl->max_depth();
618 return impl->max_local_depth();
626 return impl->max_nodes();
633 return impl->min_nodes();
648 return impl->size_local();
655 if (!
impl)
return true;
656 return impl->get_autorefine();
666 impl->set_autorefine(value);
674 if (!
impl)
return 0.0;
675 return impl->get_thresh();
685 impl->set_thresh(value);
694 return impl->get_k();
709 if (!
impl)
return *
this;
719 const std::shared_ptr< FunctionImpl<T,NDIM> >&
get_impl()
const {
737 this->impl->set_functor(functor);
738 print(
"set functor in mra.h");
746 template <
typename R>
748 impl = std::shared_ptr<implT>(
new implT(*
f.get_impl(),
f.get_pmap(), zero));
761 const std::shared_ptr< WorldDCPmapInterface< Key<NDIM> > >&
get_pmap()
const {
764 return impl->get_pmap();
812 "function must be compressed or reconstructed for norm2sq_local");
813 return impl->norm2sq_local();
827 impl->world.gop.fence();
942 template <
typename opT>
953 return impl->autorefine_square_test(key, t);
969 bool fence =
true)
const {
998 if (
impl)
impl->print_tree_json(os);
1004 os <<
"digraph G {" << std::endl;
1005 if (
impl)
impl->print_tree_graphviz(os);
1006 os <<
"}" << std::endl;
1035 template <
typename opT>
1045 template <
typename opT>
1047 bool fence =
true) {
1055 template <
typename opT>
1057 bool fence =
true) {
1084 template <
typename Q>
1114 template <
typename Q,
typename R>
1135 "gaxpy: this must be reconstructed or redundant_after_merge");
1137 "gaxpy: other must be reconstructed or redundant_after_merge");
1153 template <
typename Q>
1165 return gaxpy(
T(1.0), other,
Q(1.0),
true);
1172 template <
typename Q>
1185 return gaxpy(
T(1.0), other,
Q(-1.0),
true);
1192 template <
typename Q>
1237 if (!
impl)
return 0.0;
1239 return impl->trace_local();
1246 if (!
impl)
return 0.0;
1249 impl->world.gop.fence();
1255 template <
typename R>
1274 return impl->dot_local(*(
g.get_impl()));
1281 template<
typename R>
1287 impl->get_coeffs().clear();
1299 template<
typename opT>
1303 impl->get_coeffs().clear();
1314 impl->get_coeffs().clear();
1322 template<
typename opT>
1326 impl->get_coeffs().clear();
1339 impl->get_coeffs().clear();
1350 template<
typename opT>
1354 impl->get_coeffs().clear();
1367 impl->get_coeffs().clear();
1377 template<
size_t LDIM,
size_t KDIM,
typename opT>
1385 impl->finalize_sum();
1391 template<
size_t LDIM,
size_t KDIM>
1398 impl->finalize_sum();
1407 template <
typename R>
1416 if constexpr (std::is_same_v<T,R>) {
1427 if constexpr (std::is_same_v<R,T>) {
1429 return g.inner_on_demand(*
this);
1430 if (
g.is_on_demand())
1485 impl->world.gop.fence();
1498 const bool leaf_refine=
true)
const {
1501 T local =
impl->inner_adaptive_local(
f, leaf_refine);
1503 impl->world.gop.fence();
1511 template <
typename L>
1526 template<
typename R>
1530 constexpr std::size_t
LDIM=std::max(
NDIM/2,std::size_t(1));
1534 func->replicate_low_dim_functions(
true);
1551 template <
typename R,
size_t LDIM>
1566 this->
get_impl()->project_out(result.
get_impl().get(),
g.get_impl().get(),dim,
true);
1570 result.
get_impl()->trickle_down(
false);
1592 template <
typename Archive>
1596 long magic = 0l,
id = 0l, ndim = 0l,
k = 0l;
1598 ar &
magic &
id & ndim &
k & cell;
1608 std::ostringstream
oss;
1609 oss <<
"simulation cells inconsistent: stored cell differs from FunctionDefaults cell.\n"
1610 <<
"Call FunctionDefaults<" <<
NDIM <<
">::clear_cell() before reloading "
1611 <<
"(this will render all existing functions useless!)";
1628 template <
typename Archive>
1648 template <
typename Q,
typename opT>
1661 template <
typename Q, std::
size_t D>
1664 std::vector< std::shared_ptr< FunctionImpl<Q,D> > > r(
v.size());
1665 for (
unsigned int i=0; i<
v.size(); ++i) r[i] =
v[i].
get_impl();
1670 template <
typename opT>
1672 std::vector<implT*>
v(
vf.size(),
NULL);
1673 for (
unsigned int i=0; i<
v.size(); ++i) {
1689 template <
typename opT>
1693 const bool fence=
true) {
1695 for (
unsigned int i=0; i<
vin.size(); ++i) {
1699 for (
unsigned int i=0; i<
vout.size(); ++i) {
1710 template <
typename L,
typename R>
1718 std::vector<FunctionImpl<T,NDIM>*>
vresult(right.size());
1719 std::vector<const FunctionImpl<R,NDIM>*>
vright(right.size());
1720 for (
unsigned int i=0; i<right.size(); ++i) {
1721 result[i].set_impl(left,
false);
1722 vresult[i] = result[i].impl.get();
1723 vright[i] = right[i].get_impl().get();
1733 template<
typename L,
typename R>
1751 template <
typename R,
typename Q>
1762 template <
typename L,
typename R>
1782 impl.reset(
new implT(*
f.impl,
f.get_pmap(),
false));
1797 impl.reset(
new implT(*
f.impl,
f.get_pmap(),
false));
1812 const std::vector<long>& map,
const std::vector<long>&
mirror,
1818 for (std::size_t i=0; i<map.size(); ++i)
MADNESS_ASSERT(map[i]>=0 &&
static_cast<std::size_t
>(map[i])<
NDIM);
1820 impl.reset(
new implT(*
f.impl,
f.get_pmap(),
false));
1831 double local =
impl->check_symmetry_local();
1833 impl->world.gop.fence();
1835 if (this->
world().rank()==0)
print(
"asymmetry wrt particle",
asy);
1852 impl->chop_at_level(n,
true);
1859 template <
typename T,
typename opT, std::
size_t NDIM>
1868 template <
typename Q,
typename T, std::
size_t NDIM>
1876 result.get_impl()->scale_oop(
alpha,*
f.get_impl(),
fence);
1882 template <
typename Q,
typename T, std::
size_t NDIM>
1893 template <
typename Q,
typename T, std::
size_t NDIM>
1902 template <
typename Q,
typename T, std::
size_t NDIM>
1909 template <
typename L,
typename R,std::
size_t NDIM>
1926 template <
typename L,
typename R,std::
size_t NDIM>
1933 template <
typename L,
typename R,
typename opT, std::
size_t NDIM>
1947 template <
typename Q,
typename opT, std::
size_t NDIM>
1950 if (!
func.is_reconstructed())
func.reconstruct();
1960 template <
typename Q,
typename opT, std::
size_t NDIM>
1974 template <
typename L,
typename R, std::
size_t D>
1987 template <
typename L,
typename R, std::
size_t NDIM>
1993 return mul(left,right,
true);
1999 template<
typename T, std::
size_t KDIM, std::
size_t LDIM>
2015 std::vector<std::shared_ptr<FunctionImpl<T,KDIM>>>
vleft=
get_impl(left);
2016 std::vector<std::shared_ptr<FunctionImpl<T,LDIM>>>
vright=
get_impl(right);
2025 template<
typename T, std::
size_t KDIM, std::
size_t LDIM>
2028 typedef std::vector<Function<T,KDIM>> vector;
2033 template<
typename T, std::
size_t KDIM, std::
size_t LDIM,
typename opT>
2049 print(
"incomplete FunctionFactory in Function::hartree_product");
2058 std::vector<std::shared_ptr<FunctionImpl<T,KDIM>>>
vleft;
2059 std::vector<std::shared_ptr<FunctionImpl<T,LDIM>>>
vright;
2066 left2.world().gop.fence();
2074 template <
typename L,
typename R,std::
size_t NDIM>
2084 template <
typename L,
typename R,std::
size_t NDIM>
2094 template <
typename L,
typename R,std::
size_t NDIM>
2103 template<
typename T, std::
size_t NDIM>
2119 template <
typename L,
typename R, std::
size_t NDIM>
2134 return add(left,right,
true);
2139 template <
typename L,
typename R,std::
size_t NDIM>
2150 template <
typename L,
typename R, std::
size_t NDIM>
2162 return sub(left,right,
true);
2171 template <
typename T, std::
size_t NDIM>
2174 bool fence =
true) {
2179 result.
set_impl(std::shared_ptr<implT>(
new implT(*
f.get_impl(), pmap,
false)));
2186 template <
typename T, std::
size_t NDIM>
2195 template <
typename T, std::
size_t NDIM>
2203 result.
set_impl(std::make_shared<implT>(
world,*
f.get_impl(), pmap,
false));
2218 template <
typename T,
typename Q, std::
size_t NDIM>
2232 template <
typename T, std::
size_t NDIM>
2249 template <
typename opT,
typename T, std::
size_t LDIM>
2256 typedef std::vector<Function<T,LDIM>>
vecfuncL;
2261 bool same=(
ff1[0].get_impl()==
ff2[0].get_impl());
2268 for (
auto&
f :
f1)
f.make_nonstandard(
true,
false);
2269 for (
auto&
f :
f2)
f.make_nonstandard(
true,
false);
2281 for (
size_t i=0; i<
f1.size(); ++i)
2285 if (
op.print_timings) {
2290 result.
get_impl()->finalize_apply();
2292 if (
op.modified()) {
2293 result.
get_impl()->trickle_down(
true);
2295 result.
get_impl()->reconstruct(
true);
2305 template <
typename opT,
typename R, std::
size_t NDIM>
2310 constexpr std::size_t
OPDIM=opT::opdim;
2316 result.get_impl()->apply(
op, *
f.get_impl(),
fence);
2323 r1.set_impl(
f,
false);
2325 result.get_impl()->reset_timer();
2328 result.get_impl()->apply_source_driven(
op, *
f.get_impl(),
fence);
2351 template <
typename opT,
typename R, std::
size_t NDIM>
2366 if (
op.modified()) {
2371 ff.get_impl()->undo_redundant(
false);
2372 result.
get_impl()->trickle_down(
true);
2382 fff.get_impl()->timer_filter.print(
"filter");
2383 fff.get_impl()->timer_compress_svd.print(
"compress_svd");
2387 ff.world().gop.fence();
2400 result.
get_impl()->reconstruct(
true);
2403 if (
op.destructive()) {
2404 ff.world().gop.fence();
2417 template <
typename opT,
typename R, std::
size_t NDIM>
2428 result.get_impl()->apply_1d_realspace_push(
op,
ff.get_impl().get(),
axis,
fence);
2445 template <
typename T, std::
size_t NDIM>
2457 template <
typename T, std::
size_t NDIM>
2474 template <
typename T, std::
size_t NDIM>
2477 const std::vector<long>&
mirror,
bool fence=
true) {
2488 template <
typename T, std::
size_t NDIM>
2489 typename std::enable_if_t<NDIM%2==0, Function<T,NDIM>>
2492 std::vector<long> map(
NDIM);
2495 for (std::size_t
d=0;
d<
LDIM; ++
d) {
2514 template <
typename T, std::
size_t NDIM>
2520 std::vector<long> map(
NDIM);
2524 map[0]=3; map[1]=4; map[2]=5;
2525 map[3]=0; map[4]=1; map[5]=2;
2527 map[0]=0; map[1]=2; map[2]=1;
2528 map[3]=3; map[4]=5; map[5]=4;
2531 map[0]=2; map[1]=1; map[2]=0;
2532 map[3]=5; map[4]=4; map[5]=3;
2535 map[0]=1; map[1]=0; map[2]=2;
2536 map[3]=4; map[4]=3; map[5]=5;
2539 if (
f.world().rank()==0) {
2546 result.
get_impl()->average(*
f.get_impl());
2559 template<
typename T, std::
size_t NDIM, std::
size_t LDIM>
2585 template <
typename T, std::
size_t NDIM>
2603 template <
typename T,
typename R, std::
size_t NDIM>
2615 template<std::size_t
NDIM,
typename T, std::size_t
LDIM,
typename R, std::size_t
KDIM,
2619 const std::array<int, CDIM>
v2,
int task=0) {
2624 static_assert((
KDIM +
LDIM -
NDIM) % 2 == 0,
"faulty dimensions in inner (partial version)");
2625 static_assert(
KDIM +
LDIM - 2 *
CDIM ==
NDIM,
"faulty dimensions in inner (partial version)");
2646 f.get_impl()->compute_snorm_and_dnorm(
false);
2647 for (
auto&
g :
vg)
g.get_impl()->compute_snorm_and_dnorm(
false);
2652 std::vector<Function<resultT,NDIM>> result(
vg.size());
2655 for (
size_t i=0; i<
vg.size(); ++i) {
2657 .k(
f.k()).thresh(
f.thresh()).empty().nofence();
2658 result[i].get_impl()->partial_inner(*
f.get_impl(),*(
vg[i]).get_impl(),
v1,
v2);
2675 for (
auto it=
funcimpl.get_coeffs().begin(); it!=
funcimpl.get_coeffs().end(); ++it) {
2676 const auto& key=it->first;
2677 const auto&
node=it->second;
2685 for (
auto&
g :
vg) {
2691 f_nc.reconstruct(
false);
2705 template<std::size_t
NDIM,
typename T, std::size_t
LDIM,
typename R, std::size_t
KDIM,
2709 const std::array<int, CDIM>
v2,
int task=0) {
2717 template <
typename T, std::
size_t LDIM,
typename R, std::
size_t KDIM>
2721 std::array<int,1>({std::get<0>(
v1)}),
2722 std::array<int,1>({std::get<0>(
v2)}));
2729 template <
typename T, std::
size_t LDIM,
typename R, std::
size_t KDIM>
2733 std::array<int,2>({std::get<0>(
v1),std::get<1>(
v1)}),
2734 std::array<int,2>({std::get<0>(
v2),std::get<1>(
v2)}));
2741 template <
typename T, std::
size_t LDIM,
typename R, std::
size_t KDIM>
2745 std::array<int,3>({std::get<0>(
v1),std::get<1>(
v1),std::get<2>(
v1)}),
2746 std::array<int,3>({std::get<0>(
v2),std::get<1>(
v2),std::get<2>(
v2)}));
2759 template <
typename T,
typename opT, std::
size_t NDIM>
2762 std::shared_ptr< FunctionFunctorInterface<double,3> >
func(
new opT(
g));
2774 template <
typename T,
typename opT, std::
size_t NDIM>
2779 template <
typename T,
typename R, std::
size_t NDIM>
2782 return (
f*
R(1.0)).add_scalar(r);
2785 template <
typename T,
typename R, std::
size_t NDIM>
2788 return (
f*
R(1.0)).add_scalar(r);
2791 template <
typename T,
typename R, std::
size_t NDIM>
2794 return (
f*
R(1.0)).add_scalar(-r);
2797 template <
typename T,
typename R, std::
size_t NDIM>
2800 return (
f*
R(-1.0)).add_scalar(r);
2804 template <std::
size_t NDIM>
2814 template <std::
size_t NDIM>
2824 template <std::
size_t NDIM>
2835 template <std::
size_t NDIM>
2849 template <std::
size_t NDIM>
2855 template <std::
size_t NDIM>
2861 template <std::
size_t NDIM>
2868 template <
typename T, std::
size_t NDIM>
2872 return result.
square(
true);
2876 template <
typename T, std::
size_t NDIM>
2880 return result.
abs(
true);
2884 template <
typename T, std::
size_t NDIM>
2893 template <
typename T, std::
size_t NDIM>
2900 template <std::
size_t NDIM>
2906 template <std::
size_t NDIM>
2916 template <
typename T, std::
size_t NDIM>
2918 if (
f.is_initialized())
return f.get_impl()->get_tree_state();
2929 template <
typename T, std::
size_t NDIM>
2935 template <
typename R, std::
size_t MDIM>
2937 f1.impl.swap(
f2.impl);
2946 template <
class archiveT,
class T, std::
size_t NDIM>
2953 template <
class archiveT,
class T, std::
size_t NDIM>
2961 template <
class T, std::
size_t NDIM>
2967 template <
class T, std::
size_t NDIM>
2980 template<
typename T, std::
size_t NDIM>
double q(double t)
Definition DKops.h:18
This header should include pretty much everything needed for the parallel runtime.
long dim(int i) const
Returns the size of dimension i.
Definition basetensor.h:147
This class is used to specify boundary conditions for all operators.
Definition bc.h:72
CompositeFunctorInterface implements a wrapper of holding several functions and functors.
Definition function_interface.h:172
FunctionDefaults holds default paramaters as static class members.
Definition funcdefaults.h:100
static const double & get_thresh()
Returns the default threshold.
Definition funcdefaults.h:177
static std::shared_ptr< WorldDCPmapInterface< Key< NDIM > > > & get_pmap()
Returns the default process map that was last initialized via set_default_pmap()
Definition funcdefaults.h:401
static void set_cell(const Tensor< double > &value)
Sets the user cell for the simulation.
Definition funcdefaults.h:366
FunctionFactory implements the named-parameter idiom for Function.
Definition function_factory.h:86
virtual FunctionFactory & thresh(double thresh)
Definition function_factory.h:198
virtual FunctionFactory & k(int k)
Definition function_factory.h:193
Abstract base class interface required for functors used as input to Functions.
Definition function_interface.h:68
FunctionImpl holds all Function state to facilitate shallow copy semantics.
Definition funcimpl.h:945
const dcT & get_coeffs() const
Definition mraimpl.h:343
bool is_on_demand() const
Definition mraimpl.h:283
FunctionNode holds the coefficients, etc., at each node of the 2^NDIM-tree.
Definition funcimpl.h:127
A multiresolution adaptive numerical function.
Definition mra.h:139
void print_tree_json(std::ostream &os=std::cout) const
Definition mra.h:996
Tensor< T > eval_cube(const Tensor< double > &cell, const std::vector< long > &npt, bool eval_refine=false) const
Evaluates a cube/slice of points (probably for plotting) ... collective but no fence necessary.
Definition mra.h:410
TENSOR_RESULT_TYPE(T, R) inner(const Function< R
Returns the inner product.
Function< T, NDIM > & map_and_mirror(const Function< T, NDIM > &f, const std::vector< long > &map, const std::vector< long > &mirror, bool fence)
This is replaced with mirror(map(f)) ... private.
Definition mra.h:1811
void gaxpy_ext(const Function< L, NDIM > &left, T(*f)(const coordT &), T alpha, T beta, double tol, bool fence=true) const
Definition mra.h:1512
T inner_adaptive(const std::shared_ptr< FunctionFunctorInterface< T, NDIM > > f, const bool leaf_refine=true) const
Definition mra.h:1497
void unaryop_coeff(const opT &op, bool fence=true)
Unary operation applied inplace to the coefficients.
Definition mra.h:1046
bool is_compressed() const
Returns true if compressed, false otherwise. No communication.
Definition mra.h:542
Function< T, NDIM/2 > dirac_convolution(const bool fence=true) const
Definition mra.h:1576
return impl inner_local * g())
void set_impl(const Function< R, NDIM > &f, bool zero=true)
Replace current FunctionImpl with a new one using the same parameters & map as f.
Definition mra.h:747
bool autorefine() const
Returns value of autorefine flag. No communication.
Definition mra.h:653
TreeState operating_state
Definition mra.h:1438
Function< T, NDIM > & add_scalar(T t, bool fence=true)
Inplace add scalar. No communication except for optional fence.
Definition mra.h:1095
void print_size(const std::string name) const
print some info about this
Definition mra.h:593
Function< T, NDIM > & fill_tree(const Function< R, NDIM > &g, bool fence=true)
With this being an on-demand function, fill the MRA tree according to different criteria.
Definition mra.h:1282
void broaden(const BoundaryConditions< NDIM > &bc=FunctionDefaults< NDIM >::get_bc(), bool fence=true) const
Inplace broadens support in scaling function basis.
Definition mra.h:968
void print_info() const
Print a summary of the load balancing info.
Definition mra.h:1012
Function< T, NDIM > & scale(const Q q, bool fence=true)
Inplace, scale the function by a constant. No communication except for optional fence.
Definition mra.h:1085
void norm_tree(bool fence=true) const
Initializes information about the function norm at all length scales.
Definition mra.h:833
void load(World &world, Archive &ar)
Replaces this function with one loaded from an archive using the default processor map.
Definition mra.h:1593
Function< T, NDIM > & abs_square(bool fence=true)
Returns *this for chaining.
Definition mra.h:1222
void replicate(const DistributionType type, bool fence=true) const
Definition mra.h:771
double norm2sq_local() const
Returns the square of the norm of the local function ... no communication.
Definition mra.h:808
IsSupported< TensorTypeData< Q >, Function< T, NDIM > >::type & operator*=(const Q q)
Inplace scaling by a constant.
Definition mra.h:1194
void sum_down(bool fence=true) const
Sums scaling coeffs down tree restoring state with coeffs only at leaves. Optional fence....
Definition mra.h:930
Level depthpt(const coordT &xuser) const
Definition mra.h:486
bool is_redundant() const
Returns true if redundant, false otherwise. No communication.
Definition mra.h:572
Function< T, NDIM > & operator+=(const Function< Q, NDIM > &other)
Inplace addition of functions in the wavelet basis.
Definition mra.h:1154
Function< T, NDIM > & operator=(const Function< T, NDIM > &f)
Assignment is shallow. No communication, works in either basis.
Definition mra.h:191
void set_autorefine(bool value, bool fence=true)
Sets the value of the autorefine flag. Optional global fence.
Definition mra.h:663
World & world() const
Returns the world.
Definition mra.h:753
T trace() const
Returns global value of int(f(x),x) ... global comm required.
Definition mra.h:1244
T typeT
Definition mra.h:157
friend void swap(Function< R, MDIM > &f1, Function< R, MDIM > &f2)
implements swap algorithm
Definition mra.h:2936
Function< T, NDIM > & fill_tree(const opT &op, bool fence=true)
With this being an on-demand function, fill the MRA tree according to different criteria.
Definition mra.h:1300
const Function< T, NDIM > & change_tree_state(const TreeState finalstate, bool fence=true) const
changes tree state to given state
Definition mra.h:917
Function< typename opT::resultT, NDIM > & unary_op_coeffs(const Function< Q, NDIM > &func, const opT &op, bool fence)
This is replaced with left*right ... private.
Definition mra.h:1649
void print_tree_graphviz(std::ostream &os=std::cout) const
Process 0 prints a graphviz-formatted output of all nodes in the tree (collective)
Definition mra.h:1002
double norm2() const
Returns the 2-norm of the function ... global sum ... works in either basis.
Definition mra.h:820
Function< T, NDIM > & fill_cuspy_tree(const opT &op, const bool fence=true)
Definition mra.h:1323
void change_tensor_type(const TensorArgs &targs, bool fence=true)
change the tensor type of the coefficients in the FunctionNode
Definition mra.h:1641
const Function< T, NDIM > & refine(bool fence=true) const
Inplace autorefines the function using same test as for squaring.
Definition mra.h:962
TENSOR_RESULT_TYPE(T, R) inner_on_demand(const Function< R
Returns the inner product for one on-demand function.
T operator()(const coordT &xuser) const
Evaluates the function at a point in user coordinates. Collective operation.
Definition mra.h:451
MADNESS_ASSERT(g.is_compressed())
Function< T, NDIM > & square(bool fence=true)
Inplace squaring of function ... global comm only if not reconstructed.
Definition mra.h:1204
Function< T, NDIM > & fill_tree(bool fence=true)
With this being an on-demand function, fill the MRA tree according to different criteria.
Definition mra.h:1311
void verify_tree() const
Verifies the tree data structure ... global sync implied.
Definition mra.h:533
Future< Level > evaldepthpt(const coordT &xuser) const
Definition mra.h:332
void do_hartree_product(const std::vector< std::shared_ptr< FunctionImpl< T, LDIM > > > left, const std::vector< std::shared_ptr< FunctionImpl< T, KDIM > > > right)
perform the hartree product of f*g, invoked by result
Definition mra.h:1392
Function< T, NDIM > & mapdim(const Function< T, NDIM > &f, const std::vector< long > &map, bool fence)
This is replaced with mapdim(f) ... private.
Definition mra.h:1777
int k() const
Returns the number of multiwavelets (k). No communication.
Definition mra.h:691
const std::shared_ptr< WorldDCPmapInterface< Key< NDIM > > > & get_pmap() const
Returns a shared pointer to the process map.
Definition mra.h:761
double thresh() const
Returns value of truncation threshold. No communication.
Definition mra.h:672
Function< T, NDIM > & gaxpy_oop(T alpha, const Function< L, NDIM > &left, T beta, const Function< R, NDIM > &right, bool fence)
This is replaced with alpha*left + beta*right ... private.
Definition mra.h:1763
void set_thresh(double value, bool fence=true)
Sets the value of the truncation threshold. Optional global fence.
Definition mra.h:682
Function< T, NDIM > & multiop_values(const opT &op, const std::vector< Function< T, NDIM > > &vf)
This is replaced with op(vector of functions) ... private.
Definition mra.h:1671
void distribute(std::shared_ptr< WorldDCPmapInterface< Key< NDIM > > > newmap) const
distribute this function according to newmap
Definition mra.h:799
void unaryop(const opT &op, bool fence=true)
Inplace unary operation on function values.
Definition mra.h:1036
const std::shared_ptr< FunctionImpl< T, NDIM > > & get_impl() const
Returns a shared-pointer to the implementation.
Definition mra.h:719
void standard(bool fence=true)
Converts the function standard compressed form. Possible non-blocking comm.
Definition mra.h:880
void unaryop_node(const opT &op, bool fence=true)
Unary operation applied inplace to the nodes.
Definition mra.h:1056
Function< T, NDIM > & truncate(double tol=0.0, bool fence=true)
Truncate the function with optional fence. Compresses with fence if not compressed.
Definition mra.h:707
Function< T, NDIM > & fill_cuspy_tree(const bool fence=true)
Special refinement on 6D boxes where the electrons come close (meet)
Definition mra.h:1336
std::size_t size() const
Returns the number of coefficients in the function ... collective global sum.
Definition mra.h:638
bool is_on_demand() const
Definition mra.h:741
Function< T, NDIM > & operator-=(const Function< Q, NDIM > &other)
Inplace subtraction of functions in the wavelet basis.
Definition mra.h:1173
bool compressed
Definition mra.h:1258
const Function< T, NDIM > & reconstruct(bool fence=true) const
Reconstructs the function, transforming into scaling function basis. Possible non-blocking comm.
Definition mra.h:907
Function< T, NDIM > & abs(bool fence=true)
Returns *this for chaining.
Definition mra.h:1213
Function< T, NDIM > & reduce_rank(const double thresh=0.0, const bool fence=true)
reduce the rank of the coefficient tensors
Definition mra.h:1841
Function< T, NDIM > & mirror(const Function< T, NDIM > &f, const std::vector< long > &mirrormap, bool fence)
This is replaced with mirror(f) ... private.
Definition mra.h:1792
std::size_t max_nodes() const
Returns the max number of nodes on a processor.
Definition mra.h:623
void do_hartree_product(const std::vector< std::shared_ptr< FunctionImpl< T, LDIM > > > left, const std::vector< std::shared_ptr< FunctionImpl< T, KDIM > > > right, const opT *op)
perform the hartree product of f*g, invoked by result
Definition mra.h:1378
std::shared_ptr< FunctionImpl< T, NDIM > > impl
Definition mra.h:146
void replicate_on_hosts(bool fence=true) const
replicate this function, one copy per host
Definition mra.h:792
void replicate(bool fence=true) const
replicate this function, generating a unique pmap
Definition mra.h:782
T trace_local() const
Returns local contribution to int(f(x),x) ... no communication.
Definition mra.h:1235
void mul_on_demand(const Function< L, NDIM > &f, const Function< R, NDIM > &g, bool fence=true)
Same as operator* but with optional fence and no automatic reconstruction.
Definition mra.h:1734
Function< T, NDIM > & gaxpy(const T &alpha, const Function< Q, NDIM > &other, const R &beta, bool fence=true)
Inplace, general bi-linear operation in wavelet basis. No communication except for optional fence.
Definition mra.h:1115
Vector< double, NDIM > coordT
Type of vector holding coordinates.
Definition mra.h:156
void store(Archive &ar) const
Stores the function to an archive.
Definition mra.h:1629
std::size_t max_local_depth() const
Returns the maximum local depth of the function tree ... no communications.
Definition mra.h:615
void vtransform(const std::vector< Function< R, NDIM > > &v, const Tensor< Q > &c, std::vector< Function< T, NDIM > > &vresult, double tol, bool fence=true)
sparse transformation of a vector of functions ... private
Definition mra.h:1752
TENSOR_RESULT_TYPE(T, R) local
std::size_t max_depth() const
Returns the maximum depth of the function tree ... collective global sum.
Definition mra.h:602
Function()
Default constructor makes uninitialized function. No communication.
Definition mra.h:174
std::vector< std::pair< bool, T > > eval_local_only(const std::vector< coordT > &xuser, Level maxlevel) const
Definition mra.h:320
std::size_t tree_size() const
Returns the number of nodes in the function tree ... collective global sum.
Definition mra.h:586
Function< TENSOR_RESULT_TYPE(T, R), NDIM-LDIM > project_out(const Function< R, LDIM > &g, const int dim) const
project this on the low-dim function g: h(x) = <f(x,y) | g(y)>
Definition mra.h:1552
static std::vector< std::shared_ptr< FunctionImpl< Q, D > > > vimpl(const std::vector< Function< Q, D > > &v)
Returns vector of FunctionImpl pointers corresponding to vector of functions.
Definition mra.h:1662
FunctionImpl< T, NDIM > implT
Definition mra.h:153
void clear(bool fence=true)
Clears the function as if constructed uninitialized. Optional fence.
Definition mra.h:979
void refine_general(const opT &op, bool fence=true) const
Inplace autorefines the function. Optional fence. Possible non-blocking comm.
Definition mra.h:943
static void doconj(const Key< NDIM >, Tensor< T > &t)
Definition mra.h:1066
std::pair< bool, T > eval_local_only(const Vector< double, NDIM > &xuser, Level maxlevel) const
Evaluate function only if point is local returning (true,value); otherwise return (false,...
Definition mra.h:245
void set_functor(const std::shared_ptr< FunctionFunctorInterface< T, NDIM > > functor)
Replace the current functor with the provided new one.
Definition mra.h:736
bool impl_initialized() const
Definition mra.h:149
Function< T, NDIM > & fill_nuclear_cuspy_tree(const opT &op, const size_t particle, const bool fence=true)
Definition mra.h:1351
bool is_redundant_after_merge() const
Returns true if redundant_after_merge, false otherwise. No communication.
Definition mra.h:580
return local
Definition mra.h:1453
void eval_local_only(const std::vector< coordT > &xuser, Level maxlevel, std::vector< std::pair< bool, T > > &results) const
Batched eval_local_only writing into a caller-provided buffer.
Definition mra.h:285
auto func
Definition mra.h:1531
~Function()
Destruction of any underlying implementation is deferred to next global fence.
Definition mra.h:198
Function(const Function< T, NDIM > &f)
Copy constructor is shallow. No communication, works in either basis.
Definition mra.h:185
void set_impl(const std::shared_ptr< FunctionImpl< T, NDIM > > &impl)
Replace current FunctionImpl with provided new one.
Definition mra.h:726
T operator()(double x, double y=0, double z=0, double xx=0, double yy=0, double zz=0) const
Evaluates the function at a point in user coordinates. Collective operation.
Definition mra.h:465
T inner_ext_local(const std::shared_ptr< FunctionFunctorInterface< T, NDIM > > f, const bool leaf_refine=true, const bool keep_redundant=false) const
Definition mra.h:1464
std::size_t min_nodes() const
Returns the min number of nodes on a processor.
Definition mra.h:630
constexpr std::size_t LDIM
Definition mra.h:1530
static constexpr std::size_t dimT
Definition mra.h:158
change_tree_state(operating_state, false)
bool is_nonstandard() const
Returns true if nonstandard-compressed, false otherwise. No communication.
Definition mra.h:564
void verify() const
Asserts that the function is initialized.
Definition mra.h:162
double err(const funcT &func) const
Returns an estimate of the difference ||this-func|| ... global sum performed.
Definition mra.h:520
T inner_ext(const std::shared_ptr< FunctionFunctorInterface< T, NDIM > > f, const bool leaf_refine=true, const bool keep_redundant=false) const
Definition mra.h:1480
void make_redundant(bool fence=true)
Converts the function to redundant form, i.e. sum coefficients on all levels.
Definition mra.h:892
Function< T, NDIM > & fill_nuclear_cuspy_tree(const size_t particle, const bool fence=true)
Special refinement on 6D boxes for the nuclear potentials (regularized with cusp, non-regularized wit...
Definition mra.h:1364
double check_symmetry() const
check symmetry of a function by computing the 2nd derivative
Definition mra.h:1827
void multi_to_multi_op_values(const opT &op, const std::vector< Function< T, NDIM > > &vin, std::vector< Function< T, NDIM > > &vout, const bool fence=true)
apply op on the input vector yielding an output vector of functions
Definition mra.h:1690
FunctionFactory< T, NDIM > factoryT
Definition mra.h:155
std::size_t size_local() const
Return the number of coefficients in the function on this processor.
Definition mra.h:645
const Function< T, NDIM > & compress(bool fence=true) const
Compresses the function, transforming into wavelet basis. Possible non-blocking comm.
Definition mra.h:853
bool is_initialized() const
Returns true if the function is initialized.
Definition mra.h:167
bool is_reconstructed() const
Returns true if reconstructed, false otherwise. No communication.
Definition mra.h:553
Function< T, NDIM > & chop_at_level(const int n, const bool fence=true)
remove all nodes with level higher than n
Definition mra.h:1849
MADNESS_ASSERT(is_compressed())
void vmulXX(const Function< L, NDIM > &left, const std::vector< Function< R, NDIM > > &right, std::vector< Function< T, NDIM > > &result, double tol, bool fence)
Multiplication of function * vector of functions using recursive algorithm of mulxx.
Definition mra.h:1711
double errsq_local(const funcT &func) const
Returns an estimate of the difference ||this-func||^2 from local data.
Definition mra.h:505
void make_nonstandard(bool keepleaves, bool fence=true) const
Compresses the function retaining scaling function coeffs. Possible non-blocking comm.
Definition mra.h:866
if(VERIFY_TREE) verify_tree()
Future< T > eval(const coordT &xuser) const
Evaluates the function at a point in user coordinates. Possible non-blocking comm.
Definition mra.h:211
void print_tree(std::ostream &os=std::cout) const
Process 0 prints a summary of all nodes in the tree (collective)
Definition mra.h:989
FunctionNode< T, NDIM > nodeT
Definition mra.h:154
bool redundant
Definition mra.h:1259
Future< long > evalR(const coordT &xuser) const
Evaluates the function rank at a point in user coordinates. Possible non-blocking comm.
Definition mra.h:369
Function(const factoryT &factory)
Constructor from FunctionFactory provides named parameter idiom. Possible non-blocking communication.
Definition mra.h:178
NDIM &g const
Definition mra.h:1256
void unaryop(T(*f)(T))
Inplace unary operation on function values.
Definition mra.h:1027
Function< T, NDIM > conj(bool fence=true)
Inplace complex conjugate. No communication except for optional fence.
Definition mra.h:1074
A future is a possibly yet unevaluated value.
Definition future.h:370
remote_refT remote_ref(World &world) const
Returns a structure used to pass references to another process.
Definition future.h:672
Key is the index for a node of the 2^NDIM-tree.
Definition key.h:70
Traits class to specify support of numeric types.
Definition type_data.h:56
A tensor is a multidimensional array.
Definition tensor.h:318
Tensor< T > & emul(const Tensor< T > &t)
Inplace multiply by corresponding elements of argument Tensor.
Definition tensor.h:1800
Tensor< T > & conj()
Inplace complex conjugate.
Definition tensor.h:717
A simple, fixed dimension vector.
Definition vector.h:64
void erase(const keyT &key)
Erases entry from container (non-blocking comm if remote)
Definition worlddc.h:1392
Interface to be provided by any process map.
Definition worlddc.h:122
void fence(bool debug=false)
Synchronizes all processes in communicator AND globally ensures no pending AM or tasks.
Definition worldgop.cc:176
bool set_forbid_fence(bool value)
Set forbid_fence flag to new value and return old value.
Definition worldgop.h:677
A parallel world class.
Definition world.h:132
ProcessID rank() const
Returns the process rank in this World (same as MPI_Comm_rank()).
Definition world.h:320
ProcessID size() const
Returns the number of processes in this World (same as MPI_Comm_size()).
Definition world.h:330
WorldGopInterface & gop
Global operations.
Definition world.h:207
World * get_world() const
Returns a pointer to the world.
Definition parallel_archive.h:130
An archive for storing local or parallel data wrapping a BinaryFstreamOutputArchive.
Definition parallel_archive.h:321
Objects that implement their own parallel archive interface should derive from this class.
Definition parallel_archive.h:58
static const double R
Definition csqrt.cc:46
Declaration and initialization of tree traversal functions and generic derivative.
double(* f1)(const coord_3d &)
Definition derivatives.cc:55
double(* f2)(const coord_3d &)
Definition derivatives.cc:56
const double delta
Definition dielectric_external_field.cc:119
Provides FunctionDefaults and utilities for coordinate transformation.
Provides FunctionCommonData, FunctionImpl and FunctionFactory.
Defines/implements plotting interface for functions.
Provides typedefs to hide use of templates and to increase interoperability.
auto T(World &world, response_space &f) -> response_space
Definition global_functions.cc:28
const double beta
Definition gygi_soltion.cc:62
static const double v
Definition hatom_sf_dirac.cc:20
Tensor< double > op(const Tensor< double > &x)
Definition kain.cc:508
Multidimension Key for MRA tree and associated iterators.
Implements (2nd generation) static load/data balancing for functions.
#define MADNESS_CHECK(condition)
Check a condition — even in a release build the condition is always evaluated so it can have side eff...
Definition madness_exception.h:182
#define MADNESS_EXCEPTION(msg, value)
Macro for throwing a MADNESS exception.
Definition madness_exception.h:119
#define MADNESS_ASSERT(condition)
Assert a condition that should be free of side-effects since in release builds this might be a no-op.
Definition madness_exception.h:134
#define MADNESS_CHECK_THROW(condition, msg)
Check a condition — even in a release build the condition is always evaluated so it can have side eff...
Definition madness_exception.h:207
Header to declare stuff which has not yet found a home.
static const bool VERIFY_TREE
Definition mra.h:57
Definition potentialmanager.cc:41
Namespace for all elements and tools of MADNESS.
Definition DFParameters.h:10
double abs(double x)
Definition complexfun.h:48
Function< double, NDIM > abssq(const Function< double_complex, NDIM > &z, bool fence=true)
Returns a new function that is the square of the absolute value of the input.
Definition mra.h:2901
DistributionType
some introspection of how data is distributed
Definition worlddc.h:81
@ NodeReplicated
even if there are several ranks per node
Definition worlddc.h:84
@ RankReplicated
replicate the container over all world ranks
Definition worlddc.h:83
Function< TENSOR_RESULT_TYPE(typename opT::opT, R), NDIM > apply_1d_realspace_push(const opT &op, const Function< R, NDIM > &f, int axis, bool fence=true)
Definition mra.h:2419
Function< TENSOR_RESULT_TYPE(L, R), NDIM > sub(const Function< L, NDIM > &left, const Function< R, NDIM > &right, bool fence=true)
Same as operator- but with optional fence and no automatic compression.
Definition mra.h:2141
Function< TENSOR_RESULT_TYPE(L, R), NDIM > binary_op(const Function< L, NDIM > &left, const Function< R, NDIM > &right, const opT &op, bool fence=true)
Generate new function = op(left,right) where op acts on the function values.
Definition mra.h:1935
Function< Q, NDIM > convert(const Function< T, NDIM > &f, bool fence=true)
Type conversion implies a deep copy. No communication except for optional fence.
Definition mra.h:2219
Function< TENSOR_RESULT_TYPE(Q, T), NDIM > mul(const Q alpha, const Function< T, NDIM > &f, bool fence=true)
Returns new function equal to alpha*f(x) with optional fence.
Definition mra.h:1870
std::enable_if_t< NDIM%2==0, Function< T, NDIM > > swap_particles(const Function< T, NDIM > &f)
swap particles 1 and 2
Definition mra.h:2490
TreeState
Definition funcdefaults.h:59
@ nonstandard_after_apply
s and d coeffs, state after operator application
Definition funcdefaults.h:64
@ redundant_after_merge
s coeffs everywhere, must be summed up to yield the result
Definition funcdefaults.h:66
@ reconstructed
s coeffs at the leaves only
Definition funcdefaults.h:60
@ nonstandard
s and d coeffs in internal nodes
Definition funcdefaults.h:62
@ unknown
Definition funcdefaults.h:68
@ nonstandard_with_leaves
like nonstandard, with s coeffs at the leaves
Definition funcdefaults.h:63
static void user_to_sim(const Vector< double, NDIM > &xuser, Vector< double, NDIM > &xsim)
Convert user coords (cell[][]) to simulation coords ([0,1]^ndim)
Definition funcdefaults.h:455
std::vector< Function< TENSOR_RESULT_TYPE(T, R), NDIM > > innerXX(const Function< T, LDIM > &f, const std::vector< Function< R, KDIM > > &vg, const std::array< int, CDIM > v1, const std::array< int, CDIM > v2, int task=0)
Computes the partial scalar/inner product between two functions, returns a low-dim function.
Definition mra.h:2618
std::vector< CCPairFunction< T, NDIM > > operator*(const double fac, const std::vector< CCPairFunction< T, NDIM > > &arg)
Definition ccpairfunction.h:1089
int Level
Definition key.h:59
TreeState get_tree_state(const Function< T, NDIM > &f)
get tree state of a function
Definition mra.h:2917
std::vector< CCPairFunction< T, NDIM > > operator-(const std::vector< CCPairFunction< T, NDIM > > c1, const std::vector< CCPairFunction< T, NDIM > > &c2)
Definition ccpairfunction.h:1060
Function< TENSOR_RESULT_TYPE(L, R), NDIM > mul_sparse(const Function< L, NDIM > &left, const Function< R, NDIM > &right, double tol, bool fence=true)
Sparse multiplication — left and right must be reconstructed and if tol!=0 have tree of norms already...
Definition mra.h:1911
std::string get_mra_data_dir()
Definition startup.cc:209
Function< T, NDIM > gaxpy_oop_reconstructed(const double alpha, const Function< T, NDIM > &left, const double beta, const Function< T, NDIM > &right, const bool fence=true)
Returns new function alpha*left + beta*right optional fence, having both addends reconstructed.
Definition mra.h:2104
void print(const T &t, const Ts &... ts)
Print items to std::cout (items separated by spaces) and terminate with a new line.
Definition print.h:226
response_space apply(World &world, std::vector< std::vector< std::shared_ptr< real_convolution_3d > > > &op, response_space &f)
Definition basic_operators.cc:43
@ TT_2D
Definition gentensor.h:120
@ TT_FULL
Definition gentensor.h:120
NDIM & f
Definition mra.h:2604
Function< TENSOR_RESULT_TYPE(L, R), NDIM > add(const Function< L, NDIM > &left, const Function< R, NDIM > &right, bool fence=true)
Same as operator+ but with optional fence and no automatic compression.
Definition mra.h:2096
Function< T, NDIM > symmetrize(const Function< T, NDIM > &f, const std::string symmetry, bool fence=true)
symmetrize a function
Definition mra.h:2516
Function< TENSOR_RESULT_TYPE(typename opT::opT, R), NDIM > apply_only(const opT &op, const Function< R, NDIM > &f, bool fence=true)
Apply operator ONLY in non-standard form - required other steps missing !!
Definition mra.h:2307
double inner(response_space &a, response_space &b)
Definition response_functions.h:639
double imag(double x)
Definition complexfun.h:56
Function< typename opT::resultT, NDIM > unary_op(const Function< Q, NDIM > &func, const opT &op, bool fence=true)
Out of place application of unary operation to function values with optional fence.
Definition mra.h:1949
void startup(World &world, int argc, char **argv, bool doprint=false, bool make_stdcout_nice_to_reals=true)
initialize the internal state of the MADmra library
Definition startup.cc:64
std::string type(const PairType &n)
Definition PNOParameters.h:18
static bool print_timings
Definition SCF.cc:106
std::vector< CCPairFunction< T, NDIM > > operator+(const std::vector< CCPairFunction< T, NDIM > > c1, const std::vector< CCPairFunction< T, NDIM > > &c2)
Definition ccpairfunction.h:1052
Function< T, NDIM > multiply(const Function< T, NDIM > f, const Function< T, LDIM > g, const int particle, const bool fence=true)
multiply a high-dimensional function with a low-dimensional function
Definition mra.h:2560
void load(Function< T, NDIM > &f, const std::string name)
Definition mra.h:2968
Function< T, NDIM > project(const Function< T, NDIM > &other, int k=FunctionDefaults< NDIM >::get_k(), double thresh=FunctionDefaults< NDIM >::get_thresh(), bool fence=true)
Definition mra.h:2587
double real(double x)
Definition complexfun.h:52
static XNonlinearSolver< std::vector< Function< T, NDIM > >, T, vector_function_allocator< T, NDIM > > nonlinear_vector_solver(World &world, const long nvec)
Definition nonlinsol.h:371
std::string name(const FuncType &type, const int ex=-1)
Definition ccpairfunction.h:28
void save(const Function< T, NDIM > &f, const std::string name)
Definition mra.h:2962
Function< T, KDIM+LDIM > hartree_product(const std::vector< Function< T, KDIM > > &left, const std::vector< Function< T, LDIM > > &right)
Performs a Hartree/outer product on the two given low-dimensional function vectors.
Definition mra.h:2001
Function< T, NDIM > copy(const Function< T, NDIM > &f, const std::shared_ptr< WorldDCPmapInterface< Key< NDIM > > > &pmap, bool fence=true)
Create a new copy of the function with different distribution and optional fence.
Definition mra.h:2172
static const double d
Definition nonlinschro.cc:121
Implements most functionality of separated operators.
Implements ParallelInputArchive and ParallelOutputArchive for parallel serialization of data.
double Q(double a)
Definition relops.cc:20
static const double c
Definition relops.cc:10
static const double L
Definition rk.cc:46
Definition test_ar.cc:204
void serialize(Archive &ar)
Definition mra.h:1023
T(* f)(T)
Definition mra.h:1018
void operator()(const Key< NDIM > &key, Tensor< T > &t) const
Definition mra.h:1020
SimpleUnaryOpWrapper(T(*f)(T))
Definition mra.h:1019
void serialize(Archive &ar)
Definition mra.h:956
bool operator()(implT *impl, const Key< NDIM > &key, const nodeT &t) const
Definition mra.h:952
Definition type_data.h:146
TensorArgs holds the arguments for creating a LowRankTensor.
Definition gentensor.h:134
Default load of an object via serialize(ar, t).
Definition archive.h:667
static void store(const ParallelOutputArchive< archiveT > &ar, const Function< T, NDIM > &f)
Definition mra.h:2955
Default store of an object via serialize(ar, t).
Definition archive.h:612
Tensor< double > operator()(const Key< NDIM > &key, const Tensor< double_complex > &t) const
Definition mra.h:2838
double resultT
Definition mra.h:2837
void serialize(Archive &ar)
Definition mra.h:2843
void serialize(Archive &ar)
Definition mra.h:2832
Tensor< double > operator()(const Key< NDIM > &key, const Tensor< double_complex > &t) const
Definition mra.h:2827
double resultT
Definition mra.h:2826
Tensor< double > operator()(const Key< NDIM > &key, const Tensor< double_complex > &t) const
Definition mra.h:2817
void serialize(Archive &ar)
Definition mra.h:2821
double resultT
Definition mra.h:2816
Tensor< double > operator()(const Key< NDIM > &key, const Tensor< double_complex > &t) const
Definition mra.h:2807
double resultT
Definition mra.h:2806
void serialize(Archive &ar)
Definition mra.h:2811
Definition funcimpl.h:610
returns true if the result of a hartree_product is a leaf node (compute norm & error)
Definition funcimpl.h:500
Definition funcimpl.h:564
Definition lowrankfunction.h:336
double real(double a)
Definition tdse4.cc:78
Defines and implements most of Tensor.
#define UNARY_OPTIMIZED_ITERATOR(X, x, exp)
Definition tensor_macros.h:658
AtomicInt sum
Definition test_atomicint.cc:46
double norm(const T i1)
Definition test_cloud.cc:85
int task(int i)
Definition test_runtime.cpp:4
void e()
Definition test_sig.cc:75
static const double alpha
Definition testcosine.cc:10
constexpr std::size_t NDIM
Definition testgconv.cc:54
std::size_t axis
Definition testpdiff.cc:59
#define TENSOR_RESULT_TYPE(L, R)
This macro simplifies access to TensorResultType.
Definition type_data.h:205
Defines operations on vectors of Functions.
Implements WorldContainer.
#define PROFILE_FUNC
Definition worldprofile.h:209
#define PROFILE_MEMBER_FUNC(classname)
Definition worldprofile.h:210