FieldStatisticsBase.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 SRC_CORECOMPONENTS_PHYSICSSOLVERS_FIELDSTATISTICSBASE_HPP_
#define SRC_CORECOMPONENTS_PHYSICSSOLVERS_FIELDSTATISTICSBASE_HPP_
 
#include "events/tasks/TaskBase.hpp"
#include "physicsSolvers/PhysicsSolverManager.hpp"
#include "mesh/MeshLevel.hpp"
#include "fileIO/Outputs/OutputBase.hpp"
 
namespace geos
{
 
template< typename SOLVER >
class FieldStatisticsBase : public TaskBase
{
public:
 
  FieldStatisticsBase( const string & name,
                       Group * const parent )
    : TaskBase( name, parent ),
    m_solver( nullptr ),
    m_outputDir( joinPath( OutputBase::getOutputDirectory(), name ) )
  {
    enableLogLevelInput();
 
    string const key = SOLVER::coupledSolverAttributePrefix() + "SolverName";
    registerWrapper( key, &m_solverName ).
      setRTTypeName( rtTypes::CustomTypes::groupNameRef ).
      setInputFlag( dataRepository::InputFlags::REQUIRED ).
      setDescription( "Name of the " + SOLVER::coupledSolverAttributePrefix() + " solver" );
 
    this->registerWrapper( viewKeyStruct::writeCSVFlagString(), &m_writeCSV ).
      setApplyDefaultValue( 0 ).
      setInputFlag( dataRepository::InputFlags::OPTIONAL ).
      setDescription( "Write statistics into a CSV file" );
  }
 
  virtual bool execute( real64 const time_n,
                        real64 const dt,
                        integer const cycleNumber,
                        integer const eventCounter,
                        real64 const eventProgress,
                        DomainPartition & domain ) override = 0;
 
protected:
 
  void postInputInitialization() override
  {
    Group & problemManager = this->getGroupByPath( "/Problem" );
    Group & physicsSolverManager = problemManager.getGroup( "Solvers" );
 
    m_solver = physicsSolverManager.getGroupPointer< SOLVER >( m_solverName );
    GEOS_THROW_IF( m_solver == nullptr,
                   GEOS_FMT( "{}: Could not find solver '{}' of type {}",
                             getDataContext(),
                             m_solverName, LvArray::system::demangleType< SOLVER >() ),
                   InputError );
 
    // create dir for output
    if( m_writeCSV > 0 )
    {
      if( MpiWrapper::commRank() == 0 )
      {
        makeDirsForPath( m_outputDir );
      }
      // wait till the dir is created by rank 0
      MPI_Barrier( MPI_COMM_WORLD );
    }
  }
 
  struct viewKeyStruct
  {
    static constexpr char const * writeCSVFlagString() { return "writeCSV"; }
  };
 
  SOLVER * m_solver;
 
  // Output directory
  string const m_outputDir;
 
  // Flag to enable writing CSV output
  integer m_writeCSV;
 
private:
 
  string m_solverName;
 
};
 
 
} /* namespace geos */
 
#endif /* SRC_CORECOMPONENTS_PHYSICSSOLVERS_FIELDSTATISTICSBASE_HPP_ */