molstar
Version:
A comprehensive macromolecular library.
90 lines • 3.58 kB
JavaScript
/**
* Copyright (c) 2020-2021 mol* contributors, licensed under MIT, See LICENSE file for more info.
*
* @author Sebastian Bittrich <sebastian.bittrich@rcsb.org>
* @author Alexander Rose <alexander.rose@weirdbyte.de>
*/
import { __awaiter, __generator } from "tslib";
import { Column } from '../../../mol-data/db';
import { handleAtoms, handleBonds } from '../mol/parser';
import { Task } from '../../../mol-task';
import { ReaderResult as Result } from '../result';
import { Tokenizer, TokenBuilder } from '../common/text/tokenizer';
import { TokenColumnProvider as TokenColumn } from '../common/text/column/token';
var delimiter = '$$$$';
function handleDataItems(tokenizer) {
var dataHeader = TokenBuilder.create(tokenizer.data, 32);
var data = TokenBuilder.create(tokenizer.data, 32);
while (tokenizer.position < tokenizer.length) {
var line = Tokenizer.readLine(tokenizer);
if (line.startsWith(delimiter))
break;
if (!line)
continue;
if (line.startsWith('> ')) {
TokenBuilder.add(dataHeader, tokenizer.tokenStart + 2, tokenizer.tokenEnd);
Tokenizer.markLine(tokenizer);
var start = tokenizer.tokenStart;
var end = tokenizer.tokenEnd;
var added = false;
while (tokenizer.position < tokenizer.length) {
var line2 = Tokenizer.readLine(tokenizer);
if (!line2 || line2.startsWith(delimiter) || line2.startsWith('> ')) {
TokenBuilder.add(data, start, end);
added = true;
break;
}
end = tokenizer.tokenEnd;
}
if (!added) {
TokenBuilder.add(data, start, end);
}
}
}
return {
dataHeader: TokenColumn(dataHeader)(Column.Schema.str),
data: TokenColumn(data)(Column.Schema.str)
};
}
function handleMolFile(tokenizer) {
var title = Tokenizer.readLine(tokenizer).trim();
var program = Tokenizer.readLine(tokenizer).trim();
var comment = Tokenizer.readLine(tokenizer).trim();
var counts = Tokenizer.readLine(tokenizer);
var atomCount = +counts.substr(0, 3), bondCount = +counts.substr(3, 3);
if (Number.isNaN(atomCount) || Number.isNaN(bondCount)) {
// try to skip to next molecule
while (tokenizer.position < tokenizer.length) {
var line = Tokenizer.readLine(tokenizer);
if (line.startsWith(delimiter))
break;
}
return;
}
var atoms = handleAtoms(tokenizer, atomCount);
var bonds = handleBonds(tokenizer, bondCount);
var dataItems = handleDataItems(tokenizer);
return {
molFile: { title: title, program: program, comment: comment, atoms: atoms, bonds: bonds },
dataItems: dataItems
};
}
function parseInternal(data) {
var tokenizer = Tokenizer(data);
var compounds = [];
while (tokenizer.position < tokenizer.length) {
var c = handleMolFile(tokenizer);
if (c)
compounds.push(c);
}
return Result.success({ compounds: compounds });
}
export function parseSdf(data) {
var _this = this;
return Task.create('Parse Sdf', function () { return __awaiter(_this, void 0, void 0, function () {
return __generator(this, function (_a) {
return [2 /*return*/, parseInternal(data)];
});
}); });
}
//# sourceMappingURL=parser.js.map