Files
ThirdParty-6/ParaView-5.0.1/ParaViewCore/ClientServerCore/Rendering/vtkCubeAxesRepresentation.cxx

609 lines
20 KiB
C++

/*=========================================================================
Program: ParaView
Module: vtkCubeAxesRepresentation.cxx
Copyright (c) Kitware, Inc.
All rights reserved.
See Copyright.txt or http://www.paraview.org/HTML/Copyright.html for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
#include "vtkCubeAxesRepresentation.h"
#include "vtkAlgorithmOutput.h"
#include "vtkAxisActor.h"
#include "vtkBoundingBox.h"
#include "vtkCommand.h"
#include "vtkCompositeDataIterator.h"
#include "vtkCompositeDataSet.h"
#include "vtkCubeAxesActor.h"
#include "vtkDataSet.h"
#include "vtkFieldData.h"
#include "vtkFloatArray.h"
#include "vtkIdTypeArray.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkMath.h"
#include "vtkNew.h"
#include "vtkObjectFactory.h"
#include "vtkOutlineSource.h"
#include "vtkPolyData.h"
#include "vtkProperty.h"
#include "vtkProperty.h"
#include "vtkPVChangeOfBasisHelper.h"
#include "vtkPVRenderView.h"
#include "vtkRenderer.h"
#include "vtkSmartPointer.h"
#include "vtkStringArray.h"
#include "vtkTextProperty.h"
#include "vtkTransform.h"
vtkStandardNewMacro(vtkCubeAxesRepresentation);
//----------------------------------------------------------------------------
vtkCubeAxesRepresentation::vtkCubeAxesRepresentation()
{
this->OutlineGeometry = vtkSmartPointer<vtkPolyData>::New();
this->CubeAxesActor = vtkCubeAxesActor::New();
this->CubeAxesActor->SetPickable(0);
this->Position[0] = this->Position[1] = this->Position[2] = 0.0;
this->Orientation[0] = this->Orientation[1] = this->Orientation[2] = 0.0;
this->Scale[0] = this->Scale[1] = this->Scale[2] = 1.0;
this->CustomBounds[0] = this->CustomBounds[2] = this->CustomBounds[4] = 0.0;
this->CustomBounds[1] = this->CustomBounds[3] = this->CustomBounds[5] = 1.0;
this->CustomBoundsActive[0] = 0;
this->CustomBoundsActive[1] = 0;
this->CustomBoundsActive[2] = 0;
this->CustomRange[0] = this->CustomRange[2] = this->CustomRange[4] = 0.0;
this->CustomRange[1] = this->CustomRange[3] = this->CustomRange[5] = 1.0;
this->CustomRangeActive[0] = 0;
this->CustomRangeActive[1] = 0;
this->CustomRangeActive[2] = 0;
this->UseOrientedBounds = false;
this->UserXTitle = this->UserYTitle = this->UserZTitle = NULL;
this->UseDefaultXTitle = this->UseDefaultYTitle = this->UseDefaultZTitle = 1;
this->OriginalBoundsRangeActive[0] = 0;
this->OriginalBoundsRangeActive[1] = 0;
this->OriginalBoundsRangeActive[2] = 0;
this->RendererType = vtkPVRenderView::DEFAULT_RENDERER;
}
//----------------------------------------------------------------------------
vtkCubeAxesRepresentation::~vtkCubeAxesRepresentation()
{
this->CubeAxesActor->Delete();
this->SetUserXTitle(NULL);
this->SetUserYTitle(NULL);
this->SetUserZTitle(NULL);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetVisibility(bool val)
{
this->Superclass::SetVisibility(val);
this->CubeAxesActor->SetVisibility(val? 1 : 0);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetColor(double r, double g, double b)
{
this->CubeAxesActor->GetProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetXAxesLinesProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetYAxesLinesProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetZAxesLinesProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetXAxesGridlinesProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetYAxesGridlinesProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetZAxesGridlinesProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetXAxesInnerGridlinesProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetYAxesInnerGridlinesProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetZAxesInnerGridlinesProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetXAxesGridpolysProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetYAxesGridpolysProperty()->SetColor(r, g, b);
this->CubeAxesActor->GetZAxesGridpolysProperty()->SetColor(r, g, b);
for(int i=0; i < 3; i++)
{
this->CubeAxesActor->GetTitleTextProperty(i)->SetColor(r, g, b);
this->CubeAxesActor->GetLabelTextProperty(i)->SetColor(r, g, b);
}
}
//----------------------------------------------------------------------------
bool vtkCubeAxesRepresentation::AddToView(vtkView* view)
{
vtkPVRenderView* pvview = vtkPVRenderView::SafeDownCast(view);
if (pvview)
{
if (vtkRenderer* renderer = pvview->GetRenderer(this->RendererType))
{
renderer->AddActor(this->CubeAxesActor);
this->CubeAxesActor->SetCamera(renderer->GetActiveCamera());
this->RenderView = pvview;
return this->Superclass::AddToView(view);
}
}
return false;
}
//----------------------------------------------------------------------------
bool vtkCubeAxesRepresentation::RemoveFromView(vtkView* view)
{
vtkPVRenderView* pvview = vtkPVRenderView::SafeDownCast(view);
if (pvview)
{
if (vtkRenderer* renderer = pvview->GetRenderer(this->RendererType))
{
renderer->RemoveActor(this->CubeAxesActor);
this->CubeAxesActor->SetCamera(NULL);
this->RenderView = NULL;
return this->Superclass::RemoveFromView(view);
}
}
this->RenderView = NULL;
return false;
}
//----------------------------------------------------------------------------
int vtkCubeAxesRepresentation::FillInputPortInformation(
int vtkNotUsed(port), vtkInformation* info)
{
info->Set(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkDataSet");
info->Append(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkCompositeDataSet");
info->Set(vtkAlgorithm::INPUT_IS_OPTIONAL(), 1);
return 1;
}
//----------------------------------------------------------------------------
int vtkCubeAxesRepresentation::ProcessViewRequest(
vtkInformationRequestKey* request_type, vtkInformation* inInfo,
vtkInformation* outInfo)
{
if (!this->Superclass::ProcessViewRequest(request_type, inInfo, outInfo))
{
return 0;
}
if (request_type == vtkPVView::REQUEST_UPDATE())
{
vtkPVRenderView::SetPiece(inInfo, this, this->OutlineGeometry);
// We need to deliver to all processes to ensure we get the reduced data
// bounds on all process. Otherwise we end up with BUG #0013403.
vtkPVRenderView::SetDeliverToAllProcesses(inInfo, this, true);
}
else if (request_type == vtkPVView::REQUEST_RENDER())
{
vtkAlgorithmOutput* producerPort = vtkPVRenderView::GetPieceProducer(inInfo, this);
if (producerPort)
{
vtkAlgorithm* producer = producerPort->GetProducer();
vtkDataSet* ds = vtkDataSet::SafeDownCast(producer->GetOutputDataObject(
producerPort->GetIndex()));
if (ds)
{
// Update local bounds based on the server side information
ds->GetBounds(this->DataBounds);
// Configure the cube axes based on the fields data that have been
// passed along by the server
this->ConfigureCubeAxes(ds);
}
}
this->UpdateBounds();
}
return 1;
}
//----------------------------------------------------------------------------
int vtkCubeAxesRepresentation::RequestData(vtkInformation*,
vtkInformationVector** inputVector, vtkInformationVector*)
{
vtkMath::UninitializeBounds(this->DataBounds);
vtkFieldData* fieldDataToKeepAround = NULL;
if (inputVector[0]->GetNumberOfInformationObjects()==1)
{
vtkDataObject* input = vtkDataObject::GetData(inputVector[0], 0);
fieldDataToKeepAround = input->GetFieldData();
vtkDataSet* ds = vtkDataSet::SafeDownCast(input);
vtkCompositeDataSet* cd = vtkCompositeDataSet::SafeDownCast(input);
if (ds)
{
ds->GetBounds(this->DataBounds);
}
else
{
vtkCompositeDataIterator* iter = cd->NewIterator();
vtkBoundingBox bbox;
for (iter->InitTraversal(); !iter->IsDoneWithTraversal();
iter->GoToNextItem())
{
ds = vtkDataSet::SafeDownCast(iter->GetCurrentDataObject());
if (ds)
{
double bds[6];
ds->GetBounds(bds);
if (vtkMath::AreBoundsInitialized(bds))
{
bbox.AddBounds(bds);
}
}
}
iter->Delete();
bbox.GetBounds(this->DataBounds);
}
}
if (vtkMath::AreBoundsInitialized(this->DataBounds))
{
vtkNew<vtkOutlineSource> outlineSource;
outlineSource->SetBoxTypeToAxisAligned();
outlineSource->SetBounds(this->DataBounds);
outlineSource->Update();
this->OutlineGeometry->ShallowCopy(outlineSource->GetOutput());
}
else
{
this->OutlineGeometry->Initialize();
}
if(fieldDataToKeepAround)
{
this->OutlineGeometry->GetFieldData()->ShallowCopy(fieldDataToKeepAround);
}
// We fire UpdateDataEvent to notify the representation proxy that the
// representation was updated. The representation proxty will then call
// PostUpdateData(). We do this since now representations are not updated at
// the proxy level.
this->InvokeEvent(vtkCommand::UpdateDataEvent);
return 1;
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::UpdateBounds()
{
double *scale = this->Scale;
double *position = this->Position;
double *rotation = this->Orientation;
double bds[6];
if (scale[0] != 1.0 || scale[1] != 1.0 || scale[2] != 1.0 ||
position[0] != 0.0 || position[1] != 0.0 || position[2] != 0.0 ||
rotation[0] != 0.0 || rotation[1] != 0.0 || rotation[2] != 0.0)
{
const double *bounds = this->DataBounds;
vtkSmartPointer<vtkTransform> transform =
vtkSmartPointer<vtkTransform>::New();
transform->Translate(position);
transform->RotateZ(rotation[2]);
transform->RotateX(rotation[0]);
transform->RotateY(rotation[1]);
transform->Scale(scale);
vtkBoundingBox bbox;
int i, j, k;
double origX[3], x[3];
for (i = 0; i < 2; i++)
{
origX[0] = bounds[i];
for (j = 0; j < 2; j++)
{
origX[1] = bounds[2 + j];
for (k = 0; k < 2; k++)
{
origX[2] = bounds[4 + k];
transform->TransformPoint(origX, x);
bbox.AddPoint(x);
}
}
}
bbox.GetBounds(bds);
}
else
{
memcpy(bds, this->DataBounds, sizeof(double)*6);
}
//overload bounds with the active custom bounds
for ( int i=0; i < 3; ++i)
{
int pos = i * 2;
if ( this->CustomBoundsActive[i] )
{
bds[pos]=this->CustomBounds[pos];
bds[pos+1]=this->CustomBounds[pos+1];
}
}
this->CubeAxesActor->SetBounds(bds);
// Override the range if we have any custom value or set them as the bounds
for ( int i=0; i < 3; ++i)
{
int pos = i * 2;
double* bp = NULL;
if ( this->CustomRangeActive[i] )
{
bp = &this->CustomRange[pos];
}
else if ( !this->UseOrientedBounds && this->OriginalBoundsRangeActive[i] != 0)
{
bp = &this->DataBounds[pos];
}
else // Default setup
{
bp = &bds[pos];
}
// Use the proper SetRange method
switch(i)
{
case 0: this->CubeAxesActor->SetXAxisRange(bp); break;
case 1: this->CubeAxesActor->SetYAxisRange(bp); break;
case 2: this->CubeAxesActor->SetZAxisRange(bp); break;
}
}
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
}
//***************************************************************************
// Forwarded to internal vtkCubeAxesActor
void vtkCubeAxesRepresentation::SetFlyMode(int val)
{
this->CubeAxesActor->SetFlyMode(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetInertia(int val)
{
this->CubeAxesActor->SetInertia(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetCornerOffset(double val)
{
this->CubeAxesActor->SetCornerOffset(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetTickLocation(int val)
{
this->CubeAxesActor->SetTickLocation(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetXTitle(const char* val)
{
this->SetUserXTitle(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetXAxisVisibility(int val)
{
this->CubeAxesActor->SetXAxisVisibility(val);
this->CubeAxesActor->SetXAxisLabelVisibility(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetXAxisTickVisibility(int val)
{
this->CubeAxesActor->SetXAxisTickVisibility(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetXAxisMinorTickVisibility(int val)
{
this->CubeAxesActor->SetXAxisMinorTickVisibility(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetDrawXGridlines(int val)
{
this->CubeAxesActor->SetDrawXGridlines(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetYTitle(const char* val)
{
this->SetUserYTitle(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetYAxisVisibility(int val)
{
this->CubeAxesActor->SetYAxisVisibility(val);
this->CubeAxesActor->SetYAxisLabelVisibility(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetYAxisTickVisibility(int val)
{
this->CubeAxesActor->SetYAxisTickVisibility(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetYAxisMinorTickVisibility(int val)
{
this->CubeAxesActor->SetYAxisMinorTickVisibility(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetDrawYGridlines(int val)
{
this->CubeAxesActor->SetDrawYGridlines(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetZTitle(const char* val)
{
this->SetUserZTitle(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetZAxisVisibility(int val)
{
this->CubeAxesActor->SetZAxisVisibility(val);
this->CubeAxesActor->SetZAxisLabelVisibility(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetZAxisTickVisibility(int val)
{
this->CubeAxesActor->SetZAxisTickVisibility(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetZAxisMinorTickVisibility(int val)
{
this->CubeAxesActor->SetZAxisMinorTickVisibility(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetDrawZGridlines(int val)
{
this->CubeAxesActor->SetDrawZGridlines(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetGridLineLocation(int val)
{
this->CubeAxesActor->SetGridLineLocation(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetUseOfAxesOrigin(int val)
{
this->CubeAxesActor->SetUseAxisOrigin(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetAxesOrigin(double valX, double valY, double valZ)
{
this->CubeAxesActor->SetAxisOrigin(valX, valY, valZ);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetAxesOrigin(double val[3])
{
this->CubeAxesActor->SetAxisOrigin(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetXLabelFormat(const char* format)
{
this->CubeAxesActor->SetXLabelFormat(format);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetYLabelFormat(const char* format)
{
this->CubeAxesActor->SetYLabelFormat(format);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetZLabelFormat(const char* format)
{
this->CubeAxesActor->SetYLabelFormat(format);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetStickyAxes(int val)
{
this->CubeAxesActor->SetStickyAxes(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::SetCenterStickyAxes(int val)
{
this->CubeAxesActor->SetCenterStickyAxes(val);
}
//----------------------------------------------------------------------------
void vtkCubeAxesRepresentation::ConfigureCubeAxes(vtkDataObject* input)
{
// Update axes title informations
const char* utitle = NULL;
const char* vtitle = NULL;
const char* wtitle = NULL;
vtkPVChangeOfBasisHelper::GetBasisName(input, utitle, vtitle, wtitle);
if (this->UseDefaultXTitle == 1 && utitle != NULL)
{
this->CubeAxesActor->SetXTitle(utitle);
}
else if(this->UserXTitle)
{
this->CubeAxesActor->SetXTitle(this->UserXTitle);
}
if (this->UseDefaultYTitle == 1 && vtitle != NULL)
{
this->CubeAxesActor->SetYTitle(vtitle);
}
else if(this->UserYTitle)
{
this->CubeAxesActor->SetYTitle(this->UserYTitle);
}
if(this->UseDefaultZTitle == 1 && wtitle != NULL)
{
this->CubeAxesActor->SetZTitle(wtitle);
}
else if(this->UserZTitle)
{
this->CubeAxesActor->SetZTitle(this->UserZTitle);
}
// Update axis orientation and bounds. This is where we loose all rhyme and
// reason. vtkCubeAxesActor has a bad API to support basis vectors. It takes
// normalized basis vectors for the 3 axis only. Hence, we do this trick -- we
// pass normalized basis vectors, then scale the oriented bounds (which should
// have been called data bounds in the coordinate system defined by basis
// vectors).
vtkVector3d u(1, 0, 0), v(0, 1, 0), w(0, 0, 1);
vtkSmartPointer<vtkMatrix4x4> changeOfBasisMatrix =
vtkPVChangeOfBasisHelper::GetChangeOfBasisMatrix(input);
double unorm =1.0, vnorm = 1.0, wnorm = 1.0;
if (changeOfBasisMatrix)
{
vtkPVChangeOfBasisHelper::GetBasisVectors(changeOfBasisMatrix, u, v, w);
unorm = u.Normalize();
vnorm = v.Normalize();
wnorm = w.Normalize();
}
this->CubeAxesActor->SetAxisBaseForX(u.GetData());
this->CubeAxesActor->SetAxisBaseForY(v.GetData());
this->CubeAxesActor->SetAxisBaseForZ(w.GetData());
// Make sure we enable oriented bounding box if any
double orientedBounds[6] = {0,0,0,0,0,0};
if (vtkPVChangeOfBasisHelper::GetBoundingBoxInBasis(input, orientedBounds))
{
double minpoint[4] = {orientedBounds[0], orientedBounds[2], orientedBounds[4], 1};
double maxpoint[4] = {orientedBounds[1], orientedBounds[3], orientedBounds[5], 1};
this->UseOrientedBounds = true;
this->CubeAxesActor->SetUseOrientedBounds(true);
this->CubeAxesActor->SetOrientedBounds(
minpoint[0]*unorm, maxpoint[0]*unorm,
minpoint[1]*vnorm, maxpoint[1]*vnorm,
minpoint[2]*wnorm, maxpoint[2]*wnorm);
this->CustomRangeActive[0] = this->CustomRangeActive[1] = this->CustomRangeActive[2] = 1;
std::copy(orientedBounds, orientedBounds+6, this->CustomRange);
}
else
{
this->UseOrientedBounds = false;
this->CubeAxesActor->SetUseOrientedBounds(false);
}
}