ext/numo/linalg/blas/gen/erbpp2.rb in numo-linalg-0.1.0 vs ext/numo/linalg/blas/gen/erbpp2.rb in numo-linalg-0.1.1

@@ -1,6 +1,7 @@
 require "erb"
+require_relative "erbln"
 
 class ErbPP
 
   def initialize(parent=nil, erb_base=nil, **opts, &block)
     @parent = parent
@@ -64,11 +65,15 @@
     dirs = [dirs] if !dirs.kind_of?(Array)
     dirs.each do |x|
       Dir.glob(x).each do |dir|
         path = File.join(dir,file)
         if File.exist?(path)
-          erb = ERB.new(File.read(path), safe_level, trim_mode)
-          erb.filename = path
+          if get(:line_number)
+            erb = ERBLN.new(File.read(path), path, trim_mode)
+          else
+            erb = ERB.new(File.read(path), safe_level, trim_mode)
+            erb.filename = path
+          end
           return erb
         end
       end
     end
     raise "file not found: #{file.inspect} in #{dirs.inspect}"