Skip to content

Commit

Permalink
Merge branch 'heavy_mesons_onlinemeas_gamma5herm' of github.com:etmc/…
Browse files Browse the repository at this point in the history 
…tmLQCD into heavy_mesons_onlinemeas_gamma5herm
  • Loading branch information
@simone-romiti
simone-romiti committed Jan 26, 2024
2 parents 3926b02 + 49cb20a commit d4e5b50
Show file tree
Hide file tree
Showing 2 changed files with 65 additions and 55 deletions.
112 changes: 57 additions & 55 deletions meas/correlators.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,6 @@
#include "gettime.h"



#define TM_OMEAS_FILENAME_LENGTH 100

/******************************************************
Expand Down Expand Up @@ -253,6 +252,15 @@ void light_correlators_measurement(const int traj, const int id, const int ieo)
return;
}


void spinor_dirac_array(su3_vector* phi, spinor psi){
phi[0] = psi.s0;
phi[1] = psi.s1;
phi[2] = psi.s2;
phi[3] = psi.s3;
}


/******************************************************
*
* This routine computes the correlator matrix <G1,G2> (<source sink>),
Expand Down Expand Up @@ -473,9 +481,10 @@ void heavy_correlators_measurement(const int traj, const int id, const int ieo,
for (size_t beta = 0; beta < 4; beta++) { // spin dilution index
for (size_t F = 0; F < 2; F++) { // flavor dilution index
for (size_t i_f = 0; i_f < 2; i_f++) { // flavor index of the doublet
eo_source_spinor_field_spin_diluted_oet_ts(arr_eo_spinor[0][db][beta][F][0][i_f],
arr_eo_spinor[0][db][beta][F][1][i_f], t0,
beta, sample, traj, seed_i);
eo_source_spinor_field_spin_diluted_oet_ts(
arr_eo_spinor[0][db][beta][F][0][i_f],
arr_eo_spinor[0][db][beta][F][1][i_f], t0,
beta, sample, traj, seed_i);
}
}
}
Expand All @@ -498,15 +507,15 @@ void heavy_correlators_measurement(const int traj, const int id, const int ieo,
// stored temporarely in the propagator spinors (used as dummy)
if (F==0){
mul_one_pm_itau2_and_div_by_sqrt2(
arr_eo_spinor[1][1][beta][F][i_eo][0], arr_eo_spinor[1][1][beta][F][i_eo][1],
arr_eo_spinor[0][1][beta][F][i_eo][0], phi1, +1.0,
VOLUME / 2);
arr_eo_spinor[1][1][beta][F][i_eo][0], arr_eo_spinor[1][1][beta][F][i_eo][1],
arr_eo_spinor[0][1][beta][F][i_eo][0], phi1, +1.0,
VOLUME / 2);
}
if (F==1){
mul_one_pm_itau2_and_div_by_sqrt2(
arr_eo_spinor[1][1][beta][F][i_eo][0], arr_eo_spinor[1][1][beta][F][i_eo][1],
phi2, arr_eo_spinor[0][1][beta][F][i_eo][1], +1.0,
VOLUME / 2);
arr_eo_spinor[1][1][beta][F][i_eo][0], arr_eo_spinor[1][1][beta][F][i_eo][1],
phi2, arr_eo_spinor[0][1][beta][F][i_eo][1], +1.0,
VOLUME / 2);
}

for (size_t i_f = 0; i_f < 2; i_f++) {
Expand Down Expand Up @@ -598,9 +607,9 @@ void heavy_correlators_measurement(const int traj, const int id, const int ieo,
// (c,s) --> [(1-i*tau_2)/sqrt(2)] * (c,s)
// stored temporarely in phi1, phi2
mul_one_pm_itau2_and_div_by_sqrt2(phi1, phi2,
arr_eo_spinor[1][1][beta][F][i_eo][0], arr_eo_spinor[1][1][beta][F][i_eo][1], -1.0,
VOLUME / 2);
arr_eo_spinor[1][1][beta][F][i_eo][0], arr_eo_spinor[1][1][beta][F][i_eo][1], -1.0,
VOLUME / 2);

assign(arr_eo_spinor[1][1][beta][F][i_eo][0], phi1, VOLUME / 2);
assign(arr_eo_spinor[1][1][beta][F][i_eo][1], phi2, VOLUME / 2);
}
Expand All @@ -616,19 +625,19 @@ void heavy_correlators_measurement(const int traj, const int id, const int ieo,
}

// now we switch from even-odd representation to standard
//for (size_t i_sp = 0; i_sp < 2; i_sp++) { // source or sink
for (size_t db = 0; db < 2; db++) { // doublet: light of heavy
for (size_t beta = 0; beta < 4; beta++) { // spin dilution index
for (size_t F = 0; F < 2; F++) { // flavor projection index
for (size_t i_f = 0; i_f < 2; i_f++) { // flavor index
convert_eo_to_lexic(arr_spinor[1][db][beta][F][i_f],
arr_eo_spinor[1][db][beta][F][0][i_f],
arr_eo_spinor[1][db][beta][F][1][i_f]);
for (size_t i_sp = 0; i_sp < 2; i_sp++) { // source or sink
for (size_t db = 0; db < 2; db++) { // doublet: light of heavy
for (size_t beta = 0; beta < 4; beta++) { // spin dilution index
for (size_t F = 0; F < 2; F++) { // flavor projection index
for (size_t i_f = 0; i_f < 2; i_f++) { // flavor index
convert_eo_to_lexic(arr_spinor[i_sp][db][beta][F][i_f],
arr_eo_spinor[i_sp][db][beta][F][0][i_f],
arr_eo_spinor[i_sp][db][beta][F][1][i_f]);
}
}
}
}
}
// }

//MPI_Barrier(MPI_COMM_WORLD);
if (g_proc_id == 0){
Expand All @@ -637,29 +646,31 @@ void heavy_correlators_measurement(const int traj, const int id, const int ieo,

//MPI_Barrier(MPI_COMM_WORLD);
if (g_proc_id == 0){
printf("Checkpoint 11\n");
printf("Checkpoint 11.0\n");
}

/*
Now that I have all the propagators (all in the basis of
https://arxiv.org/pdf/1005.2042.pdf) I can build the correlators of eq. (20)
*/
const int f0 = 0; // flavor index of the up

/* now we sum only over local space for every t */
const int j_ts = t0-g_proc_coords[0]*T; // checkerboard index of the time at the source

for (t = 0; t < T; t++) {
j = g_ipt[t][0][0][0];
j = g_ipt[t][0][0][0]; // source and sink separated by "t"

// dummy variables
res = 0.0;
respa = 0.0;
resp4 = 0.0;


for (i = j; i < j + LX * LY * LZ; i++) {

// light correlators
for (size_t beta = 0; beta < 4; beta++) { // spin dilution
spinor psi_u = arr_spinor[1][0][beta][f0][f0][i];
spinor eta_u = arr_spinor[0][0][beta][f0][f0][0];

res += _spinor_prod_re(psi_u, psi_u);
_gamma0(phi, psi_u);
Expand All @@ -668,43 +679,34 @@ void heavy_correlators_measurement(const int traj, const int id, const int ieo,
resp4 += _spinor_prod_im(psi_u, phi);
}

// diagonal elements of Gamma_i*gamma_5
double diag_G1g5[4] = {1.0, 1.0, 1.0, 1.0};
double diag_G2g5[4] = {1.0, 1.0, 1.0, 1.0};

// heavy correlators
for (size_t hi = 0; hi < 2; hi++) {
for (size_t hj = 0; hj < 2; hj++) {
for (size_t g1 = 0; g1 < 2; g1++) {
if (g1 == 0) { // Gamma_1 = \mathbb{1}
diag_G1g5[2] = -1;
diag_G1g5[3] = -1;
}
for (size_t g2 = 0; g2 < 2; g2++) {
// heavy doublet spinor propagator
for (int alpha_1=0; alpha_1 < 4; alpha_1++){
spinor eta_h = arr_spinor[0][1][alpha_1][hj][hi][0];
spinor psi_u = arr_spinor[1][0][alpha_1][f0][f0][i];
for (int alpha_2=0; alpha_2 < 4; alpha_2++){
spinor eta_u = arr_spinor[0][0][alpha_2][f0][f0][0];
double dum1 = 0.0; // dummy variable
_spinor_scalar_prod(dum1, eta_u, psi_u);
double dum2 = 0.0; // dummy variable
spinor psi_h = arr_spinor[1][1][alpha_2][hj][hi][i];
if (g2 == 0) { // Gamma_2 = \mathbb{1}
diag_G2g5[2] = -1.0;
diag_G2g5[3] = -1.0;
double complex dum_tot = 0.0;
for (int alpha_1 = 0; alpha_1 < 4; alpha_1++){
for (int alpha_2 = 0; alpha_2 < 4; alpha_2++){
// S_\ell
spinor psi_h = arr_spinor[1][1][alpha_1][hj][hi][i];
if (g1 == 0){ // Gamma_1 = Id
_gamma5(psi_h, psi_h);
}
su3_vector psi_h_su3[4];
spinor_dirac_array(&psi_h_su3[0], psi_h);
spinor psi_u_star = arr_spinor[1][0][alpha_2][f0][f0][i];
if (g2 == 0){ // Gamma_2 = Id. NOTE: works because Gamma_2=Gamma_2* for Gamma_2=1,gamma_5
_gamma5(psi_u_star, psi_u_star);
}
_spinor_scalar_prod(dum2, eta_h, psi_h);
res_hihj_g1g2[hi][hj][g1][g2] += dum2*diag_G2g5[alpha_2]*dum1*diag_G1g5[alpha_1];
su3_vector psi_u_star_su3[4];
spinor_dirac_array(&psi_u_star_su3[0], psi_u_star);
complex double dum_12 = 0.0;
_colorvec_scalar_prod(dum_12, psi_u_star_su3[alpha_1], psi_h_su3[alpha_2]);
dum_tot =+ dum_12;
}
}
// restoring to default
diag_G2g5[2] = 1.0;
diag_G2g5[3] = 1.0;
res_hihj_g1g2[hi][hj][g1][g2] = creal(dum_tot); // correlators are real
}
// restoring to default
diag_G1g5[2] = 1.0;
diag_G1g5[3] = 1.0;
}
}

Expand Down
8 changes: 8 additions & 0 deletions solver/dirac_operator_eigenvectors.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,6 +265,14 @@ int cyclicDiff(int a,int b, int period);
conj((r).s3.c2) * (s).s3.c2;



#define _colorvec_scalar_prod(proj,r,s)\
(proj) = conj((r).c0) * (s).c0 + \
conj((r).c1) * (s).c1; + \
conj((r).c2) * (s).c2;



#define PROJECTSPLIT(p_plus,up_plus,col_proj,phi_o,phi_plus,col_phi)\
p_plus = 0; \
p_plus += conj(up_plus->s0.col_proj) * (phi_o->s0.col_phi); \
Expand Down

0 comments on commit d4e5b50

Please sign in to comment.