snl_fei_PointBlockMap.hpp

/*--------------------------------------------------------------------*/
/*    Copyright 2005 Sandia Corporation.                              */
/*    Under the terms of Contract DE-AC04-94AL85000, there is a       */
/*    non-exclusive license for use of this work by or on behalf      */
/*    of the U.S. Government.  Export of this program may require     */
/*    a license from the United States Government.                    */
/*--------------------------------------------------------------------*/
 
#ifndef _snl_fei_PointBlockMap_hpp_
#define _snl_fei_PointBlockMap_hpp_
 
#include <fei_macros.hpp>
 
#include <map>
 
namespace snl_fei {
  class PointBlockMap {
  public:
    PointBlockMap();

    virtual ~PointBlockMap();

   void setPtEqualBlk();

   bool ptEqualBlk() { return( ptEqualBlk_ ); }

   int setEqn(int ptEqn, int blkEqn);

   int setEqn(int ptEqn, int blkEqn, int blkSize);

   int setBlkEqnSize(int blkEqn, int size);

   int getBlkEqnSize(int blkEqn);

   int getMaxBlkEqnSize() { return( maxSize_ ); }

   void setMaxBlkEqnSize(int sz) { maxSize_ = sz; }

   int eqnToBlkEqn(int eqn) const;

   int blkEqnToPtEqn(int blkEqn) const;

   int getBlkEqnInfo(int blkEqn, int& ptEqn, int& blkSize);

   int getPtEqnInfo(int ptEqn, int& blkEqn, int& blkOffset);

   int getBlkEqnOffset(int blkEqn, int ptEqn);

   bool isExactlyBlkEqn(int ptEqn);

   std::map<int,int>* getPtEqns()
     {
       return( ptEqns_ );
     }

   std::map<int,std::pair<int,int> >* getBlkEqns()
     {
       return( blkEqns_ );
     }
 
  private:
   PointBlockMap(const PointBlockMap& src);
 
   PointBlockMap& operator=(const PointBlockMap& src);
 
   std::map<int,int>* ptEqns_; //ptEqns_ maps point-equations to block-equations
 
   std::map<int,std::pair<int,int> >* blkEqns_;
   //blkEqns_ maps block-equations to point-equations and block-sizes
   //(the point-equation stored is the first or smallest point-equation that is
   //associated with the block-equation)
 
   int maxSize_;
 
   bool ptEqualBlk_; //if this is true, then all blkSizes are 1
  };//class PointBlockMap
}//namespace snl_fei
#endif // _snl_fei_PointBlockMap_hpp_