From 5eb262ece1840ff1989b0f4d0231d2d3c51b598a Mon Sep 17 00:00:00 2001
From: Impact <pascal.mouquet@ird.fr>
Date: Fri, 22 Feb 2019 14:08:54 +0400
Subject: [PATCH] add indices argument for multiprocessing
---
sen2chain/multi_processing.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/sen2chain/multi_processing.py b/sen2chain/multi_processing.py
index 71816b3..926c907 100644
--- a/sen2chain/multi_processing.py
+++ b/sen2chain/multi_processing.py
@@ -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()
--
GitLab