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
Branches
No related tags found
No related merge requests found
...@@ -4,6 +4,7 @@ import multiprocessing, subprocess ...@@ -4,6 +4,7 @@ import multiprocessing, subprocess
import os import os
from time import sleep from time import sleep
import logging import logging
from functools import partial
#~ import psutil #~ import psutil
from .products import L1cProduct, L2aProduct from .products import L1cProduct, L2aProduct
...@@ -34,14 +35,15 @@ def l2a_multiprocessing(process_list, nb_proc=4): ...@@ -34,14 +35,15 @@ def l2a_multiprocessing(process_list, nb_proc=4):
pool.close() pool.close()
pool.join() pool.join()
def multi_cldidx(l2a_identifier): def multi_cldidx(indice_list, l2a_identifier):
l2a = L2aProduct(l2a_identifier) l2a = L2aProduct(l2a_identifier)
l2a.process_cloud_mask_v2() 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) 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.close()
pool.join() pool.join()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment