Added support for ClassicControlledOperation

src/QFRSimulator.cpp

@@ -11,16 +11,16 @@ void QFRSimulator::Simulate() {
 
     unsigned long op_num = 0;
 
-    std::map<unsigned int, bool> classic_values;
+    std::map<int, bool> classic_values;
 
     for (auto& op : *qc) {
-        if (!op->isUnitary()) {
-            if(auto* nu_op = dynamic_cast<qc::NonUnitaryOperation*>(op.get())) {
-                if (nu_op->getName()[0] == 'M') {
+        if (op->isNonUnitaryOperation()) {
+            if (auto *nu_op = dynamic_cast<qc::NonUnitaryOperation *>(op.get())) {
+                if (nu_op->getName()[0] == 'M') { // Measure starts with 'M', quite hacky though
                     auto quantum = nu_op->getControls();
                     auto classic = nu_op->getTargets();
 
-                    if(quantum.size() != classic.size()) {
+                    if (quantum.size() != classic.size()) {
                         std::cerr << "[ERROR] Measurement: Sizes of quantum and classic register mismatch.\n";
                         std::exit(1);
                     }
@@ -40,6 +40,17 @@ void QFRSimulator::Simulate() {
             }
             dd->garbageCollect();
         } else {
+            if (op->isClassicControlledOperation()) {
+                if (auto *cc_op = dynamic_cast<qc::ClassicControlledOperation *>(op.get())) {
+                    auto classic_bit_index = static_cast<short>(cc_op->getParameter().at(0)); // fp -> short *argh*
+                    if (!classic_values[classic_bit_index]) {
+                        continue;
+                    }
+                } else {
+                    std::cerr << "[ERROR] Dynamic cast to ClassicControlledOperation failed." << std::endl;
+                    std::exit(1);
+                }
+            }
             //std::clog << "[INFO] op " << op_num++ << " is " << op->getName()
             //          << " #controls=" << op->getControls().size()
             //          << " statesize=" << dd->size(root_edge) << "\n";

test/test_simple.cpp

@@ -14,6 +14,22 @@ TEST(SimulatorTest, SingleOneQubitGateOnTwoQubitCircuit) {
     ASSERT_EQ("01", m);
 }
 
+TEST(SimulatorTest, ClassicControlledOp) {
+    auto quantumComputation = std::make_unique<qc::QuantumComputation>(2);
+    quantumComputation->emplace_back<qc::StandardOperation>(2, 0, qc::X);
+    std::vector<unsigned short> qubit_to_measure = {0};
+    quantumComputation->emplace_back<qc::NonUnitaryOperation>(2, qubit_to_measure, qubit_to_measure);
+    std::unique_ptr<qc::Operation> op (new qc::StandardOperation(2, 1, qc::X));
+    quantumComputation->emplace_back<qc::ClassicControlledOperation>(op, 0);
+
+    QFRSimulator ddsim(quantumComputation);
+    ddsim.Simulate();
+
+    auto m = ddsim.MeasureAll(false);
+
+    ASSERT_EQ("11", m);
+}
+
 
 TEST(SimulatorTest, DestructiveMeasurementAll) {
     auto quantumComputation = std::make_unique<qc::QuantumComputation>(2);