ion-CSP 2.0.4__py3-none-any.whl → 2.0.8__py3-none-any.whl

ion_CSP/read_mlp_density.py

@@ -1,4 +1,5 @@
 import os
+import csv
 import shutil
 import logging
 import subprocess
@@ -76,15 +77,30 @@ class ReadMlpDensity:
             os.makedirs(backup_dir, exist_ok=True)
             for item in os.listdir(self.max_density_dir):
                 shutil.move(os.path.join(self.max_density_dir, item), os.path.join(backup_dir, item))
-        
+        numbers, mlp_densities, mlp_energies = [], [], []
         os.makedirs(self.max_density_dir, exist_ok=True)
         for density, CONTCAR_filename in sorted_filename[:n_screen]:
             # 生成新的包含密度值的文件名，并重命名文件
             # 密度转换为字符串，保留4位小数
             density_str = f'{density:.4f}'
+            mlp_densities.append(density_str)
             # 保留 CONTCAR 的序数信息，方便回推检查
             number = CONTCAR_filename.split("_")[1]
+            numbers.append(number)
             OUTCAR_filename = f'OUTCAR_{number}'
+            try:
+                with open(f"{self.folder_dir}/OUTCAR_{number}") as mlp_out:
+                    lines = mlp_out.readlines()
+                    for line in lines:
+                        if "TOTEN" in line:
+                            values = line.split()
+                            mlp_energy = round(float(values[-2]), 2)
+            except FileNotFoundError:
+                logging.error(
+                    f"  No avalible MLP OUTCAR_{number} found"
+                )
+                mlp_energy = False
+            mlp_energies.append(mlp_energy)
             new_CONTCAR_filename = f'CONTCAR_{density_str}_{number}'
             new_OUTCAR_filename = f'OUTCAR_{density_str}_{number}'
             shutil.copy(f'{self.folder_dir}/{CONTCAR_filename}', f'{self.max_density_dir}/{new_CONTCAR_filename}')
@@ -92,6 +108,30 @@ class ReadMlpDensity:
             print(f'New CONTCAR and OUTCAR of {density_str}_{number} are renamed and saved')
             logging.info(f'New CONTCAR and OUTCAR of {density_str}_{number} are renamed and saved')
 
+        with open(
+            f"{self.max_density_dir}/mlp_density_energy.csv",
+            "w",
+            newline="",
+            encoding="utf-8",
+        ) as csv_file:
+            writer = csv.writer(csv_file)
+            header = [
+                "Number",
+                "MLP_E",
+                "MLP_Density",
+            ]
+            datas = list(
+                zip(
+                    numbers,
+                    mlp_energies,
+                    mlp_densities,
+                )
+            )
+            datas.sort(key=lambda x: -float(x[-1]))
+            writer.writerow(header)
+            for data in datas:
+                writer.writerow(data)
+
     def phonopy_processing_max_density(self, specific_directory :str = None):
         """
         Use phonopy to check and generate symmetric primitive cells, reducing the complexity of subsequent optimization calculations, and preventing pyxtal.from_random from generating double proportioned supercells. 
@@ -127,14 +167,18 @@ class ReadMlpDensity:
                 # 复制对应的OUTCAR文件到primitive_cell目录下
                 density_number = new_CONTCAR_filename.split("CONTCAR_")[1]
                 new_OUTCAR_filename = f'OUTCAR_{density_number}'
-                shutil.copy(f'{self.max_density_dir}/{new_OUTCAR_filename}', f'{self.primitive_cell_dir}/{new_OUTCAR_filename}')
+                shutil.copy(f'{self.phonopy_dir}/{new_OUTCAR_filename}', f'{self.primitive_cell_dir}/{new_OUTCAR_filename}')
+            shutil.copy(
+                f"{self.phonopy_dir}/mlp_density_energy.csv",
+                f"{self.primitive_cell_dir}/mlp_density_energy.csv",
+            )
             for_vasp_opt_dir = os.path.join(self.base_dir, '3_for_vasp_opt')
             if os.path.exists(for_vasp_opt_dir):
                 shutil.rmtree(for_vasp_opt_dir)
             shutil.copytree(self.primitive_cell_dir, for_vasp_opt_dir)
             logging.info('The phonopy processing has been completed!!\nThe symmetrized primitive cells have been saved in POSCAR format to the primitive_cell folder.\nThe output content of phonopy has been saved to the phonopy.log file in the same directory.')
             # 在 phonopy 成功进行对称化处理后，删除 2_mlp_optimized/max_density 文件夹以节省空间
-            shutil.rmtree(self.max_density_dir)
+            shutil.rmtree(self.phonopy_dir)
         except FileNotFoundError:
             logging.error(
                 "There are no CONTCAR structure files after screening.\nPlease check if the ions correspond to the crystals and adjust the screening criteria"

ion_CSP/task_manager.py

@@ -125,7 +125,7 @@ class TaskManager:
     def _display_tasks(self, tasks, total, current_page, total_pages, function):
         """标准化任务显示 - Standardized tasks display"""
         display = "logs" if function == "view" else "tasks" 
-        print(f"\033cPage {current_page}/{total_pages} ({total} {display})")
+        print(f"\033cPage {current_page}/{total_pages} ({total} {display})\n")
         if function == "kill":
             for i, task in enumerate(tasks, 1):
                 print(
@@ -138,11 +138,11 @@ class TaskManager:
                 )
         else:
             raise ValueError(f"Not supported function {function}. Available function: 'view' and 'kill' ")
-        print("\nPage {} of {}".format(current_page, total_pages))
         # 分页控制
         print("\nOptions:")
         if function == "view":
             print("n) Next page | p) Previous page | f) Filter | q) Quit")
+            print("Enter number to view log in detail")
         elif function == "kill":
             print("n) Next page | p) Previous page | f) Filter | k) Kill | q) Quit")
         else:
@@ -171,10 +171,11 @@ class TaskManager:
             elif function == "kill" and choice == "K":
                 try:
                     task_num = input("Enter task number to kill: ").strip()
-                    if not task_num.isdigit():
+                    if task_num.isdigit() and 1 <= int(task_num) <= 10:
+                        # 计算全局任务索引
+                        global_index = current_page * 10 + (int(task_num) - 1)
+                    else: 
                         raise ValueError
-                    # 计算全局任务索引
-                    global_index = current_page * 10 + (int(task_num) - 1)
                     if 0 <= global_index < len(tasks):
                         selected_index = global_index
                         confirm = input(
@@ -189,10 +190,10 @@ class TaskManager:
                     else:
                         print("Invalid task number")
                         input("\nPress Enter to continue...")
-                except ValueError:
+                except (ValueError, TypeError):
                     print("Please enter a valid number")
                     input("\nPress Enter to continue...")
-            elif function == "view" and choice.isdigit():
+            elif function == "view" and choice.isdigit() and 1<= int(choice) <=10:
                 # 计算全局任务索引
                 global_index = current_page * 10 + (int(choice) - 1)
                 if 0 <= global_index < len(tasks):
@@ -255,6 +256,8 @@ class TaskManager:
             os.remove(std_log)
             std_log.symlink_to(output_log)
 
+        print('Starting task ......')
+        time.sleep(6)
         logging.info(f"Started {module} module (PID: {process.pid})")
         print(f"Task started (PID: {process.pid})")
         print(f"Normalized log file: {std_log}")
@@ -366,7 +369,7 @@ class TaskManager:
             print("q) Exit")
             print("=" * 50)
 
-            choice = input("Please enter operation: ").strip().lower()
+            choice = input("Please select an operation: ").strip().lower()
             if choice == "1":
                 work_dir = input("Enter EE working directory: ").strip()
                 self.task_runner("EE", work_dir)

{ion_csp-2.0.4.dist-info → ion_csp-2.0.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ion_CSP
-Version: 2.0.4
+Version: 2.0.8
 Summary: Crystal Structure Design Software Based on Molecular/Ionic Configuration.
 Home-page: https://github.com/bagabaga007/ion_CSP
 Author: Ze Yang
@@ -67,7 +67,7 @@ Dynamic: requires-python
 | paramiko    | 3.5.1    |
 | pyxtal      | 1.0.4    |
 | phonopy     | 2.28.0   |
-| rdkit       | 2023.03.3|
+| rdkit       | 2024.03.3|
 
 ### 安装步骤
 ```bash
@@ -172,7 +172,7 @@ This software enables efficient crystal structure screening from molecular/ion c
 | paramiko    | 3.5.1    |
 | pyxtal      | 1.0.4    |
 | phonopy     | 2.28.0   |
-| rdkit       | 2023.03.3|
+| rdkit       | 2024.03.3|
 
 ### Installation Steps
 ```bash

{ion_csp-2.0.4.dist-info → ion_csp-2.0.8.dist-info}/RECORD

@@ -6,14 +6,14 @@ ion_CSP/gen_opt.py,sha256=aVZmf2RqePCwZShgpNvzqfntNAW0I0yctWHGXoe3mgw,19463
 ion_CSP/identify_molecules.py,sha256=hFKXS0Jjd7LyMsYGc9RmnoRPu1ibXF9fYO_9lR3wTfo,4634
 ion_CSP/log_and_time.py,sha256=6RKegXF8Gc1HqzAbE-PT9ejX3Ncyuz3v3FivujJt8to,9072
 ion_CSP/mlp_opt.py,sha256=ox4Qxg4D6WzrB8dxVnUWmAngnOA_wdcInP5UhBWsH4c,5535
-ion_CSP/read_mlp_density.py,sha256=ktGNHYWLi4VrdJK_Hg2ac2-JM2QlVt9iymLxdixkLb8,9663
+ion_CSP/read_mlp_density.py,sha256=TinsFGJPF0TPVZKMCzPyxuHBGrdeqw7yOnGh-78qbqo,11209
 ion_CSP/steps_opt_monitor.sh,sha256=1klPjnK0gqkDbvI9PtjdK5qidJ5G0Mo8q1SfrlLW5xM,3330
-ion_CSP/task_manager.py,sha256=PGtlvlJb7uzb1YcJyhP7cURQqOO4nNM-VacyZwVXixs,16088
+ion_CSP/task_manager.py,sha256=S0QK7EJhKrQL1hzMqkR6a6y3Du027Ppyz-zHenWrFQ4,16241
 ion_CSP/upload_download.py,sha256=Hiz5jKOy9x26hJJdcpt-owQdVUbzzGuGOelro6JozY8,23801
 ion_CSP/vasp_processing.py,sha256=eQTMKIPBzi1X4HHTqWkHE1Lt8dvCCV-MXPCR3RtKYZ0,13977
-ion_csp-2.0.4.dist-info/licenses/LICENSE,sha256=2J6A8GT2iIf2LhuWO1_0ilgx7ijzzpQ2BXU7rHKe8Cc,1068
-ion_csp-2.0.4.dist-info/METADATA,sha256=r9ijfq_Yo5mTqMnCzNT9vYVF-FDSxN9bkHtU99FDdUw,6380
-ion_csp-2.0.4.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-ion_csp-2.0.4.dist-info/entry_points.txt,sha256=NexQJDs9f69kJA2DgoU6tsA3V8a66nadJRem1U_c_6g,54
-ion_csp-2.0.4.dist-info/top_level.txt,sha256=aYZa43dDebjLpWPN6bDIlBb6BVwA8gk4ajEjDDK9b9I,8
-ion_csp-2.0.4.dist-info/RECORD,,
+ion_csp-2.0.8.dist-info/licenses/LICENSE,sha256=2J6A8GT2iIf2LhuWO1_0ilgx7ijzzpQ2BXU7rHKe8Cc,1068
+ion_csp-2.0.8.dist-info/METADATA,sha256=pZg7XM0CD5Du6rQJvijc_VOyfvtFVlnmWAZiwBPmeIU,6380
+ion_csp-2.0.8.dist-info/WHEEL,sha256=QZxptf4Y1BKFRCEDxD4h2V0mBFQOVFLFEpvxHmIs52A,91
+ion_csp-2.0.8.dist-info/entry_points.txt,sha256=NexQJDs9f69kJA2DgoU6tsA3V8a66nadJRem1U_c_6g,54
+ion_csp-2.0.8.dist-info/top_level.txt,sha256=aYZa43dDebjLpWPN6bDIlBb6BVwA8gk4ajEjDDK9b9I,8
+ion_csp-2.0.8.dist-info/RECORD,,

{ion_csp-2.0.4.dist-info → ion_csp-2.0.8.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.3.1)
+Generator: setuptools (80.6.0)
 Root-Is-Purelib: true
 Tag: py3-none-any