00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031 #ifndef SUNDANCE_INHOMOGENEOUSNODALDOFMAP_H
00032 #define SUNDANCE_INHOMOGENEOUSNODALDOFMAP_H
00033
00034 #include "SundanceDefs.hpp"
00035 #include "SundanceDOFMapBase.hpp"
00036 #include "SundanceCellSet.hpp"
00037 #include "SundanceCellFilter.hpp"
00038 #include "SundanceBasisFamily.hpp"
00039 #include "SundanceObjectWithVerbosity.hpp"
00040
00041
00042 namespace Sundance
00043 {
00044 using namespace Teuchos;
00045
00046
00047
00048
00049 class InhomogeneousNodalDOFMap : public DOFMapBase
00050 {
00051 public:
00052
00053 InhomogeneousNodalDOFMap(const Mesh& mesh,
00054 const Array<Map<Set<int>, CellFilter> >& funcSetToDomainMap,
00055 int setupVerb);
00056
00057
00058 virtual ~InhomogeneousNodalDOFMap(){;}
00059
00060
00061 RCP<const MapStructure>
00062 getDOFsForCellBatch(int cellDim,
00063 const Array<int>& cellLID,
00064 const Set<int>& requestedFuncSet,
00065 Array<Array<int> >& dofs,
00066 Array<int>& nNodes,
00067 int verb) const ;
00068
00069 protected:
00070
00071 void getFunctionDofs(int cellDim,
00072 const Array<int>& cellLID,
00073 const Array<int>& facetLID,
00074 const Array<int>& funcs,
00075 Array<Array<int> >& dofs) const ;
00076
00077 public:
00078
00079 RCP<const Set<int> >
00080 allowedFuncsOnCellBatch(int cellDim,
00081 const Array<int>& cellLID) const ;
00082
00083
00084 const Array<CellFilter>& funcDomains() const {return funcDomains_;}
00085
00086
00087 virtual void print(std::ostream& os) const ;
00088
00089
00090 protected:
00091
00092
00093 Array<int> dofsOnCell(int cellDim, int cellLID, const Set<int>& reqFuncs) const ;
00094
00095
00096 void init();
00097
00098 void computeOffsets(int localCount) ;
00099
00100 void shareRemoteDOFs(const Array<Array<int> >& remoteNodes);
00101
00102 void assignNode(int fLID,
00103 int funcComboIndex,
00104 int dofOffset,
00105 int nFuncs,
00106 Array<Array<int> >& remoteNodes,
00107 Array<int>& hasProcessedCell,
00108 int& nextDOF) ;
00109
00110 int dim_;
00111 RCP<BasisDOFTopologyBase> basis_;
00112 int nTotalFuncs_;
00113 Array<CellFilter> funcDomains_;
00114
00115 Array<Array<int> > nodeDofs_;
00116 Array<Array<int> > elemDofs_;
00117 Array<int> nodeToFuncSetIndexMap_;
00118 Array<int> elemToFuncSetIndexMap_;
00119 Array<Set<int> > elemFuncSets_;
00120 Array<Set<int> > nodalFuncSets_;
00121 Array<int> nodeToOffsetMap_;
00122 Array<int> elemToOffsetMap_;
00123
00124 Array<Array<int> > funcIndexWithinNodeFuncSet_;
00125
00126 Array<RCP<const MapStructure> > elemStructure_;
00127 Array<RCP<const MapStructure> > nodeStructure_;
00128 };
00129 }
00130
00131
00132 #endif