Fix: add CHECK_WARNING_QUIT function for reporting no plane wave error

Author: dyzheng

source/module_base/tool_quit.cpp

@@ -128,4 +128,25 @@ void WARNING_QUIT(const std::string &file,const std::string &description)
     QUIT();
 }
 
+
+//Input judgement and communicate , if any judgement is true, do WARNING_QUIT
+void CHECK_WARNING_QUIT(bool error_in, const std::string &file,const std::string &description)
+{
+	int error = (int)error_in;
+#ifdef __NORMAL
+// only for UT, do nothing here
+#else
+#ifdef __MPI
+	int error_max = error;
+	MPI_Reduce(&error, &error_max, 1, MPI_INT, MPI_MAX, 0, MPI_COMM_WORLD);
+	MPI_Bcast(&error_max, 1, MPI_INT, 0, MPI_COMM_WORLD);
+	error = error_max;
+#endif
+#endif
+	if(error)
+	{
+		WARNING_QUIT(file, description);
+	}
+}
+
 }

source/module_base/tool_quit.h

@@ -43,6 +43,14 @@ void QUIT(void);
  */
 void WARNING_QUIT(const std::string &file, const std::string &description);
 
+/**
+ * @brief Check, if true, WARNING_QUIT
+ *
+ * @param file The file where warning happens
+ * @param description The warning information
+ */
+void CHECK_WARNING_QUIT(bool error, const std::string &file,const std::string &description);
+
 } // namespace ModuleBase
 
 #endif

source/module_pw/pw_basis_k.cpp

@@ -80,6 +80,7 @@ void PW_Basis_K::setupIndGk()
                 ++ng;
             }
         }
+        ModuleBase::CHECK_WARNING_QUIT((ng==0), "PW_Basis_K::setupIndGk", "some cores have no plane waves!");
         this->npwk[ik] = ng;
         if(ng == 0)
         {