From d5c1d8439afb474b0531d835cb77755c399ba362 Mon Sep 17 00:00:00 2001
From: Jamie Wong <jamie.lf.wong@gmail.com>
Date: Tue, 16 Jan 2018 22:27:31 -0800
Subject: [PATCH] Add utils, time import more granularly

---
 application.tsx     |  4 ++--
 flamechart-view.tsx |  2 ++
 utils.ts            | 35 +++++++++++++++++++++++++++++++++++
 3 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/application.tsx b/application.tsx
index b2c0ddc..0055bba 100644
--- a/application.tsx
+++ b/application.tsx
@@ -159,9 +159,9 @@ export class Application extends ReloadableComponent<{}, ApplicationState> {
     })
     console.timeEnd('import')
 
-    console.time('first render')
+    console.time('first setState')
     this.setState({ profile, flamechart, sortedFlamechart }, () => {
-      console.timeEnd('first render')
+      console.timeEnd('first setState')
     })
   }
 
diff --git a/flamechart-view.tsx b/flamechart-view.tsx
index 04f9a81..cd0a3c3 100644
--- a/flamechart-view.tsx
+++ b/flamechart-view.tsx
@@ -99,6 +99,7 @@ export class FlamechartPanZoomView extends ReloadableComponent<FlamechartPanZoom
 
   private preprocess(flamechart: Flamechart) {
     if (!this.canvas || !this.regl) return
+    console.time('panzoom preprocess')
     const configSpaceRects: Rect[] = []
     const colors: vec3[] = []
 
@@ -126,6 +127,7 @@ export class FlamechartPanZoomView extends ReloadableComponent<FlamechartPanZoom
     this.renderer = rectangleBatchRenderer(this.regl, configSpaceRects, colors)
     this.setConfigSpaceViewportRect(new Rect())
     this.hoveredLabel = null
+    console.timeEnd('panzoom preprocess')
   }
 
   private canvasRef = (element?: Element) => {
diff --git a/utils.ts b/utils.ts
index db2dc1b..7caa2d4 100644
--- a/utils.ts
+++ b/utils.ts
@@ -15,6 +15,41 @@ export function lastOf<T>(ts: T[]): T | null {
   return ts[ts.length-1] || null
 }
 
+export function sortBy<T>(ts: T[], key: (t: T) => number | string): void {
+  function comparator(a: T, b: T) {
+    return key(a) < key(b) ? -1 : 1
+  }
+  ts.sort(comparator)
+}
+
+export function getOrInsert<K, V>(map: Map<K, V>, k: K, fallback: (k?: K) => V): V {
+  if (!map.has(k)) map.set(k, fallback(k))
+  return map.get(k)!
+}
+
+export function getOrElse<K, V>(map: Map<K, V>, k: K, fallback: (k?: K) => V): V {
+  if (!map.has(k)) return fallback(k)
+  return map.get(k)!
+}
+
+export function* itMap<T, U>(it: Iterable<T>, f: (t: T) => U): Iterable<U> {
+  for (let t of it) {
+    yield f(t)
+  }
+}
+
+export function itForEach<T>(it: Iterable<T>, f: (t: T) => void): void {
+  for (let t of it) { f(t) }
+}
+
+export function itReduce<T, U>(it: Iterable<T>, f: (a: U, b: T) => U, init: U): U {
+  let accum: U = init
+  for (let t of it) {
+    accum = f(accum, t)
+  }
+  return accum
+}
+
 // NOTE: This blindly assumes the same result across contexts.
 const measureTextCache = new Map<string, number>()
 export function cachedMeasureTextWidth(ctx: CanvasRenderingContext2D, text: string): number {