import h5py import pyqtgraph as pg import io_utils import noise import sys, os sys.path.append(os.path.abspath('.')) from src import era if __name__ == '__main__': fnam = sys.argv[1] # read the h5 file diff, diff_ret, support, support_ret, \ good_pix, solid_unit, solid_units_ret, \ emods, econs, efids, T, T_rav, B_rav = io_utils.read_output_h5(fnam) pow = 1.0 # display the radial average of the data diff_rav = noise.rad_av(diff * good_pix) plot = pg.plot(diff_rav**pow) # display the radial average of the retrieved diff diff_ret = np.abs(np.fft.fftn(solid_units_ret[0]))**2 diff_ret_rav = noise.rad_av(diff_ret) + B_rav plot.plot(diff_rav**pow, pen=pg.mkPen('b')) # display the radial average of the background plot.plot(B_rav**pow, pen=pg.mkPen('g')) # display the radial average of the background
args = parser.parse_args() # check that args.ini exists if not os.path.exists(args.path): raise NameError('output h5 file does not exist: ' + args.path) return args if __name__ == '__main__': signal.signal(signal.SIGINT, signal.SIG_DFL) # allow Control-C args = parse_cmdline_args() if args.inout == 'input': # read the h5 file diff, support, good_pix, solid_unit, params = io_utils.read_input_h5(args.path) ex = Show_input(diff, support, good_pix, solid_unit) elif args.inout == 'output': # read the h5 file diff, diff_ret, support, support_ret, \ good_pix, solid_unit, solid_units_ret, \ emods, econs, efids, PRTF, PRTF_rav, PSD, PSD_I, B_rav = io_utils.read_output_h5(args.path) ex = Application(diff, diff_ret, support, support_ret, \ good_pix, solid_unit, solid_units_ret, \ emods, econs, efids, PRTF, PRTF_rav, PSD, PSD_I, B_rav)