measure_ring_distance.py: improve output messages
[SaveMySugar/python3-savemysugar.git] / src / measure_ring_distance.py
1 #!/usr/bin/env python3
2 #
3 # measure_ring_distance - measure the time between two rings in the same call
4 #
5 # Copyright (C) 2015  Antonio Ospite <ao2@ao2.it>
6 #
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation, either version 3 of the License, or
10 # (at your option) any later version.
11 #
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 # GNU General Public License for more details.
16 #
17 # You should have received a copy of the GNU General Public License
18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
19
20 from savemysugar.cumulative_average import cumulative_average
21 from savemysugar.Modem import Modem
22 import time
23
24
25 class Rings(object):
26     previous_time = -1
27     average_distance = 0
28     min_distance = 10000
29     max_distance = -1
30     count = 0
31
32
33 def on_ring():
34     Rings.count += 1
35
36     if Rings.previous_time == -1:
37         print("First ring")
38         Rings.previous_time = time.time()
39     else:
40         new_time = time.time()
41         distance = new_time - Rings.previous_time
42         Rings.previous_time = new_time
43         Rings.min_distance = min(Rings.min_distance, distance)
44         Rings.max_distance = max(Rings.max_distance, distance)
45         Rings.average_distance = cumulative_average(Rings.average_distance,
46                                                     Rings.count - 1, distance)
47         print()
48         print("Other ring")
49         print("Ring distance: %f" % distance)
50         print("Min ring distance: %f" % Rings.min_distance)
51         print("Max ring distance %f" % Rings.max_distance)
52         print("Average distance: %f" % Rings.average_distance)
53
54
55 def measure_ring_distance(ingoing_port, outgoing_port, destination_number):
56     ingoing_modem = Modem(ingoing_port)
57     ingoing_modem.register_callback("RING", on_ring)
58     outgoing_modem = Modem(outgoing_port)
59
60     outgoing_modem.send_command("ATDT" + destination_number + ";")
61
62     try:
63         ingoing_modem.get_response_loop()
64     except KeyboardInterrupt:
65         outgoing_modem.send_command("ATH")
66         outgoing_modem.get_response()
67
68
69 def main():
70     import sys
71     if len(sys.argv) != 4:
72         print("usage: %s" % sys.argv[0],
73               "<ingoing serial port> <outgoing serial port>",
74               "<destination number>")
75         sys.exit(1)
76
77     measure_ring_distance(sys.argv[1], sys.argv[2], sys.argv[3])
78
79
80 if __name__ == "__main__":
81     main()