RayTracingAngularQuadrature.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* 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 "MooseTypes.h"

// libMesh includes
#include "libmesh/dense_matrix.h"
#include "libmesh/point.h"

class RayTracingAngularQuadrature
{
public:
  RayTracingAngularQuadrature(const unsigned int dim,
                              const unsigned int polar_order,
                              const unsigned int azimuthal_order,
                              const Real mu_min,
                              const Real mu_max);

  static void gaussLegendre(const unsigned int order, std::vector<Real> & x, std::vector<Real> & w);

  static void chebyshev(const unsigned int order, std::vector<Real> & x, std::vector<Real> & w);

  static void rotationMatrix(const libMesh::Point & direction, libMesh::DenseMatrix<Real> & matrix);
  static libMesh::Point orthonormalVector(const libMesh::Point & v);

  void rotate(const libMesh::Point & rotation_direction);

  const libMesh::Point & getDirection(const unsigned int l) const;
  const std::vector<Real> & getWeights(const unsigned int l) const;
  Real getTotalWeight(const unsigned int l) const;
  const std::vector<Real> & getPolarSins(const unsigned int l) const;

  std::size_t numDirections() const { return _current_directions.size(); }
  bool hasDirection(const unsigned int l) const { return l < _current_directions.size(); }

  void checkDirection(const unsigned int l) const;

  std::size_t numPolar(const unsigned int l) const;

  unsigned int polarOrder() const { return _polar_order; }
  unsigned int azimuthalOrder() const { return _azimuthal_order; }
  Real muMin() const { return _mu_min; }
  Real muMax() const { return _mu_max; }

  const libMesh::Point & currentRotationDirection() const { return _current_rotation_direction; }

  unsigned int dim() const { return _dim; }

private:
  void build();

  const unsigned int _dim;
  const unsigned int _polar_order;
  const unsigned int _azimuthal_order;
  const Real _mu_min;
  const Real _mu_max;

  std::vector<Real> _phi;
  std::vector<Real> _mu;
  std::vector<Real> _polar_sin;
  std::vector<Real> _w;

  std::vector<libMesh::Point> _current_directions;
  std::vector<std::vector<Real>> _current_weights;
  std::vector<std::vector<Real>> _current_polar_sins;

  libMesh::Point _current_rotation_direction;
};