fei_Dof.hpp

/*--------------------------------------------------------------------*/
/*    Copyright 2009 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 _fei_Dof_hpp_
#define _fei_Dof_hpp_

#include <fei_macros.hpp>

namespace fei {

template<class LocalOrdinal, class GlobalOrdinal>
class Dof {
 public:
  Dof(LocalOrdinal rank, GlobalOrdinal id, LocalOrdinal field)
   : m_rank(rank), m_id(id), m_field(field) {}

  ~Dof(){}

  LocalOrdinal rank() const { return m_rank; }
  GlobalOrdinal id() const { return m_id; }
  LocalOrdinal field() const { return m_field; }

 private:
  LocalOrdinal m_rank;
  GlobalOrdinal m_id;
  LocalOrdinal m_field;
};//class Dof

template<class LocalOrdinal, class GlobalOrdinal>
struct less_rank_id_field {
  bool operator()(const Dof<LocalOrdinal,GlobalOrdinal>& dof1,
                 const Dof<LocalOrdinal,GlobalOrdinal>& dof2) const
  {
    if (dof1.rank()==dof2.rank()) {
      if (dof1.id() == dof2.id()) return dof1.field() < dof2.field();
      else return dof1.id() < dof2.id();
    }
    else {
      return dof1.rank() < dof2.rank();
    }
  }

  bool operator()(const Dof<LocalOrdinal,GlobalOrdinal>* dof1,
                 const Dof<LocalOrdinal,GlobalOrdinal>* dof2) const
  {
    if (dof1->rank()==dof2->rank()) {
      if (dof1->id() == dof2->id()) return dof1->field() < dof2->field();
      else return dof1->id() < dof2->id();
    }
    else {
      return dof1->rank() < dof2->rank();
    }
  }
};

template<class LocalOrdinal, class GlobalOrdinal>
struct less_field_rank_id {
  bool operator()(const Dof<LocalOrdinal,GlobalOrdinal>& dof1,
                 const Dof<LocalOrdinal,GlobalOrdinal>& dof2) const
  {
    if (dof1.field()==dof2.field()) {
      if (dof1.rank() == dof2.rank()) return dof1.id() < dof2.id();
      else return dof1.rank() < dof2.rank();
    }
    else {
      return dof1.field() < dof2.field();
    }
  }

  bool operator()(const Dof<LocalOrdinal,GlobalOrdinal>* dof1,
                 const Dof<LocalOrdinal,GlobalOrdinal>* dof2) const
  {
    if (dof1->field()==dof2->field()) {
      if (dof1->rank() == dof2->rank()) return dof1->id() < dof2->id();
      else return dof1->rank() < dof2->rank();
    }
    else {
      return dof1->field() < dof2->field();
    }
  }
};

}//namespace fei
#endif