mpc.hh

Go to the documentation of this file.

//==============================================================================
//==============================================================================
#ifndef MPC_HH_
#define MPC_HH_
 
#include <map>
#include <vector>
#include <iosfwd>
#include <set>
 
namespace Opm {
namespace Elasticity {
 
enum Direction { NONE = 0, X = 1, Y = 2, Z = 4,
                 XY = 1+2, XZ = 1+4, YZ = 2+4, 
                 XYZ = 1+2+4 };
 
class MPC
{
public:
 
  struct DOF
  {
    DOF() : node(0), dof(0), coeff(double(0)) {}
 
    DOF(int n, int d, double c = double(0)) : node(n), dof(d), coeff(c) {}
 
    friend std::ostream& operator<<(std::ostream& s, const DOF& d);
 
    int  node;  
    int  dof;   
    double coeff; 
  };
 
  class Less {
  public:
    bool operator()(const MPC* lhs, const MPC* rhs) const;
    static bool compareSlaveDofOnly;
  };
 
  MPC(int n, int d, double c = double(0)) : slave(n,d,c) { iceq = -1; }
 
  void addMaster(int n, int d, double c = double(1), double tol = double(1.0e-8))
  {
    if (c < -tol || c > tol) master.push_back(DOF(n,d,c));
  }
 
  void updateMaster(size_t pos, double c)
  {
    if (pos < master.size())
      master[pos].coeff = c;
  }
 
  void removeMaster(size_t pos)
  {
    if (pos < master.size())
      master.erase(master.begin()+pos);
  }
 
  void addOffset(double offset) { slave.coeff += offset; }
 
  void setSlaveCoeff(double c0) { slave.coeff = c0; }
 
  const DOF& getSlave() const          { return slave; }
  const DOF& getMaster(size_t i) const { return master[i]; }
 
  size_t getNoMaster() const { return master.size(); }
 
  friend std::ostream& operator<<(std::ostream& s, const MPC& mpc);
 
  int              iceq;   
private:
  DOF              slave;  
  std::vector<DOF> master; 
};
 
typedef std::set<MPC*,MPC::Less> MPCSet;
 
typedef std::map<int,MPC*> MPCMap;
 
}
}
 
#endif