From 9292a5f43df2819a3d2091b2bf982752474e1a60 Mon Sep 17 00:00:00 2001
From: gizak <gizak@icloud.com>
Date: Fri, 10 Apr 2015 09:57:34 -0400
Subject: [PATCH] Fix https://github.com/gizak/termui/issues/26

---
 example/grid.go |  6 ++----
 grid.go         | 25 +++++++++++++++++++++----
 2 files changed, 23 insertions(+), 8 deletions(-)

diff --git a/example/grid.go b/example/grid.go
index 84dc929..c010b4b 100644
--- a/example/grid.go
+++ b/example/grid.go
@@ -107,14 +107,12 @@ func main() {
 			if e.Type == ui.EventKey && e.Ch == 'q' {
 				return
 			}
-			if e.Type == ui.EventResize {
-				ui.Body.Width = ui.TermWidth()
-				ui.Body.Align()
-			}
 		default:
 			for _, g := range gs {
 				g.Percent = (g.Percent + 3) % 100
 			}
+			ui.Body.Width = ui.TermWidth()
+			ui.Body.Align()
 
 			draw(i)
 			i++
diff --git a/grid.go b/grid.go
index 2cc7481..6d86294 100644
--- a/grid.go
+++ b/grid.go
@@ -44,10 +44,27 @@ func (r *row) isRenderableLeaf() bool {
 
 // assign widgets' (and their parent rows') width recursively.
 func (r *row) assignWidth(w int) {
-	cw := int(float64(w*r.Span) / 12)
-	r.SetWidth(cw)
+	r.SetWidth(w)
 
-	for i := range r.Cols {
+	accW := 0                            // acc span and offset
+	calcW := make([]int, len(r.Cols))    // calculated width
+	calcOftX := make([]int, len(r.Cols)) // computated start position of x
+
+	for i, c := range r.Cols {
+		accW += c.Span + c.Offset
+		cw := int(float64(c.Span*r.Width) / 12.0)
+
+		if i >= 1 {
+			calcOftX[i] = calcOftX[i-1] +
+				calcW[i-1] +
+				int(float64(r.Cols[i-1].Offset*r.Width)/12.0)
+		}
+
+		// use up the space if it is the last col
+		if i == len(r.Cols)-1 && accW == 12 {
+			cw = r.Width - calcOftX[i]
+		}
+		calcW[i] = cw
 		r.Cols[i].assignWidth(cw)
 	}
 }
@@ -86,7 +103,7 @@ func (r *row) assignX(x int) {
 		acc := 0
 		for i, c := range r.Cols {
 			if c.Offset != 0 {
-				acc += int(float64(c.Offset*r.Width) / float64(12))
+				acc += int(float64(c.Offset*r.Width) / 12.0)
 			}
 			r.Cols[i].assignX(x + acc)
 			acc += c.Width