Change Namespace

Author: adamisntdead

src/gates.cpp

@@ -46,63 +46,64 @@ float TDagger_coef[] = {
     1, 0, 0, 0,
     0, 0, 0.7071067811865475244008443621048490393, -0.7071067811865475244008443621048490393
 };
+namespace QC {
+  namespace Gates {
+      // Pauli-X / Not Gate
+      af::array X = af::array(2, 2, (af::cfloat*) X_coef);
 
-namespace Gates {
-    // Pauli-X / Not Gate
-    af::array X = af::array(2, 2, (af::cfloat*) X_coef);
+      // Pauli-Y Gate
+      af::array Y = af::array(2, 2, (af::cfloat*) Y_coef);
 
-    // Pauli-Y Gate
-    af::array Y = af::array(2, 2, (af::cfloat*) Y_coef);
+      // Pauli-Z Gate
+      af::array Z = af::array(2, 2, (af::cfloat*) Z_coef);
 
-    // Pauli-Z Gate
-    af::array Z = af::array(2, 2, (af::cfloat*) Z_coef);
+      // Hadamard Gate
+      af::array H = (1/sqrt(2)) * af::array(2, 2, (af::cfloat*) H_coef);
 
-    // Hadamard Gate
-    af::array H = (1/sqrt(2)) * af::array(2, 2, (af::cfloat*) H_coef);
+      // Identity Gate
+      af::array Id = af::array(2, 2, (af::cfloat*) Id_coef);
 
-    // Identity Gate
-    af::array Id = af::array(2, 2, (af::cfloat*) Id_coef);
+      // S Gate
+      af::array S = af::array(2, 2, (af::cfloat*) S_coef);
 
-    // S Gate
-    af::array S = af::array(2, 2, (af::cfloat*) S_coef);
+      // S-Dagger Gate
+      af::array SDagger = af::array(2, 2, (af::cfloat*) SDagger_coef);
 
-    // S-Dagger Gate
-    af::array SDagger = af::array(2, 2, (af::cfloat*) SDagger_coef);
+      // T / Pi over 8 Gate
+      af::array T = af::array(2, 2, (af::cfloat*) T_coef);
 
-    // T / Pi over 8 Gate
-    af::array T = af::array(2, 2, (af::cfloat*) T_coef);
-
-    // T-Dagger Gate
-    af::array TDagger = af::array(2, 2, (af::cfloat*) TDagger_coef);
-}
+      // T-Dagger Gate
+      af::array TDagger = af::array(2, 2, (af::cfloat*) TDagger_coef);
+  }
 
-af::array generateGate(af::array gate, int numQubits, int a) {
-  // identity = 1
-  af::array id = af::identity(2, 2, c32);
-  af::array K;
+  af::array generateGate(af::array gate, int numQubits, int a) {
+    // identity = 1
+    af::array id = af::identity(2, 2, c32);
+    af::array K;
+
+    for (int i = 0; i < numQubits; i++) {
+      if (i == 0) {
+        if (a == 0) {
+          K = gate;
+        } else {
+          K = id;
+        }
+      } else if (i == a) {
+        try {
+          K = kron(K, gate);
+        } catch (const std::exception& e) {
+          std::cout << e.what();
+        }
 
-  for (int i = 0; i < numQubits; i++) {
-    if (i == 0) {
-      if (a == 0) {
-        K = gate;
       } else {
-        K = id;
-      }
-    } else if (i == a) {
-      try {
-        K = kron(K, gate);
-      } catch (const std::exception& e) {
-        std::cout << e.what();
-      }
-
-    } else {
-      try {
-        K = kron(K, id);
-      } catch (const std::exception& e) {
-        std::cout << e.what();
+        try {
+          K = kron(K, id);
+        } catch (const std::exception& e) {
+          std::cout << e.what();
+        }
       }
     }
-  }
 
-  return K;
-}
+    return K;
+  }
+}

src/gates.h

@@ -2,19 +2,21 @@
 #define GATES_H_
 
 #include <arrayfire.h>
+namespace QC {
+    namespace Gates {
+        extern af::array X;
+        extern af::array Y;
+        extern af::array Z;
+        extern af::array H;
+        extern af::array Id;
+        extern af::array S;
+        extern af::array SDagger;
+        extern af::array T;
+        extern af::array TDagger;
+    }
 
-namespace Gates {
-    extern af::array X;
-    extern af::array Y;
-    extern af::array Z;
-    extern af::array H;
-    extern af::array Id;
-    extern af::array S;
-    extern af::array SDagger;
-    extern af::array T;
-    extern af::array TDagger;
-}
+    af::array generateGate(af::array gate, int numQubits, int a);
 
-af::array generateGate(af::array gate, int numQubits, int a);
+}
 
 #endif

src/main.cpp

@@ -1,19 +1,19 @@
 #include <arrayfire.h>
 #include <iostream>
+
 #include "gates.h"
 #include "quantum.h"
-#include "kron.h"
 
 using namespace af;
 
 int main(int argc, char** argv) {
   //af::info();
 
-  QReg q(3);
+  QC::QReg q(3);
 
-  q.applyGate(Gates::H, 0);
-  q.applyGate(Gates::H, 1);
-  q.applyGate(Gates::H, 2);
+  q.applyGate(QC::Gates::H, 0);
+  q.applyGate(QC::Gates::H, 1);
+  q.applyGate(QC::Gates::H, 2);
 
   float results[8] = {0,0,0,0,0,0,0,0};
 

src/quantum.cpp

@@ -4,25 +4,27 @@
 
 #include "gates.h"
 #include "helper.h"
-
-QReg::QReg(int n) 
-  : numQubits(n) {
-    amplitudes = af::constant(0, 2 << (n - 1), c32);
-    amplitudes(0) = 1;
-    measured = false;
-}
-
-void QReg::applyGate(af::array gate, int a) {
-  amplitudes = af::matmul(generateGate(gate, numQubits, a), amplitudes);
+namespace QC {
+  QReg::QReg(int n) 
+    : numQubits(n) {
+      amplitudes = af::constant(0, 2 << (n - 1), c32);
+      amplitudes(0) = 1;
+      measured = false;
+  }
+
+  void QReg::applyGate(af::array gate, int a) {
+    amplitudes = af::matmul(generateGate(gate, numQubits, a), amplitudes);
+  }
+
+  int QReg::measure() {
+    af::array probabilities = af::pow(af::abs(amplitudes), 2);
+
+    return select(probabilities);
+  }
+
+  af::array QReg::getAmplitudes() {
+    return amplitudes;
+  }
 }
 
-int QReg::measure() {
-  af::array probabilities = af::pow(af::abs(amplitudes), 2);
-
-  return select(probabilities);
-}
-
-af::array QReg::getAmplitudes() {
-  return amplitudes;
-}
 

src/quantum.h

@@ -2,8 +2,8 @@
 #define QUANTUM_H_
 
 #include <arrayfire.h>
-
-class QReg {
+namespace QC {
+ class QReg {
  private:
   af::array amplitudes;
   int numQubits;
@@ -16,4 +16,6 @@ class QReg {
   af::array getAmplitudes();
 };
 
+}
+
 #endif