Skip to content

Commandline interface

main(argv)

Main function for the QDMpy command line interface.

Source code in QDMpy/cli/calculate_QDMio.py 
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
@generate_doc
def main(argv):
    """
    Main function for the QDMpy command line interface.
    """
    tstart = time.process_time()

    parser = argparse.ArgumentParser(
        description="Calculate the B111 field from ODMR data recorded with QDMio made QDM"
    )
    parser.add_argument(
        "-i",
        "--input",
        help="input path, location of the QDM data files and LED/laser images.",
        required=True,
    )
    parser.add_argument(
        "-b",
        "--binfactor",
        type=int,
        help="Binning factor of the ODMR data. Default: 1",
        default=1,
        required=False,
    )
    parser.add_argument(
        "-m",
        "--model",
        type=str,
        help="Type of model used in the experiment. Default: 'auto'",
        default='auto',
        required=False,
    )
    parser.add_argument(
        "-gf",
        "--globalfluorescence",
        type=float,
        help="Global fluorescence of the sample. Default: 0.2",
        default=0.2,
        required=False,
    )
    parser.add_argument(
        "--debug",
        help="sets logging to DEBUG level",
        action="store_true",
        default=False,
        required=False,
    )

    args = parser.parse_args()

    if args.debug:
        QDMpy.LOG.setLevel("DEBUG")
    else:
        QDMpy.LOG.setLevel("INFO")

    qdm_obj = QDM.from_qdmio(args.input, model_name=args.diamond)
    qdm_obj.bin_data(bin_factor=args.binfactor)
    qdm_obj.correct_glob_fluorescecne(glob_fluorescence=args.globalfluorescence)
    qdm_obj.fit_odmr()
    qdm_obj.export_qdmio()
    QDMpy.LOG.info(f"QDMpy finished in {time.process_time() - tstart:.2f} seconds")