UnsaturatedFlow.hpp

/** \file UnsaturatedFlow.hpp
 * \brief Mix implementation of transport element
 * \example UnsaturatedFlow.hpp
 *
 * \ingroup mofem_mix_transport_elem
 *
 */

/*
 * This file is part of MoFEM.
 * MoFEM is free software: you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the
 * Free Software Foundation, either version 3 of the License, or (at your
 * option) any later version.
 *
 * MoFEM is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
 * License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with MoFEM. If not, see <http://www.gnu.org/licenses/>. */

#ifndef __UNSATURATD_FLOW_HPP__
#define __UNSATURATD_FLOW_HPP__

namespace MixTransport {

  /**
   * \brief Generic material model for unsaturated water transport
   *
   * \note This is abstact class, no physical material model is implemented
   * here.
   */
  struct GenericMaterial {

    virtual ~GenericMaterial() {
    }

    static double ePsilon0;    ///< Regularization parameter
    static double ePsilon1;    ///< Regularization parameter
    double sCale;              ///< Scale time dependent eq.
    // double Ks;                 ///< Saturated hydraulic conductivity [m/day]

    double h;                ///< hydraulic head
    double h_t;              ///< rate of hydraulic head
    // double h_flux_residual;  ///< residual at point
    double K;                ///< Hydraulic conductivity [L/s]
    double diffK;            ///< Derivative of hydraulic conductivity [L/s * L^2/F]
    double C;                ///< Capacity [S^2/L^2]
    double diffC;            ///< Derivative of capacity [S^2/L^2 * L^2/F ]
    double tHeta;            ///< Water content
    double Se;               ///< Effective saturation

    Range tEts;              ///< Elements with this material

    double x,y,z;            ///< in meters (L)


    /**
     * \brief Initialize head
     * @return value of head
     */
    virtual double initalPcEval() const = 0;
    virtual void printMatParameters(const int id,const std::string& prefix) const = 0;

    virtual MoFEMErrorCode calK() {
      MoFEMFunctionBeginHot;
      SETERRQ(
        PETSC_COMM_SELF,MOFEM_NOT_IMPLEMENTED,
        "Not implemented how to calculate hydraulic conductivity"
      );
      MoFEMFunctionReturnHot(0);
    }

    virtual MoFEMErrorCode calDiffK() {
      MoFEMFunctionBeginHot;
      SETERRQ(
        PETSC_COMM_SELF,MOFEM_NOT_IMPLEMENTED,
        "Not implemented how to calculate derivative of hydraulic conductivity"
      );
      MoFEMFunctionReturnHot(0);
    }

    virtual MoFEMErrorCode calC() {
      MoFEMFunctionBeginHot;
      SETERRQ(
        PETSC_COMM_SELF,MOFEM_NOT_IMPLEMENTED,
        "Not implemented how to calculate capacity"
      );
      MoFEMFunctionReturnHot(0);
    }

    virtual MoFEMErrorCode calDiffC() {
      MoFEMFunctionBeginHot;
      SETERRQ(
        PETSC_COMM_SELF,MOFEM_NOT_IMPLEMENTED,
        "Not implemented how to calculate capacity"
      );
      MoFEMFunctionReturnHot(0);
    }

    virtual MoFEMErrorCode calTheta() {
      MoFEMFunctionBeginHot;
      SETERRQ(
        PETSC_COMM_SELF,MOFEM_NOT_IMPLEMENTED,
        "Not implemented how to calculate capacity"
      );
      MoFEMFunctionReturnHot(0);
    }

    virtual MoFEMErrorCode calSe() {
      MoFEMFunctionBeginHot;
      SETERRQ(
        PETSC_COMM_SELF,MOFEM_NOT_IMPLEMENTED,
        "Not implemented how to calculate capacity"
      );
      MoFEMFunctionReturnHot(0);
    }

  };

  /**
   * \brief Implementation of operators, problem and finite elements for
   * unsaturated flow
   */
  struct UnsaturatedFlowElement: public MixTransportElement {

    DM dM;  ///< Discrete manager for unsaturated flow problem

    UnsaturatedFlowElement(MoFEM::Interface &m_field):
    MixTransportElement(m_field),
    dM(PETSC_NULL),
    lastEvalBcValEnt(0),
    lastEvalBcBlockValId(-1),
    lastEvalBcFluxEnt(0),
    lastEvalBcBlockFluxId(-1) {
    }

    ~UnsaturatedFlowElement() {
      if(dM!=PETSC_NULL) {
        ierr = DMDestroy(&dM); CHKERRABORT(PETSC_COMM_WORLD,ierr);
      }
    }

    typedef std::map<int,boost::shared_ptr<GenericMaterial> > MaterialsDoubleMap;
    MaterialsDoubleMap dMatMap; ///< materials database

    /**
     * \brief For given element handle get material block Id
     * @param  ent      finite element entity handle
     * @param  block_id reference to returned block id
     * @return          error code
     */
    virtual MoFEMErrorCode getMaterial(
      const EntityHandle ent,int &block_id
    ) const {
      MoFEMFunctionBeginHot;
      for(
        MaterialsDoubleMap::const_iterator mit = dMatMap.begin();
        mit!=dMatMap.end();mit++
      ) {
        if(mit->second->tEts.find(ent)!=mit->second->tEts.end()) {
          block_id = mit->first;
          MoFEMFunctionReturnHot(0);
        }
      }
      SETERRQ(mField.get_comm(),MOFEM_DATA_INCONSISTENCY,"Element not found, no material data");
      MoFEMFunctionReturnHot(0);
    }

    /**
     * \brief Class storing information about boundary condition
     */
    struct BcData {
      Range eNts;
      double fixValue;
      boost::function<
      double (const double x,const double y,const double z)
      > hookFun;
      BcData():
      hookFun(NULL) {
      }
    };
    typedef map<int,boost::shared_ptr<BcData> > BcMap;
    BcMap bcValueMap; ///< Store boundary condition for head capillary pressure

    EntityHandle lastEvalBcValEnt;
    int lastEvalBcBlockValId;

    /**
     * \brief Get value on boundary
     * @param  ent   entity handle
     * @param  gg    number of integration point
     * @param  x     x-coordinate
     * @param  y     y-coordinate
     * @param  z     z-coordinate
     * @param  value returned value
     * @return       error code
     */
    MoFEMErrorCode getBcOnValues(
      const EntityHandle ent,const int gg,
      const double x,const double y,const double z,
      double &value
    ) {
      MoFEMFunctionBeginHot;
      int block_id = -1;
      if(lastEvalBcValEnt==ent) {
        block_id = lastEvalBcBlockValId;
      } else {
        for(BcMap::iterator it = bcValueMap.begin();it!=bcValueMap.end();it++) {
          if(it->second->eNts.find(ent)!=it->second->eNts.end()) {
            block_id = it->first;
          }
        }
        lastEvalBcValEnt = ent;
        lastEvalBcBlockValId = block_id;
      }
      if(block_id>=0) {
        if(bcValueMap.at(block_id)->hookFun) {
          value = bcValueMap.at(block_id)->hookFun(x,y,z);
        } else {
          value = bcValueMap.at(block_id)->fixValue;
        }
      } else {
        value = 0;
      }
      MoFEMFunctionReturnHot(0);
    }

    BcMap bcFluxMap;
    EntityHandle lastEvalBcFluxEnt;
    int lastEvalBcBlockFluxId;

    /**
     * \brief essential (Neumann) boundary condition (set fluxes)
     * @param  ent  handle to finite element entity
     * @param  x    coord
     * @param  y    coord
     * @param  z    coord
     * @param  flux reference to flux which is set by function
     * @return      [description]
     */
    MoFEMErrorCode getBcOnFluxes(
      const EntityHandle ent,
      const double x,const double y,const double z,
      double &flux
    ) {
      MoFEMFunctionBeginHot;
      int block_id = -1;
      if(lastEvalBcFluxEnt==ent) {
        block_id = lastEvalBcBlockFluxId;
      } else {
        for(BcMap::iterator it = bcFluxMap.begin();it!=bcFluxMap.end();it++) {
          if(it->second->eNts.find(ent)!=it->second->eNts.end()) {
            block_id = it->first;
          }
        }
        lastEvalBcFluxEnt = ent;
        lastEvalBcBlockFluxId = block_id;
      }
      if(block_id>=0) {
        if(bcFluxMap.at(block_id)->hookFun) {
          flux = bcFluxMap.at(block_id)->hookFun(x,y,z);
        } else {
          flux = bcFluxMap.at(block_id)->fixValue;
        }
      } else {
        flux = 0;
      }
      MoFEMFunctionReturnHot(0);
    }


    /**
     * \brief Evaluate boundary condition at the boundary
     */
    struct OpRhsBcOnValues: public MoFEM::FaceElementForcesAndSourcesCore::UserDataOperator {

      UnsaturatedFlowElement &cTx;
      boost::shared_ptr<MethodForForceScaling> valueScale;

      /**
       * \brief Constructor
       */
      OpRhsBcOnValues(
        UnsaturatedFlowElement &ctx,const std::string fluxes_name,
        boost::shared_ptr<MethodForForceScaling>& value_scale
      ):
      MoFEM::FaceElementForcesAndSourcesCore::UserDataOperator(fluxes_name,UserDataOperator::OPROW),
      cTx(ctx),
      valueScale(value_scale) {
      }

      VectorDouble nF;  ///< Vector of residuals

      /**
       * \brief Integrate boundary condition
       * @param  side local index of entity
       * @param  type type of entity
       * @param  data data on entity
       * @return      error code
       */
      MoFEMErrorCode doWork(
        int side,EntityType type,DataForcesAndSourcesCore::EntData &data
      ) {
        MoFEMFunctionBeginHot;
        try {
          if(data.getFieldData().size()==0) MoFEMFunctionReturnHot(0);
          // Get EntityHandle of the finite element
          EntityHandle fe_ent = getNumeredEntFiniteElementPtr()->getEnt();
          // Resize and clear vector
          nF.resize(data.getIndices().size());
          nF.clear();
          // Get number of integration points
          int nb_gauss_pts = data.getN().size1();
          for(int gg = 0;gg<nb_gauss_pts;gg++) {
            double x,y,z;
            x = getCoordsAtGaussPts()(gg,0);
            y = getCoordsAtGaussPts()(gg,1);
            z = getCoordsAtGaussPts()(gg,2);
            double value;
            // get value of boundary condition
            ierr = cTx.getBcOnValues(fe_ent,gg,x,y,z,value); CHKERRG(ierr);
            const double w = getGaussPts()(2,gg)*0.5;
            const double beta = w*(value-z);
            noalias(nF) += beta*prod(data.getVectorN<3>(gg),getNormal());
          }
          // Scale vector if history  evaluating method is given
          Vec f = getFEMethod()->ts_F;
          if(valueScale) {
            ierr = valueScale->scaleNf(getFEMethod(),nF); CHKERRG(ierr);
          }
          // Assemble vector
          ierr = VecSetValues(
            f,data.getIndices().size(),&data.getIndices()[0],&nF[0],ADD_VALUES
          ); CHKERRG(ierr);
        } catch (const std::exception& ex) {
          std::ostringstream ss;
          ss << "throw in method: " << ex.what() << std::endl;
          SETERRQ(PETSC_COMM_SELF,MOFEM_STD_EXCEPTION_THROW,ss.str().c_str());
        }
        MoFEMFunctionReturnHot(0);
      }
    };

    /**
     * \brief Assemble flux residual
     */
    struct OpResidualFlux: public MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator {

      UnsaturatedFlowElement &cTx;

      OpResidualFlux(UnsaturatedFlowElement &ctx,const std::string& flux_name):
      MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator(flux_name,UserDataOperator::OPROW),
      cTx(ctx) {}

      VectorDouble divVec,nF;
      FTensor::Index<'i',3> i;

      MoFEMErrorCode doWork(int side,EntityType type,DataForcesAndSourcesCore::EntData &data) {
        MoFEMFunctionBeginHot;
        const int nb_dofs = data.getIndices().size();
        if(nb_dofs==0) MoFEMFunctionReturnHot(0);
        nF.resize(nb_dofs,false);
        nF.clear();
        // Get EntityHandle of the finite element
        EntityHandle fe_ent = getNumeredEntFiniteElementPtr()->getEnt();
        // Get material block id
        int block_id;
        ierr = cTx.getMaterial(fe_ent,block_id); CHKERRG(ierr);
        // Get material block
        boost::shared_ptr<GenericMaterial>& block_data = cTx.dMatMap.at(block_id);
        // block_data->printMatParameters(block_id,"Read material");
        // Get base function
        auto t_n_hdiv = data.getFTensor1N<3>();
        // Get pressure
        auto t_h = getFTensor0FromVec(cTx.valuesAtGaussPts);
        // Get flux
        auto t_flux = getFTensor1FromMat<3>(cTx.fluxesAtGaussPts);
        // Coords at integration points
        auto t_coords = getFTensor1CoordsAtGaussPts();
        // Get integration weight
        auto t_w = getFTensor0IntegrationWeight();
        // Get volume
        double vol = getVolume();
        // Get material parameters
        int nb_gauss_pts = data.getN().size1();
        for(int gg = 0;gg!=nb_gauss_pts;gg++) {
          // Get divergence
          ierr = getDivergenceOfHDivBaseFunctions(side,type,data,gg,divVec); CHKERRG(ierr);
          const double alpha = t_w*vol;
          block_data->h = t_h;
          block_data->x = t_coords(0);
          block_data->y = t_coords(1);
          block_data->z = t_coords(2);
          ierr = block_data->calK(); CHKERRG(ierr);
          const double K = block_data->K;
          const double z = t_coords(2); /// z-coordinate at Gauss pt
          // Calculate pressure gradient
          noalias(nF) -= alpha*(t_h-z)*divVec;
          // Calculate presure gradient from flux
          FTensor::Tensor0<double*> t_nf(&*nF.begin());
          for(int rr = 0;rr!=nb_dofs;rr++) {
            t_nf += alpha*(1/K)*(t_n_hdiv(i)*t_flux(i));
            ++t_n_hdiv; // move to next base function
            ++t_nf; // move to next element in vector
          }
          ++t_h;  // move to next integration point
          ++t_flux;
          ++t_coords;
          ++t_w;
        }
        // Assemble residual
        ierr = VecSetValues(
          getFEMethod()->ts_F,nb_dofs,
          &*data.getIndices().begin(),&*nF.begin(),ADD_VALUES
        ); CHKERRG(ierr);
        MoFEMFunctionReturnHot(0);
      }

    };

    /**
     * Assemble mass residual
     */
    struct OpResidualMass: public MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator {

      UnsaturatedFlowElement &cTx;

      OpResidualMass(UnsaturatedFlowElement &ctx,const std::string& val_name):
      MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator(val_name,UserDataOperator::OPROW),
      cTx(ctx) {
      }

      VectorDouble nF;

      MoFEMErrorCode doWork(int side,EntityType type,DataForcesAndSourcesCore::EntData &data) {
        MoFEMFunctionBeginHot;
        MoFEMFunctionBeginHot;
        const int nb_dofs = data.getIndices().size();
        if(nb_dofs==0) MoFEMFunctionReturnHot(0);
        // Resize local element vector
        nF.resize(nb_dofs,false);
        nF.clear();
        // Get EntityHandle of the finite element
        EntityHandle fe_ent = getNumeredEntFiniteElementPtr()->getEnt();
        // Get material block id
        int block_id;
        ierr = cTx.getMaterial(fe_ent,block_id); CHKERRG(ierr);
        // Get material block
        boost::shared_ptr<GenericMaterial>& block_data = cTx.dMatMap.at(block_id);
        // Get pressure
        auto t_h = getFTensor0FromVec(cTx.valuesAtGaussPts);
        // Get pressure rate
        auto t_h_t = getFTensor0FromVec(*cTx.headRateAtGaussPts);
        // Flux divergence
        auto t_div_flux = getFTensor0FromVec(cTx.divergenceAtGaussPts);
        // Get integration weight
        auto t_w = getFTensor0IntegrationWeight();
        // Coords at integration points
        auto t_coords = getFTensor1CoordsAtGaussPts();
        // Scale eq.
        const double scale = block_data->sCale;
        // Get volume
        const double vol = getVolume();
        // Get number of integration points
        int nb_gauss_pts = data.getN().size1();
        for(int gg = 0;gg!=nb_gauss_pts;gg++) {
          const double alpha = t_w*vol*scale;
          block_data->h = t_h;
          block_data->x = t_coords(0);
          block_data->y = t_coords(1);
          block_data->z = t_coords(2);
          ierr = block_data->calC(); CHKERRG(ierr);
          const double C = block_data->C;
          // Calculate flux conservation
          noalias(nF) += (alpha*(t_div_flux+C*t_h_t))*data.getN(gg);
          ++t_h;
          ++t_h_t;
          ++t_div_flux;
          ++t_coords;
          ++t_w;
        }
        // Assemble local vector
        Vec f = getFEMethod()->ts_F;
        ierr = VecSetValues(
          f,nb_dofs,&*data.getIndices().begin(),&*nF.begin(),ADD_VALUES
        ); CHKERRG(ierr);
        MoFEMFunctionReturnHot(0);
      }

    };

    struct OpTauDotSigma_HdivHdiv: public MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator {

      UnsaturatedFlowElement &cTx;

      OpTauDotSigma_HdivHdiv(UnsaturatedFlowElement &ctx,const std::string flux_name):
      MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator(
        flux_name,flux_name,UserDataOperator::OPROWCOL
      ),
      cTx(ctx) {
        sYmm = true;
      }

      MatrixDouble nN,transNN;

      FTensor::Index<'j',3> j;

      /**
       * \brief Assemble matrix
       * @param  row_side local index of row entity on element
       * @param  col_side local index of col entity on element
       * @param  row_type type of row entity, f.e. MBVERTEX, MBEDGE, or MBTET
       * @param  col_type type of col entity, f.e. MBVERTEX, MBEDGE, or MBTET
       * @param  row_data data for row
       * @param  col_data data for col
       * @return          error code
       */
      MoFEMErrorCode doWork(
        int row_side,int col_side,
        EntityType row_type,EntityType col_type,
        DataForcesAndSourcesCore::EntData &row_data,
        DataForcesAndSourcesCore::EntData &col_data
      ) {
        MoFEMFunctionBeginHot;
        try {
          const int nb_row = row_data.getIndices().size();
          const int nb_col = col_data.getIndices().size();
          if(nb_row==0) MoFEMFunctionReturnHot(0);
          if(nb_col==0) MoFEMFunctionReturnHot(0);
          nN.resize(nb_row,nb_col,false);
          nN.clear();
          // Get EntityHandle of the finite element
          EntityHandle fe_ent = getNumeredEntFiniteElementPtr()->getEnt();
          // Get material block id
          int block_id;
          ierr = cTx.getMaterial(fe_ent,block_id); CHKERRG(ierr);
          // Get material block
          boost::shared_ptr<GenericMaterial>& block_data = cTx.dMatMap.at(block_id);
          // Get pressure
          auto t_h = getFTensor0FromVec(cTx.valuesAtGaussPts);
          // Coords at integration points
          auto t_coords = getFTensor1CoordsAtGaussPts();
          // Get base functions
          auto t_n_hdiv_row = row_data.getFTensor1N<3>();
          // Get integration weight
          auto t_w = getFTensor0IntegrationWeight();
          // Get volume
          const double vol = getVolume();
          int nb_gauss_pts = row_data.getN().size1();
          for(int gg = 0;gg!=nb_gauss_pts;gg++) {
            block_data->h = t_h;
            block_data->x = t_coords(0);
            block_data->y = t_coords(1);
            block_data->z = t_coords(2);
            ierr = block_data->calK(); CHKERRG(ierr);
            const double K = block_data->K;
            // get integration weight and multiply by element volume
            const double alpha = t_w*vol;
            const double beta = alpha*(1/K);
            FTensor::Tensor0<double*> t_a(&*nN.data().begin());
            for(int kk = 0;kk!=nb_row;kk++) {
              auto t_n_hdiv_col = col_data.getFTensor1N<3>(gg,0);
              for(int ll = 0;ll!=nb_col;ll++) {
                t_a += beta*(t_n_hdiv_row(j)*t_n_hdiv_col(j));
                ++t_n_hdiv_col;
                ++t_a;
              }
              ++t_n_hdiv_row;
            }
            ++t_coords;
            ++t_h;
            ++t_w;
          }
          Mat a = getFEMethod()->ts_B;
          ierr = MatSetValues(
            a,
            nb_row,&*row_data.getIndices().begin(),
            nb_col,&*col_data.getIndices().begin(),
            &*nN.data().begin(),ADD_VALUES
          ); CHKERRG(ierr);
          // matrix is symmetric, assemble other part
          if(row_side != col_side || row_type != col_type) {
            transNN.resize(nb_col,nb_row);
            noalias(transNN) = trans(nN);
            ierr = MatSetValues(
              a,
              nb_col,&*col_data.getIndices().begin(),
              nb_row,&*row_data.getIndices().begin(),
              &*transNN.data().begin(),ADD_VALUES
            ); CHKERRG(ierr);
          }
        } catch (const std::exception& ex) {
          std::ostringstream ss;
          ss << "throw in method: " << ex.what() << std::endl;
          SETERRQ(PETSC_COMM_SELF,MOFEM_STD_EXCEPTION_THROW,ss.str().c_str());
        }

        MoFEMFunctionReturnHot(0);
      }

    };

    struct OpVU_L2L2: public MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator {

      UnsaturatedFlowElement &cTx;

      OpVU_L2L2(UnsaturatedFlowElement &ctx,const std::string value_name):
      MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator(
        value_name,value_name,UserDataOperator::OPROWCOL
      ),
      cTx(ctx) {
        sYmm = true;
      }

      MatrixDouble nN;

      /**
       * \brief Assemble matrix
       * @param  row_side local index of row entity on element
       * @param  col_side local index of col entity on element
       * @param  row_type type of row entity, f.e. MBVERTEX, MBEDGE, or MBTET
       * @param  col_type type of col entity, f.e. MBVERTEX, MBEDGE, or MBTET
       * @param  row_data data for row
       * @param  col_data data for col
       * @return          error code
       */
      MoFEMErrorCode doWork(
        int row_side,int col_side,
        EntityType row_type,EntityType col_type,
        DataForcesAndSourcesCore::EntData &row_data,
        DataForcesAndSourcesCore::EntData &col_data
      ) {
        MoFEMFunctionBeginHot;
        try {
          int nb_row = row_data.getIndices().size();
          int nb_col = col_data.getIndices().size();
          if(nb_row==0) MoFEMFunctionReturnHot(0);
          if(nb_col==0) MoFEMFunctionReturnHot(0);
          nN.resize(nb_row,nb_col,false);
          nN.clear();
          // Get EntityHandle of the finite element
          EntityHandle fe_ent = getNumeredEntFiniteElementPtr()->getEnt();
          // Get material block id
          int block_id;
          ierr = cTx.getMaterial(fe_ent,block_id); CHKERRG(ierr);
          // Get material block
          boost::shared_ptr<GenericMaterial>& block_data = cTx.dMatMap.at(block_id);
          // Get pressure
          auto t_h = getFTensor0FromVec(cTx.valuesAtGaussPts);
          // // Get pressure
          // auto t_flux_residual = getFTensor0FromVec(*cTx.resAtGaussPts);
          // Get pressure rate
          auto t_h_t = getFTensor0FromVec(*cTx.headRateAtGaussPts);
          // Get integration weight
          auto t_w = getFTensor0IntegrationWeight();
          // Coords at integration points
          auto t_coords = getFTensor1CoordsAtGaussPts();
          // Scale eq.
          const double scale = block_data->sCale;
          // Time step factor
          double ts_a = getFEMethod()->ts_a;
          // get volume
          const double vol = getVolume();
          int nb_gauss_pts = row_data.getN().size1();
          // get base functions on rows
          auto t_n_row = row_data.getFTensor0N();
          for(int gg = 0;gg!=nb_gauss_pts;gg++) {
            // get integration weight and multiply by element volume
            double alpha = t_w*vol*scale;
            // evaluate material model at integration points
            // to calculate capacity and tangent of capacity term
            block_data->h = t_h;
            block_data->h_t = t_h_t;
            block_data->x = t_coords(0);
            block_data->y = t_coords(1);
            block_data->z = t_coords(2);
            ierr = block_data->calC(); CHKERRG(ierr);
            ierr = block_data->calDiffC(); CHKERRG(ierr);
            const double C = block_data->C;
            const double diffC = block_data->diffC;
            // assemble local entity tangent matrix
            FTensor::Tensor0<FTensor::PackPtr<double*,1> > t_a(&*nN.data().begin());
            // iterate base functions on rows
            for(int kk = 0;kk!=nb_row;kk++) {
              // get first base function on column at integration point gg
              auto t_n_col = col_data.getFTensor0N(gg,0);
              // iterate base functions on columns
              for(int ll = 0;ll!=nb_col;ll++) {
                // assemble elements of local matrix
                t_a += (alpha*(C*ts_a+diffC*t_h_t))*t_n_row*t_n_col;
                ++t_n_col;  // move to next base function on column
                ++t_a; // move to next element in local tangent matrix
              }
              ++t_n_row; // move to next base function on row
            }
            ++t_w;      // move to next integration weight
            ++t_coords; // move to next coordinate at integration point
            ++t_h;      // move to next capillary head at integration point
            // ++t_flux_residual;
            ++t_h_t;    // move to next capillary head rate at integration point
          }
          Mat a = getFEMethod()->ts_B;
          ierr = MatSetValues(
            a,
            nb_row,&row_data.getIndices()[0],
            nb_col,&col_data.getIndices()[0],
            &*nN.data().begin(),ADD_VALUES
          ); CHKERRG(ierr);
        } catch (const std::exception& ex) {
          std::ostringstream ss;
          ss << "throw in method: " << ex.what() << std::endl;
          SETERRQ(PETSC_COMM_SELF,MOFEM_STD_EXCEPTION_THROW,ss.str().c_str());
        }

        MoFEMFunctionReturnHot(0);
      }

    };

    struct OpVDivSigma_L2Hdiv: public MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator {

      UnsaturatedFlowElement &cTx;

      /**
       * \brief Constructor
       */
      OpVDivSigma_L2Hdiv(
        UnsaturatedFlowElement &ctx,
        const std::string& val_name_row,
        const std::string& flux_name_col
      ):
      MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator(
        val_name_row,flux_name_col,UserDataOperator::OPROWCOL,false
      ),
      cTx(ctx) {
      }

      MatrixDouble nN;
      VectorDouble divVec;

      /**
       * \brief Do calculations
       * @param  row_side local index of entity on row
       * @param  col_side local index of entity on column
       * @param  row_type type of row entity
       * @param  col_type type of col entity
       * @param  row_data row data structure carrying information about base functions, DOFs indices, etc.
       * @param  col_data column data structure carrying information about base functions, DOFs indices, etc.
       * @return          error code
       */
      MoFEMErrorCode doWork(
        int row_side,int col_side,
        EntityType row_type,EntityType col_type,
        DataForcesAndSourcesCore::EntData &row_data,
        DataForcesAndSourcesCore::EntData &col_data
      ) {
        MoFEMFunctionBeginHot;
        try {
          int nb_row = row_data.getFieldData().size();
          int nb_col = col_data.getFieldData().size();
          if(nb_row==0) MoFEMFunctionReturnHot(0);
          if(nb_col==0) MoFEMFunctionReturnHot(0);
          // Get EntityHandle of the finite element
          EntityHandle fe_ent = getNumeredEntFiniteElementPtr()->getEnt();
          // Get material block id
          int block_id;
          ierr = cTx.getMaterial(fe_ent,block_id); CHKERRG(ierr);
          // Get material block
          boost::shared_ptr<GenericMaterial>& block_data = cTx.dMatMap.at(block_id);
          nN.resize(nb_row,nb_col,false);
          divVec.resize(nb_col,false);
          nN.clear();
          // Scale eq.
          const double scale = block_data->sCale;
          int nb_gauss_pts = row_data.getN().size1();
          for(int gg = 0;gg<nb_gauss_pts;gg++) {
            double alpha = getGaussPts()(3,gg)*getVolume()*scale;
            ierr = getDivergenceOfHDivBaseFunctions(
              col_side,col_type,col_data,gg,divVec
            ); CHKERRG(ierr);
            noalias(nN) += alpha*outer_prod(row_data.getN(gg),divVec);
          }
          ierr = MatSetValues(
            getFEMethod()->ts_B,
            nb_row,&row_data.getIndices()[0],
            nb_col,&col_data.getIndices()[0],
            &nN(0,0),ADD_VALUES
          ); CHKERRG(ierr);
        } catch (const std::exception& ex) {
          std::ostringstream ss;
          ss << "throw in method: " << ex.what() << std::endl;
          SETERRQ(PETSC_COMM_SELF,MOFEM_STD_EXCEPTION_THROW,ss.str().c_str());
        }
        MoFEMFunctionReturnHot(0);
      }

    };

    struct OpDivTauU_HdivL2: public MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator {

      UnsaturatedFlowElement &cTx;

      /**
       * \brief Constructor
       */
      OpDivTauU_HdivL2(
        UnsaturatedFlowElement &ctx,
        const std::string& flux_name_col,
        const std::string& val_name_row
      ):
      MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator(
        flux_name_col,val_name_row,UserDataOperator::OPROWCOL,false
      ),
      cTx(ctx) {
      }

      MatrixDouble nN;
      VectorDouble divVec;
      FTensor::Index<'i',3> i;

      /**
       * \brief Do calculations
       * @param  row_side local index of entity on row
       * @param  col_side local index of entity on column
       * @param  row_type type of row entity
       * @param  col_type type of col entity
       * @param  row_data row data structure carrying information about base functions, DOFs indices, etc.
       * @param  col_data column data structure carrying information about base functions, DOFs indices, etc.
       * @return          error code
       */
      MoFEMErrorCode doWork(
        int row_side,int col_side,
        EntityType row_type,EntityType col_type,
        DataForcesAndSourcesCore::EntData &row_data,
        DataForcesAndSourcesCore::EntData &col_data
      ) {
        MoFEMFunctionBeginHot;
        try {
          int nb_row = row_data.getFieldData().size();
          int nb_col = col_data.getFieldData().size();
          if(nb_row==0) MoFEMFunctionReturnHot(0);
          if(nb_col==0) MoFEMFunctionReturnHot(0);
          nN.resize(nb_row,nb_col,false);
          divVec.resize(nb_row,false);
          nN.clear();
          // Get EntityHandle of the finite element
          EntityHandle fe_ent = getNumeredEntFiniteElementPtr()->getEnt();
          // Get material block id
          int block_id;
          ierr = cTx.getMaterial(fe_ent,block_id); CHKERRG(ierr);
          // Get material block
          boost::shared_ptr<GenericMaterial>& block_data = cTx.dMatMap.at(block_id);
          // Get pressure
          auto t_h = getFTensor0FromVec(cTx.valuesAtGaussPts);
          // Get flux
          auto t_flux = getFTensor1FromMat<3>(cTx.fluxesAtGaussPts);
          // Coords at integration points
          auto t_coords = getFTensor1CoordsAtGaussPts();
          // Get integration weight
          auto t_w = getFTensor0IntegrationWeight();
          // Get base function
          auto t_n_hdiv_row = row_data.getFTensor1N<3>();
          // Get volume
          double vol = getVolume();
          int nb_gauss_pts = row_data.getN().size1();
          for(int gg = 0;gg!=nb_gauss_pts;gg++) {
            block_data->h = t_h;
            block_data->x = t_coords(0);
            block_data->y = t_coords(1);
            block_data->z = t_coords(2);
            ierr = block_data->calK(); CHKERRG(ierr);
            ierr = block_data->calDiffK(); CHKERRG(ierr);
            const double K = block_data->K;
            // const double z = t_coords(2);
            const double KK = K*K;
            const double diffK = block_data->diffK;
            double alpha = t_w*vol;
            ierr = getDivergenceOfHDivBaseFunctions(
              row_side,row_type,row_data,gg,divVec
            ); CHKERRG(ierr);
            noalias(nN) -= alpha*outer_prod(divVec,col_data.getN(gg));
            FTensor::Tensor0<double*> t_a(&*nN.data().begin());
            for(int rr = 0;rr!=nb_row;rr++) {
              double beta = alpha*(-diffK/KK)*(t_n_hdiv_row(i)*t_flux(i));
              auto t_n_col = col_data.getFTensor0N(gg,0);
              for(int cc =0;cc!=nb_col;cc++) {
                t_a += beta*t_n_col;
                ++t_n_col;
                ++t_a;
              }
              ++t_n_hdiv_row;
            }
            ++t_w;
            ++t_coords;
            ++t_h;
            ++t_flux;
          }
          ierr = MatSetValues(
            getFEMethod()->ts_B,
            nb_row,&row_data.getIndices()[0],
            nb_col,&col_data.getIndices()[0],
            &nN(0,0),ADD_VALUES
          ); CHKERRG(ierr);
        } catch (const std::exception& ex) {
          std::ostringstream ss;
          ss << "throw in method: " << ex.what() << std::endl;
          SETERRQ(PETSC_COMM_SELF,MOFEM_STD_EXCEPTION_THROW,ss.str().c_str());
        }
        MoFEMFunctionReturnHot(0);
      }

    };

    struct OpEvaluateInitiallHead: public MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator {
      UnsaturatedFlowElement &cTx;
      OpEvaluateInitiallHead(UnsaturatedFlowElement &ctx,const std::string& val_name):
      MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator(val_name,UserDataOperator::OPROW),
      cTx(ctx) {
      }

      MatrixDouble nN;
      VectorDouble nF;

      MoFEMErrorCode doWork(int side,EntityType type,DataForcesAndSourcesCore::EntData &data) {
        MoFEMFunctionBeginHot;
        try {
          if(data.getFieldData().size()==0) MoFEMFunctionReturnHot(0);
          int nb_dofs = data.getFieldData().size();
          int nb_gauss_pts = data.getN().size1();
          if(nb_dofs!=static_cast<int>(data.getN().size2())) {
            SETERRQ(PETSC_COMM_WORLD,MOFEM_DATA_INCONSISTENCY,"wrong number of dofs");
          }
          nN.resize(nb_dofs,nb_dofs,false);
          nF.resize(nb_dofs,false);
          nN.clear();
          nF.clear();

          // Get EntityHandle of the finite element
          EntityHandle fe_ent = getFEEntityHandle();
          // Get material block id
          int block_id;
          ierr = cTx.getMaterial(fe_ent,block_id); CHKERRG(ierr);
          // Get material block
          boost::shared_ptr<GenericMaterial>& block_data = cTx.dMatMap.at(block_id);

          // loop over integration points
          for(int gg = 0;gg<nb_gauss_pts;gg++) {
            // get coordinates at integration point
            block_data->x = getCoordsAtGaussPts()(gg,0);
            block_data->y = getCoordsAtGaussPts()(gg,1);
            block_data->z = getCoordsAtGaussPts()(gg,2);
            // get weight for integration rule
            double alpha = getGaussPts()(2,gg)*getVolume();
            nN += alpha*outer_prod(data.getN(gg),data.getN(gg));
            nF += alpha*block_data->initalPcEval()*data.getN(gg);
          }

          // factor matrix
          cholesky_decompose(nN);
          // solve local problem
          cholesky_solve(nN,nF,ublas::lower());

          // set solution to vector
          ierr = VecSetValues(
            cTx.D1,nb_dofs,&*data.getIndices().begin(),
            &*nF.begin(),INSERT_VALUES
          ); CHKERRG(ierr);

        } catch (const std::exception& ex) {
          std::ostringstream ss;
          ss << "throw in method: " << ex.what() << std::endl;
          SETERRQ(PETSC_COMM_SELF,1,ss.str().c_str());
        }

        MoFEMFunctionReturnHot(0);
      }

    };

    struct OpIntegrateFluxes: public MoFEM::FaceElementForcesAndSourcesCore::UserDataOperator {
      UnsaturatedFlowElement &cTx;

      /**
       * \brief Constructor
       */
      OpIntegrateFluxes(
        UnsaturatedFlowElement &ctx,const std::string fluxes_name
      ):
      MoFEM::FaceElementForcesAndSourcesCore::UserDataOperator(fluxes_name,UserDataOperator::OPROW),
      cTx(ctx) {
      }

      FTensor::Index<'i',3> i;

      /**
       * \brief Integrate boundary condition
       * @param  side local index of entity
       * @param  type type of entity
       * @param  data data on entity
       * @return      error code
       */
      MoFEMErrorCode doWork(
        int side,EntityType type,DataForcesAndSourcesCore::EntData &data
      ) {
        MoFEMFunctionBeginHot;
        try {
          int nb_dofs = data.getFieldData().size();
          if(nb_dofs==0) MoFEMFunctionReturnHot(0);
          // Get base function
          auto t_n_hdiv = data.getFTensor1N<3>();
          // get normal of face
          auto t_normal = getFTensor1Normal();
          // Integration weight
          auto t_w = getFTensor0IntegrationWeight();
          double flux_on_entity = 0;
          int nb_gauss_pts = data.getN().size1();
          for(int gg = 0;gg<nb_gauss_pts;gg++) {
            auto t_data = data.getFTensor0FieldData();
            for(int rr = 0;rr!=nb_dofs;rr++) {
              flux_on_entity -= (0.5*t_data*t_w)*(t_n_hdiv(i)*t_normal(i));
              ++t_n_hdiv;
              ++t_data;
            }
            ++t_w;
          }
          ierr = VecSetValue(cTx.ghostFlux,0,flux_on_entity,ADD_VALUES); CHKERRG(ierr);
        } catch (const std::exception& ex) {
          std::ostringstream ss;
          ss << "throw in method: " << ex.what() << std::endl;
          SETERRQ(PETSC_COMM_SELF,MOFEM_STD_EXCEPTION_THROW,ss.str().c_str());
        }
        MoFEMFunctionReturnHot(0);
      }

    };

    /**
     * Operator used to post-process results for unsaturated infiltration problem.
     * Operator should with element for post-processing results, i.e. PostProcVolumeOnRefinedMesh
     */
    struct OpPostProcMaterial: public MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator {

      UnsaturatedFlowElement &cTx;
      moab::Interface &postProcMesh;
      std::vector<EntityHandle> &mapGaussPts;

      OpPostProcMaterial(
        UnsaturatedFlowElement &ctx,
        moab::Interface &post_proc_mesh,
        std::vector<EntityHandle> &map_gauss_pts,
        const std::string field_name
      ):
      MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator(
        field_name,ForcesAndSourcesCore::UserDataOperator::OPROW
      ),
      cTx(ctx),
      postProcMesh(post_proc_mesh),
      mapGaussPts(map_gauss_pts) {
      }

      MoFEMErrorCode doWork(
        int side,
        EntityType type,
        DataForcesAndSourcesCore::EntData &data
      ) {
        MoFEMFunctionBeginHot;
        int nb_dofs = data.getFieldData().size();
        if(nb_dofs==0) MoFEMFunctionReturnHot(0);

        // if(type != MBTET) MoFEMFunctionReturnHot(0);
        // Get EntityHandle of the finite element
        EntityHandle fe_ent = getNumeredEntFiniteElementPtr()->getEnt();
        // Get material block id
        int block_id;
        ierr = cTx.getMaterial(fe_ent,block_id); CHKERRG(ierr);
        // Get material block
        boost::shared_ptr<GenericMaterial>& block_data = cTx.dMatMap.at(block_id);

        // Set bloc Id
        Tag th_id;
        int def_block_id = -1;
        rval = postProcMesh.tag_get_handle(
          "BLOCK_ID",1,MB_TYPE_INTEGER,th_id,
          MB_TAG_CREAT|MB_TAG_SPARSE,&def_block_id
        ); CHKERRG(rval);
        rval = postProcMesh.tag_set_data(th_id,&fe_ent,1,&block_id);  CHKERRG(rval);

        // Create mesh tag. Tags are created on post-processing mesh and
        // visable in post-processor, e.g. Paraview
        double zero = 0;
        Tag th_theta;
        rval = postProcMesh.tag_get_handle(
          "THETA",1,MB_TYPE_DOUBLE,th_theta,
          MB_TAG_CREAT|MB_TAG_SPARSE,&zero
        ); CHKERRG(rval);
        Tag th_se;
        rval = postProcMesh.tag_get_handle(
          "Se",1,MB_TYPE_DOUBLE,th_se,
          MB_TAG_CREAT|MB_TAG_SPARSE,&zero
        ); CHKERRG(rval);
        // Tag th_ks;
        // rval = postProcMesh.tag_get_handle(
        //   "Ks",1,MB_TYPE_DOUBLE,th_ks,
        //   MB_TAG_CREAT|MB_TAG_SPARSE,&zero
        // ); CHKERRG(rval);
        // rval = postProcMesh.tag_set_data(th_ks,&fe_ent,1,&block_data->Ks); CHKERRG(rval);
        Tag th_k;
        rval = postProcMesh.tag_get_handle(
          "K",1,MB_TYPE_DOUBLE,th_k,
          MB_TAG_CREAT|MB_TAG_SPARSE,&zero
        ); CHKERRG(rval);
        Tag th_c;
        rval = postProcMesh.tag_get_handle(
          "C",1,MB_TYPE_DOUBLE,th_c,
          MB_TAG_CREAT|MB_TAG_SPARSE,&zero
        ); CHKERRG(rval);


        // Get pressure at integration points
        auto t_h = getFTensor0FromVec(cTx.valuesAtGaussPts);
        // Coords at integration points
        auto t_coords = getFTensor1CoordsAtGaussPts();

        int nb_gauss_pts = data.getN().size1();
        for(int gg = 0;gg<nb_gauss_pts;gg++) {
          block_data->h = t_h;
          block_data->x = t_coords(0);
          block_data->y = t_coords(1);
          block_data->z = t_coords(2);
          // Calculate theta (water content) and save it on mesh tags
          ierr = block_data->calTheta(); CHKERRG(ierr);
          double theta = block_data->tHeta;
          rval = postProcMesh.tag_set_data(th_theta,&mapGaussPts[gg],1,&theta); CHKERRG(rval);
          ierr = block_data->calSe(); CHKERRG(ierr);
          // Calculate Se (effective saturation and save it on the mesh tags)
          double Se = block_data->Se;
          rval = postProcMesh.tag_set_data(th_se,&mapGaussPts[gg],1,&Se); CHKERRG(rval);
          // Calculate K (hydraulic conductivity) and save it on the mesh tags
          ierr = block_data->calK(); CHKERRG(ierr);
          double K = block_data->K;
          rval = postProcMesh.tag_set_data(th_k,&mapGaussPts[gg],1,&K); CHKERRG(rval);
          // Calculate water capacity and save it on the mesh tags
          ierr = block_data->calC(); CHKERRG(ierr);
          double C = block_data->C;
          rval = postProcMesh.tag_set_data(th_c,&mapGaussPts[gg],1,&C); CHKERRG(rval);
          ++t_h;
          ++t_coords;
        }

        MoFEMFunctionReturnHot(0);
      }

    };

    /**
     * Finite element implementation called by TS monitor. Element calls other
     * finite elements to evaluate material properties and save results on the mesh.
     *
     * \note Element overloaded only FEMethod::postProcess methos where other elements
     * are called.
     */
    struct MonitorPostProc: public MoFEM::FEMethod {

        UnsaturatedFlowElement &cTx;
        boost::shared_ptr<PostProcVolumeOnRefinedMesh> postProc;
        boost::shared_ptr<ForcesAndSourcesCore> fluxIntegrate;

        const int fRequency;

        MonitorPostProc(
          UnsaturatedFlowElement &ctx,
          boost::shared_ptr<PostProcVolumeOnRefinedMesh>& post_proc,
          boost::shared_ptr<ForcesAndSourcesCore> flux_Integrate,
          const int frequency
        ):
        cTx(ctx),
        postProc(post_proc),
        fluxIntegrate(flux_Integrate),
        fRequency(frequency) {
        }

        MoFEMErrorCode preProcess() {
          MoFEMFunctionBeginHot;
          MoFEMFunctionReturnHot(0);
        }

        MoFEMErrorCode operator()() {
          MoFEMFunctionBeginHot;
          MoFEMFunctionReturnHot(0);
        }

        MoFEMErrorCode postProcess() {
          MoFEMFunctionBeginHot;

          // Get time step
          int step;
          ierr = TSGetTimeStepNumber(ts,&step); CHKERRG(ierr);

          if((step)%fRequency==0) {
            // Post-process results and save in the file
            PetscPrintf(PETSC_COMM_WORLD,"Output results %d - %d\n",step,fRequency);
            ierr = DMoFEMLoopFiniteElements(cTx.dM,"MIX",postProc); CHKERRG(ierr);
            ierr = postProc->writeFile(
              string("out_")+boost::lexical_cast<std::string>(step)+".h5m"
            ); CHKERRG(ierr);

          }

          // Integrate fluxes on faces where pressure head is applied
          ierr = VecZeroEntries(cTx.ghostFlux); CHKERRG(ierr);
          ierr = VecGhostUpdateBegin(cTx.ghostFlux,INSERT_VALUES,SCATTER_FORWARD); CHKERRG(ierr);
          ierr = VecGhostUpdateEnd(cTx.ghostFlux,INSERT_VALUES,SCATTER_FORWARD); CHKERRG(ierr);
          // Run finite element to integrate fluxes
          ierr = DMoFEMLoopFiniteElements(cTx.dM,"MIX_BCVALUE",fluxIntegrate); CHKERRG(ierr);
          ierr = VecAssemblyBegin(cTx.ghostFlux); CHKERRG(ierr);
          ierr = VecAssemblyEnd(cTx.ghostFlux); CHKERRG(ierr);
          // accumulate errors from processors
          ierr = VecGhostUpdateBegin(cTx.ghostFlux,ADD_VALUES,SCATTER_REVERSE); CHKERRG(ierr);
          ierr = VecGhostUpdateEnd(cTx.ghostFlux,ADD_VALUES,SCATTER_REVERSE); CHKERRG(ierr);
          // scatter errors to all processors
          ierr = VecGhostUpdateBegin(cTx.ghostFlux,INSERT_VALUES,SCATTER_FORWARD); CHKERRG(ierr);
          ierr = VecGhostUpdateEnd(cTx.ghostFlux,INSERT_VALUES,SCATTER_FORWARD); CHKERRG(ierr);
          double *ghost_flux;
          ierr = VecGetArray(cTx.ghostFlux,&ghost_flux); CHKERRG(ierr);
          ierr = PetscPrintf(
            PETSC_COMM_WORLD,"Flux at time %6.4g %6.4g\n",ts_t,ghost_flux[0]
          ); CHKERRG(ierr);
          ierr = VecRestoreArray(cTx.ghostFlux,&ghost_flux); CHKERRG(ierr);

          MoFEMFunctionReturnHot(0);
        }

      };


    /// \brief add fields
    MoFEMErrorCode addFields(const std::string &values,const std::string &fluxes,const int order) {
      MoFEMFunctionBeginHot;
      //Fields
      ierr = mField.add_field(fluxes,HDIV,DEMKOWICZ_JACOBI_BASE,1); CHKERRG(ierr);
      ierr = mField.add_field(values,L2,AINSWORTH_LEGENDRE_BASE,1); CHKERRG(ierr);
      ierr = mField.add_field(values+"_t",L2,AINSWORTH_LEGENDRE_BASE,1); CHKERRG(ierr);
      // ierr = mField.add_field(fluxes+"_residual",L2,AINSWORTH_LEGENDRE_BASE,1); CHKERRG(ierr);

      //meshset consisting all entities in mesh
      EntityHandle root_set = mField.get_moab().get_root_set();
      //add entities to field

      for(_IT_CUBITMESHSETS_BY_SET_TYPE_FOR_LOOP_(mField,BLOCKSET,it)) {
        if(it->getName().compare(0,4,"SOIL")!=0) continue;
        ierr = mField.add_ents_to_field_by_type(
          dMatMap[it->getMeshsetId()]->tEts,MBTET,fluxes
        ); CHKERRG(ierr);
        ierr = mField.add_ents_to_field_by_type(
          dMatMap[it->getMeshsetId()]->tEts,MBTET,values
        ); CHKERRG(ierr);
        ierr = mField.add_ents_to_field_by_type(
          dMatMap[it->getMeshsetId()]->tEts,MBTET,values+"_t"
        ); CHKERRG(ierr);
        // ierr = mField.add_ents_to_field_by_type(
        //   dMatMap[it->getMeshsetId()]->tEts,MBTET,fluxes+"_residual"
        // ); CHKERRG(ierr);
      }

      ierr = mField.set_field_order(root_set,MBTET,fluxes,order+1); CHKERRG(ierr);
      ierr = mField.set_field_order(root_set,MBTRI,fluxes,order+1); CHKERRG(ierr);
      ierr = mField.set_field_order(root_set,MBTET,values,order); CHKERRG(ierr);
      ierr = mField.set_field_order(root_set,MBTET,values+"_t",order); CHKERRG(ierr);
      // ierr = mField.set_field_order(root_set,MBTET,fluxes+"_residual",order); CHKERRG(ierr);
      MoFEMFunctionReturnHot(0);
    }

    /// \brief add finite elements
    MoFEMErrorCode addFiniteElements(
      const std::string &fluxes_name,const std::string &values_name
    ) {
      MoFEMFunctionBeginHot;

      // Define element "MIX". Note that this element will work with fluxes_name and
      // values_name. This reflect bilinear form for the problem
      ierr = mField.add_finite_element("MIX",MF_ZERO); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_row("MIX",fluxes_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_col("MIX",fluxes_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_row("MIX",values_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_col("MIX",values_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_data("MIX",fluxes_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_data("MIX",values_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_data("MIX",values_name+"_t"); CHKERRG(ierr);
      // ierr = mField.modify_finite_element_add_field_data("MIX",fluxes_name+"_residual"); CHKERRG(ierr);

      for(_IT_CUBITMESHSETS_BY_SET_TYPE_FOR_LOOP_(mField,BLOCKSET,it)) {
        if(it->getName().compare(0,4,"SOIL")!=0) continue;
        ierr = mField.add_ents_to_finite_element_by_type(
          dMatMap[it->getMeshsetId()]->tEts,MBTET,"MIX"
        ); CHKERRG(ierr);
      }

      // Define element to integrate natural boundary conditions, i.e. set values.
      ierr = mField.add_finite_element("MIX_BCVALUE",MF_ZERO); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_row("MIX_BCVALUE",fluxes_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_col("MIX_BCVALUE",fluxes_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_data("MIX_BCVALUE",fluxes_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_data("MIX_BCVALUE",values_name); CHKERRG(ierr);

      for(_IT_CUBITMESHSETS_BY_SET_TYPE_FOR_LOOP_(mField,BLOCKSET,it)) {
        if(it->getName().compare(0,4,"HEAD")!=0) continue;
        ierr = mField.add_ents_to_finite_element_by_type(
          bcValueMap[it->getMeshsetId()]->eNts,MBTRI,"MIX_BCVALUE"
        ); CHKERRG(ierr);
      }

      // Define element to apply essential boundary conditions.
      ierr = mField.add_finite_element("MIX_BCFLUX",MF_ZERO); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_row("MIX_BCFLUX",fluxes_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_col("MIX_BCFLUX",fluxes_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_data("MIX_BCFLUX",fluxes_name); CHKERRG(ierr);
      ierr = mField.modify_finite_element_add_field_data("MIX_BCFLUX",values_name); CHKERRG(ierr);

      for(_IT_CUBITMESHSETS_BY_SET_TYPE_FOR_LOOP_(mField,BLOCKSET,it)) {
        if(it->getName().compare(0,4,"FLUX")!=0) continue;
        ierr = mField.add_ents_to_finite_element_by_type(
          bcFluxMap[it->getMeshsetId()]->eNts,MBTRI,"MIX_BCFLUX"
        ); CHKERRG(ierr);
      }

      MoFEMFunctionReturnHot(0);
    }

    /**
     * \brief Build problem
     * @param  ref_level mesh refinement on which mesh problem you like to built.
     * @return           error code
     */
    MoFEMErrorCode buildProblem(BitRefLevel ref_level = BitRefLevel().set(0)) {
      MoFEMFunctionBeginHot;

      // Build fields
      ierr = mField.build_fields(); CHKERRG(ierr);
      // Build finite elements
      ierr = mField.build_finite_elements("MIX"); CHKERRG(ierr);
      ierr = mField.build_finite_elements("MIX_BCFLUX"); CHKERRG(ierr);
      ierr = mField.build_finite_elements("MIX_BCVALUE"); CHKERRG(ierr);
      //Build adjacencies of degrees of freedom and elements
      ierr = mField.build_adjacencies(ref_level); CHKERRG(ierr);

      //  create DM instance
      ierr = DMCreate(PETSC_COMM_WORLD,&dM);CHKERRG(ierr);
      // setting that DM is type of DMMOFEM, i.e. MOFEM implementation manages DM
      ierr = DMSetType(dM,"DMMOFEM");CHKERRG(ierr);
      // mesh is portioned, each process keeps only part of problem
      ierr = DMMoFEMSetIsPartitioned(dM,PETSC_TRUE);
      // creates problem in DM
      ierr = DMMoFEMCreateMoFEM(dM,&mField,"MIX",ref_level); CHKERRG(ierr);
      // discretised problem creates square matrix (that makes some optimizations)
      ierr = DMMoFEMSetIsPartitioned(dM,PETSC_TRUE); CHKERRG(ierr);
      // set DM options from command line
      ierr = DMSetFromOptions(dM); CHKERRG(ierr);
      // add finite elements
      ierr = DMMoFEMAddElement(dM,"MIX"); CHKERRG(ierr);
      ierr = DMMoFEMAddElement(dM,"MIX_BCFLUX"); CHKERRG(ierr);
      ierr = DMMoFEMAddElement(dM,"MIX_BCVALUE"); CHKERRG(ierr);
      // constructor data structures
      ierr = DMSetUp(dM); CHKERRG(ierr);

      PetscSection section;
      ierr = mField.getInterface<ISManager>()->sectionCreate("MIX",&section); CHKERRG(ierr);
      ierr = DMSetDefaultSection(dM,section); CHKERRG(ierr);
      ierr = DMSetDefaultGlobalSection(dM,section); CHKERRG(ierr);
      // ierr = PetscSectionView(section,PETSC_VIEWER_STDOUT_WORLD);
      ierr = PetscSectionDestroy(&section); CHKERRG(ierr);

      MoFEMFunctionReturnHot(0);
    }

    boost::shared_ptr<ForcesAndSourcesCore> feFaceBc;  ///< Elemnet to calculate essential bc
    boost::shared_ptr<ForcesAndSourcesCore> feFaceRhs; ///< Face element apply natural bc
    boost::shared_ptr<ForcesAndSourcesCore> feVolInitialPc;  ///< Calculate inital boundary conditions
    boost::shared_ptr<ForcesAndSourcesCore> feVolRhs;  ///< Assemble residual vector
    boost::shared_ptr<ForcesAndSourcesCore> feVolLhs;  ///< Assemble tangent matrix
    boost::shared_ptr<MethodForForceScaling> scaleMethodFlux; ///< Method scaling fluxes
    boost::shared_ptr<MethodForForceScaling> scaleMethodValue; ///< Method scaling values
    boost::shared_ptr<FEMethod> tsMonitor;  ///< Element used by TS monitor to postprocess results at time step

    boost::shared_ptr<VectorDouble> headRateAtGaussPts; ///< Vector keeps head rate
    // boost::shared_ptr<VectorDouble> resAtGaussPts;  ///< Residual field

    /**
    * \brief Set integration rule to volume elements
    *
    */
    struct VolRule {
      int operator()(int,int,int p_data) const {
        return 2*p_data+p_data;
      }
    };
    /**
    * \brief Set integration rule to boundary elements
    *
    */
    struct FaceRule {
      int operator()(int p_row,int p_col,int p_data) const {
        return 2*p_data;
      }
    };

    std::vector<int> bcVecIds;
    VectorDouble bcVecVals,vecValsOnBc;

    /**
     * \brief Pre-peprocessing
     * Set head pressute rate and get inital essential boundary conditions
     */
    struct preProcessVol {
      UnsaturatedFlowElement& cTx;
      boost::shared_ptr<ForcesAndSourcesCore> fePtr;
      //std::string mArk;

      preProcessVol(
        UnsaturatedFlowElement& ctx,
        boost::shared_ptr<ForcesAndSourcesCore>& fe_ptr/*,std::string mark*/
      ):
      cTx(ctx),
      fePtr(fe_ptr)/*,mArk(mark)*/ {
      }
      MoFEMErrorCode operator()() {
        MoFEMFunctionBeginHot;
        // Update pressure rates
        ierr = fePtr->mField.getInterface<VecManager>()->setOtherLocalGhostVector(
          fePtr->problemPtr,"VALUES",string("VALUES")+"_t",
          ROW,fePtr->ts_u_t,INSERT_VALUES,SCATTER_REVERSE
        ); CHKERRG(ierr);
        switch (fePtr->ts_ctx) {
          case TSMethod::CTX_TSSETIFUNCTION:
          if(!cTx.bcIndices.empty()) {
            double scale;
            ierr = cTx.scaleMethodFlux->getForceScale(fePtr->ts_t,scale); CHKERRG(ierr);
            if(cTx.bcVecIds.size()!=cTx.bcIndices.size()) {
              cTx.bcVecIds.insert(cTx.bcVecIds.begin(),cTx.bcIndices.begin(),cTx.bcIndices.end());
              cTx.bcVecVals.resize(cTx.bcVecIds.size(),false);
              cTx.vecValsOnBc.resize(cTx.bcVecIds.size(),false);
            }
            ierr = VecGetValues(
              cTx.D0,cTx.bcVecIds.size(),&*cTx.bcVecIds.begin(),&*cTx.bcVecVals.begin()
            ); CHKERRG(ierr);
            ierr = VecGetValues(
              fePtr->ts_u,cTx.bcVecIds.size(),&*cTx.bcVecIds.begin(),&*cTx.vecValsOnBc.begin()
            ); CHKERRG(ierr);
            cTx.bcVecVals *= scale;
            // cerr << mArk << endl;
            // cerr << "v " << cTx.vecValsOnBc << endl;
            // cerr << "v " << cTx.bcVecVals << endl;
            VectorDouble::iterator vit = cTx.bcVecVals.begin();
            const NumeredDofEntity *dof_ptr;
            for(
              std::vector<int>::iterator it = cTx.bcVecIds.begin();
              it!=cTx.bcVecIds.end();it++,vit++
            ) {
              ierr = fePtr->problemPtr->getColDofsByPetscGlobalDofIdx(*it,&dof_ptr); CHKERRG(ierr);
              dof_ptr->getFieldData() = *vit;
            }
          } else {
            cTx.bcVecIds.resize(0);
            cTx.bcVecVals.resize(0);
            cTx.vecValsOnBc.resize(0);
          }
          break;
          default:
          // don nothing
          break;
        }
        MoFEMFunctionReturnHot(0);
      }
    };

    /**
     * \brief Post proces method for volume element
     * Assemble vectors and matrices and apply essential boundary conditions
     */
    struct postProcessVol {
      UnsaturatedFlowElement& cTx;
      boost::shared_ptr<ForcesAndSourcesCore> fePtr;
      // std::string mArk;
      postProcessVol(
        UnsaturatedFlowElement& ctx,
        boost::shared_ptr<ForcesAndSourcesCore>& fe_ptr//,std::string mark
      ):
      cTx(ctx),
      fePtr(fe_ptr)/*,mArk(mark)*/ {
      }
      MoFEMErrorCode operator()() {
        MoFEMFunctionBeginHot;
        switch (fePtr->ts_ctx) {
          case TSMethod::CTX_TSSETIJACOBIAN: {
            ierr = MatAssemblyBegin(fePtr->ts_B,MAT_FINAL_ASSEMBLY); CHKERRG(ierr);
            ierr = MatAssemblyEnd(fePtr->ts_B,MAT_FINAL_ASSEMBLY); CHKERRG(ierr);
            // MatView(fePtr->ts_B,PETSC_VIEWER_DRAW_WORLD);
            // std::string wait;
            // std::cin >> wait;
            ierr = MatZeroRowsColumns(
              fePtr->ts_B,cTx.bcVecIds.size(),&*cTx.bcVecIds.begin(),1,PETSC_NULL,PETSC_NULL
            ); CHKERRG(ierr);
            ierr = MatAssemblyBegin(fePtr->ts_B,MAT_FINAL_ASSEMBLY); CHKERRG(ierr);
            ierr = MatAssemblyEnd(fePtr->ts_B,MAT_FINAL_ASSEMBLY); CHKERRG(ierr);
            // MatView(fePtr->ts_B,PETSC_VIEWER_DRAW_WORLD);
            // std::string wait;
            // std::cin >> wait;
          }
          break;
          case TSMethod::CTX_TSSETIFUNCTION: {
            ierr = VecAssemblyBegin(fePtr->ts_F); CHKERRG(ierr);
            ierr = VecAssemblyEnd(fePtr->ts_F); CHKERRG(ierr);
            if(!cTx.bcVecIds.empty()) {
              cTx.vecValsOnBc -= cTx.bcVecVals;
              // cerr << mArk << endl;
              // cerr << "a " << cTx.vecValsOnBc << endl;
              // cerr << "a " << cTx.bcVecVals << endl;
              ierr = VecSetValues(
                fePtr->ts_F,cTx.bcVecIds.size(),
                &*cTx.bcVecIds.begin(),
                &*cTx.vecValsOnBc.begin(),INSERT_VALUES
              ); CHKERRG(ierr);
            }
            ierr = VecAssemblyBegin(fePtr->ts_F); CHKERRG(ierr);
            ierr = VecAssemblyEnd(fePtr->ts_F); CHKERRG(ierr);
            // ierr = VecView(fePtr->ts_F,PETSC_VIEWER_STDOUT_WORLD); CHKERRG(ierr);
            // ierr = fePtr->mField.getInterface<VecManager>()->setOtherLocalGhostVector(
            //   fePtr->problemPtr,"VALUES",string("FLUXES")+"_residual",
            //   ROW,fePtr->ts_F,INSERT_VALUES,SCATTER_REVERSE
            // ); CHKERRG(ierr);
          }
          break;
          default:
          // don nothing
          break;
        }
        MoFEMFunctionReturnHot(0);
      }
    };

    /**
     * \brief Create finite element instances
     * @param  vol_rule integration rule for volume element
     * @param  face_rule integration rule for boundary element
     * @return error code
     */
    MoFEMErrorCode setFiniteElements(
      ForcesAndSourcesCore::RuleHookFun vol_rule = VolRule(),
      ForcesAndSourcesCore::RuleHookFun face_rule = FaceRule()
    ) {
      MoFEMFunctionBeginHot;

      // create finite element instances
      feFaceBc = boost::shared_ptr<ForcesAndSourcesCore>(new FaceElementForcesAndSourcesCore(mField));
      feFaceRhs = boost::shared_ptr<ForcesAndSourcesCore>(new FaceElementForcesAndSourcesCore(mField));
      feVolInitialPc = boost::shared_ptr<ForcesAndSourcesCore>(new VolumeElementForcesAndSourcesCore(mField));
      feVolLhs = boost::shared_ptr<ForcesAndSourcesCore>(new VolumeElementForcesAndSourcesCore(mField));
      feVolRhs = boost::shared_ptr<ForcesAndSourcesCore>(new VolumeElementForcesAndSourcesCore(mField));
      // set integration rule to elements
      feFaceBc->getRuleHook = face_rule;
      feFaceRhs->getRuleHook = face_rule;
      feVolInitialPc->getRuleHook = vol_rule;
      feVolLhs->getRuleHook = vol_rule;
      feVolRhs->getRuleHook = vol_rule;
      // set function hook for finite element postprocessing stage
      feVolRhs->preProcessHook = preProcessVol(*this,feVolRhs);
      feVolLhs->preProcessHook = preProcessVol(*this,feVolLhs);
      feVolRhs->postProcessHook = postProcessVol(*this,feVolRhs);
      feVolLhs->postProcessHook = postProcessVol(*this,feVolLhs);

      // create method for setting history for fluxes on boundary
      scaleMethodFlux= boost::shared_ptr<MethodForForceScaling>(
        new TimeForceScale("-flux_history",false)
      );

      // create method for setting history for presure heads on boundary
      scaleMethodValue = boost::shared_ptr<MethodForForceScaling>(
        new TimeForceScale("-head_history",false)
      );

      // Set operator to calculate essential boundary conditions
      feFaceBc->getOpPtrVector().push_back(new OpEvaluateBcOnFluxes(*this,"FLUXES"));

      // Set operator to calculate initial capillary pressure
      feVolInitialPc->getOpPtrVector().push_back(new OpEvaluateInitiallHead(*this,"VALUES"));

      // set residual face from Neumann terms, i.e. applied pressure
      feFaceRhs->getOpPtrVector().push_back(new OpRhsBcOnValues(*this,"FLUXES",scaleMethodValue));
      // set residual finite element operators
      headRateAtGaussPts = boost::make_shared<VectorDouble>();
      // resAtGaussPts = boost::make_shared<VectorDouble>();
      feVolRhs->getOpPtrVector().push_back(
        new OpCalculateScalarFieldValues(string("VALUES")+"_t",headRateAtGaussPts,MBTET)
      );
      feVolRhs->getOpPtrVector().push_back(new OpValuesAtGaussPts(*this,"VALUES"));
      feVolRhs->getOpPtrVector().push_back(new OpFluxDivergenceAtGaussPts(*this,"FLUXES"));
      feVolRhs->getOpPtrVector().push_back(new OpResidualFlux(*this,"FLUXES"));
      feVolRhs->getOpPtrVector().push_back(new OpResidualMass(*this,"VALUES"));
      feVolRhs->getOpPtrVector().back().opType = ForcesAndSourcesCore::UserDataOperator::OPROW;
      // set tangent matrix finite element operators
      feVolLhs->getOpPtrVector().push_back(
        new OpCalculateScalarFieldValues(string("VALUES")+"_t",headRateAtGaussPts,MBTET)
      );
      // feVolLhs->getOpPtrVector().push_back(
      //   new OpCalculateScalarFieldValues(string("FLUXES")+"_residual",resAtGaussPts,MBTET)
      // );
      feVolLhs->getOpPtrVector().push_back(new OpValuesAtGaussPts(*this,"VALUES"));
      feVolLhs->getOpPtrVector().push_back(new OpFluxDivergenceAtGaussPts(*this,"FLUXES"));
      feVolLhs->getOpPtrVector().push_back(new OpTauDotSigma_HdivHdiv(*this,"FLUXES"));
      feVolLhs->getOpPtrVector().push_back(new OpVU_L2L2(*this,"VALUES"));
      feVolLhs->getOpPtrVector().push_back(new OpVDivSigma_L2Hdiv(*this,"VALUES","FLUXES"));
      feVolLhs->getOpPtrVector().push_back(new OpDivTauU_HdivL2(*this,"FLUXES","VALUES"));

      // Adding finite elements to DM, time solver will ask for it to assemble
      // tangent matrices and residuals
      boost::shared_ptr<FEMethod> null;
      ierr = DMMoFEMTSSetIFunction(dM,"MIX_BCVALUE",feFaceRhs,null,null); CHKERRG(ierr);
      ierr = DMMoFEMTSSetIFunction(dM,"MIX",feVolRhs,null,null); CHKERRG(ierr);
      ierr = DMMoFEMTSSetIJacobian(dM,"MIX",feVolLhs,null,null); CHKERRG(ierr);

      // setting up post-processing
      boost::shared_ptr<PostProcVolumeOnRefinedMesh> post_process(new PostProcVolumeOnRefinedMesh(mField));
      ierr = post_process->generateReferenceElementMesh(); CHKERRG(ierr);
      ierr = post_process->addFieldValuesPostProc("VALUES"); CHKERRG(ierr);
      ierr = post_process->addFieldValuesPostProc("VALUES_t"); CHKERRG(ierr);
      // ierr = post_process->addFieldValuesPostProc("FLUXES_residual"); CHKERRG(ierr);
      ierr = post_process->addFieldValuesPostProc("FLUXES"); CHKERRG(ierr);
      post_process->getOpPtrVector().push_back(new OpValuesAtGaussPts(*this,"VALUES"));
      post_process->getOpPtrVector().push_back(
        new OpPostProcMaterial(
          *this,post_process->postProcMesh,post_process->mapGaussPts,"VALUES"
        )
      );

      // Integrate fluxes on boundary
      boost::shared_ptr<ForcesAndSourcesCore> flux_integrate;
      flux_integrate = boost::shared_ptr<ForcesAndSourcesCore>(
        new FaceElementForcesAndSourcesCore(mField)
      );
      flux_integrate->getOpPtrVector().push_back(new OpIntegrateFluxes(*this,"FLUXES"));
      int frequency = 1;
      ierr = PetscOptionsBegin(
        PETSC_COMM_WORLD,"","Monitor post proc","none"
      ); CHKERRG(ierr);
      ierr = PetscOptionsInt(
        "-how_often_output",
        "frequency how often results are dumped on hard disk","",
        frequency,&frequency,NULL
      ); CHKERRG(ierr);
      ierr = PetscOptionsEnd(); CHKERRG(ierr);

      tsMonitor = boost::shared_ptr<FEMethod>(
        new MonitorPostProc(*this,post_process,flux_integrate,frequency)
      );
      TsCtx *ts_ctx;
      DMMoFEMGetTsCtx(dM,&ts_ctx);
      ts_ctx->get_postProcess_to_do_Monitor().push_back(tsMonitor);
      MoFEMFunctionReturnHot(0);
    }

    Vec D1;  ///< Vector with inital head capillary pressure
    Vec ghostFlux;  ///< Ghost Vector of integrated fluxes

    /**
     * \brief Create vectors and matrices
     * @return Error code
     */
    MoFEMErrorCode createMatrices() {
      MoFEMFunctionBeginHot;
      ierr = DMCreateMatrix(dM,&Aij); CHKERRG(ierr);
      ierr = DMCreateGlobalVector(dM,&D0); CHKERRG(ierr);
      ierr = VecDuplicate(D0,&D1); CHKERRG(ierr);
      ierr = VecDuplicate(D0,&D); CHKERRG(ierr);
      ierr = VecDuplicate(D0,&F); CHKERRG(ierr);
      int ghosts[] = { 0 };
      int nb_locals = mField.get_comm_rank()==0?1:0;
      int nb_ghosts = mField.get_comm_rank()>0?1:0;
      ierr = VecCreateGhost(
        PETSC_COMM_WORLD,nb_locals,1,nb_ghosts,ghosts,&ghostFlux
      ); CHKERRG(ierr);
      MoFEMFunctionReturnHot(0);
    }

    /**
     * \brief Delete matrices and vector when no longer needed
     * @return error code
     */
    MoFEMErrorCode destroyMatrices() {
      MoFEMFunctionBeginHot;
      ierr = MatDestroy(&Aij); CHKERRG(ierr);
      ierr = VecDestroy(&D); CHKERRG(ierr);
      ierr = VecDestroy(&D0); CHKERRG(ierr);
      ierr = VecDestroy(&D1); CHKERRG(ierr);
      ierr = VecDestroy(&F); CHKERRG(ierr);
      ierr = VecDestroy(&ghostFlux); CHKERRG(ierr);
      MoFEMFunctionReturnHot(0);
    }

    /**
     * \brief Calculate boundary conditions for fluxes
     * @return Error code
     */
    MoFEMErrorCode calculateEssentialBc() {
      MoFEMFunctionBeginHot;
      // clear vectors
      ierr = VecZeroEntries(D0); CHKERRG(ierr);
      ierr = VecGhostUpdateBegin(D0,INSERT_VALUES,SCATTER_FORWARD); CHKERRG(ierr);
      ierr = VecGhostUpdateEnd(D0,INSERT_VALUES,SCATTER_FORWARD); CHKERRG(ierr);
      // clear essential bc indices, it could have dofs from other mesh refinement
      bcIndices.clear();
      // set operator to calculate essential boundary conditions
      ierr = DMoFEMLoopFiniteElements(dM,"MIX_BCFLUX",feFaceBc); CHKERRG(ierr);
      ierr = VecGhostUpdateBegin(D0,INSERT_VALUES,SCATTER_REVERSE); CHKERRG(ierr);
      ierr = VecGhostUpdateEnd(D0,INSERT_VALUES,SCATTER_REVERSE); CHKERRG(ierr);
      ierr = VecAssemblyBegin(D0); CHKERRG(ierr);
      ierr = VecAssemblyEnd(D0); CHKERRG(ierr);
      double norm2D0;
      ierr = VecNorm(D0,NORM_2,&norm2D0); CHKERRG(ierr);
      // ierr = VecView(D0,PETSC_VIEWER_STDOUT_WORLD); CHKERRG(ierr);
      PetscPrintf(PETSC_COMM_WORLD,"norm2D0 = %6.4e\n",norm2D0);
      MoFEMFunctionReturnHot(0);
    }

    /**
     * \brief Calculate inital pressure head distribution
     * @return Error code
     */
    MoFEMErrorCode calculateInitialPc() {
      MoFEMFunctionBeginHot;
      // clear vectors
      ierr = VecZeroEntries(D1); CHKERRG(ierr);
      ierr = VecGhostUpdateBegin(D1,INSERT_VALUES,SCATTER_FORWARD); CHKERRG(ierr);
      ierr = VecGhostUpdateEnd(D1,INSERT_VALUES,SCATTER_FORWARD); CHKERRG(ierr);
      // Calculate initial pressure head on each element
      ierr = DMoFEMLoopFiniteElements(dM,"MIX",feVolInitialPc); CHKERRG(ierr);
      // Assemble vector
      ierr = VecGhostUpdateBegin(D1,INSERT_VALUES,SCATTER_REVERSE); CHKERRG(ierr);
      ierr = VecGhostUpdateEnd(D1,INSERT_VALUES,SCATTER_REVERSE); CHKERRG(ierr);
      ierr = VecAssemblyBegin(D1); CHKERRG(ierr);
      ierr = VecAssemblyEnd(D1); CHKERRG(ierr);
      // Calculate norm
      double norm2D1;
      ierr = VecNorm(D1,NORM_2,&norm2D1); CHKERRG(ierr);
      // ierr = VecView(D0,PETSC_VIEWER_STDOUT_WORLD); CHKERRG(ierr);
      PetscPrintf(PETSC_COMM_WORLD,"norm2D1 = %6.4e\n",norm2D1);
      MoFEMFunctionReturnHot(0);
    }

    /**
     * \brief solve problem
     * @return error code
     */
    MoFEMErrorCode solveProblem(bool set_initial_pc = true) {
      MoFEMFunctionBeginHot;
      if(set_initial_pc) {
        // Set initial head
        ierr = DMoFEMMeshToLocalVector(dM,D1,INSERT_VALUES,SCATTER_REVERSE); CHKERRG(ierr);
      }

      // Initiate vector from data on the mesh
      ierr = DMoFEMMeshToLocalVector(dM,D,INSERT_VALUES,SCATTER_FORWARD); CHKERRG(ierr);

      // Create time solver
      TS ts;
      ierr = TSCreate(PETSC_COMM_WORLD,&ts); CHKERRG(ierr);
      // Use backward Euler method
      ierr = TSSetType(ts,TSBEULER); CHKERRG(ierr);
      // Set final time
      double ftime = 1;
      ierr = TSSetDuration(ts,PETSC_DEFAULT,ftime); CHKERRG(ierr);
      // Setup solver from commabd line
      ierr = TSSetFromOptions(ts); CHKERRG(ierr);
      // Set DM to TS
      ierr = TSSetDM(ts,dM); CHKERRG(ierr);
      #if PETSC_VERSION_GE(3,7,0)
      ierr = TSSetExactFinalTime(ts,TS_EXACTFINALTIME_STEPOVER); CHKERRG(ierr);
      #endif
      // Set-up monitor
      TsCtx *ts_ctx;
      DMMoFEMGetTsCtx(dM,&ts_ctx);
      ierr = TSMonitorSet(ts,f_TSMonitorSet,ts_ctx,PETSC_NULL); CHKERRG(ierr);

      //This add SNES monitor, to show error by fields. It is dirty trick
      //to add monitor, so code is hidden from doxygen
      ierr = TSSetSolution(ts,D); CHKERRG(ierr);
      ierr = TSSetUp(ts); CHKERRG(ierr);
      SNES snes;
      ierr = TSGetSNES(ts,&snes); CHKERRG(ierr);

      #if PETSC_VERSION_GE(3,7,0)
      {
        PetscViewerAndFormat *vf;
        ierr = PetscViewerAndFormatCreate(PETSC_VIEWER_STDOUT_WORLD,PETSC_VIEWER_DEFAULT,&vf);CHKERRG(ierr);
        ierr = SNESMonitorSet(
          snes,
          (MoFEMErrorCode (*)(SNES,PetscInt,PetscReal,void*))SNESMonitorFields,
          vf,(MoFEMErrorCode (*)(void**))PetscViewerAndFormatDestroy
        );CHKERRG(ierr);
      }
      #else
      {
        ierr = SNESMonitorSet(
          snes,
          (MoFEMErrorCode (*)(SNES,PetscInt,PetscReal,void*))SNESMonitorFields,0,0
        );CHKERRG(ierr);
      }
      #endif


      ierr = TSSolve(ts,D); CHKERRG(ierr);

      // Get statistic form TS and print it
      ierr = TSGetTime(ts,&ftime); CHKERRG(ierr);
      PetscInt steps,snesfails,rejects,nonlinits,linits;
      ierr = TSGetTimeStepNumber(ts,&steps); CHKERRG(ierr);
      ierr = TSGetSNESFailures(ts,&snesfails); CHKERRG(ierr);
      ierr = TSGetStepRejections(ts,&rejects); CHKERRG(ierr);
      ierr = TSGetSNESIterations(ts,&nonlinits); CHKERRG(ierr);
      ierr = TSGetKSPIterations(ts,&linits); CHKERRG(ierr);
      PetscPrintf(PETSC_COMM_WORLD,
        "steps %D (%D rejected, %D SNES fails), ftime %g, nonlinits %D, linits %D\n",
        steps,rejects,snesfails,ftime,nonlinits,linits
      );

      MoFEMFunctionReturnHot(0);
    }

  };

}

#endif //  __UNSATURATD_FLOW_HPP__