def __call__(self, a, b):
"Execute the call behavior."
ma = getmask(a)
mb = getmask(b)
d1 = filled(a, self.fillx)
d2 = filled(b, self.filly)
t = self.domain(d1, d2)
if fromnumeric.sometrue(t, None):
d2 = where(t, self.filly, d2)
mb = mask_or(mb, t)
m = mask_or(ma, mb)
result = self.f(d1, d2)
return masked_array(result, m)
def make_mask(m, copy=0, flag=0):
"""make_mask(m, copy=0, flag=0)
return m as a mask, creating a copy if necessary or requested.
Can accept any sequence of integers or nomask. Does not check
that contents must be 0s and 1s.
if flag, return nomask if m contains no true elements.
"""
if m is nomask:
return nomask
elif isinstance(m, ndarray):
if m.dtype.type is MaskType:
if copy:
result = numeric.array(m, dtype=MaskType, copy=copy)
else:
result = m
else:
result = m.astype(MaskType)
else:
result = filled(m, True).astype(MaskType)
if flag and not fromnumeric.sometrue(fromnumeric.ravel(result)):
return nomask
else:
return result
