VTK画直线带箭头和长度文字vtkLineWidget2

最新推荐文章于 2023-12-07 22:56:16 发布

ericohe

最新推荐文章于 2023-12-07 22:56:16 发布

阅读量1.6k

点赞数 2

分类专栏： VTK C++

本文链接：https://blog.csdn.net/Ericohe/article/details/114930785

版权

C++ 同时被 2 个专栏收录

69 篇文章 0 订阅

订阅专栏

VTK

40 篇文章 7 订阅

订阅专栏

#include <vtkActor.h>
#include <vtkCamera.h>
#include <vtkCommand.h>
#include <vtkLineRepresentation.h>
#include <vtkLineWidget2.h>
#include <vtkNamedColors.h>
#include <vtkNew.h>
#include <vtkPolyData.h>
#include <vtkPolyDataMapper.h>
#include <vtkProperty.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>
#include <vtkSphereSource.h>

#include <vtkAutoInit.h>
VTK_MODULE_INIT(vtkInteractionStyle)
VTK_MODULE_INIT(vtkRenderingFreeType)
VTK_MODULE_INIT(vtkRenderingOpenGL2)
VTK_MODULE_INIT(vtkRenderingVolumeOpenGL2)

namespace {
    // This does the actual work.
    // Callback for the interaction
    class vtkLineCallback : public vtkCommand
    {
    public:
        static vtkLineCallback* New()
        {
            return new vtkLineCallback;
        }

        virtual void Execute(vtkObject* caller, unsigned long, void*)
        {

            vtkLineWidget2* lineWidget = reinterpret_cast<vtkLineWidget2*>(caller);

            // Get the actual box coordinates of the line
            vtkNew<vtkPolyData> polydata;
            static_cast<vtkLineRepresentation*>(lineWidget->GetRepresentation())
                ->GetPolyData(polydata);

            // Display one of the points, just so we know it's working
            double p[3];
            polydata->GetPoint(0, p);
            std::cout << "P: " << p[0] << " " << p[1] << " " << p[2] << std::endl;
        }
        vtkLineCallback()
        {
        }
    };
} // namespace

int main(int, char*[])
{
    vtkNew<vtkNamedColors> colors;

    vtkNew<vtkSphereSource> sphereSource;
    sphereSource->Update();

    // Create a mapper and actor
    vtkNew<vtkPolyDataMapper> mapper;
    mapper->SetInputConnection(sphereSource->GetOutputPort());
    vtkNew<vtkActor> actor;
    actor->SetMapper(mapper);
    actor->GetProperty()->SetColor(colors->GetColor3d("MistyRose").GetData());

    // A renderer and render window
    vtkNew<vtkRenderer> renderer;
    vtkNew<vtkRenderWindow> renderWindow;
    renderWindow->AddRenderer(renderer);
    renderWindow->SetWindowName("LineWidget2");

    renderer->AddActor(actor);
    //renderer->SetBackground(colors->GetColor3d("MidnightBlue").GetData());
    renderer->SetBackground(colors->GetColor3d("Black").GetData());

    // An interactor
    vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
    renderWindowInteractor->SetRenderWindow(renderWindow);

    vtkNew<vtkLineWidget2> lineWidget;
    lineWidget->SetInteractor(renderWindowInteractor);
    lineWidget->CreateDefaultRepresentation();

    // You could do this if you want to set properties at this point:
    vtkNew<vtkLineRepresentation> lineRepresentation;
    lineRepresentation->SetDistanceAnnotationVisibility(true);
    lineRepresentation->SetDistanceAnnotationFormat("%-#6.2f mm");
    lineRepresentation->SetLineColor(1, 0, 0);
    lineRepresentation->SetDirectionalLine(true);

    lineRepresentation->GetEndPointProperty()->SetColor(1, 0, 0);
    //lineRepresentation->GetEndPointProperty()->SetLineWidth(.5);
    lineRepresentation->GetEndPointProperty()->SetOpacity(0.1);

    //lineRepresentation->GetEndPoint2Property()->SetLineWidth(10.);
    lineRepresentation->GetEndPoint2Property()->SetPointSize(20.0);
    
    lineRepresentation->GetEndPoint2Property()->SetColor(1, 0, 0);


    lineWidget->SetRepresentation(lineRepresentation);

    vtkNew<vtkLineCallback> lineCallback;

    lineWidget->AddObserver(vtkCommand::InteractionEvent, lineCallback);

    // Render
    renderWindow->Render();

    renderWindowInteractor->Initialize();
    renderWindow->Render();
    lineWidget->On();

    // Begin mouse interaction
    renderWindowInteractor->Start();

    return EXIT_SUCCESS;
}

效果：