Fix: memory leak in sltk_grid.cpp (#2141)

Author: dyzheng

source/module_cell/module_neighbor/sltk_grid.cpp

@@ -57,24 +57,7 @@ Grid::Grid(const int &test_grid_in):test_grid(test_grid_in)
 Grid::~Grid()
 {
 	delete[] atomlink;
-	if (this->init_cell_flag)
-	{
-		for (int i = 0;i < this->dx;i++)
-		{
-			for (int j = 0;j < this->dy;j++)
-			{
-				delete[] this->Cell[i][j];
-			}
-		}
-
-		for (int i = 0;i < this->dx;i++)
-		{
-			delete[] this->Cell[i];
-		}
-
-		delete[] this->Cell;
-		this->init_cell_flag = false;
-	}
+	this->delete_Cell();
 
 }
 
@@ -102,6 +85,7 @@ void Grid::setMemberVariables(
 {
 	ModuleBase::TITLE("SLTK_Grid", "setMemberVariables");
 
+	this->delete_Cell();
 	// mohan add 2010-09-05
 	AdjacentSet::call_times = 0;
 

source/module_cell/module_neighbor/sltk_grid.h

@@ -88,6 +88,27 @@ class Grid
 	double cell_y_length;
 	double cell_z_length;
 	CellSet ***Cell; //dx , dy ,dz is cell number in each direction,respectly.
+	void delete_Cell() //it will replace by container soon!
+	{
+		if (this->init_cell_flag)
+		{
+			for (int i = 0;i < this->dx;i++)
+			{
+				for (int j = 0;j < this->dy;j++)
+				{
+					delete[] this->Cell[i][j];
+				}
+			}
+
+			for (int i = 0;i < this->dx;i++)
+			{
+				delete[] this->Cell[i];
+			}
+
+			delete[] this->Cell;
+			this->init_cell_flag = false;
+		}
+	}
 
 	double grid_length[3];
 	double vec1[3];