From 4b40b6bdf2c8bcc56cd6d0db98f684d971454edb Mon Sep 17 00:00:00 2001
From: Folyd <lyshuhow@gmail.com>
Date: Sun, 27 Jun 2021 11:48:47 +0800
Subject: [PATCH] Support legend

---
 index.html               |  1 +
 src/SvelteHeatmap.svelte | 19 ++++++++++++++++--
 src/views/Legend.svelte  | 42 ++++++++++++++++++++++++++++++++++++++++
 3 files changed, 60 insertions(+), 2 deletions(-)
 create mode 100644 src/views/Legend.svelte

diff --git a/index.html b/index.html
index e709b88..1a3c8b2 100644
--- a/index.html
+++ b/index.html
@@ -120,6 +120,7 @@ <h1 class="text-3xl">svelte-heatmap</h1>
                     startDate: moment().subtract(5, 'months').toDate(),
                     endDate: moment().toDate(),
                     view: 'monthly',
+                    displayLegend: false,
                 },
                 target: document.querySelector('#monthly'),
             });
diff --git a/src/SvelteHeatmap.svelte b/src/SvelteHeatmap.svelte
index 23d19d8..1b48963 100644
--- a/src/SvelteHeatmap.svelte
+++ b/src/SvelteHeatmap.svelte
@@ -53,6 +53,16 @@
             {/each}
         </g>
     {/if}
+    {#if displayLegend}
+        <Legend 
+        translateX={width - (wholeColors.length + 5) * cellSize}
+        translateY={height - legendHeight / 2}
+        height={legendHeight}
+        colors={wholeColors} 
+        {cellSize}
+        {fontFamily}
+        {fontSize}/>
+    {/if}
 </svg>
 
 <script>
@@ -64,6 +74,7 @@ import {
 
 import Month from './views/Month.svelte';
 import Week from './views/Week.svelte';
+import Legend from './views/Legend.svelte';
 
 export let allowOverflow = false;
 export let cellGap = 2;
@@ -83,6 +94,8 @@ export let monthLabelHeight = 12;
 export let monthLabels = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'];
 export let startDate = null;
 export let view = 'weekly';
+export let displayLegend = true;
+export let legendHeight = 30;
 
 const isNewMonth = (chunks, index) => {
     const chunk = chunks[index];
@@ -114,9 +127,9 @@ $: chunks = view === 'monthly'
 
 $: weekRect = (7 * cellRect) - cellGap;
 
-$: height = view === 'monthly'
+$: height = (view === 'monthly'
     ? (6 * cellRect) - cellGap + monthLabelHeight // <- max of 6 rows in monthly view
-    : weekRect + monthLabelHeight;
+    : weekRect + monthLabelHeight) + (displayLegend? legendHeight: 0);
 
 $: width = view === 'monthly'
     ? ((weekRect + monthGap) * chunks.length) - monthGap
@@ -125,4 +138,6 @@ $: width = view === 'monthly'
 $: dayLabelPosition = index => {
     return (cellRect * index) + (cellRect / 2) + monthLabelHeight;
 }
+
+$: wholeColors = [emptyColor, ...colors];
 </script>
\ No newline at end of file
diff --git a/src/views/Legend.svelte b/src/views/Legend.svelte
new file mode 100644
index 0000000..21c470b
--- /dev/null
+++ b/src/views/Legend.svelte
@@ -0,0 +1,42 @@
+<g transform="translate({translateX}, {translateY})">
+    {#each colors as color, index}
+        <rect
+            fill={color}
+            height={cellSize}
+            width={cellSize}
+            x={20 + cellSize * index}
+            {y}
+            alignment-baseline="middle"
+        />
+    {/each}
+    <text
+        x="0"
+        y={cellSize}
+        font-family={fontFamily}
+        font-size={fontSize}
+        alignment-baseline="middle"
+    >
+        Less
+    </text>
+    <text
+        x={15 + cellSize * (colors.length + 1)}
+        y={cellSize}
+        font-family={fontFamily}
+        font-size={fontSize}
+        alignment-baseline="middle"
+    >
+        More
+    </text>
+</g>
+
+<script>
+    export let translateX;
+    export let translateY;
+    export let height;
+    export let colors;
+    export let cellSize;
+    export let fontSize;
+    export let fontFamily;
+
+    $: y = (height - cellSize) / 4;
+</script>
\ No newline at end of file