import multiprocessing
import time
from time import time

import numpy as np


def matrix_multi(first, second, res, start_i, stop_i, size):
    for i in range(start_i, stop_i):
        for j in range(size):
            res[i][j] = 0
            for k in range(size):
                res[i][j] += first[i][k] * second[k][j]


def do(first, second, size, threads):
    offset = int(size / threads)
    offset_last = size % threads + offset

    processes = []
    res = np.zeros((size, size))
    start_test = time()
    for i in range(threads):
        start_ = i * offset
        stop_ = start_ + offset_last if i == threads - 1 else start_ + offset

        process = multiprocessing.Process(target=matrix_multi, args=(first, second, res, start_, stop_, size))
        processes.append(process)
        process.start()

    for p in processes:
        p.join()
    stop_test = time()
    print(f'{n}x{n}, time: {stop_test - start_test}')


if __name__ == "__main__":
    threads_count = 20  # задание кол-ва потоков
    sizes = [100, 300, 500]
    test = [[5, 3, 4], [2, 8, 3], [6, 3, 4]]
    test2 = [[4, 3, 6], [2, 8, 3], [5, 6, 4]]

    threads_counts = [1, 4, 6, 8]
    for threads in threads_counts:
        print('-------------------------------------------------')
        print(f'Threads:{threads}')
        for n in sizes:
            first_matrix = np.random.randint(3, size=(n, n))
            second_matrix = np.random.randint(3, size=(n, n))
            do(first_matrix, second_matrix, n, threads)

        print('-------------------------------------------------')