From 8cb00ad7f42d71252fbb41a5673627d9bbba0c64 Mon Sep 17 00:00:00 2001 From: Saurabh Deshpande Date: Tue, 26 Jan 2021 17:29:42 +0100 Subject: [PATCH] Update post_process.py --- scripts/post_process.py | 58 ++--------------------------------------- 1 file changed, 2 insertions(+), 56 deletions(-) diff --git a/scripts/post_process.py b/scripts/post_process.py index 77a03d9..fe25ac0 100644 --- a/scripts/post_process.py +++ b/scripts/post_process.py @@ -1,3 +1,5 @@ +#---------------- For mesh visulisation of SOFA and CNN solutions- Saurabh Deshpande ---------------------------# + import Sofa import numpy as np import sys @@ -71,18 +73,6 @@ class CantileverBeam(Sofa.PythonScriptController): visuNode.createObject('OglModel', name='visual', position = "@topology_sofa.position", quads = "@topology_sofa.quads") 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): self.cube = rootNode.createChild(node_name) @@ -105,50 +95,6 @@ class CantileverBeam(Sofa.PythonScriptController): #print(commandLineArguments) 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 -- GitLab