def __init__(self, queue, geo_data, use_target_specific_qbx): from pytential.qbx.utils import ToHostTransferredGeoDataWrapper geo_data = ToHostTransferredGeoDataWrapper(queue, geo_data) self.geo_data = geo_data self.trav = geo_data.traversal() self.using_tsqbx = ( use_target_specific_qbx # None means use by default if possible or use_target_specific_qbx is None) ConstantOneExpansionWrangler.__init__(self, geo_data.tree())
def __init__(self, tree, filtered_targets): ConstantOneExpansionWrangler.__init__(self, tree) self.filtered_targets = filtered_targets