allow custom gates to be overwritten

CHANGELOG.md

@@ -25,6 +25,7 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 - `@qiskit/qiskit-sim`: add prettyMatrix method to Gate class
 - `@qiskit/qiskit-sim`: add gates to gates map in constructor
 - `@qiskit/qiskit-sim`: remove unused addGate function in grammar.jison
+- `@qiskit/qiskit-sim`: allow custom gates to be overwritten
 
 ## [0.9.0] - 2019-05-13
 

packages/qiskit-sim/lib/gates.js

@@ -43,12 +43,13 @@ const phaseShift = shift =>
   [[1, 0], [0, math.pow(math.e, math.multiply(math.i, math.PI / shift))]];
 
 const gates = new Map();
+let builtinGateNames;
 
 class Gate {
   constructor(name, matrix) {
     this.name = name;
     this.matrix = matrix;
-    if (!gates.has(name)) {
+    if (!builtinGateNames || !builtinGateNames.has(name)) {
       gates.set(name, this);
     }
   }
@@ -103,6 +104,7 @@ Gate.cs = new Gate('cs', buildControlled(Gate.s));
 Gate.cr2 = new Gate('cr2', buildControlled(Gate.r2));
 Gate.cr4 = new Gate('cr4', buildControlled(Gate.r4));
 Gate.cr8 = new Gate('cr8', buildControlled(Gate.r8));
+builtinGateNames = new Set(gates.keys());
 
 module.exports = {
   Gate,

packages/qiskit-sim/test/functional/gates.js

@@ -55,4 +55,14 @@ describe('sim:gates', () => {
     const custom = new Gate('custom', [[1, 0], [0, 1]]);
     assert.ok(gates.has(custom.name));
   });
+
+  it('custom gate should be allowed to be overwritten', () => {
+    const custom = new Gate('custom', [[1, 0], [0, 1]]);
+    const newMatrix = [[1, 0],
+                      [1, 1]];
+    const overwrite = new Gate(custom.name, newMatrix);
+    assert.ok(gates.has(overwrite.name));
+    assert.strictEqual(gates.get(custom.name).matrix, newMatrix);
+  });
+
 });