LocalToGlobalMap.h

Go to the documentation of this file.

// This file is part of the Peano's PETSc extension. For conditions of
// distribution and use, please see the copyright notice at
// www.peano-framework.org
#pragma once
 
 
#include <utility>
#include <map>
 
#include "tarch/logging/Log.h"
#include "tarch/logging/Statistics.h"
#include "tarch/logging/LogFilter.h"
#include "tarch/logging/LogFilterFileReader.h"
 
namespace petsc {
  class LocalToGlobalMap;
}
 
 
class petsc::LocalToGlobalMap {
  public:
 
    enum class Type: int{
      Vertex=0, Cell=1, Face=2
    };
 
    static constexpr int RankGlobalTreeNumber = -1;
 
    static constexpr int NoDegreeOfFreedom = -1;
 
    LocalToGlobalMap(int treeNumber = RankGlobalTreeNumber);
 
    void setTreeNumber( int treeNumber );
    
    int getTreeNumber() const;
 
    int reserveIndex(int cardinality);
 
    void merge( const LocalToGlobalMap& otherMap, Type type );
 
    int getGlobalIndex( const std::pair<int,int>& index, Type type );
 
    void computeLocalToGlobalMap();
 
    /*
    Increase count of indices we reserve for a particular facet type 
    (currently support vertices, faces, cells). Pass in the type and
    the integer
    */
    void increaseCountOfFacetType(Type type, int count);
 
    int getTotalNumberOfIndices() const;
 
    int getTotalNumberOfCells() const;
    
    int getIndicesPerSubtree( int treeNumber ) const;
 
  
  private:
    int _treeNumber;
 
    /*
    these three should sum to the same
    as _totalNumberOfIndices
    */
    int _numberOfVertices;
    int _numberOfCells;
    int _numberOfFaces;
    
    // let's make this total number of indices in this tree specifically.
    // later, we use this to make sure the merge was successful
    int _totalNumberOfIndices;
 
    // std::vector<petsc::LocalToGlobalMap> _subTrees;
    int _numberOfSubtrees;
 
    std::map<int,int> _indicesPerSubtree;
 
    static tarch::logging::Log  _log;
 
};