From f46df7794a153e0f7355db176389dd0b73343581 Mon Sep 17 00:00:00 2001
From: entonbiba <entonbiba@users.noreply.github.com>
Date: Sun, 12 Feb 2017 12:36:18 -0500
Subject: [PATCH] create new labels feature for spectrogram plugin (#984)

* create new labels feature for spectrogram plugin

- create labels parameter (true|false)
- create labels element and canvas
- create labels frequency method loadLabels();
- update createWrapper() method and apply labels feature parameter.

* create new labels feature for spectrogram plugin

create new labels feature for spectrogram plugin

- create labels parameter (true|false)
- create labels element and canvas
- create labels frequency method loadLabels();
- update createWrapper() method and apply labels feature parameter.

* remove extra spaces

* removed one more extra white space :)

* update css formatting

* remove extra space

* remove extra space
---
 plugin/wavesurfer.spectrogram.js | 90 +++++++++++++++++++++++++++++++-
 1 file changed, 89 insertions(+), 1 deletion(-)

diff --git a/plugin/wavesurfer.spectrogram.js b/plugin/wavesurfer.spectrogram.js
index a9b6b7b60..8ea4e36b5 100644
--- a/plugin/wavesurfer.spectrogram.js
+++ b/plugin/wavesurfer.spectrogram.js
@@ -73,10 +73,35 @@ WaveSurfer.Spectrogram = {
             this.container.removeChild(prevSpectrogram);
         }
 
+        // if labels are active
+        if (this.params.labels) {
+            var specLabelsdiv = document.createElement('div');
+            specLabelsdiv.setAttribute('id', 'specLabels');
+            this.drawer.style(specLabelsdiv, {
+                left: 0,
+                position: 'relative',
+                zIndex: 9
+            });
+            specLabelsdiv.innerHTML = '<canvas></canvas>';
+            this.wrapper = this.container.appendChild(
+                specLabelsdiv
+            );
+            // can be customized in next version
+            this.loadLabels('rgba(68,68,68,0.5)', '12px', '10px', '', '#fff', '#f7f7f7', 'center', '#specLabels');
+        }
+
         var wsParams = this.wavesurfer.params;
 
+        var specView = document.createElement('spectrogram');
+        // if labels are active
+        if (this.params.labels) {
+            this.drawer.style(specView, {
+                left: 0,
+                position: 'relative'
+            });
+        }
         this.wrapper = this.container.appendChild(
-            document.createElement('spectrogram')
+            specView
         );
         this.drawer.style(this.wrapper, {
             display: 'block',
@@ -204,6 +229,69 @@ WaveSurfer.Spectrogram = {
         return ajax;
     },
 
+    freqType: function (freq) {
+        return (freq >= 1000 ? (freq / 1000).toFixed(1) : Math.round(freq));
+    },
+
+    unitType: function (freq) {
+        return (freq >= 1000 ? 'KHz' : 'Hz');
+    },
+
+    loadLabels: function (bgFill, fontSizeFreq, fontSizeUnit, fontType, textColorFreq, textColorUnit, textAlign, container) {
+        var frequenciesHeight = this.height;
+        var bgFill = bgFill || 'rgba(68,68,68,0.5)';
+        var fontSizeFreq = fontSizeFreq || '12px';
+        var fontSizeUnit = fontSizeUnit || '10px';
+        var fontType = fontType || 'Helvetica';
+        var textColorFreq = textColorFreq || '#fff';
+        var textColorUnit = textColorUnit || '#fff';
+        var textAlign = textAlign || 'center';
+        var container = container || '#specLabels';
+        var getMaxY = frequenciesHeight || 512;
+        var labelIndex = 5 * (getMaxY / 256);
+        var freqStart = 0;
+        var step = ((this.wavesurfer.backend.ac.sampleRate / 2) - freqStart) / labelIndex;
+
+        var cLabel = document.querySelectorAll(container+' canvas')[0].getContext('2d');
+        document.querySelectorAll(container+' canvas')[0].height = this.height;
+        document.querySelectorAll(container+' canvas')[0].width = 55;
+
+        cLabel.fillStyle = bgFill;
+        cLabel.fillRect(0, 0, 55, getMaxY);
+        cLabel.fill();
+
+        for (var i = 0; i <= labelIndex; i++) {
+            cLabel.textAlign = textAlign;
+            cLabel.textBaseline = 'middle';
+
+            var freq = freqStart + (step * i);
+            var index = Math.round(freq / this.sampleRate / 2 * this.fftSamples);
+            var index = Math.round(freq / (this.fftSamples / 2) * this.fftSamples);
+            var percent = index / this.fftSamples/2;
+            var y = (1 - percent) * this.height;
+            var label = this.freqType(freq);
+            var units = this.unitType(freq);
+            var x = 16;
+            var yLabelOffset = 2;
+
+            if (i==0) {
+                cLabel.fillStyle = textColorUnit;
+                cLabel.font = fontSizeUnit + ' ' + fontType;
+                cLabel.fillText(units, x + 24, getMaxY + i - 10);
+                cLabel.fillStyle = textColorFreq;
+                cLabel.font = fontSizeFreq + ' ' + fontType;
+                cLabel.fillText(label, x, getMaxY + i - 10);
+            } else {
+                cLabel.fillStyle = textColorUnit;
+                cLabel.font = fontSizeUnit + ' ' + fontType;
+                cLabel.fillText(units, x + 24, getMaxY - i * 50 + yLabelOffset);
+                cLabel.fillStyle = textColorFreq;
+                cLabel.font = fontSizeFreq + ' ' + fontType;
+                cLabel.fillText(label, x, getMaxY - i * 50 + yLabelOffset);
+            }
+        }
+    },
+
     updateScroll: function(e) {
       this.wrapper.scrollLeft = e.target.scrollLeft;
     },