def rfharmo(): parser = argparse.ArgumentParser('Harmonic decomposition for extracting anisotropic and isotropic features from the radial and transverse RFs') parser.add_argument('rfpath', type=str, help="Path to PRFs") parser.add_argument('-t', help="Time window from tb to te for triming RFs, NOTE: do not insert space before this argument, defaults to -2/10", metavar='tb/te', default='-2/10') parser.add_argument('-s', help="Resample RFs with sampling interval of dt", metavar='dt', default=None, type=float) parser.add_argument('-o', help="Specify output path for saving constant component.", metavar='outpath', default=None, type=str) parser.add_argument('-p', help='Figure output path, defaults to ./', metavar='figure_path', default='./', type=str) args = parser.parse_args() rfsta = RFStation(args.rfpath) if args.s is not None: rfsta.resample(args.s) twin = [float(v) for v in args.t.split('/')] rfsta.harmonic(twin[0], twin[1]) if args.o is not None: rfsta.harmo.write_constant(args.o) rfsta.harmo.plot(outpath=args.p)
def test_sub03(): rfs = RFStation('ex-rfani/SC.LTA') rfs.resample(0.1) rfs.harmonic(-2, 12) rfs.harmo.write_constant() rfs.harmo.plot()