solver_CGNE.cpp

Go to the documentation of this file.

#include "solver_CGNE.h"

using Bridge::vout;


#ifdef USE_FACTORY
namespace {
  Solver *create_object(Fopr *fopr)
  {
    return new Solver_CGNE(fopr);
  }


  bool init = Solver::Factory::Register("CGNE", create_object);
}
#endif

//- parameter entries
namespace {
  void append_entry(Parameters& param)
  {
    param.Register_int("maximum_number_of_iteration", 0);
    param.Register_double("convergence_criterion_squared", 0.0);

    param.Register_string("verbose_level", "NULL");
  }


#ifdef USE_PARAMETERS_FACTORY
  bool init_param = ParametersFactory::Register("Solver.CGNE", append_entry);
#endif
}
//- end

//- parameters class
Parameters_Solver_CGNE::Parameters_Solver_CGNE() { append_entry(*this); }
//- end

const std::string Solver_CGNE::class_name = "Solver_CGNE";

//====================================================================
void Solver_CGNE::set_parameters(const Parameters& params)
{
  return this->Solver_CG::set_parameters(params);
}


//====================================================================
void Solver_CGNE::solve(Field& xq, const Field& b, int& Nconv, double& diff)
{
  Fopr   *fopr     = this->Solver_CG::get_fopr();
  string mode_prev = fopr->get_mode();

#pragma omp master
  {
    vout.detailed(m_vl, "%s: solver starts\n", class_name.c_str());
  }
#pragma omp barrier

  if (mode_prev == "DdagD") {
    this->Solver_CG::solve(xq, b, Nconv, diff);  // fallback to CG solver
    return;
  }

  if (!((mode_prev == "D") || (mode_prev == "Ddag"))) {
    vout.crucial(m_vl, "%s: unsupported mode for fopr: %s.",
                 class_name.c_str(), mode_prev.c_str());
    abort();
  }

  //  Field y(b);

#pragma omp master
  {
    if ((m_y.nin() != b.nin()) || (m_y.nvol() != b.nvol()) ||
        (m_y.nex() != b.nex())) {
      m_y.reset(b.nin(), b.nvol(), b.nex());
    }

    if (mode_prev == "D") {
      fopr->set_mode("DDdag");
    } else if (mode_prev == "Ddag") {
      fopr->set_mode("DdagD");
    }
  }
#pragma omp barrier

  this->Solver_CG::solve(m_y, b, Nconv, diff);

#pragma omp master
  {
    fopr->set_mode(mode_prev);
  }
#pragma omp barrier

  fopr->mult_dag(xq, m_y);  // xq = fopr->mult_dag(y);

#pragma omp barrier
}


//====================================================================
//============================================================END=====