11#ifndef CAL_R_OVERLAP_R_H
22#define CAL_R_OVERLAP_R_H
33
4- #include " source_lcao/module_ri/abfs-vector3_order.h"
54#include " source_base/sph_bessel_recursive.h"
65#include " source_base/vector3.h"
76#include " source_base/ylm.h"
1413#include " source_lcao/center2_orb-orb11.h"
1514#include " source_lcao/center2_orb-orb21.h"
1615#include " source_lcao/center2_orb.h"
16+ #include " source_lcao/module_ri/abfs-vector3_order.h"
1717
1818#include < map>
1919#include < set>
@@ -31,45 +31,39 @@ class cal_r_overlap_R
3131 double sparse_threshold = 1e-10 ;
3232 bool binary = false ;
3333
34- void init (const UnitCell& ucell,const Parallel_Orbitals& pv, const LCAO_Orbitals& orb);
35- void init_nonlocal (const UnitCell& ucell,const Parallel_Orbitals& pv, const LCAO_Orbitals& orb);
36- ModuleBase::Vector3<double > get_psi_r_psi (
37- const ModuleBase::Vector3<double >& R1 ,
38- const int & T1 ,
39- const int & L1 ,
40- const int & m1,
41- const int & N1 ,
42- const ModuleBase::Vector3<double >& R2 ,
43- const int & T2 ,
44- const int & L2 ,
45- const int & m2,
46- const int & N2
47- );
48- ModuleBase::Vector3<double > get_psi_r_gradpsi (
49- const ModuleBase::Vector3<double >& R1 ,
50- const int & T1 ,
51- const int & L1 ,
52- const int & m1,
53- const int & N1 ,
54- const ModuleBase::Vector3<double >& R2 ,
55- const int & T2 ,
56- const int & L2 ,
57- const int & m2,
58- const int & N2 ,
59- const ModuleBase::Vector3<double >& Efield,
60- const ModuleBase::Vector3<double >& dR
61- );
62- void get_psi_r_beta (
63- const UnitCell& ucell,
64- std::vector<std::vector<double >>& nlm,
65- const ModuleBase::Vector3<double >& R1 ,
66- const int & T1 ,
67- const int & L1 ,
68- const int & m1,
69- const int & N1 ,
70- const ModuleBase::Vector3<double >& R2 ,
71- const int & T2
72- );
34+ void init (const UnitCell& ucell, const Parallel_Orbitals& pv, const LCAO_Orbitals& orb);
35+ void init_nonlocal (const UnitCell& ucell, const Parallel_Orbitals& pv, const LCAO_Orbitals& orb);
36+ ModuleBase::Vector3<double > get_psi_r_psi (const ModuleBase::Vector3<double >& R1 ,
37+ const int & T1 ,
38+ const int & L1 ,
39+ const int & m1,
40+ const int & N1 ,
41+ const ModuleBase::Vector3<double >& R2 ,
42+ const int & T2 ,
43+ const int & L2 ,
44+ const int & m2,
45+ const int & N2 );
46+ ModuleBase::Vector3<double > get_psi_r_gradpsi (const ModuleBase::Vector3<double >& R1 ,
47+ const int & T1 ,
48+ const int & L1 ,
49+ const int & m1,
50+ const int & N1 ,
51+ const ModuleBase::Vector3<double >& R2 ,
52+ const int & T2 ,
53+ const int & L2 ,
54+ const int & m2,
55+ const int & N2 ,
56+ const ModuleBase::Vector3<double >& Efield,
57+ const ModuleBase::Vector3<double >& dR);
58+ void get_psi_r_beta (const UnitCell& ucell,
59+ std::vector<std::vector<double >>& nlm,
60+ const ModuleBase::Vector3<double >& R1 ,
61+ const int & T1 ,
62+ const int & L1 ,
63+ const int & m1,
64+ const int & N1 ,
65+ const ModuleBase::Vector3<double >& R2 ,
66+ const int & T2 );
7367 void out_rR (const UnitCell& ucell, const Grid_Driver& gd, const int & istep, const int precision = 16 );
7468 void out_rR_other (const UnitCell& ucell,
7569 const int & istep,
@@ -78,8 +72,8 @@ class cal_r_overlap_R
7872
7973 private:
8074 void initialize_orb_table (const UnitCell& ucell, const LCAO_Orbitals& orb);
81- void construct_orbs_and_orb_r (const UnitCell& ucell,const LCAO_Orbitals& orb);
82- void construct_orbs_and_nonlocal_and_orb_r (const UnitCell& ucell,const LCAO_Orbitals& orb);
75+ void construct_orbs_and_orb_r (const UnitCell& ucell, const LCAO_Orbitals& orb);
76+ void construct_orbs_and_nonlocal_and_orb_r (const UnitCell& ucell, const LCAO_Orbitals& orb);
8377
8478 std::vector<int > iw2ia;
8579 std::vector<int > iw2iL;
@@ -94,25 +88,15 @@ class cal_r_overlap_R
9488 std::vector<std::vector<std::vector<Numerical_Orbital_Lm>>> orbs;
9589 std::vector<std::vector<Numerical_Orbital_Lm>> orbs_nonlocal;
9690
97- std::map<
98- size_t ,
99- std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , Center2_Orb::Orb11>>>>>>
91+ std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , Center2_Orb::Orb11>>>>>>
10092 center2_orb11;
10193
102- std::map<
103- size_t ,
104- std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , Center2_Orb::Orb21>>>>>>
94+ std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , Center2_Orb::Orb21>>>>>>
10595 center2_orb21_r;
10696
107- std::map<
108- size_t ,
109- std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , Center2_Orb::Orb11>>>>>
110- center2_orb11_nonlocal;
97+ std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , Center2_Orb::Orb11>>>>> center2_orb11_nonlocal;
11198
112- std::map<
113- size_t ,
114- std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , Center2_Orb::Orb21>>>>>
115- center2_orb21_r_nonlocal;
99+ std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , std::map<size_t , Center2_Orb::Orb21>>>>> center2_orb21_r_nonlocal;
116100
117101 const Parallel_Orbitals* ParaV = nullptr ;
118102};
0 commit comments