import multiprocessing
from partition_main import partition_diversity

if __name__ == "__main__":
    cpu_cores = multiprocessing.cpu_count()
    pool = multiprocessing.Pool(processes=cpu_cores)

    tasks = range(1, 9)

    image_data_list = pool.map(partition_diversity, tasks)

    pool.close()
    pool.join()

    # # 将生成的图片数据保存到本地文件
    # for i, img_data in enumerate(image_data_list, start=1):
    #     with open(f'image_{i}.png', 'wb') as f:
    #         f.write(img_data)