(大鬼王的VTK日记)VTK显示点云并上色

#include "vtkInclude.h"

int main()
{
	//看文件有多少行
	int number = 1396;

	FILE* file;
	file = fopen("1.ply", "r");

	vtkSmartPointer<vtkPoints> points = vtkSmartPointer<vtkPoints>::New();
	vtkSmartPointer<vtkCellArray> cells = vtkSmartPointer<vtkCellArray>::New();
	vtkSmartPointer<vtkLookupTable> lookup = vtkSmartPointer<vtkLookupTable>::New();
	vtkSmartPointer<vtkPolyData> polydata = vtkSmartPointer<vtkPolyData>::New();
	vtkSmartPointer<vtkFloatArray> myscalar = vtkSmartPointer<vtkFloatArray>::New();
	lookup->SetNumberOfTableValues(number);
	lookup->Build();

	float x,y,z;
	int r,g,b;
	int i = 0;
	vtkIdType idtype;
	while (!feof(file))
	{
		fscanf(file, "%f %f %f %d %d %d", &x, &y, &z, &r, &g, &b);
		myscalar->InsertNextTuple1(1);
		idtype = points->InsertNextPoint(x, y, z);
		//为每个点计算颜色 r:0-255 g:0-255 b:0-255
		lookup->SetTableValue(i, r/255, g / 255, b/255, 1);
		cells->InsertNextCell(1, &idtype);
		i++;
	}
	polydata->SetPoints(points);
	polydata->SetVerts(cells);
	polydata->GetPointData()->SetScalars(myscalar);

	vtkSmartPointer<vtkPolyDataMapper> mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
	mapper->SetInputData(polydata);
	mapper->SetLookupTable(lookup);

	vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
	actor->SetMapper(mapper);
	actor->GetProperty()->SetPointSize(1);

	vtkSmartPointer<vtkRenderer> render = vtkSmartPointer<vtkRenderer>::New();
	render->AddActor(actor);

	vtkRenderWindow *renderWindow = vtkRenderWindow::New();
	renderWindow->AddRenderer(render);
	render->SetBackground(1, 1, 1);

	vtkRenderWindowInteractor *iren = vtkRenderWindowInteractor::New();
	iren->SetRenderWindow(renderWindow);

	vtkInteractorStyleTrackballCamera *style = vtkInteractorStyleTrackballCamera::New();
	iren->SetInteractorStyle(style);

	iren->Initialize();
	iren->Start();
	return 0;
}

include里面内容自己挑着加就好。

猜你喜欢

转载自blog.csdn.net/YuHeng_Liu/article/details/91847870
今日推荐