molstar
Version:
A comprehensive macromolecular library.
122 lines • 5.17 kB
JavaScript
;
/**
* Copyright (c) 2019-2020 mol* contributors, licensed under MIT, See LICENSE file for more info.
*
* @author Alexander Rose <alexander.rose@weirdbyte.de>
*/
Object.defineProperty(exports, "__esModule", { value: true });
exports.eachResidueAtom = exports.eachBondedAtom = exports.eachIntraBondedAtom = exports.eachInterBondedAtom = exports.connectedTo = exports.interConnectedTo = exports.intraConnectedTo = exports.bondToElementCount = exports.bondCount = exports.intraBondCount = exports.interBondCount = exports.compId = exports.altLoc = exports.atomId = exports.formalCharge = exports.typeSymbol = void 0;
var types_1 = require("../../../mol-model/structure/model/types");
var int_1 = require("../../../mol-data/int");
function typeSymbol(unit, index) {
return unit.model.atomicHierarchy.atoms.type_symbol.value(unit.elements[index]);
}
exports.typeSymbol = typeSymbol;
function formalCharge(unit, index) {
return unit.model.atomicHierarchy.atoms.pdbx_formal_charge.value(unit.elements[index]);
}
exports.formalCharge = formalCharge;
function atomId(unit, index) {
return unit.model.atomicHierarchy.atoms.label_atom_id.value(unit.elements[index]);
}
exports.atomId = atomId;
function altLoc(unit, index) {
return unit.model.atomicHierarchy.atoms.label_alt_id.value(unit.elements[index]);
}
exports.altLoc = altLoc;
function compId(unit, index) {
return unit.model.atomicHierarchy.atoms.label_comp_id.value(unit.elements[index]);
}
exports.compId = compId;
//
function interBondCount(structure, unit, index) {
var count = 0;
var indices = structure.interUnitBonds.getEdgeIndices(index, unit.id);
for (var i = 0, il = indices.length; i < il; ++i) {
var b = structure.interUnitBonds.edges[indices[i]];
if (types_1.BondType.isCovalent(b.props.flag))
count += 1;
}
return count;
}
exports.interBondCount = interBondCount;
function intraBondCount(unit, index) {
var count = 0;
var _a = unit.bonds, offset = _a.offset, flags = _a.edgeProps.flags;
for (var i = offset[index], il = offset[index + 1]; i < il; ++i) {
if (types_1.BondType.isCovalent(flags[i]))
count += 1;
}
return count;
}
exports.intraBondCount = intraBondCount;
function bondCount(structure, unit, index) {
return interBondCount(structure, unit, index) + intraBondCount(unit, index);
}
exports.bondCount = bondCount;
function bondToElementCount(structure, unit, index, element) {
var count = 0;
eachBondedAtom(structure, unit, index, function (unit, index) {
if (typeSymbol(unit, index) === element)
count += 1;
});
return count;
}
exports.bondToElementCount = bondToElementCount;
//
function intraConnectedTo(unit, indexA, indexB) {
var _a = unit.bonds, offset = _a.offset, b = _a.b, flags = _a.edgeProps.flags;
types_1.BondType.is;
for (var i = offset[indexA], il = offset[indexA + 1]; i < il; ++i) {
if (b[i] === indexB && types_1.BondType.isCovalent(flags[i]))
return true;
}
return false;
}
exports.intraConnectedTo = intraConnectedTo;
function interConnectedTo(structure, unitA, indexA, unitB, indexB) {
var b = structure.interUnitBonds.getEdge(indexA, unitA.id, indexB, unitB.id);
return b && types_1.BondType.isCovalent(b.props.flag);
}
exports.interConnectedTo = interConnectedTo;
function connectedTo(structure, unitA, indexA, unitB, indexB) {
return unitA === unitB ? intraConnectedTo(unitA, indexA, indexB) : interConnectedTo(structure, unitA, indexA, unitB, indexB);
}
exports.connectedTo = connectedTo;
//
function eachInterBondedAtom(structure, unit, index, cb) {
var indices = structure.interUnitBonds.getEdgeIndices(index, unit.id);
for (var i = 0, il = indices.length; i < il; ++i) {
var b = structure.interUnitBonds.edges[indices[i]];
var uB = structure.unitMap.get(b.unitB);
if (types_1.BondType.isCovalent(b.props.flag))
cb(uB, b.indexB);
}
}
exports.eachInterBondedAtom = eachInterBondedAtom;
function eachIntraBondedAtom(unit, index, cb) {
var _a = unit.bonds, offset = _a.offset, b = _a.b, flags = _a.edgeProps.flags;
for (var i = offset[index], il = offset[index + 1]; i < il; ++i) {
if (types_1.BondType.isCovalent(flags[i]))
cb(unit, b[i]);
}
}
exports.eachIntraBondedAtom = eachIntraBondedAtom;
function eachBondedAtom(structure, unit, index, cb) {
eachInterBondedAtom(structure, unit, index, cb);
eachIntraBondedAtom(unit, index, cb);
}
exports.eachBondedAtom = eachBondedAtom;
//
function eachResidueAtom(unit, index, cb) {
var offsets = unit.model.atomicHierarchy.residueAtomSegments.offsets;
var rI = unit.getResidueIndex(index);
for (var i = offsets[rI], il = offsets[rI + 1]; i < il; ++i) {
// TODO optimize, avoid search with .indexOf
var idx = int_1.SortedArray.indexOf(unit.elements, i);
if (idx !== -1)
cb(idx);
}
}
exports.eachResidueAtom = eachResidueAtom;
//# sourceMappingURL=util.js.map