Markov-chain-simulation/make_gif_tools.py at master · vbelz/Markov-chain-simulation · 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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
from customer_tools import Customer
import numpy as np
import imageio
import os
import glob
import cv2
from matplotlib import pyplot as plt


def save_each_state(Nb_customers):
    """"
    This function simulates Nb_customers, assign a random colour
    for each of them and save all states (per minute) in the supermarket
    as png pictures.
    """
    colors = [0,1,2]

    for customer_id in range (Nb_customers):
        #Create new customer
        new_customer = Customer(customer_id)
        #Assign a random colour
        new_color = np.random.choice(colors,2)
        unique_colors = np.unique(new_color)
        nb_colors_to_change = len(unique_colors)

        list(new_customer.gen)
        states = new_customer.history


        for s_nb,s in enumerate(states):

            #Read the picture to plot thge customer position at each state
            frame = cv2.imread('./demo/market.png')

            if s == 'entrance':
                for i in range(nb_colors_to_change):

                    frame[550:640, 730:820, unique_colors[i]] = 0

            elif s == 'checkout':

                for i in range(nb_colors_to_change):

                    frame[590:680, 205:295, unique_colors[i]] = 0

            elif s == 'dairy':

                for i in range(nb_colors_to_change):

                    frame[220:310, 310:400, unique_colors[i]] = 0

            elif s == 'drinks':

                for i in range(nb_colors_to_change):

                    frame[220:310, 80:170, unique_colors[i]] = 0

            elif s == 'fruit':

                for i in range(nb_colors_to_change):

                    frame[220:310, 770:860, unique_colors[i]] = 0

            elif s == 'spices':

                for i in range(nb_colors_to_change):

                    frame[220:310, 540:630, unique_colors[i]] = 0

            else:

                print('This state does not exist')

            plt.figure(figsize=(8,5))
            plt.imshow(frame)
            #Save all states to disk
            plt.savefig(f'./demo/customer_{customer_id}_states_{s_nb}.png')
            plt.close()

def make_gif_from_states(Nb_customers):
    """
    This function reads all customers states in the order of occurence
    and make a gif of all states (and delete the png pictures from disk).
    """
    image_directory = './demo'

    list_of_images=[]

    for customer_id in range (Nb_customers):
        path_to_read = os.path.join(image_directory,f'customer_{customer_id}_*.png')
        files = glob.glob(path_to_read)
        files.sort(key=os.path.getmtime)
        for f in files:
            im = imageio.imread(f)
            list_of_images.append(im)
            os.remove(f)

    imageio.mimsave(f'./demo/customer_simulation.gif', list_of_images, duration = 0.75)