Module: CsrMatrix::Helpers

Included in:
TwoDMatrix
Defined in:
lib/csrmatrix/helpers.rb

Instance Method Summary collapse

Instance Method Details

#count_nonzero(array) ⇒ Object

max_row



42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
 
# File 'lib/csrmatrix/helpers.rb', line 42

def count_nonzero(array) 
	# Finds all nonzero values in an array.
	# pre 	array 
	# post 	int nonzero count of array
	max_count = 0
	array.each_index do |i|
		subarray = array[i]
	  subarray.each_index do |x|
	  	if array[i][x] != 0
	  		max_count += 1
	  	end
	  end
	end
	return max_count
end

#count_total(array) ⇒ Object

depth



74
75
76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/csrmatrix/helpers.rb', line 74

def count_total(array) 
	# Counts all elements in array - assumed 2d
	# pre 	array
	# post 	int count of all elements
  max_count = 0
  array.each_index do |i|
    subarray = array[i]
    subarray.each_index do |x|
      max_count += 1
    end
  end
  return max_count
end

#depth(array) ⇒ Object

count_nonzero



58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
 
# File 'lib/csrmatrix/helpers.rb', line 58

def depth(array)
	# Code from http://stackoverflow.com/questions/9545613/getting-dimension-of-multidimensional-array-in-ruby
	# Identifies the depth of an array.
	# pre 	array
	# post 	int depth of array
   return 0 if array.class != Array
	result = 1
	array.each do |sub_a|
	  if sub_a.class == Array
	    dim = depth(sub_a)
	    result = dim + 1 if dim + 1 > result
	  end
	end
	return result
end

#max_col(array) ⇒ Object

ARRAY FUNCTIONS for pre-processing of matrix



9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
 
# File 'lib/csrmatrix/helpers.rb', line 9

def max_col(array)
	# Identifies the 'column' value of an array (eg. the number of entries in a column)
	# pre 	array
	# post 	column count of array
	values = array
	max_count = 0
	# Loop over indexes.
	values.each_index do |i|
		counter = 0
	  # Get subarray and loop over its indexes also.
	  subarray = values[i]
	  subarray.each_index do |x|
	  	counter += 1
	  end
	  if counter > max_count
	  	max_count = counter
	  end
	end
	return max_count
end

#max_row(array) ⇒ Object

max_col



30
31
32
33
34
35
36
37
38
39
40
 
# File 'lib/csrmatrix/helpers.rb', line 30

def max_row(array)
	# Identifies the 'row' value of an array (eg. the number of entries in a row)
	# pre 	array
	# post 	row count of array
	values = array
	max_count = 0
	values.each_index do |i|
	  max_count += 1
	end
	return max_count
end