chore: add filtering by type

Exabyte-io · Sep 13, 2023 · ccc21d5 · ccc21d5
1 parent ab845fd
commit ccc21d5
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/include/meta_properties/pseudopotential.js b/src/include/meta_properties/pseudopotential.js
@@ -132,6 +132,8 @@ export class Pseudopotential extends Property {
                 filteredPseudos = this.filterByAppName(filteredPseudos, fValue);
             } else if (fKey === "elements" && fValue) {
                 filteredPseudos = this.filterByElements(filteredPseudos, fValue);
+            } else if (fKey === "type" && fValue) {
+                filteredPseudos = this.filterByType(filteredPseudos, fValue);
             }
         });
         return filteredPseudos;
@@ -161,4 +163,13 @@ export class Pseudopotential extends Property {
             return 0;
         });
     }
+
+    /**
+     * @param {Pseudopotential[]} pseudos - Pseudopotentials to filter
+     * @param {string} pseudoType - Pseudopotential type (e.g. "us", "nc", ...)
+     * @returns {Pseudopotential[]}
+     */
+    static filterByType(pseudos, pseudoType) {
+        return pseudos.filter((pseudo) => pseudo.type.includes(pseudoType));
+    }
 }