add more debug, SLE solver not working?

This commit is contained in:
Josh Holtrop 2011-03-29 20:42:04 -04:00
parent a36e4f6d08
commit 9c2bb94603

View File

@ -32,13 +32,18 @@ class Sketch(object):
dim = max(len(equations), len(shape_vars)) dim = max(len(equations), len(shape_vars))
C = zeros((dim, dim)) C = zeros((dim, dim))
d = zeros(dim) d = zeros(dim)
print 'matrix size (%d, %d)' % (dim, dim)
for i in range(len(equations)): for i in range(len(equations)):
e = equations[i] e = equations[i]
coeffs = e[0] coeffs = e[0]
for c in coeffs: for c in coeffs:
var_index = shape_vars[c[1]] var_index = shape_vars[c[1]]
C[i][var_index] = c[0] C[i][var_index] = c[0]
if c[0] != 0:
print c[1][0].name, c[1][1], 'is at', i, var_index, \
'with coefficient', c[0]
d[i] = e[1] d[i] = e[1]
print 'equ', i, 'expected value:', e[1]
# for row in range(len(equations)): # for row in range(len(equations)):
# print '[', # print '[',
# for col in range(len(shape_vars)): # for col in range(len(shape_vars)):
@ -48,9 +53,10 @@ class Sketch(object):
s = SLE(C, d) s = SLE(C, d)
r = s.solve() r = s.solve()
solved_vars = r.xf solved_vars = r.xf
print 'IGNORED THINGY: ', r.ff
print "SOLVED: ", solved_vars print "SOLVED: ", solved_vars
for i in range(len(solved_vars)): for i in range(len(solved_vars)):
varid_to_shape_var[i][0].setVar(varid_to_shape_var[i][1], varid_to_shape_var[i][0].setVar(varid_to_shape_var[i][1],
solved_vars[i]) solved_vars[i])
# for s in self.shapes: for s in self.shapes:
# print 'shape:', repr(s.vars) print 'shape:', repr(s.vars)