ipyvasp 1.1.2__py2.py3-none-any.whl → 1.1.4__py2.py3-none-any.whl

ipyvasp/_lattice.py

@@ -218,7 +218,7 @@ def periodic_table(selection=None):
     names = np.array(names)[fidx]
     
     # We are adding patches, because imshow does not properly appear in PDF of latex
-    ax = ptk.get_axes((7, 3.9),left=0.01,right=0.99,top=0.99,bottom=0.01)
+    ax = ptk.get_axes(1, (7, 3.9),left=0.01,right=0.99,top=0.99,bottom=0.01)
     patches = np.array([Rectangle(offset,0.9 if i in [92,110] else 1,1) for i, offset in zip(fidx,offsets)])
     pc = PatchCollection(patches, facecolors=fc, edgecolors=ec,linewidths=(0.7,))
     ax.add_collection(pc)

ipyvasp/_version.py

@@ -1 +1 @@
-__version__ = "1.1.2"
+__version__ = "1.1.4"

ipyvasp/core/serializer.py

@@ -766,16 +766,19 @@ class BrZoneData(Dict2Data):
         d.update({"faces": faces, "vertices": vertices, "_specials": specials})
         return self.__class__(d)
     
-    def tile(self, nxyz, filter=None):
-        """Create a tiled array of BZ centers for visualization.
+    def tile(self, nxyz, filter=None, primitive=False):
+        """Create a tiled rectangular array of BZ centers for visualization. Use in `center` parameter for BZ plotting.
     
         Parameters
         ----------
         nxyz : list or tuple of 3 ints
             Number of tiles along each cartesian direction [nx, ny, nz].
             Must be 3 positive integers.
+            If primitive = False, result is not guaranteed to be same size as ``nx * ny * nz`` due to shape distortion.
         filter : callable, optional
-            Function filter(x,y,z) that takes cartesian coordinates and returns bool.
+            Function filter(x,y,z) that takes normalized (to 1) cartesian coordinates and returns bool.
+        primitive : bool
+            If True, tile points in basis rather than orthogonal geometry. Default is False.
             
         Returns
         -------
@@ -794,17 +797,56 @@ class BrZoneData(Dict2Data):
             if not isinstance(n, int) or n < 1:
                 raise ValueError(f"nxyz[{i}] must be a positive integer")
         
-        xyz = self.to_cartesian(np.indices(np.ceil(nxyz).astype(int)).reshape((3,-1)).T) 
-        # Apply filter if provided
-        if filter is not None:
-            if not callable(filter):
-                raise TypeError("filter must be callable")
-            
-            mask = np.array([filter(x, y, z) for x, y, z in xyz])
-            xyz = xyz[mask]
+        if filter is not None and not callable(filter):
+            raise ValueError("filter must be callable")
+        
+        # Tile in shape of cell if primitive
+        if primitive:
+            pts = np.indices(nxyz).reshape((3,-1)).T
+            if filter and len(pts) > 0:
+                # Per-dimension normalization for primitive
+                norms = np.array([max(1, n - 1) for n in nxyz])
+                mask = np.array([filter(*(p/norms)) for p in pts]) 
+                pts = pts[mask]
+            return pts
+        
+        # Orthogonal/Cartesian-aligned case
+        target_span = np.linalg.norm(self.basis.sum(axis=0)) * (np.array(nxyz) - 1)
+        
+        # 2. Map Cartesian box corners to Lattice Space
+        corners_cart = np.array([[i, j, k] for i in [0, 1] for j in [0, 1] for k in [0, 1]]) * target_span
+        corners_frac = self.to_fractional(corners_cart)
+        
+        # 3. Find integer bounds (Must use +1 for inclusive arange)
+        n_min = np.floor(corners_frac.min(axis=0)).astype(int)
+        n_max = np.ceil(corners_frac.max(axis=0)).astype(int)
         
-        # Convert to fractional coordinates and return
-        return self.to_fractional(xyz)
+        ranges = [np.arange(n_min[i], n_max[i] + 1) for i in range(3)]
+        grid_n = np.array(np.meshgrid(*ranges, indexing='ij')).T.reshape(-1, 3)
+        
+        all_pts = self.to_cartesian(grid_n)
+        
+        # 4. Clipping with a slightly more generous epsilon to prevent missing bridge points
+        eps = 1e-9
+        mask = (all_pts >= -eps).all(axis=1) & (all_pts <= target_span + eps).all(axis=1)
+        pts = all_pts[mask]
+        
+        N = np.cumprod(nxyz)[-1]
+        if len(pts) > N: # box becomes too much inclusive at larger points, restrict that
+            dist = np.linalg.norm(pts - pts.mean(axis=0), axis=1)
+            pts = pts[np.argsort(dist)[:N]]
+
+        # 5. Apply Global Normalized filter (preserves geometry)
+        if filter and len(pts) > 0:
+            p_min, p_max = pts.min(axis=0), pts.max(axis=0)
+            global_range = np.max(p_max - p_min) or 1.0
+            norm_pts = (pts - p_min) / global_range
+            
+            f_mask = np.array([filter(*p) for p in norm_pts])
+            pts = pts[f_mask]
+            
+        pts = pts[np.lexsort((pts[:, 0], pts[:, 1], pts[:, 2]))] # reorder for easy reading of points
+        return self.to_fractional(pts).round(12) # no need to ceil or int here, that makes empty areas, just remove values like E-17
 
 
 class CellData(Dict2Data):

{ipyvasp-1.1.2.dist-info → ipyvasp-1.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ipyvasp
-Version: 1.1.2
+Version: 1.1.4
 Summary: A processing tool for VASP DFT input/output processing in Jupyter Notebook.
 Home-page: https://github.com/massgh/ipyvasp
 Author: Abdul Saboor

{ipyvasp-1.1.2.dist-info → ipyvasp-1.1.4.dist-info}/RECORD

@@ -1,8 +1,8 @@
 ipyvasp/__init__.py,sha256=pzTqeKuf6sN2GQmaexmMgG677ggT3sxIFyQDXq_2whU,1422
 ipyvasp/__main__.py,sha256=eJV1TZSiT8mC_VqAeksNnBI2I8mKMiPkEIlwikbtOjI,216
 ipyvasp/_enplots.py,sha256=gJ7S9WBmrxvDEbmoccDRaJG01kpx9oNlRf7mozigbgY,37872
-ipyvasp/_lattice.py,sha256=Lh-ip60M6APUckR4I0bQ6GMN2aaY7gpkA1Z0uuFt1A4,108117
-ipyvasp/_version.py,sha256=JhNfc49cF1z8HnPYAyTem2s5Sn4s2XfgPp_HRwRQSfQ,23
+ipyvasp/_lattice.py,sha256=LW20S2p_jx4bth5JedFnsvAvw6iWT1wLMmoi0Acsz2Y,108120
+ipyvasp/_version.py,sha256=PZhUTNgGE_in6n-SwcZzX-gYtmYPohHIX-3AQqOiJEY,23
 ipyvasp/bsdos.py,sha256=hVHpxkdT2ImRsxwFvMSMHxRSo4LqDM90DnUhwTP8vcs,32192
 ipyvasp/cli.py,sha256=-Lf-qdTvs7WyrA4ALNLaoqxMjLsZkXdPviyQps3ezqg,6880
 ipyvasp/evals_dataframe.py,sha256=n2iSH4D4ZbrxlAV4yDTVHbcl3ycfD0zfQYmTBcxjfkE,20789
@@ -15,11 +15,11 @@ ipyvasp/widgets.py,sha256=Bpa4Y3Eopk_ZPFsVetfysClZP2q_2ONvmOwUol9vVGI,53154
 ipyvasp/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ipyvasp/core/parser.py,sha256=o-uHyL_w9W0pmxoSt3JmLwwmmT3WRHlHSs_NoiHO-hs,39401
 ipyvasp/core/plot_toolkit.py,sha256=ru8-FLJp8-X2p_Ft0F3K68qNSUJhy3jQD_S2zW50FWg,40930
-ipyvasp/core/serializer.py,sha256=9xuLfl9LtUXPxTX2_zRs7nvfZsZOZN00bnxErAxDA5w,40065
+ipyvasp/core/serializer.py,sha256=JhtvuWjqBVSWzsDodUzsO_avTxC5yBH8NkFTM95-FEk,42385
 ipyvasp/core/spatial_toolkit.py,sha256=dXowREhiFzBvvr5f_bApzFhf8IzjH2E2Ix90oCBUetY,14885
-ipyvasp-1.1.2.dist-info/LICENSE,sha256=F3SO5RiAZOMfmMGf1KOuk2g_c4ObvuBJhd9iBLDgXoQ,1263
-ipyvasp-1.1.2.dist-info/METADATA,sha256=2wWh4djtZU6uiYMOCpsCuRDFiCAIQHToqTAf3Clkp3g,3218
-ipyvasp-1.1.2.dist-info/WHEEL,sha256=Kh9pAotZVRFj97E15yTA4iADqXdQfIVTHcNaZTjxeGM,110
-ipyvasp-1.1.2.dist-info/entry_points.txt,sha256=aU-gGjQG2Q8XfxDlNc_8__cwfp8WG2K5ZgFPInTm2jg,45
-ipyvasp-1.1.2.dist-info/top_level.txt,sha256=ftziWlMWu_1VpDP1sRTFrkfBnWxAi393HYDVu4wRhUk,8
-ipyvasp-1.1.2.dist-info/RECORD,,
+ipyvasp-1.1.4.dist-info/LICENSE,sha256=F3SO5RiAZOMfmMGf1KOuk2g_c4ObvuBJhd9iBLDgXoQ,1263
+ipyvasp-1.1.4.dist-info/METADATA,sha256=pUPSVUeAInyhEuZ_XGLmVUc2ndKaW8SUtqGdvgU1Yus,3218
+ipyvasp-1.1.4.dist-info/WHEEL,sha256=Kh9pAotZVRFj97E15yTA4iADqXdQfIVTHcNaZTjxeGM,110
+ipyvasp-1.1.4.dist-info/entry_points.txt,sha256=aU-gGjQG2Q8XfxDlNc_8__cwfp8WG2K5ZgFPInTm2jg,45
+ipyvasp-1.1.4.dist-info/top_level.txt,sha256=ftziWlMWu_1VpDP1sRTFrkfBnWxAi393HYDVu4wRhUk,8
+ipyvasp-1.1.4.dist-info/RECORD,,