Use classList if the browser support it

doubleotoo · Dec 7, 2011 · 2451414 · 2451414
1 parent d934d46
commit 2451414
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/masonry.js b/masonry.js
@@ -15,20 +15,34 @@
 
   // from bonzo.js, by Dustin Diaz - https://github.com/ded/bonzo
 
+  // use classList if supported because it's way faster: http://jsperf.com/classlist-vs-bonzo - by <@calvein>
+  var supportClassList = (function() {
+    var div = document.createElement('div');
+    return 'classList' in div;
+  })();
+
   function classReg(c) {
     return new RegExp("(^|\\s+)" + c + "(\\s+|$)");
   }
 
   function hasClass(el, c) {
-    return classReg(c).test(el.className);
+    return supportClassList ? el.classList.contains(c) : classReg(c).test(el.className);
   }
   function addClass(el, c) {
-    if ( !hasClass(el, c) ) {
-      el.className = el.className + ' ' + c;
+    if ( supportClassList ) {
+      el.classList.add(c);
+    } else {
+      if ( !hasClass(el, c) ) {
+        el.className = el.className + ' ' + c;
+      }
     }
   }
   function removeClass(el, c) {
-    el.className = el.className.replace(classReg(c), ' ');
+    if ( supportClassList ) {
+      el.classList.remove(c);
+    } else {
+      el.className = el.className.replace(classReg(c), ' ');
+    }
   }
 
   // -------------------------- getStyle -------------------------- //