From 30b7ff4a35daabf850d6b1ec184c28bf4f93ea04 Mon Sep 17 00:00:00 2001
From: Elian Dib <elian.dib@inria.fr>
Date: Thu, 25 Jul 2019 20:23:43 +0200
Subject: [PATCH] Allow zero as a label and color reference

---
 colorLabels.m | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/colorLabels.m b/colorLabels.m
index f8b7029..b61a047 100644
--- a/colorLabels.m
+++ b/colorLabels.m
@@ -2,9 +2,10 @@ function labels = colorLabels(labels,varargin)
 %COLORLABELS Summary of this function goes here
 %   Detailed explanation goes here
 
-cmap = jet(max(labels(:)));
+sz = size(labels);
+cmap = jet(max(labels(:))+1);
 shuffle = false;
-gray = ones(size(labels));
+gray = ones(sz);
 coldim = 3;
 
 p = inputParser();
@@ -31,8 +32,8 @@ order = 1:(ndims(gray)+1);
 order(coldim) = ndims(gray)+1;
 order(ndims(gray)+1) = coldim;
 
-labels = cmap(labels(:),:);
-labels = reshape(labels,[size(gray),3]);
+labels = cmap(labels(:)+1,:);
+labels = reshape(labels,[sz,3]);
 labels = permute(labels,order);
 labels = labels.*gray;
 
-- 
GitLab