exposing operator name

src/viewer/helpers/selection.ts

@@ -44,6 +44,7 @@ export type Target = {
 export type SelectBase = {
     readonly modelId: string
     readonly labelAsymId: string
+    readonly operatorName?: string
 }
 export type SelectSingle = {
     readonly labelSeqId: number
@@ -186,14 +187,18 @@ const labelFromProps = (entryId: string, labelAsymId?: string, range?: number[])
         (range && range.length > 1 ? `-${range[range.length - 1]}` : '');
 };
 
-export function rangeToTest(asymId: string, residues: number[]) {
+export function rangeToTest(asymId: string, residues: number[], operatorName?: string) {
+    const chainTests: Expression[] = [MS.core.rel.eq([MS.ammp('label_asym_id'), asymId])];
+    if(operatorName)
+        chainTests.push(MS.core.rel.eq([operatorName, MS.acp('operatorName')]));
+
     if (residues.length > 0) {
         return {
-            'chain-test': MS.core.rel.eq([MS.ammp('label_asym_id'), asymId]),
+            'chain-test': MS.core.logic.and(chainTests),
             'residue-test': MS.core.set.has([MS.set(...residues), MS.ammp('label_seq_id')])
         };
     } else {
-        return { 'chain-test': MS.core.rel.eq([MS.ammp('label_asym_id'), asymId]) };
+        return { 'chain-test': MS.core.logic.and(chainTests) };
     }
 }
 

src/viewer/helpers/viewer.ts

@@ -47,10 +47,10 @@ export function select(plugin: PluginContext, targets: SelectTarget | SelectTarg
     if (modifier === 'set')
         clearSelection(plugin, mode);
     (Array.isArray(targets) ? targets : [targets]).forEach((target, n)=>{
-        const data = getStructureWithModelId(plugin.managers.structure.hierarchy.current.structures, target);
-        if (!data) return;
+        const structure = getStructureWithModelId(plugin.managers.structure.hierarchy.current.structures, target);
+        if (!structure) return;
 
-        const loci = targetToLoci(target, data);
+        const loci = targetToLoci(target, structure);
         if (!loci) return;
 
         if (mode === 'hover') {
@@ -86,7 +86,7 @@ export async function createComponent(plugin: PluginContext, componentLabel: str
 
         const residues = toResidues(target);
         const sel = StructureSelectionQuery('innerQuery_' + Math.random().toString(36).substr(2),
-            MS.struct.generator.atomGroups(rangeToTest(target.labelAsymId, residues)));
+            MS.struct.generator.atomGroups(rangeToTest(target.labelAsymId, residues, target.operatorName)));
         await plugin.managers.structure.component.add({
             selection: sel,
             options: { checkExisting: false, label: componentLabel },