Mercurial Repositories > Radon Code Metrics / file diff

--- a/RadonMetrics/MaintainabilityIndexCalculator.py	Sun Apr 09 17:15:02 2017 +0200
+++ b/RadonMetrics/MaintainabilityIndexCalculator.py	Thu Jun 22 18:38:10 2017 +0200
@@ -55,7 +55,8 @@
     return __maintainabilityIndex(file, text)
 
 
-def batchMaintainabilityIndex(argumentsList, send, fx, cancelled):
+def batchMaintainabilityIndex(argumentsList, send, fx, cancelled,
+                              maxProcesses=0):
     """
     Module function to calculate the maintainability index for a batch of
     files.
@@ -69,13 +70,19 @@
     @type str
     @param cancelled reference to function checking for a cancellation
     @type function
+    @param maxProcesses number of processes to be used
+    @type int
     """
-    try:
-        NumberOfProcesses = multiprocessing.cpu_count()
-        if NumberOfProcesses >= 1:
-            NumberOfProcesses -= 1
-    except NotImplementedError:
-        NumberOfProcesses = 1
+    if maxProcesses == 0:
+        # determine based on CPU count
+        try:
+            NumberOfProcesses = multiprocessing.cpu_count()
+            if NumberOfProcesses >= 1:
+                NumberOfProcesses -= 1
+        except NotImplementedError:
+            NumberOfProcesses = 1
+    else:
+        NumberOfProcesses = maxProcesses
 
     # Create queues
     taskQueue = multiprocessing.Queue()