Add test case 58_KP_LR_BSE

Author: Fisherd99

docs/advanced/input_files/input-main.md

@@ -2370,7 +2370,7 @@
 - **Type**: String
 - **Availability**: *[`basis_type`](#basis_type)==lcao*
 - **Description**: The directory to save files for LibRPA.
-- **Default**: "./OUT.librpa/"
+- **Default**: "OUT.librpa"
 
 ### out_pchg
 

docs/parameters.yaml

@@ -3617,7 +3617,7 @@ parameters:
     type: String
     description: |
       The directory to save files for LibRPA.
-    default_value: "\"./OUT.librpa/\""
+    default_value: "\"OUT.librpa\""
     unit: ""
     availability: basis_type==lcao
   - name: out_pchg

source/source_cell/atom_spec.h

@@ -49,7 +49,7 @@ class Atom
     int stapos_wf = 0;                           ///< start position of wave functions
 
     std::string label;                            ///< atomic symbol
-    std::vector<ModuleBase::Vector3<double>> tau; ///< Cartesian coordinates of each atom in this type
+    std::vector<ModuleBase::Vector3<double>> tau; ///< Cartesian coordinates of each atom in this type, in unit of lat0
     std::vector<ModuleBase::Vector3<double>> dis; ///< direct displacements of each atom in this type in current step (liuyu modify 2023-03-22)
     std::vector<ModuleBase::Vector3<double>> taud; ///< Direct coordinates of each atom in this type
     std::vector<ModuleBase::Vector3<int>> boundary_shift; ///< record for periodic boundary adjustment

source/source_cell/klist.h

@@ -14,9 +14,9 @@
 class K_Vectors
 {
 public:
-    std::vector<ModuleBase::Vector3<double>> kvec_c; ///< Cartesian coordinates of k points
+    std::vector<ModuleBase::Vector3<double>> kvec_c; ///< Cartesian coordinates of k points in units of 2*pi/lat0
     std::vector<ModuleBase::Vector3<double>> kvec_d; ///< Direct coordinates of k points
-    std::vector<ModuleBase::Vector3<double>> kvec_c_full; ///< Cartesian coordinates of full k mesh match with nkstot_full
+    std::vector<ModuleBase::Vector3<double>> kvec_c_full; ///< Full-mesh (nkstot_full) Cartesian coordinates of k points, in units of 2*pi/lat0
 
     std::vector<double> wk; ///< wk, weight of k points
 
@@ -328,4 +328,4 @@ class K_Vectors
     friend void KVectorUtils::kvec_mpi_k(K_Vectors& kvec);
 #endif
 };
-#endif // KVECT_H
+#endif // KVECT_H

source/source_cell/unitcell_data.h

@@ -24,7 +24,7 @@ struct Lattice
     double omega = 0.0;                ///< the volume of the unit cell
     std::vector<int> lat_axis_free{0, 0, 0};      ///< whether each lattice axis (a,b,c) is allowed to relax (0=fixed, 1=free)
 
-    ModuleBase::Matrix3 latvec = ModuleBase::Matrix3();           ///< Unitcell lattice vectors
+    ModuleBase::Matrix3 latvec = ModuleBase::Matrix3();           ///< Unit-cell lattice vectors in units of lat0
     ModuleBase::Vector3<double> a1, a2, a3;                       ///< Same as latvec, just at another form
     ModuleBase::Vector3<double> latcenter;                        ///< (a1+a2+a3)/2 the center of vector
     ModuleBase::Matrix3 latvec_supercell = ModuleBase::Matrix3(); ///< Supercell lattice vectors

source/source_io/module_parameter/input_parameter.h

@@ -465,7 +465,7 @@ struct Input_para
     bool restart_save = false;               ///< restart //Peize Lin add 2020-04-04
     bool rpa = false;                        ///< rpa calculation
     bool rpa_out_vel = false;                ///< whether to output velocity matrix for librpa
-    std::string rpa_outdir = "./OUT.librpa/";///< output directory for librpa
+    std::string rpa_outdir = "OUT.librpa";   ///< output directory for librpa
     std::vector<int> out_pchg = {};          ///< specify the bands to be calculated for partial charge
     std::vector<int> out_wfc_norm = {};      ///< specify the bands to be calculated for norm of wfc
     std::vector<int> out_wfc_re_im = {};     ///< specify the bands to be calculated for real and imaginary parts of wfc

source/source_io/module_parameter/read_inp_out.cpp

@@ -1481,7 +1481,7 @@ If EXX(exact exchange) is calculated (i.e. dft_fuctional==hse/hf/pbe0/scan0 or r
         item.category = "Output information";
         item.type = "String";
         item.description = "The directory to save files for LibRPA.";
-        item.default_value = "\"./OUT.librpa/\"";
+        item.default_value = "\"OUT.librpa\"";
         item.unit = "";
         item.set_availability("basis_type==lcao");
         read_sync_string(input.rpa_outdir);

source/source_io/test/read_input_ptest.cpp

@@ -362,7 +362,7 @@ TEST_F(InputParaTest, ParaRead)
     EXPECT_FALSE(param.inp.dft_plus_dmft);
     EXPECT_FALSE(param.inp.rpa);
     EXPECT_FALSE(param.inp.rpa_out_vel);
-    EXPECT_EQ(param.inp.rpa_outdir, "./OUT.librpa/");
+    EXPECT_EQ(param.inp.rpa_outdir, "OUT.librpa/");
     EXPECT_EQ(param.inp.imp_sol, 0);
     EXPECT_DOUBLE_EQ(param.inp.eb_k, 80.0);
     EXPECT_DOUBLE_EQ(param.inp.tau, 1.0798 * 1e-5);

source/source_lcao/module_bse/hamilt_bse.cpp

@@ -142,13 +142,13 @@ HamiltBSE<T>::HamiltBSE(const int& nspin,
 
 template <typename T>
 void HamiltBSE<T>::cal_V_for_A(){
-    ModuleBase::TITLE("HamiltBSE", "cal_V_for_A");
-    ModuleBase::timer::start("HamiltBSE", "cal_V_for_A");
-    std::cout<<"in cal_V_for_A"<<std::endl;
     if (!this->VA_local.empty()) {
         std::cout<< "V for A has been calculated, skip." <<std::endl;
         return;
     }
+    ModuleBase::TITLE("HamiltBSE", "cal_V_for_A");
+    ModuleBase::timer::start("HamiltBSE", "cal_V_for_A");
+    std::cout<<"in cal_V_for_A"<<std::endl;
     BSE_Util::print_mem_estimate("V matrix of A", this->pA.get_local_size(), sizeof(T));
     this->VA_local.resize(this->pA.get_local_size(), 0.0);
     if (this->ri_hartree_benchmark == "aims" || this->ri_hartree_benchmark == "abacus") {
@@ -171,13 +171,13 @@ void HamiltBSE<T>::cal_V_for_A(){
 
 template <typename T>
 void HamiltBSE<T>::cal_V_for_B(){
-    ModuleBase::TITLE("HamiltBSE", "cal_V_for_B");
-    ModuleBase::timer::start("HamiltBSE", "cal_V_for_B");
-    std::cout<<"in cal_V_for_B"<<std::endl;
     if (!this->VB_local.empty()) {
         std::cout<< "V for B has been calculated, skip." <<std::endl;
         return;
     }
+    ModuleBase::TITLE("HamiltBSE", "cal_V_for_B");
+    ModuleBase::timer::start("HamiltBSE", "cal_V_for_B");
+    std::cout<<"in cal_V_for_B"<<std::endl;
     BSE_Util::print_mem_estimate("V matrix of B", this->pA.get_local_size(), sizeof(T));
     this->VB_local.resize(this->pA.get_local_size(), 0.0);
     if (this->ri_hartree_benchmark == "aims" || this->ri_hartree_benchmark == "abacus") {
@@ -200,13 +200,13 @@ void HamiltBSE<T>::cal_V_for_B(){
 
 template <typename T>
 void HamiltBSE<T>::cal_W_for_A(){
-    ModuleBase::TITLE("HamiltBSE", "cal_W_for_A");
-    ModuleBase::timer::start("HamiltBSE", "cal_W_for_A");
-    std::cout<<"in cal_W_for_A"<<std::endl;
     if (!this->WA_local.empty()) {
         std::cout<< "W for A has been calculated, skip." <<std::endl;
         return;
     }
+    ModuleBase::TITLE("HamiltBSE", "cal_W_for_A");
+    ModuleBase::timer::start("HamiltBSE", "cal_W_for_A");
+    std::cout<<"in cal_W_for_A"<<std::endl;
     BSE_Util::print_mem_estimate("W matrix of A", this->pA.get_local_size(), sizeof(T));
     this->WA_local.resize(this->pA.get_local_size(), 0.0);
     this->mo_lri.cal_W_for_A(this->WA_local, this->pA);    
@@ -219,13 +219,13 @@ void HamiltBSE<T>::cal_W_for_A(){
 
 template <typename T>
 void HamiltBSE<T>::cal_W_for_B(){
-    ModuleBase::TITLE("HamiltBSE", "cal_W_for_B");
-    ModuleBase::timer::start("HamiltBSE", "cal_W_for_B");
-    std::cout<<"in cal_W_for_B"<<std::endl;
     if (!this->WB_local.empty()) {
         std::cout<< "W for B has been calculated, skip." <<std::endl;
         return;
     }
+    ModuleBase::TITLE("HamiltBSE", "cal_W_for_B");
+    ModuleBase::timer::start("HamiltBSE", "cal_W_for_B");
+    std::cout<<"in cal_W_for_B"<<std::endl;
     BSE_Util::print_mem_estimate("W matrix of B", this->pA.get_local_size(), sizeof(T));
     this->WB_local.resize(this->pA.get_local_size(), 0.0);
     this->mo_lri.cal_W_for_B(this->WB_local, this->pA);
@@ -336,7 +336,7 @@ void HamiltBSE<T>::init_bse_matrix(const bool is_full, const int & st_index){
     }
     if (this->out_bse_ab)
     {
-        this->write_AB_matrix("A_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->BSE_A_local.data(), this->ndim, this->ndim);
+        this->write_AB_matrix(this->out_dir+"A_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->BSE_A_local.data(), this->ndim, this->ndim);
     }
 
     if (is_full)
@@ -351,7 +351,7 @@ void HamiltBSE<T>::init_bse_matrix(const bool is_full, const int & st_index){
         }
         if (this->out_bse_ab)
         {
-            this->write_AB_matrix("B_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->BSE_B_local.data(), this->ndim, this->ndim);
+            this->write_AB_matrix(this->out_dir+"B_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->BSE_B_local.data(), this->ndim, this->ndim);
         }
     }
 

source/source_lcao/module_lr/utils/lr_io.cpp

@@ -323,7 +323,7 @@ void read_librpa_eigenvectors(psi::Psi<TK>& wfc_ks,
                     wfc_ks.get_pointer(), 1, 1, const_cast<int*>(pmat.desc_wfc)/*nbasis×nbands*/,
                     pv_glb.blacs_ctxt);
 #else
-        BlasConnector::copy(nbands*nlocal, wfc_ks_global.get_pointer(), 1, wfc_ks.get_pointer(), 1);
+        BlasConnector::copy(nbands*nbasis, wfc_ks_global.get_pointer(), 1, wfc_ks.get_pointer(), 1);
 #endif
     }
     ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "read librpa eigenvectors.");
@@ -423,7 +423,7 @@ void read_librpa_eigenvectors_from_band_files(psi::Psi<TK>& wfc_ks,
                     wfc_ks.get_pointer(), 1, 1, const_cast<int*>(pmat.desc_wfc)/*nbasis×nbands*/,
                     pv_glb.blacs_ctxt);
 #else
-        BlasConnector::copy(nbands*nlocal, wfc_ks_global.get_pointer(), 1, wfc_ks.get_pointer(), 1);
+        BlasConnector::copy(nbands*nbasis, wfc_ks_global.get_pointer(), 1, wfc_ks.get_pointer(), 1);
 #endif
     }
     ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "read librpa eigenvectors.");