def __init__(
self,
density_discr,
fmm_order=False,
fmm_level_to_order=None,
expansion_factory=None,
# begin undocumented arguments
# FIXME default debug=False once everything works
debug=True):
"""
:arg fmm_order: `False` for direct calculation.
"""
LayerPotentialSourceBase.__init__(self, density_discr)
self.debug = debug
if fmm_order is not False and fmm_level_to_order is not None:
raise TypeError(
"may not specify both fmm_order and fmm_level_to_order")
if fmm_level_to_order is None:
if fmm_order is not False:
def fmm_level_to_order(kernel, kernel_args, tree, level): # noqa pylint:disable=function-redefined
return fmm_order
else:
fmm_level_to_order = False
self.density_discr = density_discr
self.fmm_level_to_order = fmm_level_to_order
if expansion_factory is None:
from sumpy.expansion import DefaultExpansionFactory
expansion_factory = DefaultExpansionFactory()
self.expansion_factory = expansion_factory
def __init__(self, density_discr,
fmm_order=False,
fmm_level_to_order=None,
expansion_factory=None,
# begin undocumented arguments
# FIXME default debug=False once everything works
debug=True):
"""
:arg fmm_order: `False` for direct calculation.
"""
LayerPotentialSourceBase.__init__(self, density_discr)
self.debug = debug
if fmm_order is not False and fmm_level_to_order is not None:
raise TypeError("may not specify both fmm_order and fmm_level_to_order")
if fmm_level_to_order is None:
if fmm_order is not False:
def fmm_level_to_order(kernel, kernel_args, tree, level): # noqa pylint:disable=function-redefined
return fmm_order
else:
fmm_level_to_order = False
self.density_discr = density_discr
self.fmm_level_to_order = fmm_level_to_order
if expansion_factory is None:
from sumpy.expansion import DefaultExpansionFactory
expansion_factory = DefaultExpansionFactory()
self.expansion_factory = expansion_factory
self.debug = debug
def __init__(
self,
density_discr,
fine_order,
qbx_order=None,
fmm_order=None,
fmm_level_to_order=None,
expansion_factory=None,
target_association_tolerance=_not_provided,
# begin experimental arguments
# FIXME default debug=False once everything has matured
debug=True,
_disable_refinement=False,
_expansions_in_tree_have_extent=True,
_expansion_stick_out_factor=0.5,
_well_sep_is_n_away=2,
_max_leaf_refine_weight=None,
_box_extent_norm=None,
_from_sep_smaller_crit=None,
_from_sep_smaller_min_nsources_cumul=None,
_tree_kind="adaptive",
_use_target_specific_qbx=None,
geometry_data_inspector=None,
cost_model=None,
fmm_backend="sumpy",
target_stick_out_factor=_not_provided):
"""
:arg fine_order: The total degree to which the (upsampled)
underlying quadrature is exact.
:arg fmm_order: `False` for direct calculation. May not be given if
*fmm_level_to_order* is given.
:arg fmm_level_to_order: A function that takes arguments of
*(kernel, kernel_args, tree, level)* and returns the expansion
order to be used on a given *level* of *tree* with *kernel*, where
*kernel* is the :class:`sumpy.kernel.Kernel` being evaluated, and
*kernel_args* is a set of *(key, value)* tuples with evaluated
kernel arguments. May not be given if *fmm_order* is given.
Experimental arguments without a promise of forward compatibility:
:arg _use_target_specific_qbx: Whether to use target-specific
acceleration by default if possible. *None* means
"use if possible".
:arg cost_model: Either *None* or an object implementing the
:class:`~pytential.qbx.cost.AbstractQBXCostModel` interface, used for
gathering modeled costs if provided (experimental)
"""
# {{{ argument processing
if fine_order is None:
raise ValueError("fine_order must be provided.")
if qbx_order is None:
raise ValueError("qbx_order must be provided.")
if target_stick_out_factor is not _not_provided:
from warnings import warn
warn(
"target_stick_out_factor has been renamed to "
"target_association_tolerance. "
"Using target_stick_out_factor is deprecated "
"and will stop working in 2018.",
DeprecationWarning,
stacklevel=2)
if target_association_tolerance is not _not_provided:
raise TypeError(
"May not pass both target_association_tolerance and "
"target_stick_out_factor.")
target_association_tolerance = target_stick_out_factor
del target_stick_out_factor
if target_association_tolerance is _not_provided:
target_association_tolerance = float(
np.finfo(density_discr.real_dtype).eps) * 1e3
if fmm_order is not None and fmm_level_to_order is not None:
raise TypeError(
"may not specify both fmm_order and fmm_level_to_order")
if _box_extent_norm is None:
_box_extent_norm = "l2"
if _from_sep_smaller_crit is None:
# This seems to win no matter what the box extent norm is
# https://gitlab.tiker.net/papers/2017-qbx-fmm-3d/issues/10
_from_sep_smaller_crit = "precise_linf"
if fmm_level_to_order is None:
if fmm_order is False:
fmm_level_to_order = False
else:
def fmm_level_to_order(kernel, kernel_args, tree, level): # noqa pylint:disable=function-redefined
return fmm_order
if _max_leaf_refine_weight is None:
if density_discr.ambient_dim == 2:
# FIXME: This should be verified now that l^2 is the default.
_max_leaf_refine_weight = 64
elif density_discr.ambient_dim == 3:
# For static_linf/linf: https://gitlab.tiker.net/papers/2017-qbx-fmm-3d/issues/8#note_25009 # noqa
# For static_l2/l2: https://gitlab.tiker.net/papers/2017-qbx-fmm-3d/issues/12 # noqa
_max_leaf_refine_weight = 512
else:
# Just guessing...
_max_leaf_refine_weight = 64
if _from_sep_smaller_min_nsources_cumul is None:
# See here for the comment thread that led to these defaults:
# https://gitlab.tiker.net/inducer/boxtree/merge_requests/28#note_18661
if density_discr.dim == 1:
_from_sep_smaller_min_nsources_cumul = 15
else:
_from_sep_smaller_min_nsources_cumul = 30
# }}}
LayerPotentialSourceBase.__init__(self, density_discr)
self.fine_order = fine_order
self.qbx_order = qbx_order
self.fmm_level_to_order = fmm_level_to_order
assert target_association_tolerance is not None
self.target_association_tolerance = target_association_tolerance
self.fmm_backend = fmm_backend
if expansion_factory is None:
from sumpy.expansion import DefaultExpansionFactory
expansion_factory = DefaultExpansionFactory()
self.expansion_factory = expansion_factory
self.debug = debug
self._disable_refinement = _disable_refinement
self._expansions_in_tree_have_extent = \
_expansions_in_tree_have_extent
self._expansion_stick_out_factor = _expansion_stick_out_factor
self._well_sep_is_n_away = _well_sep_is_n_away
self._max_leaf_refine_weight = _max_leaf_refine_weight
self._box_extent_norm = _box_extent_norm
self._from_sep_smaller_crit = _from_sep_smaller_crit
self._from_sep_smaller_min_nsources_cumul = \
_from_sep_smaller_min_nsources_cumul
self._tree_kind = _tree_kind
self._use_target_specific_qbx = _use_target_specific_qbx
self.geometry_data_inspector = geometry_data_inspector
if cost_model is None:
from pytential.qbx.cost import QBXCostModel
cost_model = QBXCostModel()
self.cost_model = cost_model
def __init__(self,
density_discr,
fine_order,
qbx_order=None,
fmm_order=None,
fmm_level_to_order=None,
to_refined_connection=None,
expansion_factory=None,
target_association_tolerance=_not_provided,
# begin undocumented arguments
# FIXME default debug=False once everything has matured
debug=True,
_refined_for_global_qbx=False,
_expansions_in_tree_have_extent=True,
_expansion_stick_out_factor=0.5,
_well_sep_is_n_away=2,
_max_leaf_refine_weight=None,
_box_extent_norm=None,
_from_sep_smaller_crit=None,
_from_sep_smaller_min_nsources_cumul=None,
_tree_kind="adaptive",
geometry_data_inspector=None,
fmm_backend="sumpy",
target_stick_out_factor=_not_provided):
"""
:arg fine_order: The total degree to which the (upsampled)
underlying quadrature is exact.
:arg to_refined_connection: A connection used for resampling from
*density_discr* the fine density discretization. It is assumed
that the fine density discretization given by
*to_refined_connection.to_discr* is *not* already upsampled. May
be *None*.
:arg fmm_order: `False` for direct calculation. May not be given if
*fmm_level_to_order* is given.
:arg fmm_level_to_order: A function that takes arguments of
*(kernel, kernel_args, tree, level)* and returns the expansion
order to be used on a given *level* of *tree* with *kernel*, where
*kernel* is the :class:`sumpy.kernel.Kernel` being evaluated, and
*kernel_args* is a set of *(key, value)* tuples with evaluated
kernel arguments. May not be given if *fmm_order* is given.
"""
# {{{ argument processing
if fine_order is None:
raise ValueError("fine_order must be provided.")
if qbx_order is None:
raise ValueError("qbx_order must be provided.")
if target_stick_out_factor is not _not_provided:
from warnings import warn
warn("target_stick_out_factor has been renamed to "
"target_association_tolerance. "
"Using target_stick_out_factor is deprecated "
"and will stop working in 2018.",
DeprecationWarning, stacklevel=2)
if target_association_tolerance is not _not_provided:
raise TypeError("May not pass both target_association_tolerance and "
"target_stick_out_factor.")
target_association_tolerance = target_stick_out_factor
del target_stick_out_factor
if target_association_tolerance is _not_provided:
target_association_tolerance = float(
np.finfo(density_discr.real_dtype).eps) * 1e3
if fmm_order is not None and fmm_level_to_order is not None:
raise TypeError("may not specify both fmm_order and fmm_level_to_order")
if _box_extent_norm is None:
_box_extent_norm = "l2"
if _from_sep_smaller_crit is None:
# This seems to win no matter what the box extent norm is
# https://gitlab.tiker.net/papers/2017-qbx-fmm-3d/issues/10
_from_sep_smaller_crit = "precise_linf"
if fmm_level_to_order is None:
if fmm_order is False:
fmm_level_to_order = False
else:
def fmm_level_to_order(kernel, kernel_args, tree, level): # noqa pylint:disable=function-redefined
return fmm_order
if _max_leaf_refine_weight is None:
if density_discr.ambient_dim == 2:
# FIXME: This should be verified now that l^2 is the default.
_max_leaf_refine_weight = 64
elif density_discr.ambient_dim == 3:
# For static_linf/linf: https://gitlab.tiker.net/papers/2017-qbx-fmm-3d/issues/8#note_25009 # noqa
# For static_l2/l2: https://gitlab.tiker.net/papers/2017-qbx-fmm-3d/issues/12 # noqa
_max_leaf_refine_weight = 512
else:
# Just guessing...
_max_leaf_refine_weight = 64
if _from_sep_smaller_min_nsources_cumul is None:
# See here for the comment thread that led to these defaults:
# https://gitlab.tiker.net/inducer/boxtree/merge_requests/28#note_18661
if density_discr.dim == 1:
_from_sep_smaller_min_nsources_cumul = 15
else:
_from_sep_smaller_min_nsources_cumul = 30
# }}}
LayerPotentialSourceBase.__init__(self, density_discr)
self.fine_order = fine_order
self.qbx_order = qbx_order
self.fmm_level_to_order = fmm_level_to_order
assert target_association_tolerance is not None
self.target_association_tolerance = target_association_tolerance
self.fmm_backend = fmm_backend
# Default values are lazily provided if these are None
self._to_refined_connection = to_refined_connection
if expansion_factory is None:
from sumpy.expansion import DefaultExpansionFactory
expansion_factory = DefaultExpansionFactory()
self.expansion_factory = expansion_factory
self.debug = debug
self._refined_for_global_qbx = _refined_for_global_qbx
self._expansions_in_tree_have_extent = \
_expansions_in_tree_have_extent
self._expansion_stick_out_factor = _expansion_stick_out_factor
self._well_sep_is_n_away = _well_sep_is_n_away
self._max_leaf_refine_weight = _max_leaf_refine_weight
self._box_extent_norm = _box_extent_norm
self._from_sep_smaller_crit = _from_sep_smaller_crit
self._from_sep_smaller_min_nsources_cumul = \
_from_sep_smaller_min_nsources_cumul
self._tree_kind = _tree_kind
self.geometry_data_inspector = geometry_data_inspector