pointcloud к многогранному в vtk python - PullRequest
Новая
       41

pointcloud к многогранному в vtk python

0 голосов
/ 29 января 2019

Цель 1: взять облако точек объекта и сгенерировать многогранную геометрию

Цель 2: сохранить многогранную геометрию в базе данных PostGIS

Хорошо, поэтому давайте сосредоточимся на цели 1.Следующий скрипт отображает облако точек - фактически он берет два облака точек из базы данных и отображает их как одно.

Pointcloud - это сканирование объекта, я хочу создать твердую геометрию, то есть многогранную, из pointcloud.Я новичок в vtk, так что я прохожу учебники.Я предполагаю, что мне понадобится какой-то этап анализа Триангуляции Делоне?затем преобразовать в многогранный?Я не программист, но я стараюсь изо всех сил, поэтому, если вы видите какие-либо явные ошибки, пожалуйста, укажите на них / предложите улучшение.

Если кто-то также хочет прокомментировать цель 2, это также будет приветствоваться.

import vtk
import psycopg2

class VtkPointCloud:

    def __init__(self, zMin=-10.0, zMax=10.0, maxNumPoints=1e6):
        self.maxNumPoints = maxNumPoints
        self.vtkPolyData = vtk.vtkPolyData()
        self.clearPoints()
        mapper = vtk.vtkPolyDataMapper()
        mapper.SetInputData(self.vtkPolyData)
        mapper.SetColorModeToDefault()
        mapper.SetScalarRange(zMin, zMax)
        mapper.SetScalarVisibility(1)
        self.vtkActor = vtk.vtkActor()
        self.vtkActor.SetMapper(mapper)

    def addPoint(self, point):
        if self.vtkPoints.GetNumberOfPoints() < self.maxNumPoints:
            pointId = self.vtkPoints.InsertNextPoint(point[:])
            self.vtkDepth.InsertNextValue(point[2])
            self.vtkCells.InsertNextCell(1)
            self.vtkCells.InsertCellPoint(pointId)
        else:
            r = random.randint(0, self.maxNumPoints)
            self.vtkPoints.SetPoint(r, point[:])
        self.vtkCells.Modified()
        self.vtkPoints.Modified()
        self.vtkDepth.Modified()

    def clearPoints(self):
        self.vtkPoints = vtk.vtkPoints()
        self.vtkCells = vtk.vtkCellArray()
        self.vtkDepth = vtk.vtkDoubleArray()
        self.vtkDepth.SetName('DepthArray')
        self.vtkPolyData.SetPoints(self.vtkPoints)
        self.vtkPolyData.SetVerts(self.vtkCells)
        self.vtkPolyData.GetPointData().SetScalars(self.vtkDepth)
        self.vtkPolyData.GetPointData().SetActiveScalars('DepthArray')





conn = psycopg2.connect("<connection details>")
cur = conn.cursor()
cur.execute("""SELECT ROUND((x - 580000)::numeric,4),ROUND((y - 4275000)::numeric,4),ROUND((z)::numeric,4) FROM pc_processing.filtered_top_dist;""")
# cur.execute("""SELECT ROUND((x)::numeric,4),ROUND((y)::numeric,4),ROUND((z)::numeric,4) FROM pc_processing.topsample;""")
T = cur.fetchall()
cur.close()

conn = psycopg2.connect("<connection details>")
cur = conn.cursor()
cur.execute("""SELECT ROUND((x - 580000)::numeric,4),ROUND((y - 4275000)::numeric,4),ROUND((z)::numeric,4) FROM pc_processing.filtered_base_dist;""")
# cur.execute("""SELECT ROUND((x)::numeric,4),ROUND((y)::numeric,4),ROUND((z)::numeric,4) FROM pc_processing.topsample;""")
B = cur.fetchall()
cur.close()

#Get the columns of the tuple (must be a better way)
xi = [i[0] for i in T]
yi = [i[1] for i in T]
zi = [i[2] for i in T]

xj = [i[0] for i in B]
yj = [i[1] for i in B]
zj = [i[2] for i in B]

#collect all points from the two layers
pointCloud = VtkPointCloud()
for k in range(len(T)):
    point = xi[k],yi[k],zi[k]
    pointCloud.addPoint(point)

for k in range(len(B)):
    point = xj[k],yj[k],zj[k]
    pointCloud.addPoint(point)






# Renderer
renderer = vtk.vtkRenderer()
renderer.AddActor(pointCloud.vtkActor)
# renderer.AddActor(pointCloud2.vtkActor)

renderer.SetBackground(.2, .3, .4)
renderer.ResetCamera()

# Render Window
renderWindow = vtk.vtkRenderWindow()
renderWindow.AddRenderer(renderer)

# Interactor
renderWindowInteractor = vtk.vtkRenderWindowInteractor()
renderWindowInteractor.SetRenderWindow(renderWindow)

# Begin Interaction
renderWindow.Render()
renderWindowInteractor.Start()
...