PointTransform.h

Go to the documentation of this file.

//============================================================================
//  The contents of this file are covered by the Viskores license. See
//  LICENSE.txt for details.
//
//  By contributing to this file, all contributors agree to the Developer
//  Certificate of Origin Version 1.1 (DCO 1.1) as stated in DCO.txt.
//============================================================================
 
//============================================================================
//  Copyright (c) Kitware, Inc.
//  All rights reserved.
//  See LICENSE.txt 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.
//============================================================================
 
#ifndef viskores_filter_field_transform_PointTransform_h
#define viskores_filter_field_transform_PointTransform_h
 
#include <viskores/Matrix.h>
#include <viskores/Transform3D.h>
 
#include <viskores/filter/Filter.h>
#include <viskores/filter/field_transform/viskores_filter_field_transform_export.h>
 
namespace viskores
{
namespace filter
{
namespace field_transform
{
class VISKORES_FILTER_FIELD_TRANSFORM_EXPORT PointTransform : public viskores::filter::Filter
{
public:
  VISKORES_CONT
  PointTransform();
 
  VISKORES_CONT void SetTranslation(const viskores::FloatDefault& tx,
                                    const viskores::FloatDefault& ty,
                                    const viskores::FloatDefault& tz)
  {
    matrix = viskores::Transform3DTranslate(tx, ty, tz);
  }
 
  VISKORES_CONT void SetTranslation(const viskores::Vec3f& v) { SetTranslation(v[0], v[1], v[2]); }
 
  VISKORES_CONT void SetRotation(const viskores::FloatDefault& angleDegrees,
                                 const viskores::Vec3f& axis)
  {
    matrix = viskores::Transform3DRotate(angleDegrees, axis);
  }
 
  VISKORES_CONT void SetRotation(const viskores::FloatDefault& angleDegrees,
                                 const viskores::FloatDefault& axisX,
                                 const viskores::FloatDefault& axisY,
                                 const viskores::FloatDefault& axisZ)
  {
    SetRotation(angleDegrees, { axisX, axisY, axisZ });
  }
 
  VISKORES_CONT void SetRotationX(const viskores::FloatDefault& angleDegrees)
  {
    SetRotation(angleDegrees, 1, 0, 0);
  }
 
  VISKORES_CONT void SetRotationY(const viskores::FloatDefault& angleDegrees)
  {
    SetRotation(angleDegrees, 0, 1, 0);
  }
 
  VISKORES_CONT void SetRotationZ(const viskores::FloatDefault& angleDegrees)
  {
    SetRotation(angleDegrees, 0, 0, 1);
  }
 
  VISKORES_CONT void SetScale(const viskores::FloatDefault& s)
  {
    matrix = viskores::Transform3DScale(s, s, s);
  }
 
  VISKORES_CONT void SetScale(const viskores::FloatDefault& sx,
                              const viskores::FloatDefault& sy,
                              const viskores::FloatDefault& sz)
  {
    matrix = viskores::Transform3DScale(sx, sy, sz);
  }
 
  VISKORES_CONT void SetScale(const viskores::Vec3f& v)
  {
    matrix = viskores::Transform3DScale(v[0], v[1], v[2]);
  }
 
  VISKORES_CONT
  void SetTransform(const viskores::Matrix<viskores::FloatDefault, 4, 4>& mtx) { matrix = mtx; }
 
  void SetChangeCoordinateSystem(bool flag);
  bool GetChangeCoordinateSystem() const;
 
private:
  VISKORES_CONT viskores::cont::DataSet DoExecute(const viskores::cont::DataSet& input) override;
 
  viskores::Matrix<viskores::FloatDefault, 4, 4> matrix;
  bool ChangeCoordinateSystem = true;
};
} // namespace field_transform
} // namespace filter
} // namespace viskores
 
#endif // viskores_filter_field_transform_PointTransform_h