Added Reset for a couple of simulators

aromanro · Jan 17, 2025 · 3348aa9 · 3348aa9
1 parent af67da1
commit 3348aa9
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/QCSim/QubitRegister.h b/QCSim/QubitRegister.h
@@ -85,6 +85,11 @@ namespace QC {
 			registerStorage(NrBasisStates - 1) = OneOverSqrt2;
 		}
 
+		void Reset()
+		{
+			setToBasisState(0);
+		}	
+
 		// all states have equal amplitude, so measurement should give any state with equal probability
 		void setToEqualSuperposition()
 		{

diff --git a/QCSim/StabilizerState.h b/QCSim/StabilizerState.h
@@ -31,6 +31,29 @@ namespace QC {
 				}
 			}
 
+			void Reset()
+			{
+				for (size_t q = 0; q < stabilizerGenerators.size(); ++q)
+				{
+					for (size_t p = 0; p < stabilizerGenerators.size(); ++p)
+					{
+						if (p == q)
+						{
+							destabilizerGenerators[q].X[q] = true;
+							stabilizerGenerators[q].Z[q] = true;
+						}
+						else
+						{
+							destabilizerGenerators[q].X[p] = false;
+							stabilizerGenerators[q].Z[p] = false;
+						}
+
+						destabilizerGenerators[q].PhaseSign = false;
+						stabilizerGenerators[q].PhaseSign = false;
+					}
+				}
+			}
+
 			bool MeasureQubit(size_t qubit)
 			{
 				size_t p;