@vpalmisano/webrtcperf 4.4.12 → 4.5.1

package/src/scenarios.ts

@@ -6,9 +6,12 @@ import { Config } from './config'
 import { Auth, google } from 'googleapis'
 import { logger } from './utils'
 import { sprintf } from 'sprintf-js'
+import { PlotData, plotHtml } from './plot'
 
 const log = logger('webrtcperf:scenarios')
 
+export type StatsRow = Record<string, string | number>
+
 /**
  * It parses a CSV stats file and returns an array of objects representing each row.
  * @param filePath The path to the CSV stats file.
@@ -20,15 +23,12 @@ export async function parseStatsFile(filePath: string) {
   const lines = fileData.split('\n')
   const headers = lines[0].split(',')
   const data = lines.slice(1).map(line =>
-    line.split(',').reduce(
-      (acc, value, index) => {
-        if (value !== '') {
-          acc[headers[index]] = isNaN(Number(value)) ? value : Number(value)
-        }
-        return acc
-      },
-      {} as Record<string, string | number>,
-    ),
+    line.split(',').reduce((acc, value, index) => {
+      if (value !== '') {
+        acc[headers[index]] = isNaN(Number(value)) ? value : Number(value)
+      }
+      return acc
+    }, {} as StatsRow),
   )
   return data
 }
@@ -155,28 +155,32 @@ export async function uploadStatsToGoogleSheet(stats: StatsSummary[], spreadshee
 
 export type ThrottleDirection = 'up' | 'down' | 'bidi'
 
-export function formatBitrate(bitrate: number | undefined, prefix = ' ') {
+export function formatBitrate(bitrate: number | undefined, prefix = ' ', pad = true) {
   if (bitrate === undefined) return ''
   let suffix = 'Kbps'
   if (bitrate >= 1000) {
     bitrate /= 1000
     suffix = 'Mbps'
   }
-  return `${prefix}${sprintf('%5.4g', bitrate)}${suffix}`
+  return `${prefix}${sprintf(`%${pad ? '5' : ''}.4g`, bitrate)}${suffix}`
 }
 
-export function formatLoss(loss: number | undefined, prefix = ' ') {
-  return loss !== undefined ? `${prefix}${loss.toFixed(0).padStart(2, ' ')}%` : ''
+export function formatLoss(loss: number | undefined, prefix = ' ', pad = true) {
+  return loss !== undefined ? `${prefix}${loss.toFixed(0).padStart(pad ? 2 : 0, ' ')}%` : ''
 }
 
-export function formatDelay(delay: number | undefined, prefix = ' ') {
-  return delay !== undefined ? `${prefix}${delay.toFixed(0).padStart(3, ' ')}ms` : ''
+export function formatDelay(delay: number | undefined, prefix = ' ', pad = true) {
+  return delay !== undefined ? `${prefix}${delay.toFixed(0).padStart(pad ? 3 : 0, ' ')}ms` : ''
 }
 
-export function formatThrottleRule(throttleRule: ThrottleRule & { direction: ThrottleDirection }, human = false) {
+export function formatThrottleRule(
+  throttleRule: ThrottleRule & { direction: ThrottleDirection },
+  human = false,
+  pad = true,
+) {
   const { rate, loss, delay, direction } = throttleRule
   return human
-    ? `${direction.padEnd(4, ' ')}${formatBitrate(rate)}${formatLoss(loss)}${formatDelay(delay)}`
+    ? `${direction.padEnd(pad ? 4 : 0, ' ')}${formatBitrate(rate, ' ', pad)}${formatLoss(loss, ' ', pad)}${formatDelay(delay, ' ', pad)}`
     : `${direction}-r${rate}-l${loss}-d${delay}`
 }
 
@@ -190,6 +194,50 @@ export function parseThrottleRule(throttleDesc: string) {
   return { direction, rate, loss, delay }
 }
 
+export async function plotStatsSummary(stats: StatsSummary[]) {
+  const labels = new Set<string>()
+  const data = {} as Record<string, Record<string, FastStats>>
+  stats.forEach(s => {
+    const { id, scenario, videoRecvBitratePerPixel } = s
+    if (!videoRecvBitratePerPixel.length) return null
+    if (!data[id]) {
+      data[id] = {}
+    }
+    const scenarioFormatted = formatThrottleRule(parseThrottleRule(scenario), true, true)
+    if (!data[id][scenarioFormatted]) {
+      data[id][scenarioFormatted] = new FastStats()
+    }
+    labels.add(scenarioFormatted)
+    data[id][scenarioFormatted].push(videoRecvBitratePerPixel.percentile(95))
+  })
+  const series: PlotData[] = []
+  Object.entries(data)
+    .sort((a, b) => a[0].localeCompare(b[0]))
+    .forEach(([id, data]) => {
+      const plotData: PlotData = { label: id, data: [] }
+      Object.entries(data)
+        .sort((a, b) => a[0].localeCompare(b[0]))
+        .forEach(([scenario, stats]) => {
+          plotData.data.push({
+            x: scenario,
+            y: stats.percentile(50),
+            yMin: stats.percentile(5),
+            yMax: stats.percentile(95),
+          })
+        })
+      series.push(plotData)
+    })
+  await plotHtml(
+    {
+      type: 'barWithErrorBars',
+      xLabel: 'Scenario',
+      yLabel: 'Video Receive Bitrate per Pixel',
+      labels: Array.from(labels).sort((a, b) => a.localeCompare(b)),
+    },
+    series,
+  )
+}
+
 /**
  * It generates a test configuration with a scenario including 2 participants.
  * The first participant sends video and the second receives it.