/* Genome-wide Efficient Mixed Model Association (GEMMA) Copyright (C) 2011 Xiang Zhou This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see . */ #include #include #include #include #include #include #include #include "gsl/gsl_vector.h" #include "gsl/gsl_matrix.h" #include "gsl/gsl_linalg.h" #include "gsl/gsl_blas.h" #include "gsl/gsl_eigen.h" #include "gsl/gsl_cdf.h" #include "lapack.h" //for functions EigenDecomp #ifdef FORCE_FLOAT #include "io_float.h" //for function ReadFile_kin #include "gemma_float.h" #include "vc_float.h" #include "lm_float.h" //for LM class #include "bslmm_float.h" //for BSLMM class #include "bslmmdap_float.h" //for BSLMMDAP class #include "ldr_float.h" //for LDR class #include "lmm_float.h" //for LMM class, and functions CalcLambda, CalcPve, CalcVgVe #include "mvlmm_float.h" //for MVLMM class #include "prdt_float.h" //for PRDT class #include "varcov_float.h" //for MVLMM class #include "mathfunc_float.h" //for a few functions #else #include "io.h" #include "gemma.h" #include "vc.h" #include "lm.h" #include "bslmm.h" #include "bslmmdap.h" #include "ldr.h" #include "lmm.h" #include "mvlmm.h" #include "prdt.h" #include "varcov.h" #include "mathfunc.h" #endif using namespace std; GEMMA::GEMMA(void): version("0.96"), date("05/17/2017"), year("2017") {} void GEMMA::PrintHeader (void) { cout< indicator_all; size_t c_bv=0; for (size_t i=0; isize; i++) { d=gsl_vector_get(y_prdt, i); d=gsl_cdf_gaussian_P(d, 1.0); gsl_vector_set(y_prdt, i, d); } } cPRDT.CopyToParam(cPar); cPRDT.WriteFiles(y_prdt); gsl_vector_free(y_prdt); } //Prediction with kinship matrix only; for one or more phenotypes if (cPar.a_mode==43) { //first, use individuals with full phenotypes to obtain estimates of Vg and Ve gsl_matrix *Y=gsl_matrix_alloc (cPar.ni_test, cPar.n_ph); gsl_matrix *W=gsl_matrix_alloc (Y->size1, cPar.n_cvt); gsl_matrix *G=gsl_matrix_alloc (Y->size1, Y->size1); gsl_matrix *U=gsl_matrix_alloc (Y->size1, Y->size1); gsl_matrix *UtW=gsl_matrix_alloc (Y->size1, W->size2); gsl_matrix *UtY=gsl_matrix_alloc (Y->size1, Y->size2); gsl_vector *eval=gsl_vector_alloc (Y->size1); gsl_matrix *Y_full=gsl_matrix_alloc (cPar.ni_cvt, cPar.n_ph); gsl_matrix *W_full=gsl_matrix_alloc (Y_full->size1, cPar.n_cvt); //set covariates matrix W and phenotype matrix Y //an intercept should be included in W, cPar.CopyCvtPhen (W, Y, 0); cPar.CopyCvtPhen (W_full, Y_full, 1); gsl_matrix *Y_hat=gsl_matrix_alloc (Y_full->size1, cPar.n_ph); gsl_matrix *G_full=gsl_matrix_alloc (Y_full->size1, Y_full->size1); gsl_matrix *H_full=gsl_matrix_alloc (Y_full->size1*Y_hat->size2, Y_full->size1*Y_hat->size2); //read relatedness matrix G, and matrix G_full ReadFile_kin (cPar.file_kin, cPar.indicator_idv, cPar.mapID2num, cPar.k_mode, cPar.error, G); if (cPar.error==true) {cout<<"error! fail to read kinship/relatedness file. "<size; i++) { if (gsl_vector_get (eval, i)<1e-10) {gsl_vector_set (eval, i, 0);} cPar.trace_G+=gsl_vector_get (eval, i); } cPar.trace_G/=(double)eval->size; cPar.time_eigen=(clock()-time_start)/(double(CLOCKS_PER_SEC)*60.0); //calculate UtW and Uty CalcUtX (U, W, UtW); CalcUtX (U, Y, UtY); //calculate variance component and beta estimates //and then obtain predicted values if (cPar.n_ph==1) { gsl_vector *beta=gsl_vector_alloc (W->size2); gsl_vector *se_beta=gsl_vector_alloc (W->size2); double lambda, logl, vg, ve; gsl_vector_view UtY_col=gsl_matrix_column (UtY, 0); //obtain estimates CalcLambda ('R', eval, UtW, &UtY_col.vector, cPar.l_min, cPar.l_max, cPar.n_region, lambda, logl); CalcLmmVgVeBeta (eval, UtW, &UtY_col.vector, lambda, vg, ve, beta, se_beta); cout<<"REMLE estimate for vg in the null model = "<size1; i++) { for (size_t j=0; j<=i; j++) { cout<size1; i++) { for (size_t j=0; j<=i; j++) { cout<size1; i++) { for (size_t j=i; jsize2; j++) { cPar.Vg_remle_null.push_back(gsl_matrix_get (Vg, i, j) ); cPar.Ve_remle_null.push_back(gsl_matrix_get (Ve, i, j) ); } } //obtain Y_hat from fixed effects gsl_blas_dgemm (CblasNoTrans, CblasTrans, 1.0, W_full, B, 0.0, Y_hat); //obtain H KroneckerSym(G_full, Vg, H_full); for (size_t i=0; isize1; i++) { gsl_matrix_view H_sub=gsl_matrix_submatrix (H_full, i*Ve->size1, i*Ve->size2, Ve->size1, Ve->size2); gsl_matrix_add (&H_sub.matrix, Ve); } //free matrices gsl_matrix_free (Vg); gsl_matrix_free (Ve); gsl_matrix_free (B); gsl_matrix_free (se_B); } PRDT cPRDT; cPRDT.CopyFromParam(cPar); cout<<"Predicting Missing Phentypes ... "< setSnps_beta; map mapRS2wA, mapRS2wK; cPar.ObtainWeight(setSnps_beta, mapRS2wK); time_start=clock(); cPar.CalcS (mapRS2wA, mapRS2wK, W, A, K, &S_mat.matrix, &Svar_mat.matrix, &ns_vec.vector); cPar.time_G=(clock()-time_start)/(double(CLOCKS_PER_SEC)*60.0); if (cPar.error==true) {cout<<"error! fail to calculate the S matrix. "< vec_cat, vec_ni; vector vec_weight, vec_z2; map mapRS2weight; mapRS2weight.clear(); time_start=clock(); ReadFile_beta (cPar.file_beta, cPar.mapRS2cat, mapRS2weight, vec_cat, vec_ni, vec_weight, vec_z2, cPar.ni_total, cPar.ns_total, cPar.ns_test); cout<<"## number of total individuals = "<size1, cPar.n_cvt); //set covariates matrix W and phenotype matrix Y //an intercept should be included in W, cPar.CopyCvtPhen (W, Y, 0); //Fit LM or mvLM if (cPar.n_ph==1) { LM cLm; cLm.CopyFromParam(cPar); gsl_vector_view Y_col=gsl_matrix_column (Y, 0); if (!cPar.file_gene.empty()) { cLm.AnalyzeGene (W, &Y_col.vector); //y is the predictor, not the phenotype } else if (!cPar.file_bfile.empty()) { cLm.AnalyzePlink (W, &Y_col.vector); } else if (!cPar.file_oxford.empty()) { cLm.Analyzebgen (W, &Y_col.vector); } else { cLm.AnalyzeBimbam (W, &Y_col.vector); } cLm.WriteFiles(); cLm.CopyToParam(cPar); } /* else { MVLM cMvlm; cMvlm.CopyFromParam(cPar); if (!cPar.file_bfile.empty()) { cMvlm.AnalyzePlink (W, Y); } else { cMvlm.AnalyzeBimbam (W, Y); } cMvlm.WriteFiles(); cMvlm.CopyToParam(cPar); } */ //release all matrices and vectors gsl_matrix_free (Y); gsl_matrix_free (W); } //VC estimation with one or multiple kinship matrices //REML approach only //if file_kin or file_ku/kd is provided, then a_mode is changed to 5 already, in param.cpp //for one phenotype only; if (cPar.a_mode==61 || cPar.a_mode==62 || cPar.a_mode==63) { if (!cPar.file_beta.empty() ) { //need to obtain a common set of SNPs between beta file and the genotype file; these are saved in mapRS2wA and mapRS2wK //normalize the weight in mapRS2wK to have an average of one; each element of mapRS2wA is 1 //update indicator_snps, so that the numbers are in accordance with mapRS2wK set setSnps_beta; ReadFile_snps_header (cPar.file_beta, setSnps_beta); map mapRS2wA, mapRS2wK; cPar.ObtainWeight(setSnps_beta, mapRS2wK); cPar.UpdateSNP (mapRS2wK); //setup matrices and vectors gsl_matrix *S=gsl_matrix_alloc (cPar.n_vc*2, cPar.n_vc); gsl_matrix *Vq=gsl_matrix_alloc (cPar.n_vc, cPar.n_vc); gsl_vector *q=gsl_vector_alloc (cPar.n_vc); gsl_vector *s=gsl_vector_alloc (cPar.n_vc+1); gsl_matrix *K=gsl_matrix_alloc (cPar.ni_test, cPar.n_vc*cPar.ni_test); gsl_matrix *A=gsl_matrix_alloc (cPar.ni_test, cPar.n_vc*cPar.ni_test); gsl_vector *y=gsl_vector_alloc (cPar.ni_test); gsl_matrix *W=gsl_matrix_alloc (cPar.ni_test, cPar.n_cvt); gsl_matrix_set_zero (K); gsl_matrix_set_zero (A); gsl_matrix_set_zero(S); gsl_matrix_set_zero(Vq); gsl_vector_set_zero (q); gsl_vector_set_zero (s); cPar.CopyCvtPhen (W, y, 0); gsl_matrix_view S_mat=gsl_matrix_submatrix(S, 0, 0, cPar.n_vc, cPar.n_vc); gsl_matrix_view Svar_mat=gsl_matrix_submatrix (S, cPar.n_vc, 0, cPar.n_vc, cPar.n_vc); gsl_vector_view s_vec=gsl_vector_subvector(s, 0, cPar.n_vc); vector vec_cat, vec_ni; vector vec_weight, vec_z2; //read beta, based on the mapRS2wK ReadFile_beta (cPar.file_beta, cPar.mapRS2cat, mapRS2wK, vec_cat, vec_ni, vec_weight, vec_z2, cPar.ni_study, cPar.ns_study, cPar.ns_test); cout<<"Study Panel: "<size1, cPar.n_cvt); gsl_matrix *G=gsl_matrix_alloc (Y->size1, Y->size1*cPar.n_vc ); //set covariates matrix W and phenotype matrix Y //an intercept should be included in W, cPar.CopyCvtPhen (W, Y, 0); //read kinship matrices if (!(cPar.file_mk).empty()) { ReadFile_mk (cPar.file_mk, cPar.indicator_idv, cPar.mapID2num, cPar.k_mode, cPar.error, G); if (cPar.error==true) {cout<<"error! fail to read kinship/relatedness file. "<size1, G->size1, G->size1); CenterMatrix (&G_sub.matrix); d=0; for (size_t j=0; jsize1; j++) { d+=gsl_matrix_get (&G_sub.matrix, j, j); } d/=(double)G->size1; (cPar.v_traceG).push_back(d); } } else if (!(cPar.file_kin).empty()) { ReadFile_kin (cPar.file_kin, cPar.indicator_idv, cPar.mapID2num, cPar.k_mode, cPar.error, G); if (cPar.error==true) {cout<<"error! fail to read kinship/relatedness file. "<size1; j++) { d+=gsl_matrix_get (G, j, j); } d/=(double)G->size1; (cPar.v_traceG).push_back(d); } /* //eigen-decomposition and calculate trace_G cout<<"Start Eigen-Decomposition..."<size; i++) { if (gsl_vector_get (eval, i)<1e-10) {gsl_vector_set (eval, i, 0);} cPar.trace_G+=gsl_vector_get (eval, i); } cPar.trace_G/=(double)eval->size; cPar.time_eigen=(clock()-time_start)/(double(CLOCKS_PER_SEC)*60.0); } else { ReadFile_eigenU (cPar.file_ku, cPar.error, U); if (cPar.error==true) {cout<<"error! fail to read the U file. "<size; i++) { if (gsl_vector_get(eval, i)<1e-10) {gsl_vector_set(eval, i, 0);} cPar.trace_G+=gsl_vector_get(eval, i); } cPar.trace_G/=(double)eval->size; } */ //fit multiple variance components if (cPar.n_ph==1) { // if (cPar.n_vc==1) { /* //calculate UtW and Uty CalcUtX (U, W, UtW); CalcUtX (U, Y, UtY); gsl_vector_view beta=gsl_matrix_row (B, 0); gsl_vector_view se_beta=gsl_matrix_row (se_B, 0); gsl_vector_view UtY_col=gsl_matrix_column (UtY, 0); CalcLambda ('L', eval, UtW, &UtY_col.vector, cPar.l_min, cPar.l_max, cPar.n_region, cPar.l_mle_null, cPar.logl_mle_H0); CalcLmmVgVeBeta (eval, UtW, &UtY_col.vector, cPar.l_mle_null, cPar.vg_mle_null, cPar.ve_mle_null, &beta.vector, &se_beta.vector); cPar.beta_mle_null.clear(); cPar.se_beta_mle_null.clear(); for (size_t i=0; isize2; i++) { cPar.beta_mle_null.push_back(gsl_matrix_get(B, 0, i) ); cPar.se_beta_mle_null.push_back(gsl_matrix_get(se_B, 0, i) ); } CalcLambda ('R', eval, UtW, &UtY_col.vector, cPar.l_min, cPar.l_max, cPar.n_region, cPar.l_remle_null, cPar.logl_remle_H0); CalcLmmVgVeBeta (eval, UtW, &UtY_col.vector, cPar.l_remle_null, cPar.vg_remle_null, cPar.ve_remle_null, &beta.vector, &se_beta.vector); cPar.beta_remle_null.clear(); cPar.se_beta_remle_null.clear(); for (size_t i=0; isize2; i++) { cPar.beta_remle_null.push_back(gsl_matrix_get(B, 0, i) ); cPar.se_beta_remle_null.push_back(gsl_matrix_get(se_B, 0, i) ); } CalcPve (eval, UtW, &UtY_col.vector, cPar.l_remle_null, cPar.trace_G, cPar.pve_null, cPar.pve_se_null); cPar.PrintSummary(); //calculate and output residuals if (cPar.a_mode==5) { gsl_vector *Utu_hat=gsl_vector_alloc (Y->size1); gsl_vector *Ute_hat=gsl_vector_alloc (Y->size1); gsl_vector *u_hat=gsl_vector_alloc (Y->size1); gsl_vector *e_hat=gsl_vector_alloc (Y->size1); gsl_vector *y_hat=gsl_vector_alloc (Y->size1); //obtain Utu and Ute gsl_vector_memcpy (y_hat, &UtY_col.vector); gsl_blas_dgemv (CblasNoTrans, -1.0, UtW, &beta.vector, 1.0, y_hat); double d, u, e; for (size_t i=0; isize; i++) { d=gsl_vector_get (eval, i); u=cPar.l_remle_null*d/(cPar.l_remle_null*d+1.0)*gsl_vector_get(y_hat, i); e=1.0/(cPar.l_remle_null*d+1.0)*gsl_vector_get(y_hat, i); gsl_vector_set (Utu_hat, i, u); gsl_vector_set (Ute_hat, i, e); } //obtain u and e gsl_blas_dgemv (CblasNoTrans, 1.0, U, Utu_hat, 0.0, u_hat); gsl_blas_dgemv (CblasNoTrans, 1.0, U, Ute_hat, 0.0, e_hat); //output residuals cPar.WriteVector(u_hat, "residU"); cPar.WriteVector(e_hat, "residE"); gsl_vector_free(u_hat); gsl_vector_free(e_hat); gsl_vector_free(y_hat); } */ // } else { gsl_vector_view Y_col=gsl_matrix_column (Y, 0); VC cVc; cVc.CopyFromParam(cPar); if (cPar.a_mode==61) { cVc.CalcVChe (G, W, &Y_col.vector); } else if (cPar.a_mode==62) { cVc.CalcVCreml (cPar.noconstrain, G, W, &Y_col.vector); } else { cVc.CalcVCacl (G, W, &Y_col.vector); } cVc.CopyToParam(cPar); //obtain pve from sigma2 //obtain se_pve from se_sigma2 //} } } } //compute confidence intervals with additional summary statistics //we do not check the sign of z-scores here, but they have to be matched with the genotypes if (cPar.a_mode==66 || cPar.a_mode==67) { //read reference file first gsl_matrix *S=gsl_matrix_alloc (cPar.n_vc, cPar.n_vc); gsl_matrix *Svar=gsl_matrix_alloc (cPar.n_vc, cPar.n_vc); gsl_vector *s_ref=gsl_vector_alloc (cPar.n_vc); gsl_matrix_set_zero(S); gsl_matrix_set_zero(Svar); gsl_vector_set_zero(s_ref); if (!cPar.file_ref.empty()) { ReadFile_ref(cPar.file_ref, S, Svar, s_ref, cPar.ni_ref); } else { ReadFile_mref(cPar.file_mref, S, Svar, s_ref, cPar.ni_ref); } //need to obtain a common set of SNPs between beta file and the genotype file; these are saved in mapRS2wA and mapRS2wK //normalize the weight in mapRS2wK to have an average of one; each element of mapRS2wA is 1 set setSnps_beta; ReadFile_snps_header (cPar.file_beta, setSnps_beta); //obtain the weights for wA, which contains the SNP weights for SNPs used in the model map mapRS2wK; cPar.ObtainWeight(setSnps_beta, mapRS2wK); //set up matrices and vector gsl_matrix *Xz=gsl_matrix_alloc (cPar.ni_test, cPar.n_vc); gsl_matrix *XWz=gsl_matrix_alloc (cPar.ni_test, cPar.n_vc); gsl_matrix *XtXWz=gsl_matrix_alloc (mapRS2wK.size(), cPar.n_vc*cPar.n_vc); gsl_vector *w=gsl_vector_alloc (mapRS2wK.size()); gsl_vector *w1=gsl_vector_alloc (mapRS2wK.size()); gsl_vector *z=gsl_vector_alloc (mapRS2wK.size()); gsl_vector *s_vec=gsl_vector_alloc (cPar.n_vc); vector vec_cat, vec_size; vector vec_z; map mapRS2z, mapRS2wA; map mapRS2A1; string file_str; //update s_vec, the number of snps in each category for (size_t i=0; i::const_iterator it=mapRS2wK.begin(); it!=mapRS2wK.end(); ++it) { vec_size[cPar.mapRS2cat[it->first]]++; } for (size_t i=0; i::const_iterator it=mapRS2wK.begin(); it!=mapRS2wK.end(); ++it) { mapRS2wA[it->first]=1; } } else { cPar.UpdateWeight (0, mapRS2wK, cPar.ni_test, s_vec, mapRS2wA); } //read in z-scores based on allele 0, and save that into a vector ReadFile_beta (cPar.file_beta, mapRS2wA, mapRS2A1, mapRS2z); //update snp indicator, save weights to w, save z-scores to vec_z, save category label to vec_cat //sign of z is determined by matching alleles cPar.UpdateSNPnZ (mapRS2wA, mapRS2A1, mapRS2z, w, z, vec_cat); //compute an n by k matrix of X_iWz cout<<"Calculating Xz ... "<size1, cPar.n_cvt); gsl_matrix *B=gsl_matrix_alloc (Y->size2, W->size2); //B is a d by c matrix gsl_matrix *se_B=gsl_matrix_alloc (Y->size2, W->size2); gsl_matrix *G=gsl_matrix_alloc (Y->size1, Y->size1); gsl_matrix *U=gsl_matrix_alloc (Y->size1, Y->size1); gsl_matrix *UtW=gsl_matrix_alloc (Y->size1, W->size2); gsl_matrix *UtY=gsl_matrix_alloc (Y->size1, Y->size2); gsl_vector *eval=gsl_vector_alloc (Y->size1); gsl_vector *env=gsl_vector_alloc (Y->size1); gsl_vector *weight=gsl_vector_alloc (Y->size1); //set covariates matrix W and phenotype matrix Y //an intercept should be included in W, cPar.CopyCvtPhen (W, Y, 0); if (!cPar.file_gxe.empty()) {cPar.CopyGxe (env);} //read relatedness matrix G if (!(cPar.file_kin).empty()) { ReadFile_kin (cPar.file_kin, cPar.indicator_idv, cPar.mapID2num, cPar.k_mode, cPar.error, G); if (cPar.error==true) {cout<<"error! fail to read kinship/relatedness file. "<size1; i++) { wi=gsl_vector_get(weight, i); for (size_t j=i; jsize2; j++) { wj=gsl_vector_get(weight, j); d=gsl_matrix_get(G, i, j); if (wi<=0 || wj<=0) {d=0;} else {d/=sqrt(wi*wj);} gsl_matrix_set(G, i, j, d); if (j!=i) {gsl_matrix_set(G, j, i, d);} } } } //eigen-decomposition and calculate trace_G cout<<"Start Eigen-Decomposition..."<size1; i++) { wi=gsl_vector_get(weight, i); if (wi<=0) {wi=0;} else {wi=sqrt(wi);} gsl_vector_view Urow=gsl_matrix_row (U, i); gsl_vector_scale (&Urow.vector, wi); } } cPar.trace_G=0.0; for (size_t i=0; isize; i++) { if (gsl_vector_get (eval, i)<1e-10) {gsl_vector_set (eval, i, 0);} cPar.trace_G+=gsl_vector_get (eval, i); } cPar.trace_G/=(double)eval->size; cPar.time_eigen=(clock()-time_start)/(double(CLOCKS_PER_SEC)*60.0); } else { ReadFile_eigenU (cPar.file_ku, cPar.error, U); if (cPar.error==true) {cout<<"error! fail to read the U file. "<size; i++) { if (gsl_vector_get(eval, i)<1e-10) {gsl_vector_set(eval, i, 0);} cPar.trace_G+=gsl_vector_get(eval, i); } cPar.trace_G/=(double)eval->size; } if (cPar.a_mode==31) { cPar.WriteMatrix(U, "eigenU"); cPar.WriteVector(eval, "eigenD"); } else if (!cPar.file_gene.empty() ) { //calculate UtW and Uty CalcUtX (U, W, UtW); CalcUtX (U, Y, UtY); LMM cLmm; cLmm.CopyFromParam(cPar); gsl_vector_view Y_col=gsl_matrix_column (Y, 0); gsl_vector_view UtY_col=gsl_matrix_column (UtY, 0); cLmm.AnalyzeGene (U, eval, UtW, &UtY_col.vector, W, &Y_col.vector); //y is the predictor, not the phenotype cLmm.WriteFiles(); cLmm.CopyToParam(cPar); } else { //calculate UtW and Uty CalcUtX (U, W, UtW); CalcUtX (U, Y, UtY); //calculate REMLE/MLE estimate and pve for univariate model if (cPar.n_ph==1) { gsl_vector_view beta=gsl_matrix_row (B, 0); gsl_vector_view se_beta=gsl_matrix_row (se_B, 0); gsl_vector_view UtY_col=gsl_matrix_column (UtY, 0); CalcLambda ('L', eval, UtW, &UtY_col.vector, cPar.l_min, cPar.l_max, cPar.n_region, cPar.l_mle_null, cPar.logl_mle_H0); CalcLmmVgVeBeta (eval, UtW, &UtY_col.vector, cPar.l_mle_null, cPar.vg_mle_null, cPar.ve_mle_null, &beta.vector, &se_beta.vector); cPar.beta_mle_null.clear(); cPar.se_beta_mle_null.clear(); for (size_t i=0; isize2; i++) { cPar.beta_mle_null.push_back(gsl_matrix_get(B, 0, i) ); cPar.se_beta_mle_null.push_back(gsl_matrix_get(se_B, 0, i) ); } CalcLambda ('R', eval, UtW, &UtY_col.vector, cPar.l_min, cPar.l_max, cPar.n_region, cPar.l_remle_null, cPar.logl_remle_H0); CalcLmmVgVeBeta (eval, UtW, &UtY_col.vector, cPar.l_remle_null, cPar.vg_remle_null, cPar.ve_remle_null, &beta.vector, &se_beta.vector); cPar.beta_remle_null.clear(); cPar.se_beta_remle_null.clear(); for (size_t i=0; isize2; i++) { cPar.beta_remle_null.push_back(gsl_matrix_get(B, 0, i) ); cPar.se_beta_remle_null.push_back(gsl_matrix_get(se_B, 0, i) ); } CalcPve (eval, UtW, &UtY_col.vector, cPar.l_remle_null, cPar.trace_G, cPar.pve_null, cPar.pve_se_null); cPar.PrintSummary(); //calculate and output residuals if (cPar.a_mode==5) { gsl_vector *Utu_hat=gsl_vector_alloc (Y->size1); gsl_vector *Ute_hat=gsl_vector_alloc (Y->size1); gsl_vector *u_hat=gsl_vector_alloc (Y->size1); gsl_vector *e_hat=gsl_vector_alloc (Y->size1); gsl_vector *y_hat=gsl_vector_alloc (Y->size1); //obtain Utu and Ute gsl_vector_memcpy (y_hat, &UtY_col.vector); gsl_blas_dgemv (CblasNoTrans, -1.0, UtW, &beta.vector, 1.0, y_hat); double d, u, e; for (size_t i=0; isize; i++) { d=gsl_vector_get (eval, i); u=cPar.l_remle_null*d/(cPar.l_remle_null*d+1.0)*gsl_vector_get(y_hat, i); e=1.0/(cPar.l_remle_null*d+1.0)*gsl_vector_get(y_hat, i); gsl_vector_set (Utu_hat, i, u); gsl_vector_set (Ute_hat, i, e); } //obtain u and e gsl_blas_dgemv (CblasNoTrans, 1.0, U, Utu_hat, 0.0, u_hat); gsl_blas_dgemv (CblasNoTrans, 1.0, U, Ute_hat, 0.0, e_hat); //output residuals cPar.WriteVector(u_hat, "residU"); cPar.WriteVector(e_hat, "residE"); gsl_vector_free(u_hat); gsl_vector_free(e_hat); gsl_vector_free(y_hat); } } //Fit LMM or mvLMM if (cPar.a_mode==1 || cPar.a_mode==2 || cPar.a_mode==3 || cPar.a_mode==4) { if (cPar.n_ph==1) { LMM cLmm; cLmm.CopyFromParam(cPar); gsl_vector_view Y_col=gsl_matrix_column (Y, 0); gsl_vector_view UtY_col=gsl_matrix_column (UtY, 0); if (!cPar.file_bfile.empty()) { if (cPar.file_gxe.empty()) { cLmm.AnalyzePlink (U, eval, UtW, &UtY_col.vector, W, &Y_col.vector); } else { cLmm.AnalyzePlinkGXE (U, eval, UtW, &UtY_col.vector, W, &Y_col.vector, env); } } // WJA added else if(!cPar.file_oxford.empty()) { cLmm.Analyzebgen (U, eval, UtW, &UtY_col.vector, W, &Y_col.vector); } else { if (cPar.file_gxe.empty()) { cLmm.AnalyzeBimbam (U, eval, UtW, &UtY_col.vector, W, &Y_col.vector); } else { cLmm.AnalyzeBimbamGXE (U, eval, UtW, &UtY_col.vector, W, &Y_col.vector, env); } } cLmm.WriteFiles(); cLmm.CopyToParam(cPar); } else { MVLMM cMvlmm; cMvlmm.CopyFromParam(cPar); if (!cPar.file_bfile.empty()) { if (cPar.file_gxe.empty()) { cMvlmm.AnalyzePlink (U, eval, UtW, UtY); } else { cMvlmm.AnalyzePlinkGXE (U, eval, UtW, UtY, env); } } else if(!cPar.file_oxford.empty()) { cMvlmm.Analyzebgen (U, eval, UtW, UtY); } else { if (cPar.file_gxe.empty()) { cMvlmm.AnalyzeBimbam (U, eval, UtW, UtY); } else { cMvlmm.AnalyzeBimbamGXE (U, eval, UtW, UtY, env); } } cMvlmm.WriteFiles(); cMvlmm.CopyToParam(cPar); } } } //release all matrices and vectors gsl_matrix_free (Y); gsl_matrix_free (W); gsl_matrix_free(B); gsl_matrix_free(se_B); gsl_matrix_free (G); gsl_matrix_free (U); gsl_matrix_free (UtW); gsl_matrix_free (UtY); gsl_vector_free (eval); gsl_vector_free (env); } //BSLMM if (cPar.a_mode==11 || cPar.a_mode==12 || cPar.a_mode==13) { gsl_vector *y=gsl_vector_alloc (cPar.ni_test); gsl_matrix *W=gsl_matrix_alloc (y->size, cPar.n_cvt); gsl_matrix *G=gsl_matrix_alloc (y->size, y->size); gsl_matrix *UtX=gsl_matrix_alloc (y->size, cPar.ns_test); //set covariates matrix W and phenotype vector y //an intercept should be included in W, cPar.CopyCvtPhen (W, y, 0); //center y, even for case/control data cPar.pheno_mean=CenterVector(y); //run bvsr if rho==1 if (cPar.rho_min==1 && cPar.rho_max==1) { //read genotypes X (not UtX) cPar.ReadGenotypes (UtX, G, false); //perform BSLMM analysis BSLMM cBslmm; cBslmm.CopyFromParam(cPar); time_start=clock(); cBslmm.MCMC(UtX, y); cPar.time_opt=(clock()-time_start)/(double(CLOCKS_PER_SEC)*60.0); cBslmm.CopyToParam(cPar); //else, if rho!=1 } else { gsl_matrix *U=gsl_matrix_alloc (y->size, y->size); gsl_vector *eval=gsl_vector_alloc (y->size); gsl_matrix *UtW=gsl_matrix_alloc (y->size, W->size2); gsl_vector *Uty=gsl_vector_alloc (y->size); //read relatedness matrix G if (!(cPar.file_kin).empty()) { cPar.ReadGenotypes (UtX, G, false); //read relatedness matrix G ReadFile_kin (cPar.file_kin, cPar.indicator_idv, cPar.mapID2num, cPar.k_mode, cPar.error, G); if (cPar.error==true) {cout<<"error! fail to read kinship/relatedness file. "<size; i++) { if (gsl_vector_get (eval, i)<1e-10) {gsl_vector_set (eval, i, 0);} cPar.trace_G+=gsl_vector_get (eval, i); } cPar.trace_G/=(double)eval->size; cPar.time_eigen=(clock()-time_start)/(double(CLOCKS_PER_SEC)*60.0); //calculate UtW and Uty CalcUtX (U, W, UtW); CalcUtX (U, y, Uty); //calculate REMLE/MLE estimate and pve CalcLambda ('L', eval, UtW, Uty, cPar.l_min, cPar.l_max, cPar.n_region, cPar.l_mle_null, cPar.logl_mle_H0); CalcLambda ('R', eval, UtW, Uty, cPar.l_min, cPar.l_max, cPar.n_region, cPar.l_remle_null, cPar.logl_remle_H0); CalcPve (eval, UtW, Uty, cPar.l_remle_null, cPar.trace_G, cPar.pve_null, cPar.pve_se_null); cPar.PrintSummary(); //Creat and calcualte UtX, use a large memory cout<<"Calculating UtX..."<size, cPar.n_cvt); gsl_matrix *G=gsl_matrix_alloc (y->size, y->size); gsl_matrix *UtX=gsl_matrix_alloc (y->size, cPar.ns_test); //set covariates matrix W and phenotype vector y //an intercept should be included in W, cPar.CopyCvtPhen (W, y, 0); //center y, even for case/control data cPar.pheno_mean=CenterVector(y); //run bvsr if rho==1 if (cPar.rho_min==1 && cPar.rho_max==1) { //read genotypes X (not UtX) cPar.ReadGenotypes (UtX, G, false); //perform BSLMM analysis BSLMM cBslmm; cBslmm.CopyFromParam(cPar); time_start=clock(); cBslmm.MCMC(UtX, y); cPar.time_opt=(clock()-time_start)/(double(CLOCKS_PER_SEC)*60.0); cBslmm.CopyToParam(cPar); //else, if rho!=1 } else { gsl_matrix *U=gsl_matrix_alloc (y->size, y->size); gsl_vector *eval=gsl_vector_alloc (y->size); gsl_matrix *UtW=gsl_matrix_alloc (y->size, W->size2); gsl_vector *Uty=gsl_vector_alloc (y->size); //read relatedness matrix G if (!(cPar.file_kin).empty()) { cPar.ReadGenotypes (UtX, G, false); //read relatedness matrix G ReadFile_kin (cPar.file_kin, cPar.indicator_idv, cPar.mapID2num, cPar.k_mode, cPar.error, G); if (cPar.error==true) {cout<<"error! fail to read kinship/relatedness file. "<size; i++) { if (gsl_vector_get (eval, i)<1e-10) {gsl_vector_set (eval, i, 0);} cPar.trace_G+=gsl_vector_get (eval, i); } cPar.trace_G/=(double)eval->size; cPar.time_eigen=(clock()-time_start)/(double(CLOCKS_PER_SEC)*60.0); //calculate UtW and Uty CalcUtX (U, W, UtW); CalcUtX (U, y, Uty); //calculate REMLE/MLE estimate and pve CalcLambda ('L', eval, UtW, Uty, cPar.l_min, cPar.l_max, cPar.n_region, cPar.l_mle_null, cPar.logl_mle_H0); CalcLambda ('R', eval, UtW, Uty, cPar.l_min, cPar.l_max, cPar.n_region, cPar.l_remle_null, cPar.logl_remle_H0); CalcPve (eval, UtW, Uty, cPar.l_remle_null, cPar.trace_G, cPar.pve_null, cPar.pve_se_null); cPar.PrintSummary(); //Creat and calcualte UtX, use a large memory cout<<"Calculating UtX..."< vec_rs; vector vec_sa2, vec_sb2, wab; vector > > BF; //read hyp and bf files (functions defined in BSLMMDAP) ReadFile_hyb (cPar.file_hyp, vec_sa2, vec_sb2, wab); ReadFile_bf (cPar.file_bf, vec_rs, BF); cPar.ns_test=vec_rs.size(); if (wab.size()!=BF[0][0].size()) {cout<<"error! hyp and bf files dimension do not match"<tm_month<<":"<tm_day":"<tm_hour<<":"<tm_min<1) { outfile<<"## total pve = "<1) { outfile<<"## total pve = "<1) { outfile<<"## total pve = "<=1 && cPar.a_mode<=4) || (cPar.a_mode>=51 && cPar.a_mode<=54) ) { outfile<<"## time on optimization = "<