rbf-brain/RBF_fine.py at master · PSUCompBio/rbf-brain · 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

# coding: utf-8

# In[1]:


#get_ipython().run_line_magic('matplotlib', 'inline')
from pygem import RBFParameters, RBF, StlHandler, VtkHandler
import numpy as np
import matplotlib.pyplot as plt
import os
cwd = os.getcwd()
full_path = os.path.realpath(__file__)
path, filename = os.path.split(full_path)


# In[2]:


params = RBFParameters()
params.read_parameters('../tests/test_datasets/parameters.prm')


# In[2]:


vtk_handler = VtkHandler()
mesh = vtk_handler.parse('../tests/test_datasets/fine_mesh.vtk')


# In[4]:


rbf = RBF(params, mesh)
rbf.perform()
new_mesh_points = rbf.modified_mesh_points
vtk_handler.write(new_mesh_points, "fine_mesh_morphed.vtk")