TableData.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_COMMON_FORMAT_TABLE_TABLEDATA_HPP
#define GEOS_COMMON_FORMAT_TABLE_TABLEDATA_HPP
 
#include "common/Units.hpp"
#include "common/DataTypes.hpp"
#include "common/format/Format.hpp"
#include "TableTypes.hpp"
 
namespace geos
{
 
class TableData
{
public:
 
  struct CellData
  {
    CellType type;
    string value = "";
  };
 
  template< typename ... Args >
  void addRow( Args const & ... args );
 
  void addRow( std::vector< CellData > const & row );
 
  void addSeparator();
 
  void clear();
 
  std::vector< std::vector< CellData > > const & getTableDataRows() const;
 
  std::vector< string > const & getErrorMsgs() const;
 
private:
 
  std::vector< std::vector< CellData > > m_rows;
 
};
 
class TableData2D
{
public:
 
  using RowType = real64;
  using ColumnType = real64;
 
  struct TableDataHolder
  {
    std::vector< string > headerNames;
    TableData tableData;
  };
 
  template< typename T >
  void addCell( RowType rowValue, ColumnType columnValue, T const & value );
 
  void collectTableValues( arraySlice1d< real64 const > rowAxisValues,
                           arraySlice1d< real64 const > columnAxisValues,
                           arrayView1d< real64 const > values );
 
  TableData2D::TableDataHolder convertTable2D( arrayView1d< real64 const > const values,
                                               units::Unit const valueUnit,
                                               ArrayOfArraysView< real64 const > const coordinates,
                                               string_view rowAxisDescription,
                                               string_view columnAxisDescription );
 
  TableDataHolder buildTableData( string_view dataDescription,
                                  string_view rowFmt = "{}", string_view columnFmt = "{}" ) const;
 
private:
  std::map< RowType, std::map< ColumnType, string > > m_data;
 
  std::set< real64 > m_columnValues;
};
 
template< typename T >
constexpr bool isCellType = std::is_same_v< T, CellType >;
 
template< typename ... Args >
void TableData::addRow( Args const &... args )
{
  std::vector< CellData > cells;
  ( [&] {
    static_assert( has_formatter_v< decltype(args) > || isCellType< std::decay_t< decltype(args) > >, "Argument passed in addRow cannot be converted to string nor a CellType" );
    if constexpr (std::is_same_v< Args, CellType >) {
      if( args == CellType::Separator )
      {
        cells.push_back( {CellType::Separator} );
      }
      else
      {
        cells.push_back( {CellType::MergeNext} );
      }
    }
    else
    {
      cells.push_back( {CellType::Value, GEOS_FMT( "{}", args )} );
    }
  } (), ...);
  addRow( cells );
}
 
template< typename T >
void TableData2D::addCell( real64 const rowValue, real64 const columnValue, T const & value )
{
  static_assert( has_formatter_v< decltype(value) >, "Argument passed in addCell cannot be converted to string" );
  m_columnValues.insert( columnValue );
  m_data[rowValue][columnValue] = GEOS_FMT( "{}", value );
}
 
}
#endif /* GEOS_COMMON_FORMAT_TABLE_TABLEDATA_HPP */