special case num_singly_doubly_occupied_sites

RimuQMC · Feb 9, 2024 · 676f352 · 676f352
1 parent 2414aad
commit 676f352
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/Hamiltonians/HubbardMom1D.jl b/src/Hamiltonians/HubbardMom1D.jl
@@ -107,6 +107,11 @@ function num_singly_doubly_occupied_sites(b::SingleComponentFockAddress)
     return singlies, doublies
 end
 
+# faster method for this special case
+function num_singly_doubly_occupied_sites(b::OccupationNumberFS)
+    return num_singly_doubly_occupied_sites(onr(b))
+end
+
 function num_singly_doubly_occupied_sites(onrep::AbstractArray)
     # this one is faster by about a factor of 2 if you already have the onrep
     # returns number of singly and doubly occupied sites