vtk实战(五十四)—对直线设置颜色

#include <vtkVersion.h>
#include <vtkSmartPointer.h>
#include <vtkCellArray.h>
#include <vtkCellData.h>
#include <vtkUnsignedCharArray.h>
#include <vtkPoints.h>
#include <vtkLine.h>
#include <vtkPolyData.h>
#include <vtkPolyDataMapper.h>
#include <vtkActor.h>
#include <vtkRenderWindow.h>
#include <vtkRenderer.h>
#include <vtkRenderWindowInteractor.h>

int main()
{

    vtkSmartPointer<vtkPolyData> linesPolyData =
        vtkSmartPointer<vtkPolyData>::New();

    double origin[3] = { 0.0, 0.0, 0.0 };
    double p0[3] = { 1.0, 0.0, 0.0 };
    double p1[3] = { 0.0, 1.0, 0.0 };

    vtkSmartPointer<vtkPoints> pts =
        vtkSmartPointer<vtkPoints>::New();
    pts->InsertNextPoint(origin);
    pts->InsertNextPoint(p0);
    pts->InsertNextPoint(p1);
    linesPolyData->SetPoints(pts);


    vtkSmartPointer<vtkLine> line0 =
        vtkSmartPointer<vtkLine>::New();
    line0->GetPointIds()->SetId(0, 0); // 第二个0表示pts中的origin点
    line0->GetPointIds()->SetId(1, 1); //第二个1表示pts中的p0点

    vtkSmartPointer<vtkLine> line1 =
        vtkSmartPointer<vtkLine>::New();
    line1->GetPointIds()->SetId(0, 0); 
    line1->GetPointIds()->SetId(1, 2);

    vtkSmartPointer<vtkCellArray> lines =
        vtkSmartPointer<vtkCellArray>::New();
    lines->InsertNextCell(line0);
    lines->InsertNextCell(line1);

    linesPolyData->SetLines(lines);
    unsigned char red[3] = { 255, 0, 0 };
    unsigned char green[3] = { 0, 255, 0 };

    vtkSmartPointer<vtkUnsignedCharArray> colors =
        vtkSmartPointer<vtkUnsignedCharArray>::New();
    colors->SetNumberOfComponents(3);
    colors->InsertNextTupleValue(red);
    colors->InsertNextTupleValue(green);

    linesPolyData->GetCellData()->SetScalars(colors);

    vtkSmartPointer<vtkPolyDataMapper> mapper =
        vtkSmartPointer<vtkPolyDataMapper>::New();
    mapper->SetInput(linesPolyData);
    vtkSmartPointer<vtkActor> actor =
        vtkSmartPointer<vtkActor>::New();
    actor->SetMapper(mapper);
    vtkSmartPointer<vtkRenderer> renderer =
        vtkSmartPointer<vtkRenderer>::New();
    renderer->AddActor(actor);
    vtkSmartPointer<vtkRenderWindow> window =
        vtkSmartPointer<vtkRenderWindow>::New();
    window->AddRenderer(renderer);
    vtkSmartPointer<vtkRenderWindowInteractor> interactor =
        vtkSmartPointer<vtkRenderWindowInteractor>::New();
    interactor->SetRenderWindow(window);
    window->Render();
    interactor->Start();

    return 0;
}

这里写图片描述

以下是一个用VTK绘制带箭头的可设置角度长度的直线的示例代码: ```python import vtk # 创建箭头 arrowSource = vtk.vtkArrowSource() arrowSource.SetShaftRadius(0.02) arrowSource.SetTipLength(0.2) # 创建线 lineSource = vtk.vtkLineSource() lineSource.SetPoint1(0,0,0) lineSource.SetPoint2(1,1,0) # 创建箭头mapper和actor arrowMapper = vtk.vtkPolyDataMapper() arrowMapper.SetInputConnection(arrowSource.GetOutputPort()) arrowActor = vtk.vtkActor() arrowActor.SetMapper(arrowMapper) # 创建线mapper和actor lineMapper = vtk.vtkPolyDataMapper() lineMapper.SetInputConnection(lineSource.GetOutputPort()) lineActor = vtk.vtkActor() lineActor.SetMapper(lineMapper) # 计算箭头位置和方向 arrowPosition = lineSource.GetPoint2() arrowDirection = [lineSource.GetPoint2()[0] - lineSource.GetPoint1()[0], lineSource.GetPoint2()[1] - lineSource.GetPoint1()[1], lineSource.GetPoint2()[2] - lineSource.GetPoint1()[2]] arrowLength = vtk.vtkMath.Normalize(arrowDirection) arrowActor.SetPosition(arrowPosition) arrowActor.SetOrientation(vtk.vtkMath.RotationBetweenVectors([1,0,0], arrowDirection)) # 设置箭头长度 arrowActor.SetScale(arrowLength, arrowLength, arrowLength) # 创建渲染器和窗口 renderer = vtk.vtkRenderer() renderer.AddActor(lineActor) renderer.AddActor(arrowActor) renderWindow = vtk.vtkRenderWindow() renderWindow.AddRenderer(renderer) # 创建交互器 interactor = vtk.vtkRenderWindowInteractor() interactor.SetRenderWindow(renderWindow) # 开始渲染和交互 renderWindow.Render() interactor.Start() ``` 这个示例代码可以绘制一条从原点到(1,1,0)的直线,并在直线末端添加一个长度和方向可调的箭头。箭头方向和直线方向相同,箭头长度为直线长度的倍数。可以通过调整`lineSource.SetPoint1()`和`lineSource.SetPoint2()`来改变直线的起点和终点,通过调整`arrowSource.SetTipLength()`来改变箭头长度。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值