shells/parallel/test/MeshPartitioner.h

Go to the documentation of this file.

// -*- C++ -*- 
//
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//
//                                   Fehmi Cirak
//                        California Institute of Technology
//                           (C) 2004 All Rights Reserved
//
// <LicenseText>
//
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//
#ifndef MESHPARTITIONER_H
#define MESHPARTITIONER_H
#include "mpi.h"

#include <iostream>
#include <vector>
#include <set>


namespace parallel {
    class MeshPartitioner;
    class PointSetPartitioner;
}


class parallel::MeshPartitioner {
public:
    MeshPartitioner(const MPI_Comm& comm, std::istream& is, std::ostream& os);
    ~MeshPartitioner();

    void partition();

    void localToGlobalNodeIDs();

private:    
    MeshPartitioner(const MeshPartitioner &);
    const MeshPartitioner & operator=(const MeshPartitioner &);

private: 
    typedef std::vector<int>              _ConnectivityCont; 
    typedef std::vector<double>           _CoordinateCont;
    typedef std::vector<int>              _GhostMarkerCont; 
    typedef std::set<int>                 _LocalGlobalIDCont; 

    typedef _ConnectivityCont::iterator   _ConnectivityIt; 
    typedef _CoordinateCont::iterator     _CoordinateIt;
    typedef _GhostMarkerCont::iterator    _GhostMarkerIt;
    typedef _LocalGlobalIDCont::iterator  _LocalGlobalIDIt;

    _ConnectivityCont                     _connectivity;    // triangle connectivity
    _CoordinateCont                       _coordinate;      // nodal coordinates

    _GhostMarkerCont                      _ghostMarker;     // marker for ghost elements
                     
    _LocalGlobalIDCont                    _localGlobalNode; // mapping between global/local ID's  
                                                            // for repartitioning

    PointSetPartitioner                  *_psetPartitioner;

    std::ostream&                         _os;             
    
    private:
    template<typename OutputIt, typename InputIt>  
    void ghostAndActiveElements(InputIt elBegin, 
                                InputIt elEnd, 
                                OutputIt allElements);
    void computeElementCenters(_CoordinateCont& centerCoordinate);
    void readMeshSMF(std::istream &is);
    void writeMeshSMF();
};

#endif

/* End of file */

---