From 9cbbd7e35fac46b0ae09a9fbeeb325095e8d3709 Mon Sep 17 00:00:00 2001 From: Alexander Rose <alex.rose@rcsb.org> Date: Wed, 27 Jun 2018 15:19:15 -0700 Subject: [PATCH] added entityKey support for coarse element properties --- src/mol-model/structure/structure/properties.ts | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/src/mol-model/structure/structure/properties.ts b/src/mol-model/structure/structure/properties.ts index 5a7b9e4d8..d13f5ddd8 100644 --- a/src/mol-model/structure/structure/properties.ts +++ b/src/mol-model/structure/structure/properties.ts @@ -89,7 +89,16 @@ const coarse = { gaussian_covariance_matrix: Element.property(l => !Unit.isGaussians(l.unit) ? notCoarse('gaussians') : l.unit.coarseConformation.covariance_matrix[l.element]) } -function eK(l: Element.Location) { return !Unit.isAtomic(l.unit) ? notAtomic() : l.unit.model.atomicHierarchy.entityKey[l.unit.chainIndex[l.element]]; } +function eK(l: Element.Location) { + switch (l.unit.kind) { + case Unit.Kind.Atomic: + return l.unit.model.atomicHierarchy.entityKey[l.unit.chainIndex[l.element]] + case Unit.Kind.Spheres: + return l.unit.model.coarseHierarchy.spheres.entityKey[l.element] + case Unit.Kind.Gaussians: + return l.unit.model.coarseHierarchy.gaussians.entityKey[l.element] + } +} const entity = { key: eK, -- GitLab