RayTracingMeshOutput.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

// Local includes
#include "RayTracingCommon.h"
#include "Ray.h"

// MOOSE includes
#include "FileOutput.h"
#include "UserObjectInterface.h"

// libMesh includes
#include "libmesh/distributed_mesh.h"
#include "libmesh/equation_systems.h"
#include "libmesh/explicit_system.h"

// Forward declarations
class RayTracingStudy;
struct TraceData;

class RayTracingMeshOutput : public FileOutput, public UserObjectInterface
{
public:
  RayTracingMeshOutput(const InputParameters & parameters);

  static InputParameters validParams();

  virtual std::string filename() override;

  void output() override;

protected:
  virtual void outputMesh() = 0;

  const RayTracingStudy & _study;

  const bool _output_data;
  const std::vector<std::string> * const _output_data_names;
  const bool _output_data_nodal;
  const bool _output_aux_data;
  const std::vector<std::string> * const _output_aux_data_names;

  std::unique_ptr<MeshBase> _segment_mesh;
  std::unique_ptr<libMesh::EquationSystems> _es;
  libMesh::ExplicitSystem * _sys;

private:
  void buildBoundingBoxes();
  void buildIDMap();
  void buildSegmentMesh();
  void setupEquationSystem();
  void fillFields();

  dof_id_type neededNodes(const TraceData & trace_data) const;
  void startingIDs(const TraceData & trace_data,
                   dof_id_type & start_node_id,
                   dof_id_type & start_elem_id) const;

  unsigned int _ray_id_var;
  unsigned int _intersections_var;
  unsigned int _pid_var;
  unsigned int _processor_crossings_var;
  unsigned int _trajectory_changes_var;
  std::vector<std::pair<RayDataIndex, unsigned int>> _data_vars;
  std::vector<std::pair<RayDataIndex, unsigned int>> _aux_data_vars;

  BoundingBox _bbox;
  std::vector<BoundingBox> _inflated_bboxes;
  std::vector<std::pair<processor_id_type, BoundingBox>> _inflated_neighbor_bboxes;

  std::unordered_map<RayID, std::pair<dof_id_type, dof_id_type>> _ray_starting_id_map;
  dof_id_type _max_node_id;
  bool _segmented_rays;
};