@jbrowse/plugin-alignments 2.4.2 → 2.6.0

package/src/LinearSNPCoverageDisplay/models/model.ts

@@ -1,6 +1,5 @@
-import { addDisposer, types, cast, getEnv, getSnapshot } from 'mobx-state-tree'
-import clone from 'clone'
-import { autorun } from 'mobx'
+import { types, cast, getEnv, getSnapshot, isAlive } from 'mobx-state-tree'
+import { observable } from 'mobx'
 
 // jbrowse
 import PluginManager from '@jbrowse/core/PluginManager'
@@ -17,6 +16,8 @@ import { LinearGenomeViewModel } from '@jbrowse/plugin-linear-genome-view'
 // locals
 import Tooltip from '../components/Tooltip'
 import { FilterModel, getUniqueModificationValues } from '../../shared'
+import { createAutorun, modificationColors } from '../../util'
+import { randomColor } from '../../util'
 
 // using a map because it preserves order
 const rendererTypes = new Map([['snpcoverage', 'SNPCoverageRenderer']])
@@ -68,7 +69,8 @@ function stateModelFactory(
       }),
     )
     .volatile(() => ({
-      modificationTagMap: undefined as Record<string, string> | undefined,
+      modificationTagMap: observable.map<string, string>({}),
+      modificationsReady: false,
     }))
     .actions(self => ({
       /**
@@ -99,17 +101,14 @@ function stateModelFactory(
        * #action
        */
       updateModificationColorMap(uniqueModifications: string[]) {
-        const colorPalette = ['red', 'blue', 'green', 'orange', 'purple']
-        let i = 0
-
-        const newMap = clone(self.modificationTagMap) || {}
         uniqueModifications.forEach(value => {
-          if (!newMap[value]) {
-            const newColor = colorPalette[i++]
-            newMap[value] = newColor
+          if (!self.modificationTagMap.has(value)) {
+            self.modificationTagMap.set(
+              value,
+              modificationColors[value] || randomColor(),
+            )
           }
         })
-        self.modificationTagMap = newMap
       },
     }))
     .views(self => {
@@ -163,10 +162,23 @@ function stateModelFactory(
         /**
          * #getter
          */
-        get modificationsReady() {
-          return self.colorBy?.type === 'modifications'
-            ? self.modificationTagMap !== undefined
-            : true
+        get autorunReady() {
+          const view = getContainingView(self) as LGV
+          return (
+            view.initialized &&
+            self.featureDensityStatsReady &&
+            !self.regionTooLarge &&
+            !self.error
+          )
+        },
+
+        get renderReady() {
+          const superProps = superRenderProps()
+          return !superProps.notReady && self.modificationsReady
+        },
+
+        get ready() {
+          return this.renderReady
         },
 
         /**
@@ -177,8 +189,8 @@ function stateModelFactory(
           const { colorBy, filterBy, modificationTagMap } = self
           return {
             ...superProps,
-            notReady: superProps.notReady || !this.modificationsReady,
-            modificationTagMap: modificationTagMap,
+            notReady: !this.ready,
+            modificationTagMap: Object.fromEntries(modificationTagMap.toJSON()),
 
             // must use getSnapshot because otherwise changes to e.g. just the
             // colorBy.type are not read
@@ -189,6 +201,12 @@ function stateModelFactory(
       }
     })
     .actions(self => ({
+      /**
+       * #action
+       */
+      setModificationsReady(flag: boolean) {
+        self.modificationsReady = flag
+      },
       /**
        * #action
        */
@@ -207,41 +225,36 @@ function stateModelFactory(
       toggleDrawArcs() {
         self.drawArcs = !self.drawArcsSetting
       },
+    }))
+    .actions(self => ({
       afterAttach() {
-        addDisposer(
+        createAutorun(
           self,
-          autorun(
-            async () => {
-              try {
-                const { colorBy } = self
-                const view = getContainingView(self) as LGV
-
-                if (
-                  !view.initialized ||
-                  !self.estimatedStatsReady ||
-                  self.regionTooLarge
-                ) {
-                  return
-                }
-                const { staticBlocks } = view
-                if (colorBy?.type === 'modifications') {
-                  const adapter = getConf(self.parentTrack, 'adapter')
-                  self.updateModificationColorMap(
-                    await getUniqueModificationValues(
-                      self,
-                      adapter,
-                      colorBy,
-                      staticBlocks,
-                    ),
-                  )
-                }
-              } catch (error) {
-                console.error(error)
-                self.setError(error)
+          async () => {
+            self.setModificationsReady(false)
+            if (!self.autorunReady) {
+              return
+            }
+            const view = getContainingView(self) as LGV
+            const { staticBlocks } = view
+            const { colorBy } = self
+            if (colorBy?.type === 'modifications') {
+              const adapter = getConf(self.parentTrack, 'adapter')
+              const vals = await getUniqueModificationValues(
+                self,
+                adapter,
+                colorBy,
+                staticBlocks,
+              )
+              if (isAlive(self)) {
+                self.updateModificationColorMap(vals)
+                self.setModificationsReady(true)
               }
-            },
-            { delay: 1000 },
-          ),
+            } else {
+              self.setModificationsReady(true)
+            }
+          },
+          { delay: 1000 },
         )
       },
     }))

package/src/MismatchParser/index.ts

@@ -1,4 +1,5 @@
 import { revcom, Feature } from '@jbrowse/core/util'
+import { getTagAlt } from '../util'
 
 export interface Mismatch {
   qual?: number
@@ -116,8 +117,8 @@ export function cigarToMismatches(
 }
 
 /**
- * parse a SAM MD tag to find mismatching bases of the template versus the reference
- * @returns array of mismatches and their positions
+ * parse a SAM MD tag to find mismatching bases of the template versus the
+ * reference @returns array of mismatches and their positions
  */
 export function mdToMismatches(
   mdstring: string,
@@ -126,13 +127,13 @@ export function mdToMismatches(
   seq: string,
   qual?: Buffer,
 ) {
-  const mismatchRecords: Mismatch[] = []
   let curr: Mismatch = { start: 0, base: '', length: 0, type: 'mismatch' }
-  const skips = cigarMismatches.filter(cigar => cigar.type === 'skip')
   let lastCigar = 0
   let lastTemplateOffset = 0
   let lastRefOffset = 0
   let lastSkipPos = 0
+  const mismatchRecords: Mismatch[] = []
+  const skips = cigarMismatches.filter(cigar => cigar.type === 'skip')
 
   // convert a position on the reference sequence to a position
   // on the template sequence, taking into account hard and soft
@@ -265,6 +266,47 @@ export function* getNextRefPos(cigarOps: string[], positions: number[]) {
     }
   }
 }
+
+export function getModificationProbabilities(feature: Feature) {
+  const m = (getTagAlt(feature, 'ML', 'Ml') as number[] | string) || []
+  return m
+    ? (typeof m === 'string' ? m.split(',').map(e => +e) : m).map(e => e / 255)
+    : (getTagAlt(feature, 'MP', 'Mp') as string | undefined)
+        ?.split('')
+        .map(s => s.charCodeAt(0) - 33)
+        .map(elt => Math.min(1, elt / 50))
+}
+
+export function getMethBins(feature: Feature) {
+  const fstart = feature.get('start')
+  const fend = feature.get('end')
+  const fstrand = feature.get('strand') as -1 | 0 | 1
+  const flen = fend - fstart
+  const mm = (getTagAlt(feature, 'MM', 'Mm') as string | undefined) || ''
+  const methBins = new Array<number>(flen)
+  const methProbs = new Array<number>(flen)
+  const seq = feature.get('seq') as string | undefined
+  if (seq) {
+    const ops = parseCigar(feature.get('CIGAR'))
+    const probabilities = getModificationProbabilities(feature)
+    const modifications = getModificationPositions(mm, seq, fstrand)
+    let probIndex = 0
+    for (const { type, positions } of modifications) {
+      if (type === 'm') {
+        for (const ref of getNextRefPos(ops, positions)) {
+          const prob = probabilities?.[probIndex] || 0
+          probIndex++
+          if (ref >= 0 && ref < flen) {
+            methBins[ref] = 1
+            methProbs[ref] = prob
+          }
+        }
+      }
+    }
+  }
+  return { methBins, methProbs }
+}
+
 export function getModificationPositions(
   mm: string,
   fseq: string,

package/src/PileupRPC/methods/GetReducedFeatures.ts

@@ -1,12 +1,11 @@
 import { getAdapter } from '@jbrowse/core/data_adapters/dataAdapterCache'
-import { Region, dedupe } from '@jbrowse/core/util'
+import { Region, dedupe, groupBy } from '@jbrowse/core/util'
 import { RemoteAbortSignal } from '@jbrowse/core/rpc/remoteAbortSignals'
 import { BaseFeatureDataAdapter } from '@jbrowse/core/data_adapters/BaseAdapter'
 import { toArray } from 'rxjs/operators'
 import { firstValueFrom } from 'rxjs'
 // locals
 import { filterForPairs, getInsertSizeStats } from '../util'
-import { ReducedFeature } from '../../shared/fetchChains'
 import PileupBaseRPC from '../base'
 import { getTag } from '../../util'
 
@@ -56,15 +55,8 @@ export default class PileupGetReducedFeatures extends PileupBaseRPC {
 
     const filtered = filterForPairs(reduced)
     const stats = filtered.length ? getInsertSizeStats(filtered) : undefined
-    const chains = {} as { [key: string]: ReducedFeature[] }
+    const chains = groupBy(reduced, f => f.name)
 
-    // pair features
-    reduced.forEach(f => {
-      if (!chains[f.name]) {
-        chains[f.name] = []
-      }
-      chains[f.name].push(f)
-    })
     return {
       chains: Object.values(chains),
       stats,

package/src/PileupRenderer/PileupLayoutSession.ts

@@ -1,6 +1,6 @@
 import deepEqual from 'fast-deep-equal'
 import { LayoutSession } from '@jbrowse/core/pluggableElementTypes/renderers/BoxRendererType'
-import { AnyConfigurationModel } from '@jbrowse/core/configuration/configurationSchema'
+import { AnyConfigurationModel } from '@jbrowse/core/configuration'
 import SerializableFilterChain from '@jbrowse/core/pluggableElementTypes/renderers/util/serializableFilterChain'
 import GranularRectLayout from '@jbrowse/core/util/layouts/GranularRectLayout'
 import MultiLayout from '@jbrowse/core/util/layouts/MultiLayout'