Skip to content
Snippets Groups Projects
Commit 5eb262ec authored by pascal.mouquet_ird.fr's avatar pascal.mouquet_ird.fr
Browse files

add indices argument for multiprocessing

parent dfb3abf2
No related branches found
No related tags found
No related merge requests found
......@@ -4,6 +4,7 @@ import multiprocessing, subprocess
import os
from time import sleep
import logging
from functools import partial
#~ import psutil
from .products import L1cProduct, L2aProduct
......@@ -34,14 +35,15 @@ def l2a_multiprocessing(process_list, nb_proc=4):
pool.close()
pool.join()
def multi_cldidx(l2a_identifier):
def multi_cldidx(indice_list, l2a_identifier):
l2a = L2aProduct(l2a_identifier)
l2a.process_cloud_mask_v2()
l2a.process_indices(["NDVI", "NDWIGAO", "NDWIMCF"], True, True)
l2a.process_indices(indice_list, True, True)
def cldidx_multiprocessing(process_list, nb_proc=4):
def cldidx_multiprocessing(process_list, indice_list=["NDVI", "NDWIGAO", "NDWIMCF"], nb_proc=4):
""" """
pool = multiprocessing.Pool(nb_proc)
results = [pool.map(multi_cldidx, process_list)]
#~ function = partial(multi_cldidx, indice_list)
results = [pool.map(partial(multi_cldidx, indice_list), process_list)]
pool.close()
pool.join()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment