Skip to content

pam.report.stringify

ActColour(colour=True) #

Source code in src/pam/report/stringify.py 
19
20
21
22
23
def __init__(self, colour=True) -> None:
    if colour:
        self._queue = inf_yield(self._col_queue)
    else:
        self._queue = inf_yield(self._bw_queue)

mapping = {'travel': 232} class-attribute instance-attribute #

paint(act, text) #

Source code in src/pam/report/stringify.py 
25
26
27
28
def paint(self, act, text):
    if act not in self.mapping:
        self.mapping[act] = next(self._queue)
    return f"\033[38;5;{self.mapping[act]}m{text}\033[0m"

inf_yield(queue) #

Source code in src/pam/report/stringify.py 
 8
 9
10
11
def inf_yield(queue: list):
    while True:
        for i in queue:
            yield i

stringify_plan(plan_array, mapping, colourer) #

Source code in src/pam/report/stringify.py 
51
52
def stringify_plan(plan_array, mapping, colourer):
    return "".join([colourer.paint(mapping[i], "▇") for i in plan_array])

stringify_plans(plans_path, simplify_pt_trips=False, crop=False, colour=True, width=101) #

Source code in src/pam/report/stringify.py 
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
def stringify_plans(
    plans_path, simplify_pt_trips: bool = False, crop: bool = False, colour=True, width=101
):
    print(f"Loading plan sequences from {plans_path}.")
    encoder = PlansToCategorical(bin_size=int(86400 / width), duration=86400)
    colourer = ActColour(colour=colour)
    for person in stream_matsim_persons(plans_path, simplify_pt_trips=simplify_pt_trips, crop=crop):
        apply_jitter_to_plan(
            plan=person.plan, jitter=timedelta(minutes=30), min_duration=timedelta(minutes=5)
        )
        encoded = encoder.encode(person.plan)
        string = stringify_plan(plan_array=encoded, mapping=encoder.index_to_act, colourer=colourer)
        print(person.pid, string)

    print()
    print("Key:")
    for act in encoder.act_to_index:
        print(f"{colourer.paint(act, '▇▇▇▇▇▇▇▇▇▇')}: {act}")