molstar
Version:
A comprehensive macromolecular library.
92 lines (91 loc) • 5.22 kB
JavaScript
/**
* Copyright (c) 2018-2020 mol* contributors, licensed under MIT, See LICENSE file for more info.
*
* @author Alexander Rose <alexander.rose@weirdbyte.de>
*/
import { __awaiter, __generator } from "tslib";
import { Task } from '../../mol-task';
import { SpacegroupCell, Box3D } from '../../mol-math/geometry';
import { Tensor, Vec3 } from '../../mol-math/linear-algebra';
import { degToRad } from '../../mol-math/misc';
import { getCcp4ValueType } from '../../mol-io/reader/ccp4/parser';
import { TypedArrayValueType } from '../../mol-io/common/typed-array';
import { arrayMin, arrayRms, arrayMean, arrayMax } from '../../mol-util/array';
import { CustomProperties } from '../../mol-model/custom-property';
/** When available (e.g. in MRC files) use ORIGIN records instead of N[CRS]START */
export function getCcp4Origin(header) {
if (header.originX === 0.0 && header.originY === 0.0 && header.originZ === 0.0) {
return Vec3.create(header.NCSTART, header.NRSTART, header.NSSTART);
}
else {
return Vec3.create(header.originX / (header.xLength / header.NX), header.originY / (header.yLength / header.NY), header.originZ / (header.zLength / header.NZ));
}
}
function getTypedArrayCtor(header) {
var valueType = getCcp4ValueType(header);
switch (valueType) {
case TypedArrayValueType.Float32: return Float32Array;
case TypedArrayValueType.Int8: return Int8Array;
case TypedArrayValueType.Int16: return Int16Array;
case TypedArrayValueType.Uint16: return Uint16Array;
}
throw Error("".concat(valueType, " is not a supported value format."));
}
export function volumeFromCcp4(source, params) {
var _this = this;
return Task.create('Create Volume', function (ctx) { return __awaiter(_this, void 0, void 0, function () {
var header, values, size, angles, spacegroup, cell, axis_order_fast_to_slow, normalizeOrder, grid, extent, origin, gridOrigin, origin_frac, dimensions_frac, space, data, calcStats;
return __generator(this, function (_a) {
header = source.header, values = source.values;
size = Vec3.create(header.xLength, header.yLength, header.zLength);
if (params && params.voxelSize)
Vec3.mul(size, size, params.voxelSize);
angles = Vec3.create(degToRad(header.alpha), degToRad(header.beta), degToRad(header.gamma));
spacegroup = header.ISPG > 65536 ? 0 : header.ISPG;
cell = SpacegroupCell.create(spacegroup || 'P 1', size, angles);
axis_order_fast_to_slow = Vec3.create(header.MAPC - 1, header.MAPR - 1, header.MAPS - 1);
normalizeOrder = Tensor.convertToCanonicalAxisIndicesFastToSlow(axis_order_fast_to_slow);
grid = [header.NX, header.NY, header.NZ];
extent = normalizeOrder([header.NC, header.NR, header.NS]);
origin = getCcp4Origin(header);
if (params === null || params === void 0 ? void 0 : params.offset)
Vec3.add(origin, origin, params.offset);
gridOrigin = normalizeOrder(origin);
origin_frac = Vec3.create(gridOrigin[0] / grid[0], gridOrigin[1] / grid[1], gridOrigin[2] / grid[2]);
dimensions_frac = Vec3.create(extent[0] / grid[0], extent[1] / grid[1], extent[2] / grid[2]);
space = Tensor.Space(extent, Tensor.invertAxisOrder(axis_order_fast_to_slow), getTypedArrayCtor(header));
data = Tensor.create(space, Tensor.Data1(values));
calcStats = header.AMIN === 0 && header.AMAX === 0 && header.AMEAN === 0 && header.ARMS === 0;
return [2 /*return*/, {
label: params === null || params === void 0 ? void 0 : params.label,
entryId: params === null || params === void 0 ? void 0 : params.entryId,
grid: {
transform: { kind: 'spacegroup', cell: cell, fractionalBox: Box3D.create(origin_frac, Vec3.add(Vec3.zero(), origin_frac, dimensions_frac)) },
cells: data,
stats: {
min: (isNaN(header.AMIN) || calcStats) ? arrayMin(values) : header.AMIN,
max: (isNaN(header.AMAX) || calcStats) ? arrayMax(values) : header.AMAX,
mean: (isNaN(header.AMEAN) || calcStats) ? arrayMean(values) : header.AMEAN,
sigma: (isNaN(header.ARMS) || header.ARMS === 0) ? arrayRms(values) : header.ARMS
},
},
sourceData: Ccp4Format.create(source),
customProperties: new CustomProperties(),
_propertyData: Object.create(null),
}];
});
}); });
}
//
export { Ccp4Format };
var Ccp4Format;
(function (Ccp4Format) {
function is(x) {
return (x === null || x === void 0 ? void 0 : x.kind) === 'ccp4';
}
Ccp4Format.is = is;
function create(ccp4) {
return { kind: 'ccp4', name: ccp4.name, data: ccp4 };
}
Ccp4Format.create = create;
})(Ccp4Format || (Ccp4Format = {}));