createDistanceTableAAOMPTarget.cpp

Go to the documentation of this file.

//////////////////////////////////////////////////////////////////////////////////////
// This file is distributed under the University of Illinois/NCSA Open Source License.
// See LICENSE file in top directory for details.
//
// Copyright (c) 2016 Jeongnim Kim and QMCPACK developers.
//
// File developed by: Ken Esler, kpesler@gmail.com, University of Illinois at Urbana-Champaign
//                    Jeremy McMinnis, jmcminis@gmail.com, University of Illinois at Urbana-Champaign
//                    Jeongnim Kim, jeongnim.kim@gmail.com, University of Illinois at Urbana-Champaign
//                    Jaron T. Krogel, krogeljt@ornl.gov, Oak Ridge National Laboratory
//                    Mark A. Berrill, berrillma@ornl.gov, Oak Ridge National Laboratory
//
// File created by: Jeongnim Kim, jeongnim.kim@gmail.com, University of Illinois at Urbana-Champaign
//////////////////////////////////////////////////////////////////////////////////////


#include "Particle/createDistanceTable.h"
#include "Particle/DistanceTable.h"
#include "Particle/SoaDistanceTableAAOMPTarget.h"

namespace qmcplusplus
{
/** Adding SymmetricDTD to the list, e.g., el-el distance table
 *\param s source/target particle set
 *\return index of the distance table with the name
 */
std::unique_ptr<DistanceTable> createDistanceTableAAOMPTarget(ParticleSet& s, std::ostream& description)
{
  using RealType = OHMMS_PRECISION;
  enum
  {
    DIM = OHMMS_DIM
  };
  const int sc = s.getLattice().SuperCellEnum;
  std::unique_ptr<DistanceTable> dt;
  std::ostringstream o;
  o << "  Distance table for similar particles (A-A):" << std::endl;
  o << "    source/target: " << s.getName() << std::endl;
  o << "    Using structure-of-arrays (SoA) data layout and OpenMP offload" << std::endl;

  if (sc == SUPERCELL_BULK)
  {
    if (s.getLattice().DiagonalOnly)
    {
      o << "    Distance computations use orthorhombic periodic cell in 3D." << std::endl;
      dt = std::make_unique<SoaDistanceTableAAOMPTarget<RealType, DIM, PPPO + SOA_OFFSET>>(s);
    }
    else
    {
      if (s.getLattice().WignerSeitzRadius > s.getLattice().SimulationCellRadius)
      {
        o << "    Distance computations use general periodic cell in 3D with corner image checks." << std::endl;
        dt = std::make_unique<SoaDistanceTableAAOMPTarget<RealType, DIM, PPPG + SOA_OFFSET>>(s);
      }
      else
      {
        o << "    Distance computations use general periodic cell in 3D without corner image checks." << std::endl;
        dt = std::make_unique<SoaDistanceTableAAOMPTarget<RealType, DIM, PPPS + SOA_OFFSET>>(s);
      }
    }
  }
  else if (sc == SUPERCELL_SLAB)
  {
    if (s.getLattice().DiagonalOnly)
    {
      o << "    Distance computations use orthorhombic code for periodic cell in 2D." << std::endl;
      dt = std::make_unique<SoaDistanceTableAAOMPTarget<RealType, DIM, PPNO + SOA_OFFSET>>(s);
    }
    else
    {
      if (s.getLattice().WignerSeitzRadius > s.getLattice().SimulationCellRadius)
      {
        o << "    Distance computations use general periodic cell in 2D with corner image checks." << std::endl;
        dt = std::make_unique<SoaDistanceTableAAOMPTarget<RealType, DIM, PPNG + SOA_OFFSET>>(s);
      }
      else
      {
        o << "    Distance computations use general periodic cell in 2D without corner image checks." << std::endl;
        dt = std::make_unique<SoaDistanceTableAAOMPTarget<RealType, DIM, PPNS + SOA_OFFSET>>(s);
      }
    }
  }
  else if (sc == SUPERCELL_WIRE)
  {
    o << "    Distance computations use periodic cell in one dimension." << std::endl;
    dt = std::make_unique<SoaDistanceTableAAOMPTarget<RealType, DIM, SUPERCELL_WIRE + SOA_OFFSET>>(s);
  }
  else //open boundary condition
  {
    o << "    Distance computations use open boundary conditions in 3D." << std::endl;
    dt = std::make_unique<SoaDistanceTableAAOMPTarget<RealType, DIM, SUPERCELL_OPEN + SOA_OFFSET>>(s);
  }

  description << o.str() << std::endl;
  return dt;
}

} //namespace qmcplusplus