mpc.hh

Go to the documentation of this file.

//==============================================================================
//==============================================================================
#ifndef MPC_HH_
#define MPC_HH_

#include <map>
#include <vector>
#include <iostream>
#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& dof)
    {
      return s <<"u_"<< char('w'+dof.dof) <<" in node "<< dof.node;
    }

    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