From bd719688dec8ffdaff8c990d985f854e3f105896 Mon Sep 17 00:00:00 2001
From: Bala Clark <balaclark@gmail.com>
Date: Thu, 1 Sep 2011 00:35:14 +0200
Subject: [PATCH] make sure layout button always has the current type as class
 name

---
 lib/ComicBook.js | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/lib/ComicBook.js b/lib/ComicBook.js
index 85dce82..b73a0b9 100755
--- a/lib/ComicBook.js
+++ b/lib/ComicBook.js
@@ -184,7 +184,7 @@ function ComicBook(id, srcs, opts) {
 			.append(
 				$(document.createElement("button"))
 					.text("layout")
-					.addClass(options.displayMode)
+					.addClass("cb-layout " + options.displayMode)
 					.click(function(){
 						ComicBook.prototype.toggleLayout();
 					})
@@ -334,10 +334,13 @@ function ComicBook(id, srcs, opts) {
 
 	ComicBook.prototype.toggleLayout = function() {
 		if (options.displayMode === "double") {
+			$("#cb-toolbar .cb-layout").removeClass("double");
 			options.displayMode = "single";
 		} else {
+			$("#cb-toolbar .cb-layout").removeClass("single");
 			options.displayMode = "double";
 		}
+		$("#cb-toolbar .cb-layout").addClass(options.displayMode);
 		ComicBook.prototype.draw();
 	};