Improved exportToGenerator method

Author: perak

lib/quantum-circuit.js

@@ -5042,7 +5042,6 @@ QuantumCircuit.prototype.exportToGenerator = function(options) {
 	var qasm = this.exportQASM(null, false, null, false, true);
 	var usedGates = this.usedGates({ noCustomGates: true, unitariesOnly: true });
 	var couplingMap = this.getCouplingMap({ unitariesOnly: true });
-	var numSamples = options.numSamples || (this.numQubits < 5 ? ((1 << this.numQubits) + 1) : (this.numQubits < 9 ? (this.numQubits * 4) : this.numQubits));
 
 	var output = {
 		source: { 
@@ -5064,40 +5063,60 @@ QuantumCircuit.prototype.exportToGenerator = function(options) {
 	var circuit = new QuantumCircuit();
 	circuit.load(this.save(true));
 
-	var stateCircuit = new QuantumCircuit();
+	if(options.fromMatrix) {
+		var matrix = circuit.circuitMatrix();
+		var unitary = [];
 
-	for(var i = 0; i < numSamples + 1; i++) {
+		matrix.map(function(row) {
+			row.map(function(value) {
+				if(typeof value == "object") {
+					unitary.push([value.re, value.im]);
+				} else {
+					unitary.push(value);
+				}
+			});
+		});
 
-		var set = { input: [], output: [] };
+		output.problem.push({
+			unitary: unitary
+		});
+	} else {
+		var stateCircuit = new QuantumCircuit();
 
-		var initialVector = [];
+		var numSamples = options.numSamples || (this.numQubits < 3 ? 8 : (this.numQubits < 5 ? ((1 << this.numQubits) + 1) : (this.numQubits < 9 ? (this.numQubits * 4) : this.numQubits)));
+		for(var i = 0; i < numSamples + 1; i++) {
 
-		if(i > 0) {
-			stateCircuit.randomCircuit(this.numQubits, 20, { useGates: ["u3", "cx"], noClassicControl: true, noMeasure: true, noReset: true });
-			stateCircuit.run();
+			var set = { input: [], output: [] };
 
-			stateCircuit.stateAsArray().map(function(state) {
-				set.input.push([ state.amplitude.re, state.amplitude.im ]);
-			});
+			var initialVector = [];
 
-			initialVector = stateCircuit.state;
-		} else {
-			circuit.initState();
+			if(i > 0) {
+				stateCircuit.randomCircuit(this.numQubits, 20, { useGates: ["u3", "cx"], noClassicControl: true, noMeasure: true, noReset: true });
+				stateCircuit.run();
 
-			circuit.stateAsArray().map(function(state) {
-				set.input.push([ state.amplitude.re, state.amplitude.im ]);
-			});
+				stateCircuit.stateAsArray().map(function(state) {
+					set.input.push([ state.amplitude.re, state.amplitude.im ]);
+				});
 
-			initialVector = circuit.state;
-		}
+				initialVector = stateCircuit.state;
+			} else {
+				circuit.initState();
 
-		circuit.run(null, { initialState: initialVector });
+				circuit.stateAsArray().map(function(state) {
+					set.input.push([ state.amplitude.re, state.amplitude.im ]);
+				});
 
-		circuit.stateAsArray().map(function(state) {
-			set.output.push([ state.amplitude.re, state.amplitude.im ]);
-		});
+				initialVector = circuit.state;
+			}
 
-		output.problem.push(set);
+			circuit.run(null, { initialState: initialVector });
+
+			circuit.stateAsArray().map(function(state) {
+				set.output.push([ state.amplitude.re, state.amplitude.im ]);
+			});
+
+			output.problem.push(set);
+		}
 	}
 
 	return output;