Mercurial > repos > agpetit > calculate_diameter
annotate estimate_nb_sub_trajectories.py @ 10:afd0260e676f draft
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit 3111e03da3d8644ceaf94e9796f6c4206d3fe440-dirty"
author | agpetit |
---|---|
date | Tue, 07 Jun 2022 12:04:45 +0000 |
parents | e5cf7698a2af |
children | f5064c93f7ab |
rev | line source |
---|---|
5
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
1 #!/usr/bin/env python3 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
2 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
3 # coding: utf-8 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
4 """ |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
5 The script allows to estimate the number of sub-trajectories to obtain. |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
6 It also allows to split the trajectory into a number of sub-trajectories. |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
7 # USAGE : estimate_nb_sub_trajectories.py -c : file obtaining with gmx check |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
8 -log : name of log file (optional) -d : output directory (optional) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
9 -o : name of output file (optional) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
10 -f : desired number of frames per sub-trajectory (optional) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
11 -start : start time of the trajectory (optional) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
12 -end : end time of the trajectory (optional) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
13 """ |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
14 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
15 __all__ = [] |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
16 __author__ = "Agnès-Elisabeth Petit" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
17 __date__ = "30/05/2022" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
18 __version__ = "0.8" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
19 __copyright__ = "(c) 2022 CC-BY-NC-SA" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
20 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
21 # Library import |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
22 import argparse |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
23 import logging |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
24 import os |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
25 import sys |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
26 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
27 import numpy as np |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
28 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
29 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
30 def test_setup(): |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
31 global args |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
32 args = parse_arguments() |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
33 args.verbose = True |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
34 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
35 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
36 def parse_arguments(): |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
37 parser = argparse.ArgumentParser( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
38 description="The script allows to estimate the number of " |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
39 "sub-trajectories to obtain. It also allows to split" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
40 " the trajectory into a number of sub-trajectories.", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
41 formatter_class=argparse.ArgumentDefaultsHelpFormatter, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
42 prefix_chars="-", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
43 add_help=True, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
44 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
45 parser.add_argument( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
46 "-v", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
47 "--verbose", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
48 action="store_true", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
49 help="""Information messages to stderr""", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
50 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
51 parser.add_argument( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
52 "-c", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
53 "--input_check", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
54 type=str, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
55 nargs=1, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
56 help=""".txt file obtained with gmx check -f. |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
57 It contains information about the trajectory""", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
58 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
59 parser.add_argument( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
60 "-log", "--log_output", type=str, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
61 default="log/estimate_nb_sub_trajectories.log", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
62 help="""Output for log file. Default : |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
63 log/estimate_nb_sub_trajectories.log""" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
64 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
65 parser.add_argument( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
66 "-d", "--output_directory", type=str, nargs=1, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
67 default="./", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
68 help="""It's output Directory. Default : ./""" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
69 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
70 parser.add_argument( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
71 "-o", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
72 "--output_file", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
73 type=str, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
74 default="estimated_number_of_sub_trajectories.tsv", |
9
e5cf7698a2af
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit 434f3e872798e786f0c6a88939455692f711496c"
agpetit
parents:
5
diff
changeset
|
75 help="""Output file. Default : |
e5cf7698a2af
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit 434f3e872798e786f0c6a88939455692f711496c"
agpetit
parents:
5
diff
changeset
|
76 estimated_number_of_sub_trajectories.tsv""", |
5
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
77 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
78 parser.add_argument( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
79 "-f", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
80 "--nb_frames", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
81 type=int, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
82 default=30, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
83 help="""Number of frames per sub-trajectory. Default : 30""", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
84 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
85 parser.add_argument( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
86 "-start", "--start_traj", type=str, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
87 help="""Start of the trajectory to be cut""" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
88 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
89 parser.add_argument( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
90 "-end", "--end_traj", type=str, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
91 help="""End of the trajectory to be cut""" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
92 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
93 return parser.parse_args() |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
94 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
95 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
96 def search_nbr_steps_time_step(txt_file): |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
97 """ |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
98 Description : Keeping the number of frames of the complete trajectory and |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
99 the time between each frame. |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
100 param txt_file: file obtained with gmx check. |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
101 return: list that contains the number of frames of the complete trajectory |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
102 and time between each frame. |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
103 """ |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
104 if args.verbose: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
105 logging.info("\nFunction search_nbr_steps_time_step") |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
106 logging.info("The input file is " + txt_file) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
107 with open(txt_file, "r") as f: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
108 len_traj = "" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
109 time_step = "" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
110 for li in f: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
111 li = li.rstrip() |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
112 if li.startswith("Step"): |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
113 li2 = li.split() |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
114 len_traj = int(li2[1]) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
115 time_step = int(li2[2]) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
116 if args.verbose: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
117 logging.info("The length of the trajectory is " + str(len_traj)) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
118 logging.info( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
119 "The elapsed time between two steps is : " + str(time_step) + " ps" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
120 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
121 logging.info("End search_nbr_steps_time_step functions") |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
122 return len_traj, time_step |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
123 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
124 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
125 def estimate_nbr_sub_trajectories(nbr_step_time_step, nbr_frames_traj, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
126 out_file): |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
127 """ |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
128 Description: Creation of a tsv file that contains the number of frames |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
129 of the complete trajectory, the number of sub-trajectories to create, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
130 the duration between each frame and the number of frames |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
131 per sub-trajectory. |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
132 param nbr_step_time_step: list which contains the number of frames of |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
133 the complete trajectory and the time between each frame |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
134 param nbr_frames_traj: number of frames per sub-trajectory |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
135 param out_file: output file name |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
136 return: list that contains the number of frames of the complete trajectory, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
137 the number of sub-trajectories to create, the duration between |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
138 each frame and the number of frames per sub-trajectory. |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
139 output: tsv file |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
140 """ |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
141 if args.verbose: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
142 logging.info("\nFunction estimate_nbr_means") |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
143 logging.info("The length of the trajectory is " + |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
144 str(nbr_step_time_step[0])) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
145 logging.info( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
146 "The elapsed time between two steps is : " |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
147 + str(nbr_step_time_step[1]) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
148 + " ps" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
149 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
150 logging.info("The output file is " + str(out_file)) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
151 name_columns = [ |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
152 "Length_trajectory (frames)", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
153 "Number_sub_trajectories", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
154 "Time_steps (ps)", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
155 "Number_frames_per_sub_trajectory", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
156 "Start_trajectory (frames)", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
157 "End_trajectory (frames)", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
158 ] |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
159 time_step = nbr_step_time_step[1] |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
160 if nbr_step_time_step[2] is not None: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
161 start_traj = nbr_step_time_step[2] |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
162 else: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
163 start_traj = 0 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
164 if nbr_step_time_step[3] is not None: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
165 end_traj = nbr_step_time_step[3] |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
166 else: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
167 end_traj = nbr_step_time_step[0] - 1 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
168 if args.verbose: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
169 logging.info( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
170 "The first frame of the trajectory is the number " |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
171 + str(start_traj) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
172 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
173 logging.info("The first frame of the trajectory is the number " |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
174 + str(end_traj)) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
175 len_traj = end_traj - start_traj + 1 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
176 n_sub_traj = len_traj // nbr_frames_traj |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
177 if args.verbose: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
178 logging.info("The estimated number of sub-trajectories is : " |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
179 + str(n_sub_traj)) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
180 list_values = [ |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
181 str(len_traj), |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
182 str(n_sub_traj), |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
183 str(time_step), |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
184 str(nbr_frames_traj), |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
185 str(start_traj), |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
186 str(end_traj), |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
187 ] |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
188 tab_values = np.asarray([name_columns, list_values]) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
189 np.savetxt(out_file, tab_values, delimiter="\t", fmt="%s") |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
190 if args.verbose: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
191 logging.info("Save table in the file : " + str(out_file)) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
192 logging.info("End estimate_nbr_sub_trajectories function") |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
193 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
194 |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
195 if __name__ == "__main__": |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
196 args = parse_arguments() |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
197 if args.output_directory[0].endswith("/"): |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
198 out_directory = args.output_directory[0] |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
199 else: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
200 out_directory = args.output_directory[0] + "/" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
201 if args.verbose: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
202 if "/" in args.log_output: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
203 log_dir = args.log_output.rsplit("/", 1)[0] |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
204 else: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
205 log_dir = "log/" |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
206 log_file = args.log_output |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
207 if not os.path.exists(log_dir): |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
208 os.makedirs(log_dir) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
209 if os.path.isfile(log_file): |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
210 os.remove(log_file) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
211 if args.log_output: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
212 logging.basicConfig( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
213 filename=log_file, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
214 format="%(levelname)s - %(message)s", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
215 level=logging.INFO, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
216 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
217 else: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
218 logging.basicConfig( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
219 filename=log_file, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
220 format="%(levelname)s - %(message)s", |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
221 level=logging.INFO, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
222 ) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
223 logging.info("verbose mode on") |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
224 nb_frames_traj = args.nb_frames |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
225 if args.start_traj and args.start_traj != "": |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
226 start_trajectory = int(args.start_traj) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
227 else: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
228 start_trajectory = None |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
229 if args.end_traj and args.end_traj != "": |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
230 end_trajectory = int(args.end_traj) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
231 else: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
232 end_trajectory = None |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
233 if args.verbose: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
234 logging.info("Start estimate number sub-trajectories") |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
235 output_file = args.output_file |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
236 if not args.input_check: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
237 print("Please enter the file created with 'gmx_check -f'") |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
238 if args.verbose: |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
239 logging.error("Please enter the file created with 'gmx_check -f'") |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
240 sys.exit(1) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
241 nb_steps_time_step = list(search_nbr_steps_time_step( |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
242 args.input_check[0])) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
243 nb_steps_time_step.append(start_trajectory) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
244 nb_steps_time_step.append(end_trajectory) |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
245 output_file = out_directory + output_file |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
246 estimate_nbr_sub_trajectories(nb_steps_time_step, nb_frames_traj, |
e504457035e5
"planemo upload for repository https://github.com/mesocentre-clermont-auvergne/aubi_piaf commit b6488400d4478d46697019485e912c38ea2202a5-dirty"
agpetit
parents:
diff
changeset
|
247 output_file) |