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)