Skip to content
Snippets Groups Projects
Select Git revision
  • 56d0e15e60bed51a3d01c9ebeb96b9f07a9c9732
  • master default protected
  • rednatco-v2
  • rednatco
  • test
  • ntc-tube-uniform-color
  • ntc-tube-missing-atoms
  • restore-vertex-array-per-program
  • watlas2
  • dnatco_new
  • cleanup-old-nodejs
  • webmmb
  • fix_auth_seq_id
  • update_deps
  • ext_dev
  • ntc_balls
  • nci-2
  • plugin
  • bugfix-0.4.5
  • nci
  • servers
  • v0.5.0-dev.1
  • v0.4.5
  • v0.4.4
  • v0.4.3
  • v0.4.2
  • v0.4.1
  • v0.4.0
  • v0.3.12
  • v0.3.11
  • v0.3.10
  • v0.3.9
  • v0.3.8
  • v0.3.7
  • v0.3.6
  • v0.3.5
  • v0.3.4
  • v0.3.3
  • v0.3.2
  • v0.3.1
  • v0.3.0
41 results

transation.ts

Blame
  • mol-script.ts 3.78 KiB
    import { MolScriptBuilder } from 'mol-script/language/builder';
    import { compile, QuerySymbolRuntime, DefaultQueryRuntimeTable } from 'mol-script/runtime/query/compiler';
    import { QueryContext, Structure, StructureQuery, ModelPropertyDescriptor } from 'mol-model/structure';
    import { readCifFile, getModelsAndStructure } from '../apps/structure-info/model';
    import { CustomPropSymbol } from 'mol-script/language/symbol';
    import Type from 'mol-script/language/type';
    import { parseMolScript } from 'mol-script/language/parser';
    import * as util from 'util'
    import { transpileMolScript } from 'mol-script/script/mol-script/symbols';
    import { formatMolScript } from 'mol-script/language/expression-formatter';
    import { StructureQualityReport } from 'mol-model-props/pdbe/structure-quality-report';
    import fetch from 'node-fetch'; 
    
    // import Examples from 'mol-script/script/mol-script/examples'
    // import { parseMolScript } from 'mol-script/script/mol-script/parser'
    // //import { compileAST } from 'mol-script/script/mol-script/compile';
    
    // for (const e of Examples) {
    //     const expr = parseMolScript(e.value)[0];
    //     console.log(e.name, util.inspect(expr, true, 10, true));
    // }
    // const exprs = parseMolScript(`(sel.atom.atom-groups
    //     :residue-test (= atom.auth_comp_id ALA)
    //     ;; ho ho ho
    //     :atom-test (set.has { _C _N } atom.el)) ; comm
    //     ;; this is a comment
    //     ((hi) (ho))`);
    
    //;; :residue-test (= atom.label_comp_id REA)
    const exprs = parseMolScript(`(sel.atom.atom-groups
        :residue-test (> pdbe.structure-quality.issue-count 0)
        :atom-test (= atom.el _C))`);
    
    const tsp = transpileMolScript(exprs[0]);
    
    //console.log(util.inspect(exprs, true, 10, true));
    
    console.log(util.inspect(tsp, true, 10, true));
    
    console.log(formatMolScript);
    console.log(formatMolScript(tsp));
    // //console.log(expr);
    
    const expr = MolScriptBuilder.core.math.add([1, 2, 3]);
    const compiled = compile<number>(expr);
    const result = compiled(new QueryContext(Structure.Empty));
    console.log(result);
    
    const CustomProp = ModelPropertyDescriptor({
        name: 'test_prop',
        isStatic: true,
        cifExport: { prefix: '', categories: [ ]},
        symbols: {
            residueIndex: QuerySymbolRuntime.Dynamic(CustomPropSymbol('custom.test-prop', 'residue-index', Type.Num), ctx => {
                const e = ctx.element;
                //console.log(e.element, e.unit.model.atomicHierarchy.residueAtomSegments.index[e.element])
                return e.unit.model.atomicHierarchy.residueAtomSegments.index[e.element];
            })
        }
    });
    
    DefaultQueryRuntimeTable.addCustomProp(CustomProp);
    
    DefaultQueryRuntimeTable.addCustomProp(StructureQualityReport.Descriptor);
    
    export async function testQ() {
        const frame = await readCifFile('e:/test/quick/1cbs_updated.cif');
        const { structure } = await getModelsAndStructure(frame);
    
        await StructureQualityReport.attachFromCifOrApi(structure.models[0], {
            PDBe_apiSourceJson: async model => {
                const rawData = await fetch(`https://www.ebi.ac.uk/pdbe/api/validation/residuewise_outlier_summary/entry/${model.label.toLowerCase()}`, { timeout: 1500 });
                return await rawData.json();
            }
        })
    
        let expr = MolScriptBuilder.struct.generator.atomGroups({
            'atom-test': MolScriptBuilder.core.rel.eq([
                MolScriptBuilder.struct.atomProperty.core.elementSymbol(),
                MolScriptBuilder.es('C')
            ]),
            // 'residue-test': MolScriptBuilder.core.rel.eq([
            //     MolScriptBuilder.struct.atomProperty.macromolecular.label_comp_id(),
            //     'REA'
            // ])
            'residue-test': MolScriptBuilder.core.rel.inRange([CustomProp.symbols.residueIndex.symbol(), 1, 5])
        });
    
        expr = tsp;
    
        const compiled = compile<StructureQuery>(expr);
        const result = compiled(new QueryContext(structure));
    
        console.log(result);
    }
    
    testQ();