1. add class Parallel_LRI_Equally_Filter

Author: PeizeLin

include/RI/parallel/Parallel_LRI_Equally_Filter.h

@@ -0,0 +1,35 @@
+// ===================
+//  Author: Peize Lin
+//  date: 2023.03.07
+// ===================
+
+#pragma once
+
+#include "Parallel_LRI_Equally.h"
+
+namespace RI
+{
+
+template<typename TA, typename Tcell, std::size_t Ndim, typename Tdata>
+class Parallel_LRI_Equally_Filter: public Parallel_LRI_Equally<TA,Tcell,Ndim,Tdata>
+{
+public:
+	using TC = std::array<Tcell,Ndim>;
+	using TAC = std::pair<TA,TC>;
+	using Tatom_pos = std::array<double,Ndim>;		// tmp
+
+	const std::vector<TAC>& get_list_Ab2 (const TA &Aa01, const TAC &Aa2, const TAC &Ab01) const override
+	{
+		return this->list_Ab2_filter.at(Ab01);
+	}
+
+	void filter_Ab2 (const std::map<TA, std::map<TAC, Tensor<Tdata>>> &Ds_b);
+
+// private:
+public:
+	std::map<TAC,std::vector<TAC>> list_Ab2_filter;
+};
+
+}
+
+#include "Parallel_LRI_Equally_Filter.hpp"

include/RI/parallel/Parallel_LRI_Equally_Filter.hpp

@@ -0,0 +1,31 @@
+// ===================
+//  Author: Peize Lin
+//  date: 2023.03.07
+// ===================
+
+#pragma once
+
+#include "Parallel_LRI_Equally_Filter.h"
+
+namespace RI
+{
+
+template<typename TA, typename Tcell, std::size_t Ndim, typename Tdata>
+void Parallel_LRI_Equally_Filter<TA,Tcell,Ndim,Tdata>::filter_Ab2 (
+	const std::map<TA, std::map<TAC, Tensor<Tdata>>> &Ds_b)
+{
+	using namespace Array_Operator;
+	for(const TAC &Ab01 : this->list_Ab01)
+	{
+		for(const TAC &Ab2 : this->list_Ab2)
+		{
+			const Tensor<Tdata> D_b = Global_Func::find(
+				Ds_b,
+				Ab01.first, TAC{Ab2.first, (Ab2.second-Ab01.second)%this->period});
+			if(!D_b.empty())
+				this->list_Ab2_filter[Ab01].push_back(Ab2);
+		}
+	}
+}
+
+}

include/RI/ri/LRI-cal.hpp

@@ -24,6 +24,8 @@ void LRI<TA,Tcell,Ndim,Tdata>::cal(
 {
 	using namespace Array_Operator;
 
+	dynamic_cast<Parallel_LRI_Equally_Filter<TA,Tcell,Ndim,Tdata>*>(this->parallel.get())->filter_Ab2(this->Ds_ab[Label::ab::b]);
+
 	const bool flag_D_b_transpose = [&labels]() -> bool
 	{
 		for(const Label::ab_ab &label : labels)

include/RI/ri/LRI.h

@@ -9,7 +9,7 @@
 #include "../ri/Label.h"
 #include "../global/Tensor.h"
 #include "CS_Matrix.h"
-#include "../parallel/Parallel_LRI_Equally.h"
+#include "../parallel/Parallel_LRI_Equally_Filter.h"
 #include "RI_Tools.h"
 #include "../global/Global_Func-2.h"
 #include "Save_Load.h"
@@ -59,7 +59,7 @@ class LRI
 
 public:
 	std::shared_ptr<Parallel_LRI<TA,Tcell,Ndim,Tdata>>
-		parallel = std::make_shared<Parallel_LRI_Equally<TA,Tcell,Ndim,Tdata>>();
+		parallel = std::make_shared<Parallel_LRI_Equally_Filter<TA,Tcell,Ndim,Tdata>>();
 	std::unordered_map< Label::ab, RI_Tools::T_filter_func<Tdata> >
 		filter_funcs;
 	std::vector<std::function<Tdata (const Label::ab_ab &label, const TA &Aa01, const TAC &Aa2, const TAC &Ab01, const TAC &Ab2)>>