composite_criterion.hpp

Go to the documentation of this file.

//
//   Project Name:        KratosSolidMechanicsApplication $
//   Created by:          $Author:            JMCarbonell $
//   Last modified by:    $Co-Author:                     $
//   Date:                $Date:                 May 2018 $
//   Revision:            $Revision:                  0.0 $
//
//
 
#if !defined(KRATOS_COMPOSITE_CRITERION_H_INCLUDED )
#define  KRATOS_COMPOSITE_CRITERION_H_INCLUDED
 
// System includes
 
// External includes
 
// Project includes
#include "custom_solvers/convergence_criteria/convergence_criterion.hpp"
 
namespace Kratos
{
 
 
 
 
 
 
 
template<class TSparseSpace,class TDenseSpace>
class CompositeCriterion : public ConvergenceCriterion< TSparseSpace, TDenseSpace >
{
 public:
 
  typedef ConvergenceCriterion<TSparseSpace,TDenseSpace>      BaseType;
  typedef typename BaseType::LocalFlagType               LocalFlagType;
  typedef typename BaseType::DataType                         DataType;
  typedef typename BaseType::DofsArrayType               DofsArrayType;
  typedef typename BaseType::SystemMatrixType         SystemMatrixType;
  typedef typename BaseType::SystemVectorType         SystemVectorType;
 
  typedef typename BaseType::Pointer                   BasePointerType;
  typedef std::vector<BasePointerType>  ConvergenceCriterionVectorType;
 
  KRATOS_CLASS_POINTER_DEFINITION(CompositeCriterion);
 
 
 
  CompositeCriterion(BasePointerType pFirstCriterion, BasePointerType pSecondCriterion) : BaseType()
  {
    mConvergenceCriteria.push_back(pFirstCriterion);
    mConvergenceCriteria.push_back(pSecondCriterion);
  }
 
  CompositeCriterion(ConvergenceCriterionVectorType& rConvergenceCriteria) : BaseType(), mConvergenceCriteria(rConvergenceCriteria)
  {
  }
 
  CompositeCriterion(CompositeCriterion const& rOther)
      :BaseType(rOther)
      ,mConvergenceCriteria(rOther.mConvergenceCriteria)
  {
  }
 
  ~CompositeCriterion () override {}
 
 
  void SetEchoLevel(int Level) override
  {
    BaseType::SetEchoLevel(Level);
    for(typename ConvergenceCriterionVectorType::iterator it=mConvergenceCriteria.begin();
        it!=mConvergenceCriteria.end(); ++it)
      (*it)->SetEchoLevel(Level);
  }
 
  bool PreCriteria(ModelPart& rModelPart,
                   DofsArrayType& rDofSet,
                   const SystemMatrixType& rA,
                   const SystemVectorType& rDx,
                   const SystemVectorType& rb) override
  {
    if(this->Is(LocalFlagType::AND)){
 
      for(typename ConvergenceCriterionVectorType::iterator it=mConvergenceCriteria.begin();
          it!=mConvergenceCriteria.end(); ++it)
        if( !((*it)->PreCriteria(rModelPart,rDofSet,rA,rDx,rb)) )
          return false;
 
      return true;
    }
    else if (this->Is(LocalFlagType::OR)){
 
      for(typename ConvergenceCriterionVectorType::iterator it=mConvergenceCriteria.begin();
          it!=mConvergenceCriteria.end(); ++it)
        if( ((*it)->PreCriteria(rModelPart,rDofSet,rA,rDx,rb)) )
          return true;
 
      return false;
    }
 
    return false;
  }
 
  bool PostCriteria(ModelPart& rModelPart,
                    DofsArrayType& rDofSet,
                    const SystemMatrixType& rA,
                    const SystemVectorType& rDx,
                    const SystemVectorType& rb) override
  {
    if(this->Is(LocalFlagType::AND)){
 
      for(typename ConvergenceCriterionVectorType::iterator it=mConvergenceCriteria.begin();
          it!=mConvergenceCriteria.end(); ++it)
        if( !((*it)->PostCriteria(rModelPart,rDofSet,rA,rDx,rb)) )
          return false;
 
      return true;
    }
    else if (this->Is(LocalFlagType::OR)){
 
      for(typename ConvergenceCriterionVectorType::iterator it=mConvergenceCriteria.begin();
          it!=mConvergenceCriteria.end(); ++it)
        if( ((*it)->PostCriteria(rModelPart,rDofSet,rA,rDx,rb)) )
          return true;
 
      return false;
    }
 
    return false;
  }
 
  void InitializeSolutionStep(ModelPart& rModelPart,
                              DofsArrayType& rDofSet,
                              const SystemMatrixType& rA,
                              const SystemVectorType& rDx,
                              const SystemVectorType& rb) override
  {
    for(typename ConvergenceCriterionVectorType::iterator it=mConvergenceCriteria.begin();
        it!=mConvergenceCriteria.end(); ++it)
      (*it)->InitializeSolutionStep(rModelPart,rDofSet,rA,rDx,rb);
  }
 
  void FinalizeSolutionStep(ModelPart& rModelPart,
                            DofsArrayType& rDofSet,
                            const SystemMatrixType& rA,
                            const SystemVectorType& rDx,
                            const SystemVectorType& rb) override
  {
    for(typename ConvergenceCriterionVectorType::iterator it=mConvergenceCriteria.begin();
        it!=mConvergenceCriteria.end(); ++it)
      (*it)->FinalizeSolutionStep(rModelPart,rDofSet,rA,rDx,rb);
  }
 
 
  int Check(ModelPart& rModelPart) override
  {
    KRATOS_TRY
 
    if( !(this->Is(LocalFlagType::AND) || this->Is(LocalFlagType::OR)) )
      KRATOS_ERROR << "Flag AND or OR is not set for the Composite Convergence Criterion "<< std::endl;
 
    for(typename ConvergenceCriterionVectorType::iterator it=mConvergenceCriteria.begin();
        it!=mConvergenceCriteria.end(); ++it)
      if( ((*it)->Check(rModelPart)) != 0 )
        return 1;
 
    return 0;
 
    KRATOS_CATCH("")
  }
 
 
 
 
 
 
 protected:
 
 
 
 
 
 
 
 
 private:
 
 
 
  ConvergenceCriterionVectorType mConvergenceCriteria;
 
 
 
 
 
 
 
};  // Class CompositeCriterion
 
 
 
 
 
 
 
}  // namespace Kratos.
 
#endif // KRATOS_COMPOSITE_CRITERION_H_INCLUDED  defined