Area.py

import numpy as np
import random
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt

launchingArea = [
    [
        [15, 1],
        [15, 3],
        [17, 1],
        [17, 3],
    ],
    [
        [7, 10],
        [7, 11],
        [8, 10],
        [8, 11],
    ],
    [
        [23, 29],
        [23, 30],
        [24, 29],
        [24, 30],
    ]
]

destinationArea = [
    [
        [29, 9],
        [29, 10],
        [30, 9],
        [30, 10],
    ],
    [
        [7, 22],
        [7, 25],
        [9, 22],
        [9, 25],
    ],
    [
        [25, 25],
        [25, 26],
        [26, 25],
        [26, 26],
    ]
]

blockArea = [
    [15, 5],
    [16, 4],
    [8, 8],
    [8, 12],
    [25, 27],
    [23, 27],
    [15, 15],
    [13, 15],
    [5, 26],
    [18, 6],
]

class Area:
    def __init__(self):
        self.la = np.concatenate(launchingArea, axis=0)
        self.da = np.concatenate(destinationArea, axis=0)
        self.ba = blockArea

    def getLaunchPoint(self, n, low=0, high=1):
        np.random.shuffle(self.la)
        if n == 'random':
            n = int(np.round(np.random.uniform(0, len(self.la))))
        if n == 'all':
            n = len(self.la)
        result = []
        for n in range(n):
            point = random.choice(self.la)
            point = np.append(point, np.random.uniform(low, high))
            result.append(np.round(point, decimals=2))
        return np.array(result)
    
    def getDestination(self, n):
        np.random.shuffle(self.da)
        if n == 'all':
            return self.da
        else:
            point = random.choice(self.da)
            return point[0], point[1]
    
    def getBlockPoint(self, n):
        np.random.shuffle(self.ba)
        if n == 'random':
            n = int(np.round(np.random.uniform(0, len(self.ba))))
        if n == 'all':
            return self.ba
        result = []
        for n in range(n):
            point = random.choice(self.ba)
            result.append(point)
        return np.array(result)
    
    def image(self, size, save=False):
        fig, axs = plt.subplots(1, 3, figsize=(10, 3))
        plt.gray()
        for ax, title, area in zip(axs, ['launch', 'destination', 'block'], [self.la, self.da, self.ba]):
            A = np.zeros((size,size))
            for p in area:
                A[p[0], p[1]] = 1
            ax.imshow(A)
            ax.set_title(title)
            ax.get_xaxis().set_visible(False)
            ax.get_yaxis().set_visible(True)
        if save == True:
            plt.savefig("img/result.png")

a = Area()
# lp = a.getLaunchPoint('random')
# dp = a.getDestination('random')
# bp = a.getBlockPoint('random')

a.image(32)