Переглянути джерело

make sure only structure objects are filtered in for export

Yana Rose 4 роки тому
батько
коміт
e17f663c9d
1 змінених файлів з 3 додано та 1 видалено
  1. 3 1
      src/viewer/helpers/export.ts

+ 3 - 1
src/viewer/helpers/export.ts

@@ -63,7 +63,9 @@ function extractStructureDataFromState(plugin: PluginContext): { [k: string]: St
         const c = cells[i];
         const nodeRef = getDecorator(plugin, c.transform.ref);
         const children = plugin.state.data.tree.children.get(nodeRef).toArray()
-            .map(x => plugin.state.data.select(StateSelection.Generators.byRef(x!))[0].obj!.data as Structure);
+            .map(x => plugin.state.data.select(StateSelection.Generators.byRef(x!))[0])
+            .filter(c => c.obj?.type === PluginStateObject.Molecule.Structure.type)
+            .map(x => x.obj!.data as Structure);
         const sele = StructureSelection.Sequence(c.obj!.data, children);
         const structure = StructureSelection.unionStructure(sele);
         const name = `${i + 1}-${structure.model.entryId}`;