From 699db497b23e0e1878c31f9de74d4efff37e9f3d Mon Sep 17 00:00:00 2001
From: Robert Goldmann <deadlocker@gmx.de>
Date: Sat, 29 Jul 2017 18:00:04 +0200
Subject: [PATCH] Fixed #117 in chartsmonth as well

---
 .../chartGenerators/MonthChartGenerator.java    | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/de/deadlocker8/budgetmaster/logic/chartGenerators/MonthChartGenerator.java b/src/de/deadlocker8/budgetmaster/logic/chartGenerators/MonthChartGenerator.java
index 35ae4150c..927397b20 100644
--- a/src/de/deadlocker8/budgetmaster/logic/chartGenerators/MonthChartGenerator.java
+++ b/src/de/deadlocker8/budgetmaster/logic/chartGenerators/MonthChartGenerator.java
@@ -145,12 +145,21 @@ public class MonthChartGenerator
 			legendItems.add(getLegendItem(label, currentItem.getColor()));
 		}
 
-		int legendWidth = (int)Math.ceil(Math.sqrt(legendItems.size()));		
-
-		for(int i = 0; i < legendItems.size(); i++)
+		int legendWidth;
+		int numberOfItems = legendItems.size();
+		if(numberOfItems <= 3)
+		{
+			legendWidth = numberOfItems;
+		}
+		else
+		{
+			legendWidth = (int)Math.ceil(Math.sqrt(numberOfItems));
+		}
+		
+		for(int i = 0; i < numberOfItems; i++)
 		{
 			int columnIndex = i % legendWidth;
-			int rowIndex = i / 4;
+			int rowIndex = i / legendWidth;			
 			legend.add(legendItems.get(i), columnIndex, rowIndex);
 		}
 
-- 
GitLab