def computeJacobian(self):
""" Compute the global Jacobian """
self.nJ = PUBSlib.computejnnz(self.nsurf,self.nedge,self.ngroup,self.nvert,self.surf_edge,self.edge_group,self.group_k,self.group_n,self.edge_count)
Ja, Ji, Jj = PUBSlib.computejacobian(self.nJ, self.nT, self.nD, self.nsurf, self.nedge, self.ngroup, self.nvert, self.surf_vert, self.surf_edge, self.edge_group, self.group_k, self.group_m, self.group_n, self.group_d, self.surf_index_P, self.edge_index_P, self.surf_index_C, self.edge_index_C, self.knot_index, self.edge_count, self.T)
self.J = scipy.sparse.csc_matrix((Ja,(Ji,Jj)))
self.Nuv = PUBSlib.getsurfacesizes(self.nsurf, self.nedge, self.ngroup, self.surf_edge, self.edge_group, self.group_n)
self.Np0 = numpy.zeros(self.nsurf+1)
for s in range(self.nsurf+1):
self.Np0[s] = int(sum(self.Nuv[:s,0]*self.Nuv[:s,1]))
surf = numpy.zeros(self.Np0[-1],int)
u = numpy.zeros(self.Np0[-1])
v = numpy.zeros(self.Np0[-1])
for s in range(self.nsurf):
T = PUBSlib.getsurfacet(s+1, self.Nuv[s,0], self.Nuv[s,1], self.nT, self.nsurf, self.nedge, self.surf_edge, self.surf_index_P, self.edge_index_P, self.T)
surf[self.Np0[s]:self.Np0[s+1]] = s
u[self.Np0[s]:self.Np0[s+1]] = T[:,:,0].flatten(order='F')
v[self.Np0[s]:self.Np0[s+1]] = T[:,:,1].flatten(order='F')
self.J0 = self.evaluateBases(surf, u, v, 0, 0)
self.Ju = self.evaluateBases(surf, u, v, 1, 0)
self.Jv = self.evaluateBases(surf, u, v, 0, 1)
self.nM = PUBSlib.computemnnz(self.nsurf,self.nedge,self.ngroup,self.nvert,self.surf_edge,self.edge_group,self.group_m,self.surf_index_C, self.edge_index_Q, self.vert_index_Q, self.edge_count, self.surf_c1, self.edge_c1)
Ma, Mi, Mj = PUBSlib.computedofmapping(self.nM,self.nsurf,self.nedge,self.ngroup,self.nvert,self.surf_vert,self.surf_edge,self.edge_group,self.group_m,self.surf_index_C,self.edge_index_C,self.edge_index_Q,self.vert_index_Q, self.edge_count,self.surf_c1,self.edge_c1)
self.M = scipy.sparse.csc_matrix((Ma,(Mi,Mj)))
self.JM = self.J.dot(self.M)
if self.printInfo:
print '# Jacobian non-zeros =',self.JM.nnz
def computeJacobian(self):
""" Compute the global Jacobian """
self.nJ = PUBSlib.computejnnz(self.nsurf, self.nedge, self.ngroup,
self.nvert, self.surf_edge,
self.edge_group, self.group_k,
self.group_n, self.edge_count)
Ja, Ji, Jj = PUBSlib.computejacobian(
self.nJ, self.nT, self.nD, self.nsurf, self.nedge, self.ngroup,
self.nvert, self.surf_vert, self.surf_edge, self.edge_group,
self.group_k, self.group_m, self.group_n, self.group_d,
self.surf_index_P, self.edge_index_P, self.surf_index_C,
self.edge_index_C, self.knot_index, self.edge_count, self.T)
self.J = scipy.sparse.csc_matrix((Ja, (Ji, Jj)))
self.Nuv = PUBSlib.getsurfacesizes(self.nsurf, self.nedge, self.ngroup,
self.surf_edge, self.edge_group,
self.group_n)
self.Np0 = numpy.zeros(self.nsurf + 1)
for s in range(self.nsurf + 1):
self.Np0[s] = int(sum(self.Nuv[:s, 0] * self.Nuv[:s, 1]))
surf = numpy.zeros(self.Np0[-1], int)
u = numpy.zeros(self.Np0[-1])
v = numpy.zeros(self.Np0[-1])
for s in range(self.nsurf):
T = PUBSlib.getsurfacet(s + 1, self.Nuv[s, 0], self.Nuv[s, 1],
self.nT, self.nsurf, self.nedge,
self.surf_edge, self.surf_index_P,
self.edge_index_P, self.T)
surf[self.Np0[s]:self.Np0[s + 1]] = s
u[self.Np0[s]:self.Np0[s + 1]] = T[:, :, 0].flatten(order='F')
v[self.Np0[s]:self.Np0[s + 1]] = T[:, :, 1].flatten(order='F')
self.J0 = self.evaluateBases(surf, u, v, 0, 0)
self.Ju = self.evaluateBases(surf, u, v, 1, 0)
self.Jv = self.evaluateBases(surf, u, v, 0, 1)
self.nM = PUBSlib.computemnnz(self.nsurf, self.nedge, self.ngroup,
self.nvert, self.surf_edge,
self.edge_group, self.group_m,
self.surf_index_C, self.edge_index_Q,
self.vert_index_Q, self.edge_count,
self.surf_c1, self.edge_c1)
Ma, Mi, Mj = PUBSlib.computedofmapping(
self.nM, self.nsurf, self.nedge, self.ngroup, self.nvert,
self.surf_vert, self.surf_edge, self.edge_group, self.group_m,
self.surf_index_C, self.edge_index_C, self.edge_index_Q,
self.vert_index_Q, self.edge_count, self.surf_c1, self.edge_c1)
self.M = scipy.sparse.csc_matrix((Ma, (Mi, Mj)))
self.JM = self.J.dot(self.M)
if self.printInfo:
print '# Jacobian non-zeros =', self.JM.nnz
