1 | """ |
---|
2 | Genetrack file transformer. |
---|
3 | |
---|
4 | The program may be invoked in multiple ways. As a standalone script:: |
---|
5 | |
---|
6 | python hdf_loader.py |
---|
7 | |
---|
8 | As a python module:: |
---|
9 | |
---|
10 | python -m genetrack.scripts.hdf_loader |
---|
11 | |
---|
12 | Or in other python scripts:: |
---|
13 | |
---|
14 | >>> |
---|
15 | >>> from genetrack.scripts import hdf_loader |
---|
16 | >>> hdf_loader.transform(inpname) |
---|
17 | >>> |
---|
18 | |
---|
19 | Run the script with no parameters to see the options that it takes. |
---|
20 | |
---|
21 | **Observed runtime**: insertion rate of 6 million lines per minute |
---|
22 | |
---|
23 | """ |
---|
24 | import os, sys, csv |
---|
25 | from genetrack import logger, conf, util, hdflib |
---|
26 | |
---|
27 | |
---|
28 | def transform(inpname, workdir=None, update=False): |
---|
29 | """ |
---|
30 | Creates a transform from a genetrack input file |
---|
31 | """ |
---|
32 | index = hdflib.PositionalData(fname=inpname, workdir=workdir, update=update) |
---|
33 | return index |
---|
34 | |
---|
35 | if __name__ == '__main__': |
---|
36 | import optparse |
---|
37 | |
---|
38 | usage = "usage: %prog -i inputfile" |
---|
39 | |
---|
40 | parser = optparse.OptionParser(usage=usage) |
---|
41 | |
---|
42 | # setting the input file name |
---|
43 | parser.add_option( |
---|
44 | '-i', '--input', action="store", |
---|
45 | dest="inpname", type='str', default=None, |
---|
46 | help="the input file name (required)" |
---|
47 | ) |
---|
48 | |
---|
49 | # setting the output file name |
---|
50 | parser.add_option( |
---|
51 | '-w', '--workdir', action="store", |
---|
52 | dest="workdir", type='str', default=None, |
---|
53 | help="work directory (optional)" |
---|
54 | ) |
---|
55 | |
---|
56 | # verbosity can be 0,1 and 2 (increasing verbosity) |
---|
57 | parser.add_option( |
---|
58 | '-v', '--verbosity', action="store", |
---|
59 | dest="verbosity", type="int", default=1, |
---|
60 | help="sets the verbosity (0, 1) (default=1)", |
---|
61 | ) |
---|
62 | |
---|
63 | parser.add_option("-u", "--update", |
---|
64 | action="store_true", dest="update", default=False, |
---|
65 | help="recreates the index even if it exists") |
---|
66 | |
---|
67 | options, args = parser.parse_args() |
---|
68 | |
---|
69 | # set verbosity |
---|
70 | logger.disable( options.verbosity ) |
---|
71 | |
---|
72 | # missing input file name |
---|
73 | if not options.inpname: |
---|
74 | parser.print_help() |
---|
75 | else: |
---|
76 | transform(inpname=options.inpname, workdir=options.workdir, update=options.update) |
---|