SolutionCheckKernelsHelpers.hpp

Go to the documentation of this file.

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2016-2024 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2024 TotalEnergies
 * Copyright (c) 2018-2024 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2023-2024 Chevron
 * Copyright (c) 2019-     GEOS/GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
#ifndef GEOS_PHYSICSSOLVERS_FLUIDFLOW_SOLUTIONCHECKKERNELSHELPERS_HPP
#define GEOS_PHYSICSSOLVERS_FLUIDFLOW_SOLUTIONCHECKKERNELSHELPERS_HPP
 
#include "common/DataTypes.hpp"
 
namespace geos
{
 
class ElementsReporterBuffer;
 
struct ElementReport
{
  globalIndex m_id;
  real64 m_value;
};
 
class ElementsReporterCollector
{
  friend class ElementsReporterBuffer;
public:
 
  using ElementCount = int32_t;
 
 
  ElementsReporterCollector( ElementsReporterCollector const & other ) = default;
 
  ElementsReporterCollector( ElementsReporterCollector && other ) = default;
 
  ElementsReporterCollector & operator=( ElementsReporterCollector const & other ) = default;
 
  ElementsReporterCollector & operator=( ElementsReporterCollector && other ) = default;
 
 
  static ElementsReporterCollector disabled()
  {
    return ElementsReporterCollector( arrayView1d< ElementCount >(),
                                      arrayView1d< ElementReport >(),
                                      arrayView1d< globalIndex const >() );
  }
 
  template< typename CollectorAtomicPolicy >
  GEOS_HOST_DEVICE
  void collectElement( CollectorAtomicPolicy, ElementReport const & report ) const
  {
    if( !m_elementsCounter.empty() )
    {
      ElementCount const outputStart = RAJA::atomicInc< CollectorAtomicPolicy >( &m_elementsCounter[0] );
 
      if( outputStart < m_elementsBuffer.size() )
      {
        m_elementsBuffer[outputStart].m_id = m_localToGlobalId[report.m_id];
        m_elementsBuffer[outputStart].m_value = report.m_value;
      }
    }
  }
 
  // // currently unused version for adding multiple ids from a given kernel
  // template< typename AddedArray, typename ElementCount >
  // GEOS_HOST_DEVICE
  // void collectIds( AddedArray const & newIds, ElementCount newIdsCount )
  // {
  //   ElementCount const outputStart = RAJA::atomicAdd< CollectorAtomicPolicy >( &m_elementsCounter[0], newIdsCount );
  //   ElementCount const maxNbIdToAdd = ElementCount( m_elementsBuffer.size() - outputStart );
  //   newIdsCount = LvArray::math::min( newIdsCount, maxNbIdToAdd );
  //   for( ElementCount i = 0; i < newIdsCount; ++i )
  //   {
  //     m_elementsBuffer[outputStart + i] = newIds[i];
  //   }
  // }
 
private:
 
  arrayView1d< ElementCount > m_elementsCounter;
 
  arrayView1d< ElementReport > m_elementsBuffer;
 
  arrayView1d< globalIndex const > m_localToGlobalId;
 
  ElementsReporterCollector( arrayView1d< ElementCount > const & elementsCounter,
                             arrayView1d< ElementReport > const & elementsBuffer,
                             arrayView1d< globalIndex const > const & localToGlobalId ):
    m_elementsCounter( elementsCounter ),
    m_elementsBuffer( elementsBuffer ),
    m_localToGlobalId( localToGlobalId )
  {}
 
};
 
} // namespace geos
 
 
#endif //GEOS_PHYSICSSOLVERS_FLUIDFLOW_SOLUTIONCHECKKERNELSHELPERS_HPP