Use same NaN condition for amplitude display as for chance display

adgt · Mar 24, 2019 · 5cdbbaa · 5cdbbaa
1 parent 3ad0ce5
commit 5cdbbaa
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/gates/AmplitudeDisplay.js b/src/gates/AmplitudeDisplay.js
@@ -113,7 +113,7 @@ function amplitudeDisplayStatTextures(stateKet, controls, controlsTexture, range
  * @param {!int} span
  * @param {!Array.<!Float32Array>} pixelGroups
  * @param {!CircuitDefinition} circuitDefinition
- * @returns {!{quality: !float, ket: undefined|!Matrix, phaseLockIndex: undefined|!int}}
+ * @returns {!{quality: !number, ket: !Matrix, phaseLockIndex: !int,incoherentKet: !Matrix}}
  */
 function processOutputs(span, pixelGroups, circuitDefinition) {
     let [ketPixels, qualityPixels, rawIncoherentKetPixels] = pixelGroups;
@@ -128,9 +128,19 @@ function processOutputs(span, pixelGroups, circuitDefinition) {
         unity += e*e;
     }
     let incoherentKetPixels = new Float32Array(w * h * 2);
+    let incoherentUnity = 0;
     for (let i = 0; i < n; i++) {
+        incoherentUnity += rawIncoherentKetPixels[i];
         incoherentKetPixels[i << 1] = Math.sqrt(rawIncoherentKetPixels[i]);
     }
+    if (isNaN(unity) || unity < 0.000001) {
+        return {
+            quality: 0.0,
+            ket: Matrix.zero(w, h).times(NaN),
+            phaseLockIndex: 0,
+            incoherentKet: Matrix.zero(w, h).times(NaN),
+        };
+    }
 
     let phaseIndex = span === circuitDefinition.numWires ? undefined : _processOutputs_pickPhaseLockIndex(ketPixels);
     let phase = phaseIndex === undefined ? 0 : Math.atan2(ketPixels[phaseIndex*2+1], ketPixels[phaseIndex*2]);
@@ -322,7 +332,7 @@ function paintErrorIfPresent(args, indicatorAlpha) {
 }
 
 /**
- * @param {!{quality: !float, ket: undefined|!Matrix, phaseLockIndex: undefined|!int}} customStats
+ * @param {!{quality: !number, ket: !Matrix, phaseLockIndex: !int,incoherentKet: !Matrix}} customStats
  */
 function customStatsToJsonData(customStats) {
     let {quality, ket, phaseLockIndex, incoherentKet} = customStats;