molstar
Version:
A comprehensive macromolecular library.
258 lines • 14.7 kB
JavaScript
;
/**
* Copyright (c) 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.trajectoryFromCifCore = exports.CifCoreFormat = void 0;
var tslib_1 = require("tslib");
var db_1 = require("../../mol-data/db");
var mol_task_1 = require("../../mol-task");
var parser_1 = require("./basic/parser");
var schema_1 = require("./basic/schema");
var component_1 = require("./common/component");
var entity_1 = require("./common/entity");
var cif_1 = require("../../mol-io/reader/cif");
var geometry_1 = require("../../mol-math/geometry");
var linear_algebra_1 = require("../../mol-math/linear-algebra");
var symmetry_1 = require("./property/symmetry");
var index_pair_1 = require("./property/bonds/index-pair");
var anisotropic_1 = require("./property/anisotropic");
var util_1 = require("./util");
var util_2 = require("../../mol-data/util");
function getSpacegroupNameOrNumber(space_group) {
var groupNumber = space_group.it_number.value(0);
var groupName = space_group['name_h-m_full'].value(0).replace('-', ' ');
if (!space_group.it_number.isDefined)
return groupName;
if (!space_group['name_h-m_full'].isDefined)
return groupNumber;
return groupNumber;
}
function getSymmetry(db) {
var cell = db.cell, space_group = db.space_group;
var nameOrNumber = getSpacegroupNameOrNumber(space_group);
var spaceCell = geometry_1.SpacegroupCell.create(nameOrNumber, linear_algebra_1.Vec3.create(cell.length_a.value(0), cell.length_b.value(0), cell.length_c.value(0)), linear_algebra_1.Vec3.scale((0, linear_algebra_1.Vec3)(), linear_algebra_1.Vec3.create(cell.angle_alpha.value(0), cell.angle_beta.value(0), cell.angle_gamma.value(0)), Math.PI / 180));
return {
spacegroup: geometry_1.Spacegroup.create(spaceCell),
assemblies: [],
isNonStandardCrystalFrame: false,
ncsOperators: []
};
}
function getModels(db, format, ctx) {
var _a;
return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
var atomCount, MOL, A, seq_id, symmetry, m, _b, fract_x, fract_y, fract_z, x, y, z, v, i, _c, type_symbol, label, typeSymbol, formalCharge, element_symbol, formal_charge, i, ts, n, element_symbol, i, atom_site, name, entityBuilder, componentBuilder, basics, models, first, bondCount, labelIndexMap, label_1, i, il, bond_type, indexA, indexB, order, dist, flag, included, j, _d, atom_site_label_1, atom_site_label_2, valence, distance, i, iA, iB, id, t;
return (0, tslib_1.__generator)(this, function (_e) {
switch (_e.label) {
case 0:
atomCount = db.atom_site._rowCount;
MOL = db_1.Column.ofConst('MOL', atomCount, db_1.Column.Schema.str);
A = db_1.Column.ofConst('A', atomCount, db_1.Column.Schema.str);
seq_id = db_1.Column.ofConst(1, atomCount, db_1.Column.Schema.int);
symmetry = getSymmetry(db);
m = symmetry.spacegroup.cell.fromFractional;
_b = db.atom_site, fract_x = _b.fract_x, fract_y = _b.fract_y, fract_z = _b.fract_z;
x = new Float32Array(atomCount);
y = new Float32Array(atomCount);
z = new Float32Array(atomCount);
v = (0, linear_algebra_1.Vec3)();
for (i = 0; i < atomCount; ++i) {
linear_algebra_1.Vec3.set(v, fract_x.value(i), fract_y.value(i), fract_z.value(i));
linear_algebra_1.Vec3.transformMat4(v, v, m);
x[i] = v[0], y[i] = v[1], z[i] = v[2];
}
_c = db.atom_site, type_symbol = _c.type_symbol, label = _c.label;
if (type_symbol.isDefined) {
element_symbol = new Array(atomCount);
formal_charge = new Int8Array(atomCount);
for (i = 0; i < atomCount; ++i) {
ts = type_symbol.value(i);
n = ts.length;
if (ts[n - 1] === '+') {
element_symbol[i] = ts.substring(0, n - 2);
formal_charge[i] = parseInt(ts[n - 2]);
}
else if (ts[n - 2] === '+') {
element_symbol[i] = ts.substring(0, n - 2);
formal_charge[i] = parseInt(ts[n - 1]);
}
else if (ts[n - 1] === '-') {
element_symbol[i] = ts.substring(0, n - 2);
formal_charge[i] = -parseInt(ts[n - 2]);
}
else if (ts[n - 2] === '-') {
element_symbol[i] = ts.substring(0, n - 2);
formal_charge[i] = -parseInt(ts[n - 1]);
}
else {
element_symbol[i] = ts;
formal_charge[i] = 0;
}
}
typeSymbol = db_1.Column.ofStringArray(element_symbol);
formalCharge = db_1.Column.ofIntArray(formal_charge);
}
else {
element_symbol = new Array(atomCount);
for (i = 0; i < atomCount; ++i) {
// TODO can take as is if type_symbol not given?
element_symbol[i] = (0, util_1.guessElementSymbolString)(label.value(i));
}
typeSymbol = db_1.Column.ofStringArray(element_symbol);
formalCharge = db_1.Column.Undefined(atomCount, db_1.Column.Schema.int);
}
atom_site = db_1.Table.ofPartialColumns(schema_1.BasicSchema.atom_site, {
auth_asym_id: A,
auth_atom_id: label,
auth_comp_id: MOL,
auth_seq_id: seq_id,
Cartn_x: db_1.Column.ofFloatArray(x),
Cartn_y: db_1.Column.ofFloatArray(y),
Cartn_z: db_1.Column.ofFloatArray(z),
id: db_1.Column.range(0, atomCount - 1),
label_asym_id: A,
label_atom_id: label,
label_comp_id: MOL,
label_seq_id: seq_id,
label_entity_id: db_1.Column.ofConst('1', atomCount, db_1.Column.Schema.str),
occupancy: db.atom_site.occupancy.isDefined
? db.atom_site.occupancy
: db_1.Column.ofConst(1, atomCount, db_1.Column.Schema.float),
type_symbol: typeSymbol,
pdbx_formal_charge: formalCharge,
pdbx_PDB_model_num: db_1.Column.ofConst(1, atomCount, db_1.Column.Schema.int),
B_iso_or_equiv: db.atom_site.u_iso_or_equiv,
}, atomCount);
name = (db.chemical.name_common.value(0) ||
db.chemical.name_systematic.value(0) ||
db.chemical_formula.sum.value(0));
entityBuilder = new entity_1.EntityBuilder();
entityBuilder.setNames([['MOL', name || 'Unknown Entity']]);
entityBuilder.getEntityId('MOL', 0 /* Unknown */, 'A');
componentBuilder = new component_1.ComponentBuilder(seq_id, db.atom_site.type_symbol);
componentBuilder.setNames([['MOL', name || 'Unknown Molecule']]);
componentBuilder.add('MOL', 0);
basics = (0, schema_1.createBasic)({
entity: entityBuilder.getEntityTable(),
chem_comp: componentBuilder.getChemCompTable(),
atom_site: atom_site
});
return [4 /*yield*/, (0, parser_1.createModels)(basics, format, ctx)];
case 1:
models = _e.sent();
if (models.frameCount > 0) {
first = models.representative;
symmetry_1.ModelSymmetry.Provider.set(first, symmetry);
bondCount = db.geom_bond._rowCount;
if (bondCount > 0) {
labelIndexMap = {};
label_1 = db.atom_site.label;
for (i = 0, il = label_1.rowCount; i < il; ++i) {
labelIndexMap[label_1.value(i)] = i;
}
bond_type = (_a = format.data.frame.categories.ccdc_geom_bond_type) === null || _a === void 0 ? void 0 : _a.getField('');
indexA = [];
indexB = [];
order = [];
dist = [];
flag = [];
included = new Set();
j = 0;
_d = db.geom_bond, atom_site_label_1 = _d.atom_site_label_1, atom_site_label_2 = _d.atom_site_label_2, valence = _d.valence, distance = _d.distance;
for (i = 0; i < bondCount; ++i) {
iA = labelIndexMap[atom_site_label_1.value(i)];
iB = labelIndexMap[atom_site_label_2.value(i)];
id = iA < iB ? (0, util_2.cantorPairing)(iA, iB) : (0, util_2.cantorPairing)(iB, iA);
if (included.has(id))
continue;
included.add(id);
indexA[j] = iA;
indexB[j] = iB;
dist[j] = distance.value(i) || -1;
if (bond_type) {
t = bond_type.str(i);
if (t === 'D') {
order[j] = 2;
flag[j] = 1 /* Covalent */;
}
else if (t === 'A') {
order[j] = 1;
flag[j] = 1 /* Covalent */ | 16 /* Aromatic */;
}
else if (t === 'S') {
order[j] = 1;
flag[j] = 1 /* Covalent */;
}
else {
order[j] = 1;
flag[j] = 1 /* Covalent */;
}
}
else {
flag[j] = 1 /* Covalent */;
// TODO derive order from bond length if undefined
order[j] = valence.isDefined ? valence.value(i) : 1;
}
j += 1;
}
index_pair_1.IndexPairBonds.Provider.set(first, index_pair_1.IndexPairBonds.fromData({ pairs: {
indexA: db_1.Column.ofIntArray(indexA),
indexB: db_1.Column.ofIntArray(indexB),
order: db_1.Column.ofIntArray(order),
distance: db_1.Column.ofFloatArray(dist),
flag: db_1.Column.ofIntArray(flag)
}, count: atomCount }));
}
}
return [2 /*return*/, models];
}
});
});
}
function atomSiteAnisotropFromCifCore(model) {
if (!CifCoreFormat.is(model.sourceData))
return;
var _a = model.sourceData.data.db, atom_site = _a.atom_site, atom_site_aniso = _a.atom_site_aniso;
var data = db_1.Table.ofPartialColumns(anisotropic_1.AtomSiteAnisotrop.Schema, {
U: atom_site_aniso.u,
}, atom_site_aniso._rowCount);
var elementToAnsiotrop = anisotropic_1.AtomSiteAnisotrop.getElementToAnsiotropFromLabel(atom_site.label, atom_site_aniso.label);
return { data: data, elementToAnsiotrop: elementToAnsiotrop };
}
function atomSiteAnisotropApplicableCifCore(model) {
if (!CifCoreFormat.is(model.sourceData))
return false;
return model.sourceData.data.db.atom_site_aniso.u.isDefined;
}
anisotropic_1.AtomSiteAnisotrop.Provider.formatRegistry.add('cifCore', atomSiteAnisotropFromCifCore, atomSiteAnisotropApplicableCifCore);
var CifCoreFormat;
(function (CifCoreFormat) {
function is(x) {
return (x === null || x === void 0 ? void 0 : x.kind) === 'cifCore';
}
CifCoreFormat.is = is;
function fromFrame(frame, db) {
if (!db)
db = cif_1.CIF.schema.cifCore(frame);
var name = (db.database_code.depnum_ccdc_archive.value(0) ||
db.database_code.depnum_ccdc_fiz.value(0) ||
db.database_code.icsd.value(0) ||
db.database_code.mdf.value(0) ||
db.database_code.nbs.value(0) ||
db.database_code.csd.value(0) ||
db.database_code.cod.value(0) ||
db._name);
return { kind: 'cifCore', name: name, data: { db: db, frame: frame } };
}
CifCoreFormat.fromFrame = fromFrame;
})(CifCoreFormat || (CifCoreFormat = {}));
exports.CifCoreFormat = CifCoreFormat;
function trajectoryFromCifCore(frame) {
var format = CifCoreFormat.fromFrame(frame);
return mol_task_1.Task.create('Parse CIF Core', function (ctx) { return getModels(format.data.db, format, ctx); });
}
exports.trajectoryFromCifCore = trajectoryFromCifCore;
//# sourceMappingURL=cif-core.js.map