Add missing Rank for foldmason jobs

soedinglab · Jul 30, 2024 · c9eaaaf · c9eaaaf
1 parent 1932eff
commit c9eaaaf
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/backend/foldmasonmsa.go b/backend/foldmasonmsa.go
@@ -27,6 +27,13 @@ func (r FoldMasonMSAJob) Hash() Id {
 	return Id(base64.URLEncoding.WithPadding(base64.NoPadding).EncodeToString(bs))
 }
 
+func (r FoldMasonMSAJob) Rank() float64 {
+	if len(r.Queries) == 0 {
+		return 0
+	}
+	return float64(len(r.Queries) * len(r.Queries[0]))
+}
+
 func (r FoldMasonMSAJob) WritePDB(path string) error {
 	var pdbDir = filepath.Join(path, "pdbs")
 	os.Mkdir(pdbDir, os.ModePerm)