nmr-learning/src/preprocess/maybridge.js

Learn a database of chemical shift and coupling constants assignments

const FS = require('fs');

const SD = require('spectra-data');

function loadFile(filename) {
  return FS.readFileSync(filename).toString();
}

function load(path, datasetName, options) {
  let OCLE = options.OCLE;
  // var keepMolfile = false || options.keepMolfile;
  // var keepMolecule = false || options.keepMolecule;
  var filter = { filter: '.txt' };
  if (typeof options.filter === 'object') {
    filter = options.filter;
  }

  var parts = FS.readdirSync(path).filter((line) => {
    return line.indexOf(filter.filter) > 0;
  });

  var result = [];
  for (var p = 0; p < parts.length; p++) {
    let fileContent = loadFile(path + parts[p]).split('\n');
    var max = fileContent.length - 1;
    // we could now loop on the sdf to add the int index
    for (var i = 1; i < max; i++) {
      let row = fileContent[i].split('\t');
      // result.push(row);
      // try {
      // var sdfi = {dataset: datasetName, id: p + "_" + i + "_" + molFiles[i].catalogID};
      var molfile = row[1].replace(/\\n/g, '\n');
      var molecule = OCLE.Molecule.fromMolfile(molfile);
      // let ocl = {value: molecule};

      molecule.addImplicitHydrogens();
      var nH = molecule.getMolecularFormula().formula.replace(/.*H([0-9]+).*/, '$1') * 1;
      var diaIDs = molecule.getGroupedDiastereotopicAtomIDs();
      diaIDs.sort(function (a, b) {
        if (a.atomLabel === b.atomLabel) {
          return b.counter - a.counter;
        }
        return a.atomLabel < b.atomLabel ? 1 : -1;
      });

      const linksOH = molecule.getAllPaths({
        fromLabel: 'H',
        toLabel: 'O',
        minLength: 1,
        maxLength: 1
      });
      const linksNH = molecule.getAllPaths({
        fromLabel: 'H',
        toLabel: 'N',
        minLength: 1,
        maxLength: 1
      });
      const linksClH = molecule.getAllPaths({
        fromLabel: 'H',
        toLabel: 'Cl',
        minLength: 1,
        maxLength: 1
      });
      const atoms = {};
      const levels = [6, 5, 4, 3];
      let hasLabile = false;
      for (const diaId of diaIDs) {
        delete diaId._highlight;
        diaId.hose = OCLE.Util.getHoseCodesFromDiastereotopicID(diaId.oclID, {
          maxSphereSize: levels[0],
          type: 0
        });

        for (const atomID of diaId.atoms) {
          atoms[atomID] = diaId.oclID;
        }

        diaId.isLabile = false;

        for (const linkOH of linksOH) {
          if (diaId.oclID === linkOH.fromDiaID) {
            diaId.isLabile = true;
            hasLabile = true;
            break;
          }
        }
        for (const linkNH of linksNH) {
          if (diaId.oclID === linkNH.fromDiaID) {
            diaId.isLabile = true;
            hasLabile = true;
            break;
          }
        }
        for (const linkClH of linksClH) {
          if (diaId.oclID === linkClH.fromDiaID) {
            diaId.isLabile = true;
            hasLabile = true;
            break;
          }
        }
      }
      var spectraData1H = SD.NMR.fromJcamp(row[2].replace(/\\n/g, '\n'));
      var signals = spectraData1H.getRanges(
        {
          nH: nH,
          realTop: true,
          thresholdFactor: 1,
          // minMaxRatio:0.020,
          clean: true,
          compile: true,
          format: 'new'
        }
      );

      let sum = 0;
      for (var j = signals.length - 1; j >= 0; j--) {
        if (signals[j].from < 0 || signals[j].from > 11.8) {
          signals.splice(j, 1);
        } else {
          if (signals[j].from > 2.48 && signals[j].to < 2.59 && signals[j].signal[0].multiplicity === 'quint') {
            signals.splice(j, 1);
          } else
          if (signals[j].from > 7.10 && signals[j].to < 7.30 && signals[j].signal[0].multiplicity === 's') {
            signals.splice(j, 1);
          } else {
            sum += signals[j].integral;
          }
        }
      }
      // Restore the integral to nH
      /* for (var j = signals.length - 1; j >= 0; j--) {
                signals[j].integral *= nH / sum;
            }*/
      for (var j = signals.length - 2; j >= 0; j--) {
        if (Math.abs((signals[j].to + signals[j].from) - (signals[j + 1].to + signals[j + 1].from)) <
                    (Math.abs(signals[j].to - signals[j].from) + Math.abs(signals[j + 1].to - signals[j + 1].from))) {
          signals[j].from = Math.min(signals[j].from, signals[j + 1].from);
          signals[j].to = Math.max(signals[j].to, signals[j + 1].to);
          signals[j].integral += signals[j + 1].integral;
          signals[j].signal[0].multiplicity = 'm';
          signals[j].signal[0].delta = (signals[j].from + signals[j].to) / 2;
          signals.splice(j + 1, 1);
        }
      }

      signals.forEach((range, index) => {
        range.signalID = `1H_${index}`;
      });

      let sample = {
        general: { ocl: { id: molecule.getIDCode(), atom: atoms, diaId: diaIDs, nH: nH, hasLabile } },
        spectra: {
          nmr: [
            {
              nucleus: 'H',
              experiment: '1d',
              range: signals,
              solvent: spectraData1H.getParamString('.SOLVENT NAME', 'unknown')
            }
          ]
        }
      };

      // {nucleus: ["H", "H"],  experiment: "cosy", region: cosyZones, solvent: cosy.getParamString(".SOLVENT NAME", "unknown")}
      result.push(sample);
      // }
      // catch (e) {
      //    console.log("Could not load the entry " + p + " " + i + " " + e);
      // }
    }
  }
  return result;
}

module.exports = { load: load };