From 676f3521364df064574781ba7eb71ad1735ba475 Mon Sep 17 00:00:00 2001
From: Joachim Brand <joachim.brand@gmail.com>
Date: Sat, 10 Feb 2024 01:11:00 +1300
Subject: [PATCH] special case num_singly_doubly_occupied_sites

---
 src/Hamiltonians/HubbardMom1D.jl | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/Hamiltonians/HubbardMom1D.jl b/src/Hamiltonians/HubbardMom1D.jl
index 26ec8aa4f..ea645806a 100644
--- 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