change ucell in module_io/write_dmr.cpp

Author: A-006

source/module_esolver/esolver_ks_lcao.cpp

@@ -991,6 +991,8 @@ void ESolver_KS_LCAO<TK, TR>::after_scf(UnitCell& ucell, const int istep)
                         PARAM.inp.out_dm1,
                         false,
                         PARAM.inp.out_app_flag,
+                        ucell.get_iat2iwt(),
+                        &ucell.nat,
                         istep);
 
     //! 7) write density matrix

source/module_io/write_dmr.cpp

@@ -55,6 +55,8 @@ void write_dmr(const std::vector<hamilt::HContainer<double>*> dmr,
                const bool out_csr,
                const bool out_npz,
                const bool append,
+               const int* iat2iwt,
+               const int* nat,
                const int istep)
 {
     if (!out_csr && !out_npz)
@@ -73,7 +75,7 @@ void write_dmr(const std::vector<hamilt::HContainer<double>*> dmr,
             Parallel_Orbitals serialV;
             serialV.init(nbasis, nbasis, nbasis, paraV.comm());
             serialV.set_serial(nbasis, nbasis);
-            serialV.set_atomic_trace(GlobalC::ucell.get_iat2iwt(), GlobalC::ucell.nat, nbasis);
+            serialV.set_atomic_trace(iat2iwt, *nat, nbasis);
             hamilt::HContainer<double> dm_serial(&serialV);
             hamilt::gatherParallels(*dmr[ispin], &dm_serial, 0);
 #else

source/module_io/write_dmr.h

@@ -45,6 +45,8 @@ void write_dmr(const std::vector<hamilt::HContainer<double>*> dmr,
                const bool out_csr,
                const bool out_npz,
                const bool append,
+               const int* iat2iwt,
+               const int* nat,
                const int istep);
 } // namespace ModuleIO