// Gmsh - Copyright (C) 1997-2021 C. Geuzaine, J.-F. Remacle
//
// See the LICENSE.txt file for license information. Please report all
// issues on https://gitlab.onelab.info/gmsh/gmsh/issues.
//
// Contributor(s):
//   Eric Bechet
//

#ifndef QUADRATURERULES_H
#define QUADRATURERULES_H

#include "GaussIntegration.h"
#include "MElement.h"

class QuadratureBase {
public:
  virtual ~QuadratureBase() {}
  virtual int getIntPoints(MElement *e, IntPt **GP) = 0;
};

// For rigid contact no need of Gauss'integration
// but to use clasical get function in term npts and IntPt are needed
// so use a empty gaussQuadrature rule
class QuadratureVoid : public QuadratureBase {
public:
  QuadratureVoid() : QuadratureBase() {}
  ~QuadratureVoid() {}
  int getIntPoints(MElement *e, IntPt **GP)
  {
    GP = nullptr;
    return 0;
  }
};

class GaussQuadrature : public QuadratureBase {
public:
  enum IntegCases { Other, Val, Grad, ValVal, GradGrad };

private:
  int order;
  IntegCases info;

public:
  GaussQuadrature(int order_ = 0) : order(order_), info(Other) {}
  GaussQuadrature(IntegCases info_) : order(0), info(info_) {}
  virtual ~GaussQuadrature() {}
  virtual int getIntPoints(MElement *e, IntPt **GP)
  {
    int integrationOrder;
    int npts;
    int geoorder = e->getPolynomialOrder();
    switch(info) {
    case Other: integrationOrder = order; break;
    case Val: integrationOrder = geoorder + 1; break;
    case Grad: integrationOrder = geoorder; break;
    case ValVal: integrationOrder = 2 * geoorder; break;
    case GradGrad: integrationOrder = 3 * (geoorder - 1) + 1; break;
    default: integrationOrder = 1;
    }
    e->getIntegrationPoints(integrationOrder, &npts, GP);
    return npts;
  }
  // copy constructor
  GaussQuadrature(const GaussQuadrature &other)
    : order(other.order), info(other.info)
  {
  }
};

#endif //_QUADRATURERULES_H_
