Partition.hpp

Go to the documentation of this file.

/*--------------------------------------------------------------------*/
/*    Copyright 2001, 2002 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.                    */
/*--------------------------------------------------------------------*/

// Copyright 2001, 2002 Sandia Corporation, Albuquerque, NM.

#ifndef stk_rebalance_Partition_hpp
#define stk_rebalance_Partition_hpp

#include <stdexcept>

// STL components
#include <vector>
#include <utility>

#include <stk_mesh/base/Entity.hpp>
#include <stk_mesh/base/Field.hpp>
#include <stk_mesh/base/Types.hpp>
#include <stk_mesh/fem/CoordinateSystems.hpp>
#include <stk_mesh/fem/TopologyDimensions.hpp>

namespace stk_classic {
namespace rebalance {

typedef mesh::Field<double, mesh::Cartesian>  VectorField ;
typedef mesh::Field<double>                   ScalarField ;


class Partition {

public:

  Partition(stk_classic::ParallelMachine comm);
private:
  Partition(const Partition &p);
public:

  virtual void set_mesh_info ( const std::vector<mesh::Entity *> &mesh_entities,
                               const VectorField   * nodal_coord_ref,
                               const ScalarField   * elem_weight_ref=NULL) = 0;

  virtual ~Partition();

  ParallelMachine parallel() const { return comm_; }

  virtual unsigned num_elems() const = 0;

  virtual void determine_new_partition(bool &RebalancingNeeded) = 0;

  virtual int get_new_partition(stk_classic::mesh::EntityProcVec &new_partition) = 0;

  virtual bool partition_dependents_needed() const = 0;


protected:

  const stk_classic::ParallelMachine comm_;
};

}
} // namespace stk_classic

#endif