From 25b2d0d545ceb90696c130b838eaefeab25cb1e3 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Sat, 5 Jun 2021 14:44:21 -0400
Subject: [PATCH] color map the spectrogram

---
 vocodes/src/Utils.ts                          |  7 ++++-
 .../src/v2/tts_result_view/SpectrogramFc.tsx  | 26 ++++++++++++++++---
 2 files changed, 28 insertions(+), 5 deletions(-)

diff --git a/vocodes/src/Utils.ts b/vocodes/src/Utils.ts
index d27f545769..8378c0241b 100644
--- a/vocodes/src/Utils.ts
+++ b/vocodes/src/Utils.ts
@@ -5,4 +5,9 @@ function getRandomInt(min: number, max: number) : number {
   return Math.floor(Math.random() * (max - min)) + min;
 }
 
-export { getRandomInt }
+function getRandomArrayValue<T>(array: Array<T>) : T {
+  let index = getRandomInt(0, array.length);
+  return array[index];
+}
+
+export { getRandomInt, getRandomArrayValue }
diff --git a/vocodes/src/v2/tts_result_view/SpectrogramFc.tsx b/vocodes/src/v2/tts_result_view/SpectrogramFc.tsx
index cfe1f9405d..eb8dfa60c2 100644
--- a/vocodes/src/v2/tts_result_view/SpectrogramFc.tsx
+++ b/vocodes/src/v2/tts_result_view/SpectrogramFc.tsx
@@ -1,4 +1,7 @@
-import React, { useEffect, useRef, useState }  from 'react';
+import React, { useEffect, useRef }  from 'react';
+import { VIRIDIS, MAGMA } from "@colormap/presets";
+import { createColorMap, linearScale } from "@colormap/core";
+import { getRandomArrayValue } from "../../Utils";
 
 interface Props {
   spectrogramJsonLink: string  
@@ -10,6 +13,15 @@ interface SpectrogramResponse {
   mel_for_scaling: Array<Array<number>>,
 }
 
+const COLOR_MAP_PRESETS = [
+  VIRIDIS, 
+  MAGMA,
+  //CIVIDIS,
+  //BLACK_WHITE,
+  //INFERNO,
+  //PLASMA,  
+];
+
 function SpectrogramFc(props: Props) {
 
   const canvasRef = useRef(null);
@@ -23,13 +35,19 @@ function SpectrogramFc(props: Props) {
 
     let k = 0;
 
+    let colorMapScale = linearScale([0, 255], [0, 1]);
+    let colorMapColors = getRandomArrayValue(COLOR_MAP_PRESETS);
+    let colorMap = createColorMap(colorMapColors, colorMapScale);
+
     for (let j = 0; j < height; j++) {
       for (let i = 0; i < width; i++) {
         let value = image[i][j];
 
-        bytes[k] = value;
-        bytes[k+1] = value;
-        bytes[k+2] = value;
+        let mapped = colorMap(value);
+
+        bytes[k] = Math.floor(mapped[0] * 255);
+        bytes[k+1] = Math.floor(mapped[1] * 255);
+        bytes[k+2] = Math.floor(mapped[2] * 255);
         bytes[k+3] = 255;
 
         k += 4;