add error handling

Author: root

dpgen/generator/lib/abacus_scf.py

@@ -218,6 +218,23 @@ def make_abacus_scf_stru(
     if type_map is None:
         type_map = sys_data_copy["atom_names"]
 
+    missing_atoms = set(sys_data_copy["atom_names"]) - set(type_map)
+    if len(missing_atoms) > 0:
+        raise ValueError(
+            f"Some atoms in sys_data are not in type_map: {missing_atoms}. "
+            "Please provide a valid type_map."
+        )
+    
+    if len(fp_pp_files) != len(type_map):
+        raise ValueError(
+            "The length of fp_pp_files should be equal to the length of type_map."
+        )
+    if fp_orb_files is not None and len(fp_orb_files) != len(type_map):
+        raise ValueError(
+            "The length of fp_orb_files should be equal to the length of type_map."
+        )
+          
+    
     fp_pp_files = [
         os.path.join(pporb, fp_pp_files[type_map.index(atom_name)])
         for atom_name in sys_data_copy["atom_names"]