code.py

import matplotlib.pyplot as plt
import numpy as np

def rotate(points, angle, origin=(0, 0)):
    angle = np.radians(angle)
    rot_matrix = np.array([[np.cos(angle), -np.sin(angle)],
                           [np.sin(angle), np.cos(angle)]])
    return np.dot(points - origin, rot_matrix.T) + origin

def should_clear_generator(p1, p2):
    forbidden_segments = [
        (np.array([0, 0]), np.array([1, 0])),
        (np.array([2, 0]), np.array([3, 0]))
    ]
    for seg in forbidden_segments:
        if np.allclose(p1, seg[0]) and np.allclose(p2, seg[1]) and  np.allclose(p2, seg[0]) and  np.allclose(p1, seg[1]):
            return True
    return False

def draw_fractal(ax, p1, p2, generator, depth, color):
    if  depth == 0:
        ax.plot([p1[0], p2[0]], [p1[1], p2[1]], color=color)
        return

    vector = p2 - p1
    length = np.linalg.norm(vector)
    scaled_generator = generator * (length / 3)

    angle = np.degrees(np.arctan2(vector[1], vector[0]))
    rotated_generator = rotate(scaled_generator, angle)
    translated_generator = rotated_generator + p1

    for i in range(len(translated_generator) - 1):
        draw_fractal(ax, translated_generator[i], translated_generator[i + 1], generator, depth - 1, color)

def main():
    fig, ax = plt.subplots(figsize=(8, 8))
    ax.set_aspect('equal')
    ax.axis('off')

    depth = int(input("Введіть глибину рекурсії: "))

    generator = np.array([
        [0, 0], [0, 1], [1, 1], [2, 1], [2, 0], [1, 0], [1, -1], [2, -1], [3, -1], [3, 0]
    ])

    square = np.array([
        [0, 0], [0, 1], [1, 1], [1, 0], [0, 0]
    ])

    for i in range(4):
        p1, p2 = square[i], square[i + 1]
        draw_fractal(ax, p1, p2, generator, depth, color='black')
    plt.show()

if __name__ == "__main__":
    main()