diff --git a/FunctionSpace/FunctionSpace.cpp b/FunctionSpace/FunctionSpace.cpp index ede8dada0ef57741dfc4dad1b3539bdc674e1f6e..ec6f1e1a1c1a1be901aed3db4a8a52419eed26da 100644 --- a/FunctionSpace/FunctionSpace.cpp +++ b/FunctionSpace/FunctionSpace.cpp @@ -3,7 +3,7 @@ using namespace std; FunctionSpace::FunctionSpace(void){ - ebLookUp = new map<Element*, Basis*, ElementComparator>; + ebLookUp = new map<MElement*, Basis*, ElementComparator>; } FunctionSpace::~FunctionSpace(void){ diff --git a/FunctionSpace/FunctionSpace.h b/FunctionSpace/FunctionSpace.h index cd5aaa420323082b24f266e2ae0ecb7d92ad609e..571a55773b0d55306495595a66ec938604910238 100644 --- a/FunctionSpace/FunctionSpace.h +++ b/FunctionSpace/FunctionSpace.h @@ -2,7 +2,7 @@ #define _FUNCTIONSPACE_H_ #include <map> -#include "Element.h" +#include "MElement.h" #include "Jacobian.h" #include "Basis.h" @@ -21,39 +21,39 @@ class FunctionSpace{ private: - + class ElementComparator{ public: - bool operator()(const Element* a, const Element* b) const; + bool operator()(const MElement* a, const MElement* b) const; }; - - std::map<Element*, Basis*, ElementComparator>* ebLookUp; // Element to Basis Lookup + + std::map<MElement*, Basis*, ElementComparator>* ebLookUp; // Element to Basis Lookup public: FunctionSpace(void); ~FunctionSpace(void); - void associate(Element& element, Basis& basis); + void associate(MElement& element, Basis& basis); void associate(int physical, Basis& basis); - Basis& getBasis(Element& element) const; + Basis& getBasis(MElement& element) const; }; ////////////////////// // Inline Functions // ////////////////////// -inline void FunctionSpace::associate(Element& element, Basis& basis){ - ebLookUp->insert(std::pair<Element*, Basis*>(&element, &basis)); +inline void FunctionSpace::associate(MElement& element, Basis& basis){ + ebLookUp->insert(std::pair<MElement*, Basis*>(&element, &basis)); } -inline Basis& FunctionSpace::getBasis(Element& element) const{ +inline Basis& FunctionSpace::getBasis(MElement& element) const{ return *(ebLookUp->find(&element)->second); } inline bool FunctionSpace::ElementComparator::operator() -(const Element* a, const Element* b) const{ - return a->getId() < b->getId(); +(const MElement* a, const MElement* b) const{ + return a->getNum() < b->getNum(); } #endif