def to_point(self, centers, falloff):
n = len(centers)
if n == 1:
sfield = SvScalarFieldPointDistance(centers[0], falloff=falloff)
vfield = SvVectorFieldPointDistance(centers[0], falloff=falloff)
elif self.point_mode == 'AVG':
sfields = [
SvScalarFieldPointDistance(center, falloff=falloff)
for center in centers
]
sfield = SvMergedScalarField('AVG', sfields)
vfields = [
SvVectorFieldPointDistance(center, falloff=falloff)
for center in centers
]
vfield = SvAverageVectorField(vfields)
elif self.point_mode == 'MIN':
kdt = kdtree.KDTree(len(centers))
for i, v in enumerate(centers):
kdt.insert(v, i)
kdt.balance()
vfield = SvKdtVectorField(kdt=kdt, falloff=falloff)
sfield = SvKdtScalarField(kdt=kdt, falloff=falloff)
else: # SEP
sfield = [
SvScalarFieldPointDistance(center, falloff=falloff)
for center in centers
]
vfield = [
SvVectorFieldPointDistance(center, falloff=falloff)
for center in centers
]
return vfield, sfield
def to_point(self, centers, falloff):
if self.metric == 'DISTANCE':
metric_single = 'EUCLIDEAN'
else:
metric_single = self.metric
n = len(centers)
if n == 1:
sfield = SvScalarFieldPointDistance(centers[0],
falloff=falloff,
metric=metric_single,
power=self.get_power())
vfield = SvVectorFieldPointDistance(centers[0],
falloff=falloff,
metric=metric_single,
power=self.get_power())
elif self.merge_mode == 'AVG':
sfields = [
SvScalarFieldPointDistance(center,
falloff=falloff,
metric=metric_single,
power=self.get_power())
for center in centers
]
sfield = SvMergedScalarField('AVG', sfields)
vfields = [
SvVectorFieldPointDistance(center,
falloff=falloff,
metric=metric_single,
power=self.get_power())
for center in centers
]
vfield = SvAverageVectorField(vfields)
elif self.merge_mode == 'MIN':
kdt = SvKdTree.new(SvKdTree.best_available_implementation(),
centers,
power=self.get_power())
vfield = SvKdtVectorField(kdt=kdt, falloff=falloff)
sfield = SvKdtScalarField(kdt=kdt, falloff=falloff)
else: # SEP
sfield = [
SvScalarFieldPointDistance(center,
falloff=falloff,
metric=metric_single,
power=self.get_power())
for center in centers
]
vfield = [
SvVectorFieldPointDistance(center,
falloff=falloff,
metric=metric_single,
power=self.get_power())
for center in centers
]
return vfield, sfield