NekSpatialBinUserObject.h

Go to the documentation of this file.

/********************************************************************/
/* SOFTWARE COPYRIGHT NOTIFICATION */
/* Cardinal */
/* */
/* (c) 2021 UChicago Argonne, LLC */
/* ALL RIGHTS RESERVED */
/* */
/* Prepared by UChicago Argonne, LLC */
/* Under Contract No. DE-AC02-06CH11357 */
/* With the U. S. Department of Energy */
/* */
/* Prepared by Battelle Energy Alliance, LLC */
/* Under Contract No. DE-AC07-05ID14517 */
/* With the U. S. Department of Energy */
/* */
/* See LICENSE for full restrictions */
/********************************************************************/
 
#pragma once
 
#include "NekUserObject.h"
#include "SpatialBinUserObject.h"
 
class NekSpatialBinUserObject : public NekUserObject
{
public:
 static InputParameters validParams();
 
 NekSpatialBinUserObject(const InputParameters & parameters);
 
 virtual ~NekSpatialBinUserObject();
 
 virtual Real spatialValue(const Point & p) const override final;
 
 virtual Real spatialValue(const Point & p, const unsigned int & component) const = 0;
 
 virtual const unsigned int bin(const Point & p) const;
 
 virtual const unsigned int num_bins() const;
 
 virtual const std::vector<Point> spatialPoints() const override { return _points; }
 
 virtual void computeBinVolumes() final;
 
 virtual void getBinVolumes() = 0;
 
 const std::vector<unsigned int> unrolledBin(const unsigned int & total_bin_index) const;
 
 const field::NekFieldEnum & field() const { return _field; }
 
 Point nekPoint(const int & local_elem_id, const int & local_node_id) const;
 
protected:
 void computePoints1D();
 
 void computePoints2D();
 
 void computePoints3D();
 
 void resetPartialStorage();
 
 void fillCoordinates(const std::vector<unsigned int> & indices, Point & p) const;
 
 const std::vector<UserObjectName> & _bin_names;
 
 const field::NekFieldEnum _field;
 
 const bool & _map_space_by_qp;
 
 const bool & _check_zero_contributions;
 
 std::vector<const SpatialBinUserObject *> _bins;
 
 std::vector<bool> _has_direction;
 
 std::vector<unsigned int> _bin_providing_direction;
 
 component::BinnedVelocityComponentEnum _velocity_component;
 
 unsigned int _n_bins;
 
 std::vector<Point> _points;
 
 std::vector<Point> _velocity_bin_directions;
 
 double * _bin_values;
 
 double * _bin_values_x;
 double * _bin_values_y;
 double * _bin_values_z;
 
 double * _bin_volumes;
 
 int * _bin_counts;
 
 double * _bin_partial_values;
 
 int * _bin_partial_counts;
};