mdkits 0.1a4__py3-none-any.whl → 0.1a6__py3-none-any.whl

mdkits/cli/density.py

@@ -5,13 +5,18 @@ import click
 import MDAnalysis
 from MDAnalysis import Universe
 from MDAnalysis.analysis.base import AnalysisBase
-from mdkits.util import cp2k_input_parsing, numpy_geo, encapsulated_mda
+from mdkits.util import (
+    arg_type,
+    numpy_geo,
+    encapsulated_mda,
+    os_operation,
+)
 import warnings
 warnings.filterwarnings("ignore")
 
 
 class Density_distribution(AnalysisBase):
-    def __init__(self, filename, cell, o, distance_judg, angle_judg, dt=0.001, bin_size=0.2, return_index=False):
+    def __init__(self, filename, cell, o, element, density, update_water, distance_judg, angle_judg, surface, dt=0.001, bin_size=0.2, return_index=False):
         u = Universe(filename)
         u.trajectory.ts.dt = dt
         u.dimensions = cell
@@ -25,21 +30,31 @@ class Density_distribution(AnalysisBase):
         self.bin_size = bin_size
         self.frame_count = 0
         self.return_index = return_index
+        self.surface = surface
+        self.element = element
+        self.density = density
+        self.update_water = update_water
+
+        if surface is not None:
+            self.surface_group = self.atomgroup.select_atoms(f"{surface}")
+        else:
+            self.surface_group = False
 
         super(Density_distribution, self).__init__(self.atomgroup.universe.trajectory, verbose=True)
 
     def _prepare(self):
         self.bin_num = int(self.u.dimensions[2] / self.bin_size) + 2
         self.density_distribution = np.zeros(self.bin_num, dtype=np.float64)
-        if self.surface:
-            self.surface_pos = ()
+        if self.surface_group:
+            self.surface_pos = np.zeros(2)
 
     def _append(self, z):
         bins = np.floor(z / self.bin_size).astype(int) + 1
         np.add.at(self.density_distribution, bins, 1)
 
+
     def _single_frame(self):
-        if self.water:
+        if self.update_water:
             o_group = self.atomgroup.select_atoms("name O")
             h_group = self.atomgroup.select_atoms("name H")
 
@@ -48,40 +63,58 @@ class Density_distribution(AnalysisBase):
             self._append(o.positions[:, 2])
 
         else:
-            group = self.atomgroup.select_atoms(f"name {self.element}")
+            group = self.atomgroup.select_atoms(f"{self.element}")
             self._append(group.positions[:, 2])
 
-        self.
+        if self.surface_group:
+            lower_z, upper_z = numpy_geo.find_surface(self.surface_group.positions[:, 2], layer_tolerance=1, surface_tolerance=5)
+            self.surface_pos[0] += lower_z
+            self.surface_pos[1] += upper_z
+
         self.frame_count += 1
 
     def _conclude(self):
         if self.frame_count > 0:
             V = self.u.dimensions[0] * self.u.dimensions[1] * self.bin_size
 
-            if self.water:
-                density_distribution = (self.density_distribution * (15.999+1.008*2) * 1.660539 / V) / self.frame_count
+            if self.density:
+                density_distribution = (self.density_distribution * self.density * 1.660539 / V) / self.frame_count
             else:
                 density_distribution = (self.density_distribution * (10000/6.02) / V) / self.frame_count
 
             bins_z = np.arange(len(self.density_distribution)) * self.bin_size
 
-            surface = self.find_surface(self.atomgroup.select_atoms("name Pt"))
+            if self.surface:
+                lower_z = self.surface_pos[0] / self.frame_count
+                upper_z = self.surface_pos[1] / self.frame_count
 
-            lower_z, upper_z = surface
-            mask = (bins_z >= lower_z) & (bins_z <= upper_z)
-            filtered_bins_z = bins_z[mask] - lower_z
-            filtered_density_distribution = density_distribution[mask]
-            conbined_data = np.column_stack((filtered_bins_z, filtered_density_distribution))
+                mask = (bins_z >= lower_z) & (bins_z <= upper_z)
+                filtered_bins_z = bins_z[mask] - lower_z
+                filtered_density_distribution = density_distribution[mask]
+
+                conbined_data = np.column_stack((filtered_bins_z, filtered_density_distribution))
+            else:
+                conbined_data = np.column_stack((bins_z, density_distribution))
 
             np.savetxt(self.o, conbined_data, header="Z\tdensity", fmt='%.5f', delimiter='\t')
 
 @click.command(name='density')
 @click.argument('filename', type=click.Path(exists=True), default=os_operation.default_file_name('*-pos-1.xyz', last=True))
-@click.option('--cell', type=arg_type.Cell, help='set xyz file cell, --cell x,y,z,a,b,c')
 @click.option('--cell', type=arg_type.Cell, help='set cell from cp2k input file or a list of lattice: --cell x,y,z or x,y,z,a,b,c', default='input.inp', show_default=True)
+@click.option('--element', type=str, help='element to analysis')
+@click.option('--density', type=float, help='output density unit (g/cm3), should give atomic mass of element, else is concentration unit (mol/L)')
 @click.option('-o', type=str, help='output file name', default='density.dat', show_default=True)
-def main(filename, cell, o):
-    density_dist = Density_distribution(filename, cell, o=o)
+@click.option('--update_water', is_flag=True, help='update water with distance or angle judgment')
+@click.option('--distance', type=float, help='update water distance judgment', default=1.2, show_default=True)
+@click.option('--angle', type=(float), help='update water angle judgment')
+@click.option('--surface', type=str, help='surface element')
+def main(filename, cell, o, element, density, update_water, distance, angle, surface):
+    """
+    analysis density or concentration of element in a trajectory file
+    """
+
+    density_dist = Density_distribution(filename, cell, o=o, distance_judg=distance, angle_judg=angle, element=element, density=density, update_water=update_water, surface=surface)
+
     density_dist.run()
 
 

mdkits/mdkits.py

@@ -5,6 +5,7 @@ from mdkits.cli import (
     extract,
     data,
     plot,
+    density
 )
 
 
@@ -12,7 +13,6 @@ from mdkits.cli import (
 @click.pass_context
 @click.version_option()
 def cli(ctx):
-    print(ctx)
     """tools for md or dft"""
     pass
 
@@ -22,6 +22,7 @@ cli.add_command(wrap.main)
 cli.add_command(extract.main)
 cli.add_command(data.main)
 cli.add_command(plot.main)
+cli.add_command(density.main)
 
 
 if __name__ == '__main__':

mdkits/util/numpy_geo.py

@@ -97,7 +97,7 @@ def unwrap(atom1, atom2, coefficients, max=0, total=False):
 
 
 def find_surface(surface_group:np.ndarray, layer_tolerance=1, surface_tolerance=5):
-    sort_group = surface_group.sort()
+    sort_group = np.sort(surface_group)
     layer_mean = []
     current_layer = [sort_group[0]]
     for i in range(1, len(sort_group)):
@@ -106,6 +106,7 @@ def find_surface(surface_group:np.ndarray, layer_tolerance=1, surface_tolerance=
         else:
             layer_mean.append(np.mean(current_layer))
             current_layer = [sort_group[i]]
+        layer_mean.append(np.mean(current_layer))
     
     if len(current_layer) == 1:
         return layer_mean[0]
@@ -115,4 +116,4 @@ def find_surface(surface_group:np.ndarray, layer_tolerance=1, surface_tolerance=
         index = np.argmax(diff > 5)
         return (layer_mean[index], layer_mean[index + 1])
     else:
-        return layer_mean[-1]
+        return (layer_mean[-1], layer_mean[0])

{mdkits-0.1a4.dist-info → mdkits-0.1a6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mdkits
-Version: 0.1a4
+Version: 0.1a6
 Summary: tools for md or dft
 License: MIT
 Keywords: molecular dynamics,density functional theory

{mdkits-0.1a4.dist-info → mdkits-0.1a6.dist-info}/RECORD

@@ -11,7 +11,7 @@ mdkits/cli/cp2k_input.py,sha256=EskBUSaAOLmrrUYGruTwZjEy9EksdEhWwR5ys9juIbg,1113
 mdkits/cli/cube.py,sha256=XcJfsJ5Y2d5MQfD45p06_gV1fTJbDSrNhCnZ3Sz2Vb0,2233
 mdkits/cli/cut_surface.py,sha256=D1naCWj0QJE3AiInzy3SeGO37butothE3BS1rZsZ5MU,1425
 mdkits/cli/data.py,sha256=FGA4S9Cfo6WUJBSPWKOJrrZXHo_Qza-jNG1P_Dw7yi4,3262
-mdkits/cli/density.py,sha256=ISn1MMfFHmw4sPTSF8B5f-e1rZohyBWxjnpQU53pxgk,3530
+mdkits/cli/density.py,sha256=Icxj-IMvUEFe15VKT13TrX5sHDmyczMnqo-VP46gPlI,5030
 mdkits/cli/density2.py,sha256=I1OtC7s979Ks7GOH47hTKUGE1h6tUFYvNZQ7Qbifs6I,3458
 mdkits/cli/extract.py,sha256=bqqJBmSaVyPYyEseGpUJcMBufIfDLTNRdmUfJ0txE5E,2498
 mdkits/cli/hartree_potential.py,sha256=XcJfsJ5Y2d5MQfD45p06_gV1fTJbDSrNhCnZ3Sz2Vb0,2233
@@ -26,18 +26,18 @@ mdkits/cli/supercell.py,sha256=r5iddLw1NNzj7NTFlR2j_jpD1AMfrsxhA08bPwQvVvg,2059
 mdkits/cli/wrap.py,sha256=AUxGISuiCfEjdMYl-TKc2VMCPHSybWKrMIOTn_6kSp0,1043
 mdkits/config/__init__.py,sha256=ZSwmnPK02LxJLMgcYmNb-tIOk8fEuHf5jpqD3SDHWLg,1039
 mdkits/config/settings.yml,sha256=PY7u0PbFLuxSnd54H5tI9oMjUf-mzyADqSZtm99BwG0,71
-mdkits/mdkits.py,sha256=SEJiJ8kqZIBBIE2pdOKwBSL8LqadWgA2_F0bk6e4xCg,443
+mdkits/mdkits.py,sha256=sBA61-E1lk-RJlOr_WcTkd7GjfxcHlBW_ASyg01s5os,471
 mdkits/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mdkits/util/arg_type.py,sha256=YUFhYShTYfUe7xajgUCiCuaeNLH-wxMXvlcMRax4zRM,1469
 mdkits/util/cp2k_input_parsing.py,sha256=6BFVsbBYxdauaUPrjr5h8JXk_R0bu7V3cbG2DEWWFlQ,1291
 mdkits/util/encapsulated_ase.py,sha256=dZU1i9Kw8rIyiQCCSDEXOs3q_z4dCMIo6bf5ooY9u0c,4089
 mdkits/util/encapsulated_mda.py,sha256=td3H24u3eHOIS2nwPucfIaMxeaVxI77oFI8nnNhw7vo,2217
 mdkits/util/fig_operation.py,sha256=FwffNUtXorMl6qE04FipgzcVljEQii7wrNJUCJMyY3E,1045
-mdkits/util/numpy_geo.py,sha256=ngSNoXkt3bHjOdv77ZlvQGbj1V75IyjjUWlDi1J0V6g,3549
+mdkits/util/numpy_geo.py,sha256=1Op8THoQeyqybSZAi7hVxohYCr4SzY6ndZC8_gAGXDA,3619
 mdkits/util/os_operation.py,sha256=tNRjniDwFqARMy5Rf6MUNmaQGpJlyifCpuN16r8aB2g,828
 mdkits/util/structure_parsing.py,sha256=mRPMJeih3O-ST7HeETDvBEkfV-1psT-XgxyYgDadV0U,4152
-mdkits-0.1a4.dist-info/entry_points.txt,sha256=xoWWZ_yL87S501AzCO2ZjpnVuYkElC6z-8J3tmuIGXQ,44
-mdkits-0.1a4.dist-info/LICENSE,sha256=VLaqyB0r_H7y3hUntfpPWcE3OATTedHWI983htLftcQ,1081
-mdkits-0.1a4.dist-info/METADATA,sha256=5uPRWLj3rBwpqoKGxCgcaPT8sLDyUB1hkT_Nksueivg,4284
-mdkits-0.1a4.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-mdkits-0.1a4.dist-info/RECORD,,
+mdkits-0.1a6.dist-info/entry_points.txt,sha256=xoWWZ_yL87S501AzCO2ZjpnVuYkElC6z-8J3tmuIGXQ,44
+mdkits-0.1a6.dist-info/LICENSE,sha256=VLaqyB0r_H7y3hUntfpPWcE3OATTedHWI983htLftcQ,1081
+mdkits-0.1a6.dist-info/METADATA,sha256=OM9IB6t1RlnqX5CM007d5oWafnkVTVpMGUrfOnxbjtA,4284
+mdkits-0.1a6.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+mdkits-0.1a6.dist-info/RECORD,,