예제 #1
0
def main(argv):
    # --------------------------------------------------------------------------
    #                    PARSE INPUT ARGUMENTS
    # --------------------------------------------------------------------------

    # intitialize default arguments
    which_arms = ['0', '1']
    which_charge = ['0', '1']
    simulate_error = False
    narrow_fit = False
    d_nbins = 60

    # get/parse command line arguments and set variables accordingly
    try:
        opts, args = getopt.getopt(argv, "sb:nha:c:",
                                   ['verbose', 'dnbins=',
                                    'narrow', 'simerr'])
        for opt, arg in opts:
            if opt == '-h':  # help
                print 'gpr_fitting.py -a <armoption> -c <chargeoption>'
                sys.exit()
            elif opt == '-a':  # arm option
                # the expected input argument is 0, 1 or 2 for south
                # north or both respectively
                # which_arms will end up being a list of characters
                # containing one or both of '0' or '1'
                if int(arg) < 2:
                    which_arms = [arg]
                elif int(arg) == 2:
                    which_arms = ['0', '1']
                else:
                    print 'ERROR: bad arm option!'
                    sys.exit()
            elif opt == '-c':  # charge option
                # the expected input argument is 0, 1 or 2 for negative
                # positive or both respectively
                # which_charge will end up being a list of characters
                # containing one or both of '0' or '1'
                if int(arg) < 2:
                    which_charge = [arg]
                elif int(arg) > 2:
                    which_charge = ['0', '1']
                else:
                    print 'ERROR: bad charge option!'
                    sys.exit()
            elif opt in ('-s', '--simerr'):
                simulate_error = True  # read previous results of fit
            elif opt in ('-n', '--narrow'):
                narrow_fit = True  # use -.1 < dw23 < .1
            elif opt in ('-b', '--dnbins'):
                d_nbins = int(arg)  # number of dw23 bins
    except getopt.GetoptError:
        print 'test.py -a <armoption> -c <chargeoption>'
        sys.exit(2)

    for a in which_arms:
        for c in which_charge:
            # initialize data objects
            data_obj = data_instance(a, c,
                                     d_nbins=d_nbins,
                                     narrow_fit=narrow_fit,
                                     simulate_error=simulate_error)

            data_obj.read_results()

            chi2, rev_chi2 = data_obj.get_chi2ndf()
            nar_chi2, nar_rev_chi2 = data_obj.get_chi2ndf('narrow')
            wing_chi2, wing_rev_chi2 = data_obj.get_chi2ndf('wings')
            print 'ac=%s chi2ndf=%.4f narchi2ndf=%.4f wingchi2ndf=%.4f' % \
                (a+c, chi2, nar_chi2, wing_chi2)
            print 'reverse ac=%s chi2ndf=%.4f narchi2ndf=%.4f wingchi2ndf=%.4f'\
                % (a+c, rev_chi2, nar_rev_chi2, wing_rev_chi2)

            data_obj.plot_result_dw23_slices()

    sys.exit(0)