lurking_navigation/static_heatmap.py at master · WSU-RAS/lurking_navigation · GitHub

1
2
3
4
5
6
7
8
9
10
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
72
73
74
75
76
77
78
79
80
"""
    Static Heatmap
"""
import sys

from config import Config
from heatmap import Heatmap, DataLine
from weighted_average import WeightedAverage

class StaticHeatmap(Heatmap):
    def __init__(self, sensor_list_filepath, smarthome_data_filepath, config):
        Heatmap.__init__(self, sensor_list_filepath, config)
        self._load_from_file(smarthome_data_filepath)

    def _load_from_file(self, smarthome_data_filepath):
        """ load heatmap from filepath

        get all the data out from a file and load into our heatmap
        then normalize our heatmap.

        """
        # build the heatmap
        smarthome_data_file = open(smarthome_data_filepath, 'r')

        next_point = self._get_next_point(smarthome_data_file, self.sensor_map, self.ignored_sensors)

        while next_point is not None:
            # add to the sensor map
            if next_point in self.point_map:
                self.point_map[next_point] += 1
            else:
                self.point_map[next_point] = 1

            # get the next point
            next_point = self._get_next_point(smarthome_data_file, self.sensor_map, self.ignored_sensors)

        # normalize the heatmap
        self._normalize_heatmap()

    def _get_next_point(self, smarthome_data_file, sensor_map, ignored_sensors):
        while True:
            next_line = smarthome_data_file.readline()

            # check if the file is empty
            if len(next_line) <= 0:
                return None

            data = DataLine(next_line)

            if data.sensor_type == "Control4-Motion" and data.message == "ON" and data.sensor_name not in ignored_sensors:
                # found a motion sensor trigger on the 1st floor
                point = sensor_map[data.sensor_name]
                return point

    def _normalize_heatmap(self):
        """ normalize the heatmap

        divide each value of the point_map by the sum of all values.

        """
        # sum up all the values
        sum = 0
        for key, value in self.point_map.iteritems():
            sum += value

        # divide each entry by the sum
        for key, value in self.point_map.iteritems():
            new_value = value / float(sum)
            self.point_map[key] = new_value

if __name__ == "__main__":
    # Take in the file we will construct a heatmap from
    sensor_list_filepath = sys.argv[1]
    smarthome_data_filepath = sys.argv[2]
    config_filepath = sys.argv[3]

    config = Config(config_filepath)

    static_heatmap = StaticHeatmap(sensor_list_filepath, smarthome_data_filepath, config)
    static_heatmap.display_heatmap()