From ef17f4eacbe65f4c2b0710a036f5a569aee9b7f5 Mon Sep 17 00:00:00 2001
From: Justin Greywolf <justin.greywolf@changepoint.com>
Date: Thu, 2 Apr 2020 09:50:27 -0700
Subject: [PATCH] 1274-adjust diagram viewbox

---
 src/diagrams/class/classRenderer.js | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/diagrams/class/classRenderer.js b/src/diagrams/class/classRenderer.js
index 3943b889b..acff7bab2 100644
--- a/src/diagrams/class/classRenderer.js
+++ b/src/diagrams/class/classRenderer.js
@@ -9,6 +9,7 @@ import svgDraw from './svgDraw';
 parser.yy = classDb;
 
 let idCache = {};
+const padding = 20;
 
 const conf = {
   dividerMargin: 10,
@@ -227,7 +228,13 @@ export const draw = function(text, id) {
 
   diagram.attr('height', g.graph().height + 40);
   diagram.attr('width', g.graph().width * 1.5 + 20);
-  diagram.attr('viewBox', '-10 -10 ' + (g.graph().width + 20) + ' ' + (g.graph().height + 20));
+
+  const svgBounds = diagram.node().getBBox();
+  const width = svgBounds.width + padding * 2;
+  const height = svgBounds.height + padding * 2;
+
+  const vBox = `${svgBounds.x - padding} ${svgBounds.y - padding} ${width} ${height}`;
+  diagram.attr('viewBox', vBox);
 };
 
 export default {