File u1base.h

File List > abstractmodel > u1base.h

Go to the documentation of this file

#ifndef TEMPLAT_COSMOINTERFACE_ABSTRACTMODEL_U1BASE_H
#define TEMPLAT_COSMOINTERFACE_ABSTRACTMODEL_U1BASE_H

#include "TempLat/lattice/algebra/complexalgebra/complexfield.h"
#include "TempLat/lattice/field/collections/vectorfieldcollection.h"
#include "CosmoInterface/fieldsnumbering.h"
#include "CosmoInterface/initializers/initialconditionstype.h"
#include "CosmoInterface/runparameters.h"

namespace TempLat
{

  template <int NDIM, typename T, size_t NU1FLDS, size_t NS, size_t NC, typename SCALARU1AXIONCOUPLINGS> class U1Base
  {
  public:
    static constexpr size_t NU1 = NU1FLDS;

    // U(1) gauge fields
    VectorFieldCollection<Field<T, NDIM>, NU1FLDS> fldU1;
    VectorFieldCollection<Field<T, NDIM>, NU1FLDS> piU1;

    // Averages U1
    T U1Mag2AvI, U1Mag2AvSI;
    T U1pi2AvSI, U1pi2AvSIM, U1pi2AvIM, U1pi2AvI;

    InitialConditionsType::U1 getU1IC()
    {
      if constexpr (NC > 0)
        return InitialConditionsType::U1::RandomWithMatter;
      else if constexpr (NS > 0 && SCALARU1AXIONCOUPLINGS::howManyCouples() > 0)
        return InitialConditionsType::U1::BunchDavisTransverseU1;
      else
        return InitialConditionsType::U1::PlaneWavesZeroB;
    }

  protected:
    U1Base(device::memory::host_ptr<MemoryToolBox<NDIM>> toolBox, const LatticeParameters<T> &par)
        : fldU1("U1", toolBox, par), piU1("pi_U1", toolBox, par),
          // Initialize average fields
          U1Mag2AvI(0), U1Mag2AvSI(0), U1pi2AvSI(0), U1pi2AvSIM(0), U1pi2AvIM(0), U1pi2AvI(0)
    {
    }
  };

} // namespace TempLat

#endif