molstar
Version:
A comprehensive macromolecular library.
255 lines (254 loc) • 9.74 kB
JavaScript
/**
* Copyright (c) 2017-2021 mol* contributors, licensed under MIT, See LICENSE file for more info.
*
* Adapted from CIFTools.js (https://github.com/dsehnal/CIFTools.js)
*
* @author David Sehnal <david.sehnal@gmail.com>
*/
Object.defineProperty(exports, "__esModule", { value: true });
exports.TextEncoder = void 0;
const string_builder_1 = require("../../../../mol-util/string-builder");
const encoder_1 = require("../encoder");
const util_1 = require("./util");
class TextEncoder {
constructor() {
this.builder = string_builder_1.StringBuilder.create();
this.encoded = false;
this.dataBlockCreated = false;
this.filter = encoder_1.Category.DefaultFilter;
this.formatter = encoder_1.Category.DefaultFormatter;
this.isBinary = false;
this.binaryEncodingProvider = void 0;
}
setFilter(filter) {
this.filter = filter || encoder_1.Category.DefaultFilter;
}
isCategoryIncluded(name) {
return this.filter.includeCategory(name);
}
setFormatter(formatter) {
this.formatter = formatter || encoder_1.Category.DefaultFormatter;
}
startDataBlock(header) {
this.dataBlockCreated = true;
string_builder_1.StringBuilder.write(this.builder, `data_${(header || '').replace(/[ \n\t]/g, '').toUpperCase()}\n#\n`);
}
writeCategory(category, context, options) {
if (this.encoded) {
throw new Error('The writer contents have already been encoded, no more writing.');
}
if (!this.dataBlockCreated) {
throw new Error('No data block created.');
}
if (!(options === null || options === void 0 ? void 0 : options.ignoreFilter) && !this.filter.includeCategory(category.name))
return;
const { instance, rowCount, source } = (0, util_1.getCategoryInstanceData)(category, context);
if (!rowCount)
return;
if (rowCount === 1) {
writeCifSingleRecord(category, instance, source, this.builder, this.filter, this.formatter);
}
else {
writeCifLoop(category, instance, source, this.builder, this.filter, this.formatter);
}
}
encode() {
this.encoded = true;
}
writeTo(stream) {
const chunks = string_builder_1.StringBuilder.getChunks(this.builder);
for (let i = 0, _i = chunks.length; i < _i; i++) {
stream.writeString(chunks[i]);
}
}
getSize() {
return string_builder_1.StringBuilder.getSize(this.builder);
}
getData() {
return string_builder_1.StringBuilder.getString(this.builder);
}
}
exports.TextEncoder = TextEncoder;
function writeValue(builder, data, key, f, floatPrecision, index) {
const kind = f.valueKind;
const p = kind ? kind(key, data) : 0 /* Column.ValueKinds.Present */;
if (p !== 0 /* Column.ValueKinds.Present */) {
if (p === 1 /* Column.ValueKinds.NotPresent */)
writeNotPresent(builder);
else
writeUnknown(builder);
}
else {
const val = f.value(key, data, index);
const t = f.type;
if (t === 0 /* Field.Type.Str */) {
if (isMultiline(val)) {
writeMultiline(builder, val);
return true;
}
else {
return writeChecked(builder, val);
}
}
else if (t === 1 /* Field.Type.Int */) {
writeInteger(builder, val);
}
else {
writeFloat(builder, val, floatPrecision);
}
}
return false;
}
function getFloatPrecisions(categoryName, fields, formatter) {
const ret = [];
for (const f of fields) {
const format = formatter.getFormat(categoryName, f.name);
if (format && typeof format.digitCount !== 'undefined')
ret[ret.length] = f.type === 2 /* Field.Type.Float */ ? Math.pow(10, Math.max(0, Math.min(format.digitCount, 15))) : 0;
else
ret[ret.length] = f.type === 2 /* Field.Type.Float */ ? Math.pow(10, (0, util_1.getFieldDigitCount)(f)) : 0;
}
return ret;
}
function writeCifSingleRecord(category, instance, source, builder, filter, formatter) {
const fields = (0, util_1.getIncludedFields)(instance);
const src = source[0];
const data = src.data;
let width = fields.reduce((w, f) => filter.includeField(category.name, f.name) ? Math.max(w, f.name.length) : 0, 0);
// this means no field from this category is included.
if (width === 0)
return;
width += category.name.length + 6;
const it = src.keys();
const key = it.move();
const precisions = getFloatPrecisions(category.name, instance.fields, formatter);
for (let _f = 0; _f < fields.length; _f++) {
const f = fields[_f];
if (!filter.includeField(category.name, f.name))
continue;
string_builder_1.StringBuilder.writePadRight(builder, `_${category.name}.${f.name}`, width);
const multiline = writeValue(builder, data, key, f, precisions[_f], 0);
if (!multiline)
string_builder_1.StringBuilder.newline(builder);
}
string_builder_1.StringBuilder.write(builder, '#\n');
}
function writeCifLoop(category, instance, source, builder, filter, formatter) {
const fieldSource = (0, util_1.getIncludedFields)(instance);
const fields = filter === encoder_1.Category.DefaultFilter ? fieldSource : fieldSource.filter(f => filter.includeField(category.name, f.name));
const fieldCount = fields.length;
if (fieldCount === 0)
return;
const precisions = getFloatPrecisions(category.name, fields, formatter);
writeLine(builder, 'loop_');
for (let i = 0; i < fieldCount; i++) {
writeLine(builder, `_${category.name}.${fields[i].name}`);
}
let index = 0;
for (let _c = 0; _c < source.length; _c++) {
const src = source[_c];
const data = src.data;
if (src.rowCount === 0)
continue;
const it = src.keys();
while (it.hasNext) {
const key = it.move();
let multiline = false;
for (let _f = 0; _f < fieldCount; _f++) {
multiline = writeValue(builder, data, key, fields[_f], precisions[_f], index);
}
if (!multiline)
string_builder_1.StringBuilder.newline(builder);
index++;
}
}
string_builder_1.StringBuilder.write(builder, '#\n');
}
function isMultiline(value) {
return typeof value === 'string' && value.indexOf('\n') >= 0;
}
function writeLine(builder, val) {
string_builder_1.StringBuilder.write(builder, val);
string_builder_1.StringBuilder.newline(builder);
}
function writeInteger(builder, val) {
string_builder_1.StringBuilder.writeInteger(builder, val);
string_builder_1.StringBuilder.whitespace1(builder);
}
function writeFloat(builder, val, precisionMultiplier) {
string_builder_1.StringBuilder.writeFloat(builder, val, precisionMultiplier);
string_builder_1.StringBuilder.whitespace1(builder);
}
function writeNotPresent(builder) {
string_builder_1.StringBuilder.writeSafe(builder, '. ');
}
function writeUnknown(builder) {
string_builder_1.StringBuilder.writeSafe(builder, '? ');
}
function writeChecked(builder, val) {
if (!val) {
string_builder_1.StringBuilder.writeSafe(builder, '. ');
return false;
}
const fst = val.charCodeAt(0);
let escape = false;
let escapeKind = 0; // 0 => ', 1 => "
let hasSingleQuote = false, hasDoubleQuote = false;
for (let i = 0, _l = val.length - 1; i <= _l; i++) {
const c = val.charCodeAt(i);
switch (c) {
case 9: // \t
escape = true;
break;
case 10: // \n
writeMultiline(builder, val);
return true;
case 32: // ' '
escape = true;
break;
case 34: // "
// no need to escape quote if it's the last char and the length is > 1
if (i && i === _l)
break;
if (hasSingleQuote) {
// the string already has a " => use multiline value
writeMultiline(builder, val);
return true;
}
hasDoubleQuote = true;
escape = true;
escapeKind = 0;
break;
case 39: // '
// no need to escape quote if it's the last char and the length is > 1
if (i && i === _l)
break;
if (hasDoubleQuote) {
writeMultiline(builder, val);
return true;
}
hasSingleQuote = true;
escape = true;
escapeKind = 1;
break;
}
}
if (!escape && (fst === 35 /* # */ || fst === 36 /* $ */ || fst === 59 /* ; */ || fst === 91 /* [ */ || fst === 93 /* ] */ || fst === 95 /* _ */)) {
escape = true;
}
if (escape) {
string_builder_1.StringBuilder.writeSafe(builder, escapeKind ? '"' : '\'');
string_builder_1.StringBuilder.writeSafe(builder, val);
string_builder_1.StringBuilder.writeSafe(builder, escapeKind ? '" ' : '\' ');
}
else {
string_builder_1.StringBuilder.writeSafe(builder, val);
string_builder_1.StringBuilder.writeSafe(builder, ' ');
}
return false;
}
function writeMultiline(builder, val) {
string_builder_1.StringBuilder.writeSafe(builder, '\n;' + val);
string_builder_1.StringBuilder.writeSafe(builder, '\n;\n');
}
;