bcl 0.5.3 → 0.5.4

data/lib/bcl/core_ext.rb

@@ -1,15 +1,47 @@
+######################################################################
+#  Copyright (c) 2008-2014, Alliance for Sustainable Energy.
+#  All rights reserved.
+#
+#  This library is free software; you can redistribute it and/or
+#  modify it under the terms of the GNU Lesser General Public
+#  License as published by the Free Software Foundation; either
+#  version 2.1 of the License, or (at your option) any later version.
+#
+#  This library is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+#  Lesser General Public License for more details.
+#
+#  You should have received a copy of the GNU Lesser General Public
+#  License along with this library; if not, write to the Free Software
+#  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+######################################################################
+
 class String
   def to_underscore
-    self.gsub(/::/, '/').
+    gsub(/::/, '/').
         gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2').
         gsub(/([a-z\d])([A-Z])/, '\1_\2').
-        tr("-", "_").
+        tr('-', '_').
         downcase
   end
 
-  # simple method to create titles
+  # simple method to create titles -- very custom to catch known inflections
   def titleize
-    arr = ['a', 'an', 'the']
-    self.gsub('_', ' ').gsub(/\w+/) { |match| arr.include?(match) ? match : match.capitalize }
+    arr = %w(a an the by to)
+    upcase_arr = %w(DX EDA AEDG LPD COP)
+    r = gsub('_', ' ').gsub(/\w+/) { |match|
+      match_result = match
+      if upcase_arr.include?(match.upcase)
+        match_result = upcase_arr[upcase_arr.index(match.upcase)]
+      elsif arr.include?(match)
+        match_result = match
+      else
+        match_result = match.capitalize
+      end
+      match_result
+    }
+
+    r
   end
-end
+end