Revert figures to static images on WebGL context loss

Author: jonmmease

packages/javascript/jupyterlab-plotly/src/javascript-renderer-extension.ts

@@ -148,6 +148,16 @@ export class RenderedPlotly extends Widget implements IRenderMime.IRenderer {
           }
         });
       }
+
+
+      (<Plotly.PlotlyHTMLElement>(this.node)).on('plotly_webglcontextlost', () => {
+            const png_data = <string>model.data['image/png'];
+            if(png_data !== undefined && png_data !== null) {
+              // We have PNG data, use it
+              this.createImage(png_data);
+              return Promise.resolve();
+            }
+          });
     });
   }
 

packages/javascript/jupyterlab-plotly/src/lib.d.ts

@@ -3,4 +3,8 @@ declare module 'plotly.js/dist/plotly' {
   export type Frame = { [key: string]: any };
   export function addFrames(root: Plotly.Root, frames: Frame[]): Promise<void>;
   export function animate(root: Plotly.Root): void;
+
+  export interface PlotlyHTMLElement extends HTMLElement {
+    on(event: 'plotly_webglcontextlost', callback: () => void): void;
+  }
 }