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_