diff --git a/CHANGELOG.md b/CHANGELOG.md index ae05abb51ae48b596525bbb267797a132c186c40..0e022d7ac4ace871e72769c72f3fc3c297ad04d3 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -16,6 +16,7 @@ Note that since we don't clearly distinguish between a public and private interf - Add per-pixel object clipping - Fix `QualityAssessment` assignment bug for structures with different auth vs label sequence numbering - Refresh `ApplyActionControl`'s param definition when toggling expanded state +- Fix `struct_conn` bond assignment for ions ## [v3.26.0] - 2022-12-04 diff --git a/src/mol-model/structure/structure/unit/bonds/inter-compute.ts b/src/mol-model/structure/structure/unit/bonds/inter-compute.ts index ab11f3f7b8598817c9c0ed68a1c953c5395e1686..69f51afb6240cc719a83338a62fda29a821b91a0 100644 --- a/src/mol-model/structure/structure/unit/bonds/inter-compute.ts +++ b/src/mol-model/structure/structure/unit/bonds/inter-compute.ts @@ -249,13 +249,29 @@ function computeInterUnitBonds(structure: Structure, props?: Partial<InterBondCo (!Unit.isAtomic(a) || mtA[a.residueIndex[a.elements[0]]] !== MoleculeType.Water) && (!Unit.isAtomic(b) || mtB[b.residueIndex[b.elements[0]]] !== MoleculeType.Water) ); - const notIon = ( - (!Unit.isAtomic(a) || mtA[a.residueIndex[a.elements[0]]] !== MoleculeType.Ion) && - (!Unit.isAtomic(b) || mtB[b.residueIndex[b.elements[0]]] !== MoleculeType.Ion) - ); + + const sameModel = a.model === b.model; + const notIonA = (!Unit.isAtomic(a) || mtA[a.residueIndex[a.elements[0]]] !== MoleculeType.Ion) || (sameModel && hasStructConnRecord(a)); + const notIonB = (!Unit.isAtomic(b) || mtB[b.residueIndex[b.elements[0]]] !== MoleculeType.Ion) || (sameModel && hasStructConnRecord(b)); + const notIon = notIonA && notIonB; return Structure.validUnitPair(s, a, b) && (notWater || !p.ignoreWater) && (notIon || !p.ignoreIon); }), }); } +function hasStructConnRecord(unit: Unit) { + if (!Unit.isAtomic(unit)) return false; + + const elements = unit.elements; + const structConn = StructConn.Provider.get(unit.model); + if (structConn) { + for (let i = 0, _i = elements.length; i < _i; i++) { + if (structConn.byAtomIndex.get(elements[i])) { + return true; + } + } + } + return false; +} + export { computeInterUnitBonds };