stl取点

stl取点

import vtk
import numpy as np
#右键选点，按下中键取消点的选择。按一次取消上一次选点，选取的点坐标和模型文件在同一个文件夹中


def getfilename():
    filename= input()  # C:/Users/big tiger/Desktop/foot/GTQ pyinstaller -F C:\Users\big tiger\PycharmProjects\POINTpicker\pointpicker.py
    return filename

def GetstlData(filename):
    stlReader = vtk.vtkSTLReader()
    stlReader.SetFileName(filename+".stl")
    stlReader.Update()
    data = stlReader.GetOutput()
    print("nodes number：", data.GetNumberOfPoints())
    if data.GetNumberOfPoints() == 0:
        raise ValueError("No point data could be loaded from " + filename)
        return None

    return data

class interactor(vtk.vtkInteractorStyleTrackballCamera):

    def __init__(self,parent=None):
        self.AddObserver("RightButtonPressEvent",self.RightButtonPressEvent)
        self.AddObserver("MiddleButtonPressEvent", self.MiddleButtonP