Changeset 1119 for framspy

Timestamp:

03/31/21 03:06:34 (4 years ago)

Author:

Maciej Komosinski

Message:

Cache a function reference for better performance

File:

• framspy/FramsticksLib.py (modified) (1 diff)

framspy/FramsticksLib.py

@@ -143 +143 @@
                 n = len(genotype_list)
                 square_matrix = np.zeros((n, n))
-                genos = []  # prepare an array of Geno objects so we don't need to convert raw strings to Geno objects all the time
+                genos = []  # prepare an array of Geno objects so that we don't need to convert raw strings to Geno objects all the time in loops
                 for g in genotype_list:
                         genos.append(frams.Geno.newFromString(g))
+                frams_evaluateDistance = frams.SimilMeasure.evaluateDistance  # cache function reference for better performance in loops
                 for i in range(n):
                         for j in range(n):  # maybe calculate only one triangle if you really need a 2x speedup
-                                square_matrix[i][j] = frams.SimilMeasure.evaluateDistance(genos[i], genos[j])._double()
+                                square_matrix[i][j] = frams_evaluateDistance(genos[i], genos[j])._double()
 
                 for i in range(n):