get proc id using two arrays

Author: jishnub

Project.toml

@@ -1,7 +1,7 @@
 name = "ParallelUtilities"
 uuid = "fad6cfc8-4f83-11e9-06cc-151124046ad0"
 authors = ["Jishnu Bhattacharya <[email protected]>"]
-version = "0.1.1"
+version = "0.1.0"
 
 [deps]
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"

src/ParallelUtilities.jl

@@ -106,31 +106,34 @@ function get_processor_id_from_split_array(iter,val::Tuple,num_procs)
 	return 0
 end
 
-function get_processor_range_from_split_array(arr₁::AbstractVector,arr₂::AbstractVector,
-	iter_section,num_procs::Integer)
+get_processor_id_from_split_array(arr₁::AbstractVector,arr₂::AbstractVector,
+	val::Tuple{Any,Any},num_procs::Integer) = 
+	get_processor_id_from_split_array(Iterators.product(arr₁,arr₂),
+		val,num_procs)
+
+function get_processor_range_from_split_array(iter,iter_section,num_procs::Integer)
 
 	if isempty(iter_section)
 		return 0:-1 # empty range
 	end
 
-	tasks_arr = collect(iter_section)
-	proc_id_start = get_processor_id_from_split_array(arr₁,arr₂,first(tasks_arr),num_procs)
-	proc_id_end = get_processor_id_from_split_array(arr₁,arr₂,last(tasks_arr),num_procs)
-	return proc_id_start:proc_id_end
-end
+	first_task = first(iter_section) 
+	proc_id_start = get_processor_id_from_split_array(iter,first_task,num_procs)
 
-function get_processor_range_from_split_array(iter,iter_section,num_procs::Integer)
-	
-	if isempty(iter_section)
-		return 0:-1 # empty range
+	last_task = first_task
+	for t in iter_section
+		last_task = t
 	end
 
-	tasks_arr = collect(iter_section)
-	proc_id_start = get_processor_id_from_split_array(iter,first(tasks_arr),num_procs)
-	proc_id_end = get_processor_id_from_split_array(iter,last(tasks_arr),num_procs)
+	proc_id_end = get_processor_id_from_split_array(iter,last_task,num_procs)
 	return proc_id_start:proc_id_end
 end
 
+get_processor_range_from_split_array(arr₁::AbstractVector,arr₂::AbstractVector,
+	iter_section,num_procs::Integer) =
+	get_processor_range_from_split_array(Iterators.product(arr₁,arr₂),
+		iter_section,num_procs)
+
 function get_index_in_split_array(iter_section,val::Tuple)
 	if isnothing(iter_section)
 		return nothing