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;
 
  virtual std::string fileExtension() const = 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;
};