tab-model/sremove.py at main · blevine37/tab-model · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
def rstates(rank,eseg,bcore,nzthresh):   # comments
	"""I really should define this better"""

	import numpy as np
	import sys
	import random

	icstates = [] 	# list of unique states that have been found to have 0 elements
			# in eseg in line with the current bcore element

	p = bcore[0]
	q = bcore[1]
	sum = []

	dimH = rank

	# Though horribly inefficient, repeating the original algorithm to avoid errors, then appending this
	        # loop over the row (changes column index)
	rowstates = []
	j = 0
	while j < dimH:
		if (j != p):
			if (eseg[p][j] <= nzthresh):
				rowstates.append(j)
			pass
		pass
		j = j + 1
	pass

        # loop over the column (changes row index)
	colstates = []
	i = 0
	while i < dimH:
		if (i != q):
			if (eseg[i][q] <= nzthresh):
				colstates.append(i)
			pass
		pass
		i = i + 1
	pass

	sum = rowstates + colstates


	i = 0
	while i < dimH -1:
		j = i + 1
		while j < dimH:
			if (eseg[i][j] <= nzthresh):
				if (i == p or i == q):
					if (j != p and j != q):
						sum.append(j)
					pass
				else:
					if (j == p or j == q):
						sum.append(i)
					else:
						k = 0
						flag = 'none'
						check2 = len(sum)
						while k < check2:
							if (i == sum[k]):
								flag = 'i'
							pass
							if (j == sum[k]):
								flag = 'j'
							k = k + 1
						pass
						if (flag == 'none'):
							check = random.random()
							if (check <= 0.5):
								sum.append(i)
							else:
								sum.append(j)
							pass
						elif (flag == 'i'):
							sum.append(i)
						elif (flag == 'j'):
							sum.append(j)
						pass
					pass
				pass
			pass
			j = j + 1
		pass
		i = i + 1
	pass

	temp3 = sum.sort()
	usum = []

	usum.append(sum[0])
	i = 1
	while i < len(sum):
		j = 0
		add = 0
		while j < len(usum):
			if (usum[j] == sum[i]):
				add = 1
			pass
			j = j + 1
		pass
		if (add == 0):
			usum.append(sum[i])
		pass
		i = i + 1
	pass

#	print 'sum in sremove'
#	print sum
#	print 'usum'
#	print usum
	icstates = usum[:]

#	print 'icstates in sremove'
#	print icstates

	return icstates