Commit 8cb00ad7 authored by Saurabh Deshpande's avatar Saurabh Deshpande

Update post_process.py

parent 851909b0
#---------------- For mesh visulisation of SOFA and CNN solutions- Saurabh Deshpande ---------------------------#
import Sofa import Sofa
import numpy as np import numpy as np
import sys import sys
...@@ -71,18 +73,6 @@ class CantileverBeam(Sofa.PythonScriptController): ...@@ -71,18 +73,6 @@ class CantileverBeam(Sofa.PythonScriptController):
visuNode.createObject('OglModel', name='visual', position = "@topology_sofa.position", quads = "@topology_sofa.quads") visuNode.createObject('OglModel', name='visual', position = "@topology_sofa.position", quads = "@topology_sofa.quads")
visuNode.createObject('VisualStyle',displayFlags='showWireframe') visuNode.createObject('VisualStyle',displayFlags='showWireframe')
# df_step = pd.read_csv('Data/first_step.csv', names=['f', 'u'], header = None)
# step_one = df['u'].values
# self.step_node = self.cube.createChild('step_node')
# self.step_topology = self.nn_node.createObject('MeshTopology', name = 'topology_nn', position = step_one.tolist(), quads ='@../topology.quads' )
# # self.nn_node.createObject('MechanicalObject', template = "Vec3d", name = "dofs", position = "@topology_nn.position")
# visuNode = self.nn_node.createChild('visu')
# visuNode.createObject('OglModel', name='visual', position = "@topology_nn.position", quads = "@topology_nn.quads")
# #visuNode.createObject('IdentityMapping')
def createGraph(self, rootNode, node_name,force): def createGraph(self, rootNode, node_name,force):
self.cube = rootNode.createChild(node_name) self.cube = rootNode.createChild(node_name)
...@@ -105,50 +95,6 @@ class CantileverBeam(Sofa.PythonScriptController): ...@@ -105,50 +95,6 @@ class CantileverBeam(Sofa.PythonScriptController):
#print(commandLineArguments) #print(commandLineArguments)
pass pass
# first_step = np.array(self.dofs.position)
# np.savetxt("Data/first_step.csv",first_step,delimiter=",")
# init_position = np.array(self.dofs.rest_position)
# if self.node_name == self.n_total -1:
#print('And the stored values are ')
#print(self.store)
#print(self.store)
# store = store - init_position #works only when intitial entry is zero
# #print(self.store)
# #print(init_position)
# store = store.reshape(12096,1)
# f_value = np.zeros((store.shape[0],store.shape[1]))
# f_value[4] = self.force[1] #same as the input force
# # for i in range(self.n_total):
# # f_value[12096*i] = f[i]
# data=np.hstack((f_value,store))
# df = pd.DataFrame(data, index = None, columns = None)
#print(df)
#for saving underformed
#np.savetxt("Data/undeformed.csv", init_position, delimiter=",")
# with open('Data/test_topology.csv', 'a') as f:
# df.to_csv(f, sep = ',', header = False ,index=False)
# with open('Data/SOFA_data_test10.csv','a') as fd:
# fd.write(data)
#np.savetxt("Data/SOFA_data_test"+ str(int(self.force[1]))+ ".csv", data, delimiter=",")
#sys.exit(0)
#sys.exit(0)
# self.node.animate = False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment