molstar
Version:
A comprehensive macromolecular library.
92 lines • 3.73 kB
JavaScript
;
/**
* Copyright (c) 2021 mol* contributors, licensed under MIT, See LICENSE file for more info.
*
* @author David Sehnal <david.sehnal@gmail.com>
*/
Object.defineProperty(exports, "__esModule", { value: true });
exports.trajectoryFromXyz = exports.XyzFormat = void 0;
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");
function getModels(mol, ctx) {
var molecules = mol.molecules;
var count = 0;
for (var _i = 0, molecules_1 = molecules; _i < molecules_1.length; _i++) {
var m = molecules_1[_i];
count += m.count;
}
var type_symbols = new Array(count);
var id = new Int32Array(count);
var x = new Float32Array(count);
var y = new Float32Array(count);
var z = new Float32Array(count);
var model_num = new Int32Array(count);
var offset = 0;
for (var i = 0; i < molecules.length; i++) {
var m = molecules[i];
for (var j = 0; j < m.count; j++) {
type_symbols[offset] = m.type_symbol.value(j);
x[offset] = m.x.value(j);
y[offset] = m.y.value(j);
z[offset] = m.z.value(j);
id[offset] = j;
model_num[offset] = i;
offset++;
}
}
var MOL = db_1.Column.ofConst('MOL', count, db_1.Column.Schema.str);
var A = db_1.Column.ofConst('A', count, db_1.Column.Schema.str);
var seq_id = db_1.Column.ofConst(1, count, db_1.Column.Schema.int);
var type_symbol = db_1.Column.ofStringArray(type_symbols);
var atom_site = db_1.Table.ofPartialColumns(schema_1.BasicSchema.atom_site, {
auth_asym_id: A,
auth_atom_id: type_symbol,
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.ofIntArray(id),
label_asym_id: A,
label_atom_id: type_symbol,
label_comp_id: MOL,
label_seq_id: seq_id,
label_entity_id: db_1.Column.ofConst('1', count, db_1.Column.Schema.str),
occupancy: db_1.Column.ofConst(1, count, db_1.Column.Schema.float),
type_symbol: type_symbol,
pdbx_PDB_model_num: db_1.Column.ofIntArray(model_num),
}, count);
var entityBuilder = new entity_1.EntityBuilder();
entityBuilder.setNames([['MOL', 'Unknown Entity']]);
entityBuilder.getEntityId('MOL', 0 /* Unknown */, 'A');
var componentBuilder = new component_1.ComponentBuilder(seq_id, type_symbol);
componentBuilder.setNames([['MOL', 'Unknown Molecule']]);
componentBuilder.add('MOL', 0);
var basics = (0, schema_1.createBasic)({
entity: entityBuilder.getEntityTable(),
chem_comp: componentBuilder.getChemCompTable(),
atom_site: atom_site
});
return (0, parser_1.createModels)(basics, XyzFormat.create(mol), ctx);
}
var XyzFormat;
(function (XyzFormat) {
function is(x) {
return (x === null || x === void 0 ? void 0 : x.kind) === 'xyz';
}
XyzFormat.is = is;
function create(mol) {
return { kind: 'xyz', name: 'xyz', data: mol };
}
XyzFormat.create = create;
})(XyzFormat || (XyzFormat = {}));
exports.XyzFormat = XyzFormat;
function trajectoryFromXyz(mol) {
return mol_task_1.Task.create('Parse XYZ', function (ctx) { return getModels(mol, ctx); });
}
exports.trajectoryFromXyz = trajectoryFromXyz;
//# sourceMappingURL=xyz.js.map