kim-tools 0.3.0__py3-none-any.whl → 0.3.1__py3-none-any.whl

kim_tools/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.3.0"
+__version__ = "0.3.1"
 
 from .aflow_util import *
 from .aflow_util import __all__ as aflow_all

kim_tools/aflow_util/core.py

@@ -6,6 +6,7 @@ import os
 import subprocess
 from curses.ascii import isalpha, isdigit
 from dataclasses import dataclass
+from itertools import permutations
 from math import acos, cos, degrees, radians, sqrt
 from os import PathLike
 from tempfile import NamedTemporaryFile
@@ -343,6 +344,30 @@ def get_stoich_reduced_list_from_prototype(prototype_label: str) -> List[int]:
     return stoich_reduced_list
 
 
+def build_abstract_formula_from_stoich_reduced_list(
+    stoich_reduced_list: List[int],
+) -> str:
+    """
+    Get abstract chemical formula from numerical list of stoichiometry
+    i.e. [1,3] -> "AB3"
+
+    Args:
+        stoich_reduced_list:
+            List of reduced stoichiometric numbers
+
+    Returns:
+        Abstract chemical formula
+    """
+    formula = ""
+    for spec_num, stoich_num in enumerate(stoich_reduced_list):
+        assert isinstance(stoich_num, int)
+        assert stoich_num > 0
+        formula += chr(65 + spec_num)
+        if stoich_num > 1:
+            formula += str(stoich_num)
+    return formula
+
+
 def get_wyckoff_lists_from_prototype(prototype_label: str) -> List[str]:
     """
     Expand the list of Wyckoff letters in the prototype to account for each individual
@@ -350,7 +375,7 @@ def get_wyckoff_lists_from_prototype(prototype_label: str) -> List[str]:
     e.g. A2B3C_mC48_15_aef_3f_2e -> ['aef','fff','ee']
     """
     expanded_wyckoff_letters = []
-    prototype_label_split = prototype_label.split("_")
+    prototype_label_split = prototype_label.split("-")[0].split("_")
     for species_wyckoff_string in prototype_label_split[3:]:
         expanded_wyckoff_letters.append("")
         curr_wyckoff_count = 0
@@ -371,14 +396,14 @@ def prototype_labels_are_equivalent(
     prototype_label_1: str,
     prototype_label_2: str,
     allow_enantiomorph: bool = False,
-    allow_species_permutation: bool = False,
 ) -> bool:
     """
-    Checks if two prototype labels are equivalent
+    Checks if two prototype labels are equivalent (species permutations not allowed)
+
+    Args:
+        allow_enantiomorph:
+            Whether to consider enantiomorphic pairs of space groups to be equivalent
     """
-    if allow_species_permutation:
-        # TODO: Possibly add this (for checking library prototype labels)
-        raise NotImplementedError("Species permutations not implemented")
 
     if prototype_label_1 == prototype_label_2:
         return True
@@ -494,6 +519,58 @@ def prototype_labels_are_equivalent(
         return True
 
 
+def find_species_permutation_between_prototype_labels(
+    prototype_label_1: str,
+    prototype_label_2: str,
+    allow_enantiomorph: bool = False,
+) -> Optional[Tuple[int]]:
+    """
+    Find the permutation of species required to match two prototype labels
+
+    Args:
+        allow_enantiomorph:
+            Whether to consider enantiomorphic pairs of space groups to be equivalent
+
+    Returns:
+        The permutation of species of ``prototype_label_1`` required to match
+        ``prototype_label_2``, or None if no match is found
+    """
+    # Disassemble prototype_label_1
+    stoich_reduced_list_1 = get_stoich_reduced_list_from_prototype(prototype_label_1)
+    pearson_1 = get_pearson_symbol_from_prototype(prototype_label_1)
+    space_group_1 = str(get_space_group_number_from_prototype(prototype_label_1))
+    species_wyckoff_sections_1 = prototype_label_1.split("-")[0].split("_")[3:]
+
+    nspecies = len(stoich_reduced_list_1)
+    assert nspecies == len(species_wyckoff_sections_1)
+
+    permutation_candidates = permutations(tuple(range(nspecies)))
+    for permutation in permutation_candidates:
+        # Permute the species
+        stoich_reduced_list_1_permuted = [stoich_reduced_list_1[i] for i in permutation]
+        species_wyckoff_sections_1_permuted = [
+            species_wyckoff_sections_1[i] for i in permutation
+        ]
+
+        # Reassemble prototype_label_1_permuted
+        abstract_formula_1_permuted = build_abstract_formula_from_stoich_reduced_list(
+            stoich_reduced_list_1_permuted
+        )
+        prototype_label_1_permuted_list = [
+            abstract_formula_1_permuted,
+            pearson_1,
+            space_group_1,
+        ] + species_wyckoff_sections_1_permuted
+        prototype_label_1_permuted = "_".join(prototype_label_1_permuted_list)
+        if prototype_labels_are_equivalent(
+            prototype_label_1=prototype_label_1_permuted,
+            prototype_label_2=prototype_label_2,
+            allow_enantiomorph=allow_enantiomorph,
+        ):
+            return permutation
+    return None
+
+
 def get_space_group_number_from_prototype(prototype_label: str) -> int:
     return int(prototype_label.split("_")[2])
 

{kim_tools-0.3.0.dist-info → kim_tools-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kim-tools
-Version: 0.3.0
+Version: 0.3.1
 Summary: Base classes and helper routines for writing KIM Tests
 Author-email: ilia Nikiforov <nikif002@umn.edu>, Ellad Tadmor <tadmor@umn.edu>, Claire Waters <bwaters@umn.edu>, "Daniel S. Karls" <karl0100umn@gmail.com>, Matt Bierbaum <matt.bierbaum@gmail.com>, Eric Fuemmeler <efuemmel@umn.edu>, Philipp Hoellmer <ph2484@nyu.edu>, Guanming Zhang <gz2241@nyu.edu>, Tom Egg <tje3676@nyu.edu>
 Maintainer-email: ilia Nikiforov <nikif002@umn.edu>

{kim_tools-0.3.0.dist-info → kim_tools-0.3.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
-kim_tools/__init__.py,sha256=1_-Nx2m2jdCABSg3wVEK_v139McyduMthh8VNWZSsd0,433
+kim_tools/__init__.py,sha256=dOQsrWeoyK-9Fr4gPNLzdouLMm0J0hhh6qd86cPECns,433
 kim_tools/kimunits.py,sha256=jOxBv9gRVhxPE6ygAIUxOzCAfPI6tT6sBaF_FNl9m-M,5387
 kim_tools/aflow_util/__init__.py,sha256=lJnQ8fZCma80QVRQeKvY4MQ87oCWu-9KATV3dKJfpDc,80
-kim_tools/aflow_util/core.py,sha256=1Rv9LSoOgr_9-XFSK4UnHZ2XXQF0MUreMrRh7WWDTQw,72854
+kim_tools/aflow_util/core.py,sha256=NULg8zJG4B5lh4p-Q7yPsJvgzPG_BgSRjgrLpyX3UUI,75526
 kim_tools/aflow_util/aflow_prototype_encyclopedia/data/A108B24C11D24_cP334_222_h4i_i_bf_i-001/info.json,sha256=IsFiO9X2Ko7yoq2QkDurUVP7k1BE4WFgblu7oxl6iZs,2013
 kim_tools/aflow_util/aflow_prototype_encyclopedia/data/A10B11_tI84_139_dehim_eh2n-001/info.json,sha256=f1EdtouuSL2y9NNw40Rvz2J9ZZcsqQBcyEmlHj6XoW8,1186
 kim_tools/aflow_util/aflow_prototype_encyclopedia/data/A10B2C_hP39_171_5c_c_a-001/info.json,sha256=vD1xjZKWShL0E6XNsSlmIhilGcGNefl56oQDLQlHO1M,1596
@@ -2015,8 +2015,8 @@ kim_tools/test_driver/__init__.py,sha256=KOiceeZNqkfrgZ66CiRiUdniceDrCmmDXQkOw0w
 kim_tools/test_driver/core.py,sha256=r4hiZcV-PkWcOo0uEqfqeP1-YGGpPepaAjmR4LJog0w,89208
 kim_tools/vc/__init__.py,sha256=zXjhxXCKVMLBMXXWYG3if7VOpBnsFrn_RjVpnohDm5c,74
 kim_tools/vc/core.py,sha256=BIjzEExnQAL2S90a_npptRm3ACqAo4fZBtvTDBMWMdw,13963
-kim_tools-0.3.0.dist-info/licenses/LICENSE.CDDL,sha256=I2luEED_SHjuZ01B4rYG-AF_135amL24JpHvZ1Jhqe8,16373
-kim_tools-0.3.0.dist-info/METADATA,sha256=MAODKb510nvTCelHGq9-IHuosDJSGjTEdl1DPnP19mA,1962
-kim_tools-0.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kim_tools-0.3.0.dist-info/top_level.txt,sha256=w_YCpJ5ERigj9te74ln7k64tqj1VumOzM_s9dsalIWY,10
-kim_tools-0.3.0.dist-info/RECORD,,
+kim_tools-0.3.1.dist-info/licenses/LICENSE.CDDL,sha256=I2luEED_SHjuZ01B4rYG-AF_135amL24JpHvZ1Jhqe8,16373
+kim_tools-0.3.1.dist-info/METADATA,sha256=scnOzhP0hy9JNn5NS6W9lSESU6G0PS2rPoCdkwIIVSE,1962
+kim_tools-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kim_tools-0.3.1.dist-info/top_level.txt,sha256=w_YCpJ5ERigj9te74ln7k64tqj1VumOzM_s9dsalIWY,10
+kim_tools-0.3.1.dist-info/RECORD,,