FEI Version of the Day
fei_NodeDatabase.hpp
00001 /*
00002 // @HEADER
00003 // ************************************************************************
00004 //             FEI: Finite Element Interface to Linear Solvers
00005 //                  Copyright (2005) Sandia Corporation.
00006 //
00007 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation, the
00008 // U.S. Government retains certain rights in this software.
00009 //
00010 // Redistribution and use in source and binary forms, with or without
00011 // modification, are permitted provided that the following conditions are
00012 // met:
00013 //
00014 // 1. Redistributions of source code must retain the above copyright
00015 // notice, this list of conditions and the following disclaimer.
00016 //
00017 // 2. Redistributions in binary form must reproduce the above copyright
00018 // notice, this list of conditions and the following disclaimer in the
00019 // documentation and/or other materials provided with the distribution.
00020 //
00021 // 3. Neither the name of the Corporation nor the names of the
00022 // contributors may be used to endorse or promote products derived from
00023 // this software without specific prior written permission.
00024 //
00025 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
00026 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
00027 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
00028 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
00029 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
00030 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
00031 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
00032 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
00033 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
00034 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
00035 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00036 //
00037 // Questions? Contact Alan Williams (william@sandia.gov) 
00038 //
00039 // ************************************************************************
00040 // @HEADER
00041 */
00042 
00043 #ifndef _NodeDatabase_hpp_
00044 #define _NodeDatabase_hpp_
00045 
00046 
00047 #include "fei_macros.hpp"
00048 #include "fei_fwd.hpp"
00049 #include "fei_defs.h"
00050 #include "fei_NodeDescriptor.hpp"
00051 #include "fei_Pool_alloc.hpp"
00052 #include "fei_mpi.h"
00053 
00054 #include <map>
00055 
00101 class NodeDatabase {
00102  public:
00104   NodeDatabase(std::map<int,int>* fieldDatabase,
00105          NodeCommMgr* nodeCommMgr);
00106 
00108   virtual ~NodeDatabase();
00109 
00113   int getNumNodeDescriptors() const { return( nodePtrs_.size() ); };
00114 
00118   std::map<GlobalID,int>& getNodeIDs() { return( nodeIDs_ ); };
00119 
00129   int getNodeWithID(GlobalID nodeID, const NodeDescriptor*& node) const;
00130   int getNodeWithID(GlobalID nodeID, NodeDescriptor*& node);
00131 
00141   int getNodeWithNumber(int nodeNumber, const NodeDescriptor*& node) const;
00142 
00149   int getNodeWithEqn(int eqnNumber, const NodeDescriptor*& node) const;
00150 
00156   void getNodeAtIndex(int i, const NodeDescriptor*& node) const;
00157   void getNodeAtIndex(int i, NodeDescriptor*& node);
00158 
00164   int countLocalNodalEqns(int localRank);
00165 
00172   int countLocalNodeDescriptors(int localRank);
00173 
00179   int getIndexOfID(GlobalID nodeID) const;
00180 
00188   int initNodeID(GlobalID nodeID);
00189 
00197   int initNodeIDs(GlobalID* nodeIDs, int numNodes);
00198 
00212   int synchronize(int firstLocalNodeNumber,
00213       int firstLocalEqn,
00214       int localRank,
00215       MPI_Comm comm);
00216 
00220   int getAssociatedNodeNumber(int eqnNumber);
00221 
00225   int getAssociatedFieldID(int eqnNumber);
00226 
00228   bool isSynchronized() { return( synchronized_ ); };
00229 
00230  private:
00231   NodeDatabase(const NodeDatabase& src);
00232   NodeDatabase& operator=(const NodeDatabase& src);
00233 
00234   void deleteMemory();
00235 
00236   std::vector<NodeDescriptor*> nodePtrs_;
00237 
00238   std::vector<int> eqnNumbers_;  //eqnNumbers_ will be a sorted list of the
00239                                   //first global equation number at each node
00240                                   //in nodePtrs_.
00241                                   //the relationship between eqnNumbers_ and
00242   std::vector<int> eqnNodeIndices_;  //eqnNodeIndices_ is like this:
00243                                   //if eqn == eqnNumbers_[i], then
00244                                   //  nodePtrs_[eqnNodeIndices_[i]] points to
00245                                   //  the node with 'eqn'
00246 
00247   std::map<GlobalID,int> nodeIDs_; //nodeIDs_ maps node-ID to an index into
00248                                 //the nodePtrs_ array of NodeDescriptors.
00249 
00250   std::map<int,int> nodeNumbers_;
00251 
00252   bool synchronized_;
00253   bool need_to_alloc_and_sync_;
00254 
00255   std::map<int,int>* fieldDB_;
00256   NodeCommMgr* nodeCommMgr_;
00257 
00258   int numLocalNodes_;
00259   int firstLocalNodeNumber_, lastLocalNodeNumber_;
00260 
00261   fei_Pool_alloc<NodeDescriptor> nodePool_;
00262 };
00263 
00264 #endif
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Friends