Generate image data

Author: edwardhorsford

app/lib/generators/event-generator.js

@@ -5,6 +5,7 @@ const { faker } = require('@faker-js/faker')
 const weighted = require('weighted')
 const dayjs = require('dayjs')
 const config = require('../../config')
+const { generateMammogramImages } = require('./mammogram-generator')
 
 const NOT_SCREENED_REASONS = [
   'Recent mammogram at different facility',
@@ -114,9 +115,6 @@ const generateEvent = ({ slot, participant, clinic, outcomeWeights, forceStatus
       ...eventBase,
       details: {
         ...eventBase.details,
-        imagesTaken: eventStatus === 'event_complete'
-          ? ['RCC', 'LCC', 'RMLO', 'LMLO']
-          : null,
         notScreenedReason: eventStatus === 'event_attended_not_screened'
           ? faker.helpers.arrayElement(NOT_SCREENED_REASONS)
           : null,
@@ -139,6 +137,12 @@ const generateEvent = ({ slot, participant, clinic, outcomeWeights, forceStatus
         actualEndTime: actualEndTime.toISOString(),
         actualDuration: actualEndTime.diff(actualStartTime, 'minute'),
       }
+
+      // Add mammogram images for completed events
+      event.mammogramData = generateMammogramImages({
+        startTime: actualStartTime,
+        isSeedData: true
+      })
     }
 
     return event

app/lib/generators/mammogram-generator.js

@@ -0,0 +1,145 @@
+// app/lib/generators/mammogram-generator.js
+
+const { faker } = require('@faker-js/faker')
+const generateId = require('../utils/id-generator')
+const dayjs = require('dayjs')
+const weighted = require('weighted')
+
+const STANDARD_VIEWS = [
+  { side: 'right', view: 'mediolateral oblique' },
+  { side: 'right', view: 'craniocaudal' },
+  { side: 'left', view: 'craniocaudal' },
+  { side: 'left', view: 'mediolateral oblique' }
+]
+
+const REPEAT_REASONS = [
+  'patient movement',
+  'positioning issue',
+  'exposure issue',
+  'breast not fully imaged',
+  'blurred image',
+  'technical fault'
+]
+
+// Probability settings for repeats
+const REPEAT_PROBABILITIES = {
+  needsRepeat: 0.15, // 15% chance of needing any repeat
+  multipleRepeats: 0.2 // 20% chance of needing more than one repeat if already repeating
+}
+
+const generateViewKey = (side, view) => {
+  const prefix = side === 'right' ? 'right' : 'left'
+  const viewName = view === 'mediolateral oblique' ? 'MediolateralOblique' : 'Craniocaudal'
+  return `${prefix}${viewName}`
+}
+
+const generateImageUrl = (side, view, accessionNumber) => {
+  const sideCode = side === 'right' ? 'R' : 'L'
+  const viewCode = view === 'mediolateral oblique' ? 'MLO' : 'CC'
+  return `/images/mammograms/${sideCode}-${viewCode}-${accessionNumber.replace('/', '-')}.dcm`
+}
+
+/**
+ * Generate images for a single view
+ * @param {Object} params - Parameters for image generation
+ * @param {string} params.side - Breast side ('right' or 'left')
+ * @param {string} params.view - View type ('mediolateral oblique' or 'craniocaudal')
+ * @param {string} params.accessionBase - Base accession number
+ * @param {number} params.startIndex - Starting index for image numbering
+ * @param {string} params.startTime - Start timestamp
+ * @param {boolean} params.isSeedData - Whether generating seed data or live data
+ * @returns {Object} View data with images
+ */
+const generateViewImages = ({ side, view, accessionBase, startIndex, startTime, isSeedData }) => {
+  let currentIndex = startIndex
+  let currentTime = dayjs(startTime)
+  const images = []
+  const needsRepeat = Math.random() < REPEAT_PROBABILITIES.needsRepeat
+
+  // Generate initial image
+  images.push({
+    timestamp: currentTime.toISOString(),
+    accessionNumber: `${accessionBase}/${currentIndex}`,
+    url: generateImageUrl(side, view, `${accessionBase}/${currentIndex}`)
+  })
+
+  // Generate repeats if needed (and if we're in seed mode or randomly need them)
+  if (needsRepeat && isSeedData) {
+    const repeatCount = Math.random() < REPEAT_PROBABILITIES.multipleRepeats ? 2 : 1
+
+    for (let i = 0; i < repeatCount; i++) {
+      currentIndex++
+      currentTime = currentTime.add(faker.number.int({ min: 25, max: 50 }), 'seconds')
+
+      images.push({
+        timestamp: currentTime.toISOString(),
+        accessionNumber: `${accessionBase}/${currentIndex}`,
+        url: generateImageUrl(side, view, `${accessionBase}/${currentIndex}`)
+      })
+    }
+  }
+
+  return {
+    side,
+    view,
+    images,
+    isRepeat: needsRepeat && isSeedData,
+    repeatReason: needsRepeat && isSeedData ? faker.helpers.arrayElement(REPEAT_REASONS) : null
+  }
+}
+
+/**
+ * Generate a complete set of mammogram images
+ * @param {Object} options - Generation options
+ * @param {Date|string} [options.startTime] - Starting timestamp (defaults to now)
+ * @param {boolean} [options.isSeedData=false] - Whether generating seed data
+ * @returns {Object} Complete mammogram data
+ */
+const generateMammogramImages = ({ startTime = new Date(), isSeedData = false } = {}) => {
+  const accessionBase = faker.number.int({ min: 100000000, max: 999999999 }).toString()
+  let currentIndex = 1
+  let currentTime = dayjs(startTime)
+  const views = {}
+
+  // Generate each standard view
+  STANDARD_VIEWS.forEach(({ side, view }) => {
+    const viewKey = generateViewKey(side, view)
+    const viewData = generateViewImages({
+      side,
+      view,
+      accessionBase,
+      startIndex: currentIndex,
+      startTime: currentTime.toISOString(),
+      isSeedData
+    })
+
+    views[viewKey] = viewData
+
+    // Update counters for next view
+    currentIndex += viewData.images.length
+    currentTime = currentTime.add(faker.number.int({ min: 45, max: 70 }), 'seconds')
+  })
+
+  // Calculate metadata
+  const totalImages = Object.values(views).reduce((sum, view) => sum + view.images.length, 0)
+  const allTimestamps = Object.values(views)
+    .flatMap(view => view.images.map(img => img.timestamp))
+    .sort()
+
+  return {
+    accessionBase,
+    views,
+    metadata: {
+      totalImages,
+      standardViewsCompleted: true,
+      startTime: allTimestamps[0],
+      endTime: allTimestamps[allTimestamps.length - 1]
+    }
+  }
+}
+
+module.exports = {
+  generateMammogramImages,
+  STANDARD_VIEWS,
+  REPEAT_REASONS
+}