Update hill_cipher.py

Author: lighting9999

ciphers/hill_cipher.py

@@ -19,9 +19,7 @@
 """
 
 import string
-
 import numpy as np
-
 from maths.greatest_common_divisor import greatest_common_divisor
 
 
@@ -145,9 +143,10 @@ def check_determinant(self) -> None:
             ValueError: determinant modular 36 of encryption key(0) is not co prime
             w.r.t 36. Try another key.
         """
-        det = round(np.linalg.det(self.encrypt_key))
-        det = int(det)  # Convert to int after rounding float
-
+        # Optimized determinant calculation to avoid redundant rounding
+        det_value = np.linalg.det(self.encrypt_key)
+        det = int(round(det_value)) if not det_value.is_integer() else int(det_value)
+        
         if det < 0:
             det = det % len(self.key_string)
 
@@ -158,7 +157,6 @@ def check_determinant(self) -> None:
                 f"w.r.t {req_l}.\nTry another key."
             )
             raise ValueError(msg)
-
     def process_text(self, text: str) -> str:
         """
         Prepare text for encryption/decryption by:
@@ -185,17 +183,16 @@ def process_text(self, text: str) -> str:
             'ABCC'
         """
         chars = [char for char in text.upper() if char in self.key_string]
-
+        
         # Handle empty input case
         if not chars:
             return ""
-
+            
         last = chars[-1]
         while len(chars) % self.break_key != 0:
             chars.append(last)
-
+            
         return "".join(chars)
-
     def encrypt(self, text: str) -> str:
         """
         Encrypt plaintext using Hill Cipher.
@@ -223,21 +220,21 @@ def encrypt(self, text: str) -> str:
         text = self.process_text(text.upper())
         if not text:
             return ""
-
+            
         encrypted = ""
 
         for i in range(0, len(text) - self.break_key + 1, self.break_key):
             # Extract batch of characters
             batch = text[i : i + self.break_key]
-
+            
             # Convert to numerical vector
             vec = [self.replace_letters(char) for char in batch]
             batch_vec = np.array([vec]).T
-
+            
             # Matrix multiplication and mod 36
             product = self.encrypt_key.dot(batch_vec)
             batch_encrypted = self.modulus(product).T.tolist()[0]
-
+            
             # Convert back to characters
             encrypted_batch = "".join(
                 self.replace_digits(num) for num in batch_encrypted
@@ -262,7 +259,7 @@ def make_decrypt_key(self) -> np.ndarray:
             >>> cipher.make_decrypt_key()
             array([[ 6, 25],
                    [ 5, 26]])
-
+            
             >>> key3x3 = np.array([[1,2,3],[4,5,6],[7,8,9]])
             >>> cipher3 = HillCipher(key3x3)
             >>> cipher3.make_decrypt_key()  # Determinant 0 should be invalid
@@ -271,12 +268,13 @@ def make_decrypt_key(self) -> np.ndarray:
             ValueError: determinant modular 36 of encryption key(0) is not co prime
             w.r.t 36. Try another key.
         """
-        det = round(np.linalg.det(self.encrypt_key))
-        det = int(det)  # Convert to int after rounding float
-
+        # Optimized determinant calculation to avoid redundant rounding
+        det_value = np.linalg.det(self.encrypt_key)
+        det = int(round(det_value)) if not det_value.is_integer() else int(det_value)
+        
         if det < 0:
             det = det % len(self.key_string)
-
+        
         det_inv: int | None = None
         for i in range(len(self.key_string)):
             if (det * i) % len(self.key_string) == 1:
@@ -319,22 +317,22 @@ def decrypt(self, text: str) -> str:
         text = self.process_text(text.upper())
         if not text:
             return ""
-
+            
         decrypt_key = self.make_decrypt_key()
         decrypted = ""
 
         for i in range(0, len(text) - self.break_key + 1, self.break_key):
             # Extract batch of characters
             batch = text[i : i + self.break_key]
-
+            
             # Convert to numerical vector
             vec = [self.replace_letters(char) for char in batch]
             batch_vec = np.array([vec]).T
-
+            
             # Matrix multiplication and mod 36
             product = decrypt_key.dot(batch_vec)
             batch_decrypted = self.modulus(product).T.tolist()[0]
-
+            
             # Convert back to characters
             decrypted_batch = "".join(
                 self.replace_digits(num) for num in batch_decrypted
@@ -344,10 +342,11 @@ def decrypt(self, text: str) -> str:
         return decrypted
 
 
+
 def main() -> None:
     """
     Command-line interface for Hill Cipher operations.
-
+    
     Steps:
     1. User inputs encryption key size
     2. User inputs encryption key matrix rows
@@ -360,14 +359,14 @@ def main() -> None:
 
     print("Enter each row of the encryption key with space separated integers")
     for i in range(n):
-        row = [int(x) for x in input(f"Row {i + 1}: ").split()]
+        row = [int(x) for x in input(f"Row {i+1}: ").split()]
         hill_matrix.append(row)
 
     hc = HillCipher(np.array(hill_matrix))
 
     print("\nWould you like to encrypt or decrypt some text?")
     option = input("1. Encrypt\n2. Decrypt\nEnter choice (1/2): ")
-
+    
     if option == "1":
         text = input("\nEnter text to encrypt: ")
         print("\nEncrypted text:")
@@ -379,24 +378,22 @@ def main() -> None:
     else:
         print("Invalid option selected")
 
-
 if __name__ == "__main__":
     import doctest
-
     doctest.testmod()
-
+    
     print("\nRunning sample tests...")
     key = np.array([[2, 5], [1, 6]])
     cipher = HillCipher(key)
-
+    
     # Test encryption/decryption round trip
     plaintext = "HELLO123"
     encrypted = cipher.encrypt(plaintext)
     decrypted = cipher.decrypt(encrypted)
-
+    
     print(f"\nOriginal text: {plaintext}")
     print(f"Encrypted text: {encrypted}")
     print(f"Decrypted text: {decrypted}")
-
+    
     # Run CLI interface
     main()