CAE_Solution/CAE_Solution/src/VTK/QVTKRenderer/qvtkrenderer.cpp

#include "qvtkrenderer.h"

/*
 * @brief: the constructor
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230321
 */
QVTKRenderer::QVTKRenderer()
{
    qvtkWidget = NULL;
    renderWin = NULL;
    renderer = NULL;
    camera_ = NULL;
    curRenderMode_ = 1;
    surfaceVisible_ = false;
    axes = NULL;
    omWidget = NULL;
}

/*
 * @brief: the destructor
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230321
 */
QVTKRenderer::~QVTKRenderer()
{
    if(this->renderer){
        this->renderer->Delete();
        this->renderer = NULL;
    }
    if(this->renderWin){
        this->renderWin->Delete();
        this->renderWin = NULL;
    }
    if(this->camera_){
        this->camera_->Delete();
        this->camera_ = NULL;
    }
    if(this->qvtkWidget){
        delete this->qvtkWidget;
        this->qvtkWidget = NULL;
    }

    if(axes){
        axes->Delete();
        axes = NULL;
    }
    if(omWidget){
        omWidget->Delete();
        omWidget = NULL;
    }
}

/*
 * @brief: init(初始化)
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230406
 */
void QVTKRenderer::init()
{
    qvtkWidget = new QVTKOpenGLNativeWidget();
    this->renderWin = vtkGenericOpenGLRenderWindow::New();
    renderer = vtkRenderer::New();
    renderer->SetBackground(1.0,1.0,1.0);
    this->renderWin->AddRenderer(this->renderer);
    qvtkWidget->SetRenderWindow(this->renderWin);
    camera_ = vtkVISCameraManager::New();
    camera_->SetRenderTo(this->renderer);

    surfaceVisible_ = true;
    curRenderMode_ = 1;
    curSelMode_ = EntitySelection;


    axes = vtkAxesActor::New();//左下角坐标轴
    axes->SetPosition(0, 0, 0);
    axes->SetTotalLength(2, 2, 2);
    axes->SetShaftType(0);
    axes->SetCylinderRadius(0.02);
    omWidget = vtkOrientationMarkerWidget::New();
    omWidget->SetOrientationMarker(axes);
    omWidget->SetInteractor(renderWin->GetInteractor());
    omWidget->SetEnabled(1);
    omWidget->InteractiveOn();

    //widgetFlag = WidgetModel::NoModel;
}

/*
 * @brief: Capture widget Images(截取窗口图片)
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230406
 */
void QVTKRenderer::CaptureWindowImages()
{
    if(qvtkWidget == NULL) return;
    QPixmap pix = qvtkWidget->grab();
    QByteArray bytes;
    QBuffer buffer(&bytes);
    buffer.open(QIODevice::WriteOnly);
    pix.save(&buffer,"PNG");
    img = QString(bytes.toBase64());//转base64
    //emit sendImgFile(img); //发送图片(信号)
}

/*
 * @brief: get the widget
 * @param: void
 * @ret: widget
 * @birth: created by czm in 20230321
 */
QWidget *QVTKRenderer::GetWidget()
{
    return this->qvtkWidget;
}

/*
 * @brief: get the renderer
 * @param: void
 * @ret: renderer
 * @birth: created by czm in 20230321
 */
vtkRenderer *QVTKRenderer::GetRenderer()
{
    return this->renderer;
}

/*
 * @brief: show render widget maximum.
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230321
 */
void QVTKRenderer::Show()
{
    qvtkWidget->showMaximized();
}

/*
 * @brief: update this view
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230321
 */
void QVTKRenderer::Update()
{
    this->renderWin->Render();
}

/*
 * @brief: set smooth transite.
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230321
 */
void QVTKRenderer::EnableSmoothTransition(bool onOff)
{
    Q_UNUSED(onOff);
}

/*
 * @brief: set renderer axis visible or not.
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230321
 */
void QVTKRenderer::SetAxisVisible(bool onOff)
{
    Q_UNUSED(onOff);
}

/*
 * @brief: zoom to extents
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::ZoomToExtents()
{
    camera_->GetCamera();
    this->camera_->Relocate();
    Update();
}

/*
 * @brief:
  * @brief set mouse action to default.
  * With no key press, orbits.
  * With the Shift key down, pans.
  * With the Ctrl key down, zooms.
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::Op_Default()
{

}

/*
 * @brief: set mouse action zoom to window
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::Op_ZoomToWindow()
{

}

/*
 * @brief: set mouse action orbit.
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::Op_Orbit()
{

}

/*
 * @brief: set mouse action pan.
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::Op_Pan()
{

}

/*
 * @brief: set mouse action zoom.
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::Op_Zoom()
{

}

/*
 * @brief: set mouse action to single selection.
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::Op_SingleClickSelect()
{

}

/*
 * @brief: set mouse action to window selection.
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::Op_WindowSelect()
{

}

/*
 * @brief: set view direction to front
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::FrontView()
{

}

/*
 * @brief: set view direction to back
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::BackView()
{

}

/*
 * @brief: set view direction to top
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::TopView()
{

}

/*
 * @brief: set view direction to bottom
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::BottomView()
{

}

/*
 * @brief: set view direction to left
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::LeftView()
{

}

/*
 * @brief: set view direction to right
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::RightView()
{

}

/*
 * @brief: set view direction to isometric
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::IsoMetricView()
{

}

/*
 * @brief: set render mode
 * 0: wireframe
 * 1: shaded
 * 2: shaded with edges
 * others: hide faces
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::SetRenderMode(int mode)
{
    if(mode>=0&&mode<=2)this->curRenderMode_ = mode;
    vtkActorCollection* collection=this->renderer->GetActors();
    int num = collection->GetNumberOfItems();
    collection->InitTraversal();
    for(int i=0;i<num;i++){
        if(this->curRenderMode_ == 0){
            vtkActor* actor = collection->GetNextActor();
            actor->GetProperty()->SetRepresentationToWireframe();
            actor->GetProperty()->EdgeVisibilityOff();
            Update();
        }else if(this->curRenderMode_ == 1){
            vtkActor* actor = collection->GetNextActor();
            actor->GetProperty()->SetRepresentationToSurface();
            actor->GetProperty()->EdgeVisibilityOff();
            Update();
        }else if(this->curRenderMode_ == 2){
            vtkActor * actor = collection->GetNextActor();
            actor->GetProperty()->SetRepresentationToSurface();
            actor->GetProperty()->EdgeVisibilityOn();
            Update();
        }
    }
}

/*
 * @brief: set geometry-point visible or not.
 * @param: bool
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::SetPointVisible(bool onOff)
{
    Q_UNUSED(onOff);
}

/*
 * @brief: set geometry-curve visible or not.
 * @param: bool
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::SetCurveVisible(bool onOff)
{
    Q_UNUSED(onOff);
}

/*
 * @brief: set geometry-surface visible or not.
 * @param: bool
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::SetSurfaceVisible(bool onOff)
{
    surfaceVisible_ = true;
    if (onOff)  SetRenderMode(curRenderMode_);
    else        SetRenderMode(-1);
    surfaceVisible_ = onOff;
}

/*
 * @brief: set the point selection mode
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::SetSelectionLevelToPoint()
{
    if (curSelMode_ == PointSelection)  return;
    else
    {
        SetPointVisible(true);
        setSelectionLevelToEntity();


        curSelMode_ = PointSelection;
    }
}

/*
 * @brief: set the surface selection mode
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::SetSelectionLevelToSurface()
{
    if (curSelMode_ == SurfaceSelection)  return;
    else
    {
        SetSurfaceVisible(true);
        this->setSelectionLevelToEntity();


        curSelMode_ = SurfaceSelection;
    }
}

/*
 * @brief: set the entity selection mode
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::SetSelectionLevelToEntity()
{
    if (curSelMode_ == EntitySelection)  return;
    else
    {
        this->setSelectionLevelToEntity();


        curSelMode_ = EntitySelection;
    }
}

/*
 * @brief: set the entity selection level
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::setSelectionLevelToEntity()
{
}

/*
 * @brief: set the subentity selection level.
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::setSelectionLevelToSubentity()
{
}

/*
 * @brief: set point size in render window.
 * @param: size
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::SetPointSize(float size)
{
    Q_UNUSED(size);
}

/*
 * @brief: set line width in render window.
 * @param: width
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::SetLineWidth(float width)
{
    Q_UNUSED(width);
}

/*
 * @brief: hide all actors
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::HideAllActors()
{
    vtkPropCollection* pc = renderer->GetViewProps();
    pc->InitTraversal();
    for(int i = 0; i < pc->GetNumberOfItems(); i ++)
    {
        pc->GetNextProp()->VisibilityOff();
    }
    //0425 append
    renderer->ResetCamera();
    renderWin->Render();
}

/*
 * @brief: remove all actors
 * @param: void
 * @ret: void
 * @birth: created by czm in 20230322
 */
void QVTKRenderer::RemoveAllActors()
{
    //this->renderer->RemoveAllViewProps();
    vtkActorCollection * allActor = renderer->GetActors();

    vtkCollectionSimpleIterator pit;
    vtkActor *currActor = nullptr;
    for (allActor->InitTraversal(pit);(currActor = allActor->GetNextActor(pit)); )
    {
        renderer->RemoveActor(currActor);
    }

}