ImageSampler.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://www.mooseframework.org
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#pragma once

// MOOSE includes
#include "FileRangeBuilder.h"
#include "ConsoleStream.h"

#include "libmesh/bounding_box.h"
#include <array>

// VTK includes
#ifdef LIBMESH_HAVE_VTK

// Some VTK header files have extra semi-colons in them, and clang
// loves to warn about it...
#include "libmesh/ignore_warnings.h"

#include "vtkSmartPointer.h"
#include "vtkPNGReader.h"
#include "vtkTIFFReader.h"
#include "vtkImageData.h"
#include "vtkStringArray.h"
#include "vtkImageThreshold.h"
#include "vtkImageNormalize.h"
#include "vtkImageCast.h"
#include "vtkImageShiftScale.h"
#include "vtkImageMagnitude.h"
#include "vtkImageFlip.h"

#include "libmesh/restore_warnings.h"

#endif

class MooseMesh;

class ImageSampler : public FileRangeBuilder
{
public:
  static InputParameters validParams();

  ImageSampler(const InputParameters & parameters);

  virtual Real sample(const Point & p) const;

  virtual void setupImageSampler(MooseMesh & mesh);

protected:
  void vtkShiftAndScale();

  void vtkThreshold();

  void vtkMagnitude();

private:
#ifdef LIBMESH_HAVE_VTK

  vtkSmartPointer<vtkStringArray> _files;

  vtkImageData * _data;

  vtkAlgorithmOutput * _algorithm;

  vtkSmartPointer<vtkImageReader2> _image;

  vtkSmartPointer<vtkImageThreshold> _image_threshold;

  vtkSmartPointer<vtkImageShiftScale> _shift_scale_filter;

  vtkSmartPointer<vtkImageMagnitude> _magnitude_filter;
#endif

  Point _origin;

  std::vector<int> _dims;

  Point _physical_dims;

  std::vector<double> _voxel;

#ifdef LIBMESH_HAVE_VTK
  unsigned int _component;
#endif

  BoundingBox _bounding_box;

  const InputParameters & _is_pars;

  ConsoleStream _is_console;

  std::array<bool, 3> _flip;
};