vtk圆柱管道上色的两种方案

int main(int, char* [])
{


	// Create a sphere
	vtkSmartPointer<vtkLineSource> lineSource =
		vtkSmartPointer<vtkLineSource>::New();
	lineSource->SetPoint1(5.0, 0.0, 0.0);
	lineSource->SetPoint2(-5.0, 0.0, 0.0);
	lineSource->SetResolution(10);
	
	vtkSmartPointer<vtkTubeFilter> tubeFilter = vtkSmartPointer<vtkTubeFilter>::New();
	tubeFilter->SetInputConnection(lineSource->GetOutputPort());
	tubeFilter->SetRadius(2.0);
	tubeFilter->SetNumberOfSides(20);
	tubeFilter->CappingOn();
	
	
	tubeFilter->SetGenerateTCoords(1);
	tubeFilter->Update();
	cout << tubeFilter->GetGenerateTCoordsAsString()<<endl;
	cout << tubeFilter->GetGenerateTCoords()<<endl;
	cout << tubeFilter->GetGenerateTCoordsMinValue() << endl;
	cout << tubeFilter->GetGenerateTCoordsMaxValue() << endl;
	//cout << tubeFilter->GenerateTCoords() << endl;

	vtkIdType tuples = tubeFilter->GetOutput()->GetPointData()->GetNumberOfTuples();
	vtkFloatArray* ColorField = vtkFloatArray::New();
	ColorField->SetName("ColorField");
	ColorField->SetNumberOfComponents(1);
	ColorField->SetNumberOfTuples(tuples);
	int i = 0;
	int j = 0;
	vtkIdType vtkId = 0;
	for (; i < 11; i++) {
		for (; j < 20 * (i + 1); j++) {
			ColorField->SetTuple1(vtkId, i);
			vtkId++;
		}
	}
	tubeFilter->GetOutput()->GetPointData()->AddArray(ColorField);


	vtkSmartPointer<vtkBMPReader> Reader = vtkSmartPointer<vtkBMPReader>::New();
	Reader->SetFileName("G:/1.bmp");
	Reader->Update();

	vtkSmartPointer<vtkTexture> texture = vtkSmartPointer<vtkTexture>::New();
	texture->SetInputConnection(Reader->GetOutputPort());
	texture->InterpolateOn();


	vtkSmartPointer<vtkTextureMapToCylinder> textureMap =
		vtkSmartPointer<vtkTextureMapToCylinder>::New();
	textureMap->SetInputConnection(tubeFilter->GetOutputPort());

	vtkSmartPointer<vtkTransformTextureCoords> transformTextureCoords =
		vtkSmartPointer<vtkTransformTextureCoords>::New();
	transformTextureCoords->SetInputConnection(textureMap->GetOutputPort());
	//transformTextureCoords->SetScale(tubeFilter->GenerateTCoords());
	
	vtkNew<vtkCleanPolyData> clean;
	clean->SetInputData(tubeFilter->GetOutput());
	clean->Update();
	vtkSmartPointer<vtkPolyData> poly = clean->GetOutput();
	


	vtkSmartPointer<vtkPolyDataWriter> writer = vtkSmartPointer<vtkPolyDataWriter>::New();
	//writer->SetInputConnection(transformTextureCoords->GetOutputPort());
	writer->SetInputData(clean->GetOutput());
	writer->SetFileName("G:/AppendCylinder.vtk");
	writer->Update();

	// Create a mapper and actor
	vtkNew<vtkPolyDataMapper> mapper;
	//mapper->SetInputData(tubeFilter->GetOutput());
	mapper->SetInputConnection(transformTextureCoords->GetOutputPort());
		

	vtkNew<vtkActor> actor;
	vtkNew<vtkNamedColors> colors;
	actor->GetProperty()->SetColor(colors->GetColor3d("Cornsilk").GetData());
	actor->SetMapper(mapper);
	actor->SetTexture(texture);

	// Create a renderer, render window, and interactor
	vtkNew<vtkRenderer> renderer;
	vtkNew<vtkRenderWindow> renderWindow;
	renderWindow->SetWindowName("Cylinder");
	renderWindow->AddRenderer(renderer);
	vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
	renderWindowInteractor->SetRenderWindow(renderWindow);

	// Add the actor to the scene
	renderer->AddActor(actor);
	renderer->SetBackground(colors->GetColor3d("DarkGreen").GetData());

	// Render and interact
	renderWindow->Render();
	renderWindowInteractor->Start();

	return EXIT_SUCCESS;
}

两种,一种是给管道侧截面上划线并给线上的点赋值(输出文件),另一种是直接使用纹理(actor形式),想要自己设置纹理颜色:

//line source;
	vtkNew<vtkPoints> points;
	points->InsertNextPoint(start[0], start[1], start[2]);
	points->InsertNextPoint(end[0], end[1], end[2]);
	m_LineSource->SetPoints(points);
	m_LineSource->SetResolution(5);


	//tube filter;
	m_tubeFilter->SetRadius(1);
	m_tubeFilter->SetNumberOfSides(10);
	m_tubeFilter->SetInputConnection(m_LineSource->GetOutputPort());
	m_tubeFilter->SetTextureLength(10);
	m_tubeFilter->SetGenerateTCoordsToUseLength();
	m_tubeFilter->Update();

	vtkSmartPointer<vtkDataArray> pTCoords = m_tubeFilter->GetOutput()->GetPointData()->GetTCoords();

	
	int nNumberOfTuples = pTCoords->GetNumberOfTuples();
	for (int i = 0; i < nNumberOfTuples; i++)
	{
		double *pTuples = pTCoords->GetTuple2(i);
		Base::Console().Message("%03d, %5.2f, %5.2f\n", i, pTuples[0], pTuples[1]);
	}



	m_Mapper->SetInputConnection(m_tubeFilter->GetOutputPort());
	m_Actor->SetMapper(m_Mapper);

	//纹理;
	vtkNew<vtkImageData> image;
	vtkNew<vtkTexture> texture;
	image->SetDimensions(256, 1, 1);
	image->AllocateScalars(VTK_UNSIGNED_CHAR, 3);
	image->SetExtent(0, 255, 0, 0, 0, 0);
	unsigned char* pixel;
	pixel = static_cast<unsigned char*>(image->GetScalarPointer());
	unsigned char on = 255;
	unsigned char off = 0;
	for (int i = 0; i < 256; ++i)
	{
		*(pixel + 3 * i + 0) = i;
		*(pixel + 3 * i + 1) = 0;//255-i;
		*(pixel + 3 * i + 2) = 0;// abs(127 - i) * 2;


	}
	texture->SetInputData(image);
	texture->InterpolateOn();
	texture->RepeatOn();
	
	m_Actor->SetTexture(texture);

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值