review-peg 0.1.0

data/lib/review/sec_counter.rb

@@ -0,0 +1,69 @@
+# encoding: utf-8
+#
+# Copyright (c) 2008-2014 Minero Aoki, Kenshi Muto, Masayoshi Takahashi,
+#                         KADO Masanori
+#
+# This program is free software.
+# You can distribute or modify this program under the terms of
+# the GNU LGPL, Lesser General Public License version 2.1.
+#
+
+module ReVIEW
+
+  # Secion Counter class
+  #
+  class SecCounter
+    def initialize(n, chapter)
+      @chapter = chapter
+      reset(n)
+    end
+
+    def reset(n)
+      @counter = Array.new(n, 0)
+    end
+
+    def inc(level)
+      n = level - 2
+      if n >= 0
+        @counter[n] += 1
+      end
+      if @counter.size > n
+        (n+1 .. @counter.size).each do |i|
+          @counter[i] = 0
+        end
+      end
+    end
+
+    def anchor(level)
+      str = "#{@chapter.format_number(false)}"
+      0.upto(level-2) do |i|
+        str << "-#{@counter[i]}"
+      end
+      str
+    end
+
+    def prefix(level, secnolevel)
+      return nil if @chapter.number.blank?
+
+      if level == 1
+        if secnolevel >= 1
+          if @chapter.is_a? ReVIEW::Book::Part
+            num = @chapter.number
+            return "#{I18n.t('part', num)}#{I18n.t("chapter_postfix")}"
+          else
+            return "#{@chapter.format_number}#{I18n.t("chapter_postfix")}"
+          end
+        end
+      else
+        if secnolevel >= level
+          prefix = @chapter.format_number(false)
+          0.upto(level - 2) do |i|
+            prefix << ".#{@counter[i]}"
+          end
+          prefix << I18n.t("chapter_postfix")
+          return prefix
+        end
+      end
+    end
+  end
+end

data/lib/review/template.rb

@@ -0,0 +1,21 @@
+require 'erb'
+module ReVIEW
+  class Template
+    TEMPLATE_DIR = File.join(File.dirname(__FILE__), "../../templates")
+
+    def self.load(filename, mode = 1)
+      self.new(filename, mode)
+    end
+
+    def initialize(filename = nil, mode = nil)
+      if filename
+        content = File.read(filename)
+        @erb = ERB.new(content, nil, mode)
+      end
+    end
+
+    def result(bind_data = nil)
+      @erb.result(bind_data)
+    end
+  end
+end

data/lib/review/textbuilder.rb

@@ -0,0 +1,17 @@
+# textbuilder.rb
+#
+# Copyright (c) 2010 Kenshi Muto
+#
+# This program is free software.
+# You can distribute or modify this program under the terms of
+# the GNU LGPL, Lesser General Public License version 2.1.
+#
+
+require 'review/topbuilder'
+
+module ReVIEW
+
+  class TEXTBuilder < TOPBuilder
+  end
+
+end # module ReVIEW

data/lib/review/textutils.rb

@@ -0,0 +1,16 @@
+# $Id: textutils.rb 2192 2005-11-13 11:55:42Z aamine $
+require 'nkf'
+
+module ReVIEW
+  module TextUtils
+    def detab(str, ts = 8)
+      add = 0
+      len = nil
+      str.gsub(/\t/) {
+        len = ts - ($`.size + add) % ts
+        add += len - 1
+        ' ' * len
+      }
+    end
+  end
+end

data/lib/review/tocparser.rb

@@ -0,0 +1,348 @@
+#
+# $Id: tocparser.rb 4268 2009-05-27 04:17:08Z kmuto $
+#
+# Copyright (c) 2002-2007 Minero Aoki
+#               2008-2009 Minero Aoki, Kenshi Muto
+#
+# This program is free software.
+# You can distribute or modify this program under the terms of
+# the GNU LGPL, Lesser General Public License version 2.1.
+# For details of the GNU LGPL, see the file "COPYING".
+#
+
+require 'review/preprocessor'
+require 'review/book'
+require 'review/textbuilder'
+
+module ReVIEW
+
+  class TOCParser
+    def TOCParser.parse(chap)
+      chap.open {|f|
+        stream = Preprocessor::Strip.new(f)
+        new.parse(stream, chap.id, chap.path, chap).map {|root|
+          root.number = chap.number
+          root
+        }
+      }
+    end
+
+    def parse(f, id, filename, chap)
+      roots = []
+      path = []
+
+      while line = f.gets
+        line.sub!(/\A\xEF\xBB\xBF/u, '') # remove BOM
+        case line
+        when /\A\#@/
+          ;
+        when /\A\s*\z/
+          ;
+        when /\A(={2,})[\[\s\{]/
+          lev = $1.size
+          error! filename, f.lineno, "section level too deep: #{lev}" if lev > 5
+          if path.empty?
+            # missing chapter label
+            path.push Chapter.new(get_label(line), id, filename, chap.book.page_metric)
+            roots.push path.first
+          end
+          next if get_label(line) =~ /\A\[\// # ex) "[/column]"
+          new = Section.new(lev, get_label(line).gsub(/\A\{.*?\}\s?/, ""))
+          until path.last.level < new.level
+            path.pop
+          end
+          path.last.add_child new
+          path.push new
+
+        when /\A= /
+          path.clear
+          path.push Chapter.new(get_label(line), id, filename, chap.book.page_metric)
+          roots.push path.first
+
+        when %r<\A//\w+(?:\[.*?\])*\{\s*\z>
+          if path.empty?
+            error! filename, f.lineno, 'list found before section label'
+          end
+          path.last.add_child(list = List.new)
+          beg = f.lineno
+          list.add line
+          while line = f.gets
+            break if %r<\A//\}> =~ line
+            list.add line
+          end
+          error! filename, beg, 'unterminated list' unless line
+
+        when %r<\A//\w>
+          ;
+        else
+          #if path.empty?
+          #  error! filename, f.lineno, 'text found before section label'
+          #end
+          next if path.empty?
+          path.last.add_child(par = Paragraph.new(chap.book.page_metric))
+          par.add line
+          while line = f.gets
+            break if /\A\s*\z/ =~ line
+            par.add line
+          end
+        end
+      end
+
+      roots
+    end
+
+    def get_label(line)
+      line = line.strip.sub(/\A=+\s*/, '')
+      compile_label(line)
+    end
+
+    def compile_label(line)
+      b = ReVIEW::TEXTBuilder.new
+      dummy_book = ReVIEW::Book::Base.load
+      dummy_chapter = ReVIEW::Book::Chapter.new(dummy_book, 1, '-', nil, StringIO.new)
+      dummy_loc = Location.new("", StringIO.new)
+      b.bind(ReVIEW::Compiler.new(b), dummy_chapter, dummy_loc)
+      b.compile_inline(line)
+    end
+
+    def error!(filename, lineno, msg)
+      raise "#{filename}:#{lineno}: #{msg}"
+    end
+
+
+    class Node
+
+      def initialize(children = [])
+        @children = children
+      end
+
+      attr_reader :children
+
+      def add_child(c)
+        @children.push c
+      end
+
+      def each_node(&block)
+        @children.each do |c|
+          yield c
+          c.each(&block)
+        end
+      end
+
+      def each_child(&block)
+        @children.each(&block)
+      end
+
+      def chapter?
+        false
+      end
+
+      def each_section(&block)
+        @children.each do |n|
+          n.yield_section(&block)
+        end
+      end
+
+      def each_section_with_index
+        i = 0
+        each_section do |n|
+          yield n, i
+          i += 1
+        end
+      end
+
+      def n_sections
+        cnt = 0
+        @children.each do |n|
+          n.yield_section { cnt += 1 }
+        end
+        cnt
+      end
+
+    end
+
+
+    class Section < Node
+
+      def initialize(level, label, path = nil)
+        super()
+        @level = level
+        @label = label
+        @filename = (path ? real_filename(path) : nil)
+      end
+
+      def real_filename(path)
+        if FileTest.symlink?(path)
+          File.basename(File.readlink(path))
+        else
+          File.basename(path)
+        end
+      end
+      private :real_filename
+
+      attr_reader :level
+      attr_reader :label
+
+      def display_label
+        if @filename
+          @label + ' ' + @filename
+        else
+          @label
+        end
+      end
+
+      def estimated_lines
+        @children.inject(0) {|sum, n| sum + n.estimated_lines }
+      end
+
+      def yield_section
+        yield self
+      end
+
+      def inspect
+        "\#<#{self.class} level=#{@level} #{@label}>"
+      end
+
+    end
+
+
+    class Chapter < Section
+
+      def initialize(label, id, path, page_metric)
+        super 1, label, path
+        @chapter_id = id
+        @path = path
+        @page_metric = page_metric
+        @volume = nil
+        @number = nil
+      end
+
+      attr_accessor :number
+
+      def chapter?
+        true
+      end
+
+      attr_reader :chapter_id
+
+      def volume
+        return @volume if @volume
+        return Book::Volume.dummy unless @path
+        @volume = Book::Volume.count_file(@path)
+        @volume.page_per_kbyte = @page_metric.page_per_kbyte
+        @volume.lines = estimated_lines()
+        @volume
+      end
+
+      def inspect
+        "\#<#{self.class} #{@filename}>"
+      end
+
+    end
+
+
+    class Paragraph < Node
+
+      def initialize(page_metric)
+        @bytes = 0
+        @page_metric = page_metric
+      end
+
+      def inspect
+        "\#<#{self.class}>"
+      end
+
+      def add(line)
+        @bytes += line.strip.bytesize
+      end
+
+      def estimated_lines
+        (@bytes + 2) / @page_metric.text.n_columns + 1
+      end
+
+      def yield_section
+      end
+
+    end
+
+
+    class List < Node
+
+      def initialize
+        @lines = 0
+      end
+
+      def inspect
+        "\#<#{self.class}>"
+      end
+
+      def add(line)
+        @lines += 1
+      end
+
+      def estimated_lines
+        @lines + 2
+      end
+
+      def yield_section
+      end
+
+    end
+
+  end
+
+
+  module TOCRoot
+    def level
+      0
+    end
+
+    def chapter?
+      false
+    end
+
+    def each_section_with_index
+      idx = -1
+      each_section do |node|
+        yield node, (idx += 1)
+      end
+    end
+
+    def each_section(&block)
+      each_chapter do |chap|
+        yield chap.toc
+      end
+    end
+
+    def n_sections
+      chapters.size
+    end
+
+    def estimated_lines
+      chapters.inject(0) {|sum, chap| sum + chap.toc.estimated_lines }
+    end
+  end
+
+  class Book::Base # reopen
+    include TOCRoot
+  end
+
+  class Book::ChapterSet # reopen
+    include TOCRoot
+  end
+
+  class Book::Part
+    include TOCRoot
+  end
+
+  class Book::Chapter # reopen
+    def toc
+      @toc ||= TOCParser.parse(self)
+      unless @toc.size == 1
+        $stderr.puts "warning: chapter #{@toc.join} contains more than 1 chapter"
+      end
+      @toc.first
+    end
+  end
+
+end