#ifndef WEAK_EQUATION_H
#define WEAK_EQUATION_H

#include <string>
#include <set>

#include "ATC_TypeDefs.h"

namespace ATC{

class Material;

 /**
   *  @class  WeakEquation
   *  @brief  WeakEquation is a template for adapting FE weak equations
   *  for the FE_Engine. It wrappers the correct material functions for
   *  integration over elements
   */

class WeakEquation {

  public:
  /** type of equation */
  enum PDE_Type {DYNAMIC_PDE=0, STATIC_PDE, EIGENVALUE_PDE, PROJECTION_PDE};

  // constructor
  WeakEquation(PDE_Type type, FieldName fieldName, int fieldSize) :
  type_(type), fieldName_(fieldName), fieldSize_(fieldSize), nsd_(3) {};

  // destructor
  virtual ~WeakEquation(){};

  /** integrand  that used to form the energy */
  virtual bool has_E_integrand(void) const {return false;}
  virtual void E_integrand(const FIELD_MATS & /* fields */,
                           const GRAD_FIELD_MATS & /* grad_fields */,
                           const Material * /* material */,
                           DENS_MAT & /* energy */ ) const {};

  /** density that used to form the mass matrix */
  virtual bool has_M_integrand(void) const {return false;}
  virtual void M_integrand(const FIELD_MATS & /* fields */,
                           const Material * /* material */,
                           DENS_MAT & /* density */ ) const {};

  /** flux that is integrated with B = Grad N as its weight */
  virtual bool has_B_integrand(void) const {return false;}
  virtual void B_integrand(const FIELD_MATS & /* fields */,
                           const GRAD_FIELD_MATS & /* grad_fields */,
                           const Material * /* material */,
                           DENS_MAT_VEC & /* flux */) const {};

  /** flux that is integrated with N as its weight */
  virtual bool has_N_integrand(void) const {return false;}
  // N_integrand bool is for masking in FE_Engine
  virtual bool N_integrand(const FIELD_MATS & /* fields */,
                           const GRAD_FIELD_MATS & /* grad_fields */,
                           const Material * /* material */,
                           DENS_MAT & /* flux */) const {return false;};

  /** stiffness matrix */

  // linear
  virtual void BB_tangent_coefficients(const FieldName /* field */,
                                       const Material * /* material */,
                                       DENS_MAT & /* coefs */) const {};
  virtual void NN_tangent_coefficients(const FieldName /* field */,
                                       const Material * /* material */,
                                       DENS_MAT & /* coefs */) const {};
  // non-linear
  virtual bool has_BB_tangent_coefficients(void) const {return false;}
  virtual void BB_tangent_coefficients(const FieldName /* field */,
                                       const FIELD_MATS & /* fields */,
                                       const Material * /* material */,
                                       DENS_MAT & /* coefs */) const {};
  virtual bool has_NN_tangent_coefficients(void) const {return false;}
  virtual void NN_tangent_coefficients(const FieldName /* field */,
                                       const FIELD_MATS & /* fields */,
                                       const Material * /* material */,
                                       DENS_MAT & /* coefs */) const {};

  /** type of equation */
  PDE_Type type(void) const {return type_;}

  /** primary field */
  FieldName field_name(void) const {return fieldName_;}
  int field_size(void) const {return fieldSize_;}

  /** list of require interfaces */
  virtual std::set<std::string> needs_material_functions(void) const = 0;

  protected:
  /** type of equation */
  PDE_Type type_;

  /** field */
  FieldName fieldName_;

  /** field size */
  int fieldSize_;

  /** number of spatial dimensions */
  int nsd_;

};

}
#endif
