setColorByResidue

nglviewer · Jun 20, 2024 · f3ee76d · f3ee76d
1 parent 03f77d0
commit f3ee76d
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/js/src/widget_ngl.ts b/js/src/widget_ngl.ts
@@ -741,16 +741,17 @@ class NGLView extends widgets.DOMWidgetView{
             }
         }
     }
-    setColorByResidue(colors, component_index, repr_index){
-        var repr = this.stage.compList[component_index].reprList[repr_index];
-        var schemeId = NGL.ColormakerRegistry.addScheme(function(params){
+
+    setColorByResidue(colors, componentIndex, reprIndex){
+        const repr = this.stage.compList[componentIndex]?.reprList[reprIndex];
+        const schemeId = NGL.ColormakerRegistry.addScheme(function(params){
             this.atomColor = function(atom){
-                 var color = colors[atom.residueIndex];
-                 return color
+                const color = colors[atom.residueIndex];
+                return color;
             };
             params; // to pass eslint; ack;
         });
-        repr.setColor(schemeId);
+        repr?.setColor(schemeId);
     }
 
     addShape(name: string, shapes: any[]) {