1. add ModuleBase::bcast_data_cereal() for bcasting arbitrary class

Author: PeizeLin

source/src_io/read_txt_input_process.cpp

@@ -8,12 +8,8 @@
 #include "src_io/read_txt_tools.h"
 #include "module_base/global_variable.h"
 
-#include <mpi.h>
-#include <sstream>
-
 #ifdef USE_CEREAL_SERIALIZATION
 #include "src_lcao/serialization_cereal.h"
-#include <cereal/archives/binary.hpp>
 #endif
 
 namespace Read_Txt_Input
@@ -130,30 +126,7 @@ namespace Read_Txt_Input
 	void Input_Process::bcast()
 	{
 #ifdef USE_CEREAL_SERIALIZATION
-		if(GlobalV::MY_RANK==0)
-		{
-			std::stringstream ss;
-			{
-				cereal::BinaryOutputArchive ar(ss);
-				ar(this->input.list);
-			}
-			const int size = ss.str().size();
-			MPI_Bcast( const_cast<int*>(&size), 1, MPI_INT, 0, MPI_COMM_WORLD );
-			MPI_Bcast( const_cast<char*>(ss.str().c_str()), size, MPI_CHAR, 0, MPI_COMM_WORLD ); 
-		}
-		else
-		{
-			int size;
-			MPI_Bcast( &size, 1, MPI_INT, 0, MPI_COMM_WORLD );
-			std::vector<char> c(size);
-			MPI_Bcast( c.data(), size, MPI_CHAR, 0, MPI_COMM_WORLD );   
-			std::stringstream ss;  
-			ss.rdbuf()->pubsetbuf(c.data(),size);
-			{
-				cereal::BinaryInputArchive ar(ss);
-				ar(this->input.list);
-			}
-		}
+		ModuleBase::bcast_data_cereal(this->input.list, MPI_COMM_WORLD, 0);
 #else
 #error Input_Process::bcast() needs cereal
 #endif

source/src_io/read_txt_stru.cpp

@@ -79,7 +79,6 @@ namespace Read_Txt_Stru
                     atoms_info.push_back(std::move(atom0));
                     atoms_info.insert(atoms_info.end(), stru_one.begin()+1, stru_one.end());
                 }
-                
             }
             else
             {

source/src_lcao/serialization_cereal.h

@@ -21,42 +21,79 @@ template<class Archive, typename T> void serialize( Archive & ar, Abfs::Vector3_
 
 namespace ModuleBase
 {
+	template<class Archive, typename T> void serialize( Archive & ar, Vector3<T> & v ){ ar(v.x); ar(v.y); ar(v.z); }
 
-template<class Archive, typename T> void serialize( Archive & ar, Vector3<T> & v ){ ar(v.x); ar(v.y); ar(v.z); }
-
-template<class Archive> void save( Archive & ar, const matrix & m )
-{
-	ar(m.nr);	ar(m.nc);
-	ar(cereal::binary_data(m.c, m.nr*m.nc*sizeof(double)));
-}
-template<class Archive> void load( Archive & ar, matrix & m )
-{
-	int nr, nc;
-	ar(nr);	ar(nc);
-	m.create(nr,nc);
-	ar(cereal::binary_data(m.c, m.nr*m.nc*sizeof(double)));
-}
+	template<class Archive> void save( Archive & ar, const matrix & m )
+	{
+		ar(m.nr);	ar(m.nc);
+		ar(cereal::binary_data(m.c, m.nr*m.nc*sizeof(double)));
+	}
+	template<class Archive> void load( Archive & ar, matrix & m )
+	{
+		int nr, nc;
+		ar(nr);	ar(nc);
+		m.create(nr,nc);
+		ar(cereal::binary_data(m.c, m.nr*m.nc*sizeof(double)));
+	}
 }
 
 namespace Read_Txt_Input
 {
-template<class Archive> void Input_Value::serialize( Archive & ar )
-{
-	ar( this->b );
-	ar( this->i );
-	ar( this->d );
-	ar( this->s );
+	template<class Archive> void Input_Value::serialize( Archive & ar )
+	{
+		ar( this->b );
+		ar( this->i );
+		ar( this->d );
+		ar( this->s );
+	}
+	template<class Archive> void Input_Item::serialize( Archive & ar )
+	{
+		ar( this->annotation );
+		ar( this->values );
+		ar( this->values_type );
+		ar( this->label );
+		ar( this->values_size_read );
+		ar( this->values_size_lower_limit );
+		ar( this->values_size_upper_limit );
+	}
 }
-template<class Archive> void Input_Item::serialize( Archive & ar )
+
+
+
+#include <mpi.h>
+#include <sstream>
+
+namespace ModuleBase
 {
-	ar( this->annotation );
-	ar( this->values );
-	ar( this->values_type );
-	ar( this->label );
-	ar( this->values_size_read );
-	ar( this->values_size_lower_limit );
-	ar( this->values_size_upper_limit );
-}
+	template<typename T>
+	void bcast_data_cereal(T &data, const MPI_Comm &mpi_comm, const int &rank_bcast)
+	{
+		int my_rank;	MPI_Comm_rank( mpi_comm, &my_rank );
+		if(my_rank==rank_bcast)
+		{
+			std::stringstream ss;
+			{
+				cereal::BinaryOutputArchive ar(ss);
+				ar(data);
+			}
+			const int size = ss.str().size();
+			MPI_Bcast( const_cast<int*>(&size), 1, MPI_INT, rank_bcast, mpi_comm );
+			MPI_Bcast( const_cast<char*>(ss.str().c_str()), size, MPI_CHAR, rank_bcast, mpi_comm ); 
+		}
+		else
+		{
+			int size;
+			MPI_Bcast( &size, 1, MPI_INT, rank_bcast, mpi_comm );
+			std::vector<char> c(size);
+			MPI_Bcast( c.data(), size, MPI_CHAR, rank_bcast, mpi_comm );   
+			std::stringstream ss;  
+			ss.rdbuf()->pubsetbuf(c.data(),size);
+			{
+				cereal::BinaryInputArchive ar(ss);
+				ar(data);
+			}
+		}	
+	}
 }
 
 #endif