def solve(w,m): if w.ndim < 4: return m/w elif w.ndim == 4: # This is slower, but handles low-hit areas near the edge better iw = array_ops.eigpow(w,-1,axes=[0,1]) return enmap.samewcs(array_ops.matmul(iw,m,axes=[0,1]), m) #return array_ops.solve_masked(w,m,axes=[0,1]) else: raise NotImplementedError("Only 2d, 3d or 4d weight maps understood")
def solve(w,m): if w.ndim < 4: return m/w elif w.ndim == 4: # This is slower, but handles low-hit areas near the edge better iw = array_ops.eigpow(w,-1,axes=[0,1]) return enmap.samewcs(array_ops.matmul(iw,m,axes=[0,1]), m) #return array_ops.solve_masked(w,m,axes=[0,1]) else: raise NotImplementedError("Only 2d, 3d or 4d weight maps understood")
def mul(w, m): if w.ndim < 4: return m * w elif w.ndim == 4: return enmap.samewcs(array_ops.matmul(w, m, axes=[0, 1]), m) else: raise NotImplementedError("Only 2d, 3d or 4d weight maps understood")
return enmap.samewcs(map[utils.transpose_inds(dets, nrow, ncol)], map) # Read our map, and give each row a weight pickup = enmap.read_map(args.pickup_map) pickup = reorder(pickup, nrow, ncol, d.dets) weight = np.median((pickup[:, 1:] - pickup[:, :-1])**2, -1) weight[weight > 0] = 1 / weight[weight > 0] # Find the output pixel for each input pixel baz = pickup[:1].posmap()[1, 0] bel = baz * 0 + args.el * utils.degree ipoint = np.array([baz, bel]) opoint = ipoint[:, None, :] + d.point_offset.T[:, :, None] opix = template.sky2pix(opoint[::-1]).astype(int) # [{y,x},ndet,naz] opix = np.rollaxis(opix, 1) # [ndet,{y,x},naz] omap = enmap.zeros((3, ) + template.shape[-2:], template.wcs) odiv = enmap.zeros((3, 3) + template.shape[-2:], template.wcs) for det in range(d.ndet): omap += utils.bin_multi(opix[det], template.shape[-2:], weight[det] * pickup[det]) * d.det_comps[det, :, None, None] odiv += utils.bin_multi( opix[det], template.shape[-2:], weight[det]) * d.det_comps[ det, :, None, None, None] * d.det_comps[det, None, :, None, None] odiv = enmap.samewcs(array_ops.eigpow(odiv, -1, axes=[0, 1]), odiv) omap = enmap.samewcs(array_ops.matmul(odiv, omap, axes=[0, 1]), omap) enmap.write_map(args.ofile, omap)
def M(self, x): map = self.dof.unzip(x) map[:] = array_ops.matmul(self.prec, map, [0, 1]) return self.dof.zip(map)
def __call__(self, m): for idtile, mtile in zip(self.idiv.tiles, m.tiles): mtile[:] = array_ops.matmul(idtile, mtile, axes=[0, 1])
def __call__(self, m): m[:] = array_ops.matmul(self.idiv, m, axes=[0, 1])
def mul(mat, vec, axes=[0, 1]): return enmap.samewcs( array_ops.matmul(mat.astype(vec.dtype), vec, axes=axes), mat, vec)
def mul(w,m): if w.ndim < 4: return m*w elif w.ndim == 4: return enmap.samewcs(array_ops.matmul(w,m, axes=[0,1]),m) else: raise NotImplementedError("Only 2d, 3d or 4d weight maps understood")
def mul(mat, vec, axes): return array_ops.matmul(mat.astype(vec.dtype), vec, axes=axes)
def M(self, x): map = self.dof.unzip(x) map[:] = array_ops.matmul(self.prec, map, [0,1]) return self.dof.zip(map)
d.boresight[2] = args.el # In degrees, calibrated in next step d = actdata.calibrate(d, exclude=["autocut"]) def reorder(map, nrow, ncol, dets): return enmap.samewcs(map[utils.transpose_inds(dets,nrow,ncol)],map) # Read our map, and give each row a weight pickup = enmap.read_map(args.pickup_map) pickup = reorder(pickup, nrow, ncol, d.dets) weight = np.median((pickup[:,1:]-pickup[:,:-1])**2,-1) weight[weight>0] = 1/weight[weight>0] # Find the output pixel for each input pixel baz = pickup[:1].posmap()[1,0] bel = baz*0 + args.el * utils.degree ipoint = np.array([baz,bel]) opoint = ipoint[:,None,:] + d.point_offset.T[:,:,None] opix = template.sky2pix(opoint[::-1]).astype(int) # [{y,x},ndet,naz] opix = np.rollaxis(opix, 1) # [ndet,{y,x},naz] omap = enmap.zeros((3,)+template.shape[-2:], template.wcs) odiv = enmap.zeros((3,3)+template.shape[-2:], template.wcs) for det in range(d.ndet): omap += utils.bin_multi(opix[det], template.shape[-2:], weight[det]*pickup[det]) * d.det_comps[det,:,None,None] odiv += utils.bin_multi(opix[det], template.shape[-2:], weight[det]) * d.det_comps[det,:,None,None,None] * d.det_comps[det,None,:,None,None] odiv = enmap.samewcs(array_ops.eigpow(odiv, -1, axes=[0,1]), odiv) omap = enmap.samewcs(array_ops.matmul(odiv, omap, axes=[0,1]), omap) enmap.write_map(args.ofile, omap)