def compute(self, **kwargs):
a = super(GenotypeDaskArray, self).compute(**kwargs)
g = _ndarray.GenotypeArray(a)
if self.mask:
m = self.mask.compute(**kwargs)
g.mask = m
return g
def __getitem__(self, *args):
out = super(GenotypeChunkedArray, self).__getitem__(*args)
if hasattr(out, 'shape') \
and len(self.shape) == len(out.shape) \
and self.shape[2] == out.shape[2]:
# dimensionality and ploidy preserved
out = _ndarray.GenotypeArray(out)
if self.mask is not None:
# attempt to slice mask too
m = self.mask.__getitem__(*args)
out.mask = m
return out
def f(block, bmapping):
g = _ndarray.GenotypeArray(block)
m = bmapping[:, 0, :]
return g.map_alleles(m)
def f(block, bmask):
g = _ndarray.GenotypeArray(block)
g.mask = bmask[:, :, 0]
return g.count_alleles(max_allele=max_allele)[:, None, :]
def f(block):
gb = _ndarray.GenotypeArray(block)
return gb.count_alleles(max_allele=max_allele)[:, None, :]
def f(block, bmask):
g = _ndarray.GenotypeArray(block)
g.mask = bmask[:, :, 0]
method = getattr(g, method_name)
return method(**kwargs)
def f(block):
g = _ndarray.GenotypeArray(block)
method = getattr(g, method_name)
return method(**kwargs)