folder = args.folder file_in = args.file_in save = args.save init = args.init folder_init = args.folder_init N = args.niter # ============================================================================== # OPEN PSF AND DIRTY CUBE - SKY to check results # ============================================================================== genname = os.path.join(folder, file_in) psf_name = genname + '_psf.fits' drt_name = genname + '_dirty.fits' L = 5 cube_psf = fix_dim(fits.getdata(psf_name, ext=0))[:, :, 0:L] cube_dirty = fix_dim(fits.getdata(drt_name, ext=0))[:, :, 0:L] sky_name = genname + '_sky.fits' sky = fix_dim(fits.getdata(sky_name, ext=0)) sky = np.transpose(sky)[:, :, 0:L] sky2 = np.sum(sky * sky) fig = pl.figure() ax = fig.add_subplot(1, 3, 1) ax.imshow(cube_psf[:, :, 1]) ax = fig.add_subplot(1, 3, 2) ax.imshow(sky[:, :, 1]) ax = fig.add_subplot(1, 3, 3) ax.imshow(cube_dirty[:, :, 1])
parser=argparse.ArgumentParser() parser.add_argument('-d','--dirty',help='add path to dirty image fits file') parser.add_argument('-p','--psf',help='add path to psf fits file') parser.add_argument('-N','--niter',help='add number of iterations', type=int) parser.add_argument('-s','--save',help='save output', type=int) parser.add_argument('-C','--channels',help='number of channels', type=int) args = parser.parse_args() dirtyim=args.dirty psfim=args.psf niter=args.niter Save=args.save L=args.channels psffits=fix_dim(fits.getdata(psfim, ext=0))[:,:,0:L] dirtyfits=fix_dim(fits.getdata(dirtyim, ext=0))[:,:,0:L] b= EasyMuffinSURE(psf=psffits,dirty=dirtyfits,save=Save) b.loop(niter) #u=np.load('u.npy',allow_pickle=True) #v=np.load('v.npy', allow_pickle=True) #x0=np.load('x0_tst.npy', allow_pickle=True) x1=b.x x0=np.transpose(x1) #print(x0)