rsyntaxtree 0.9.3 → 1.0.6

data/lib/rsyntaxtree/base_graph.rb

@@ -0,0 +1,262 @@
+#!/usr/bin/env ruby
+# -*- coding: utf-8 -*-
+
+#==========================
+# graph.rb
+#==========================
+#
+# Image utility functions to inspect text font metrics
+# Copyright (c) 2007-2021 Yoichiro Hasebe <yohasebe@gmail.com>
+
+require 'utils'
+
+module RSyntaxTree
+  class BaseGraph
+
+    def initialize(element_list, params)
+
+      @element_list  = element_list
+      @symmetrize = params[:symmetrize]
+
+      if params[:color]
+        @col_node  = "blue"
+        @col_leaf  = "green"
+        @col_path = "purple"
+      else
+        @col_node  = "black"
+        @col_leaf  = "black"
+        @col_path = "black"
+      end
+
+      @col_bg   = "none"
+      @col_fg   = "black"
+      @col_line = "black"
+
+      @leafstyle  = params[:leafstyle]
+      @fontset = params[:fontset]
+      @fontsize = params[:fontsize]
+    end
+
+    def calculate_level
+      @element_list.get_elements.select{|e| e.type == 2}.each do |e|
+        e.level = @element_list.get_id(e.parent).level + 1
+      end
+    end
+
+    def calculate_width(id = 1)
+      target = @element_list.get_id(id)
+      if target.children.empty?
+        target.width = target.content_width + $h_gap_between_nodes * 4
+
+        parent = @element_list.get_id(target.parent)
+        while parent && parent.children.size == 1
+          w = parent.content_width
+          target.width = w + $h_gap_between_nodes * 4 if w > target.content_width
+          parent = @element_list.get_id(parent.parent)
+        end
+        return target.width
+      else
+        if target.width != 0
+          return target.width
+        else
+          accum_array = []
+          target.children.each do |c|
+            accum_array << calculate_width(c)
+          end
+          if @symmetrize
+            accum_width = accum_array.max * target.children.size
+          else
+            accum_width = accum_array.sum
+          end
+
+          target.width = [accum_width, target.content_width].max
+        end
+
+        return target.width
+      end
+    end
+
+    def calculate_height(id = 1)
+      target = @element_list.get_id(id)
+      if id == 1
+        target.vertical_indent = 0
+      else
+        parent = @element_list.get_id(target.parent)
+
+        if !target.triangle && @leafstyle == "nothing" && ETYPE_LEAF == target.type && parent.children.size == 1
+          vertical_indent = parent.vertical_indent + parent.content_height
+        else
+          vertical_indent = parent.vertical_indent + parent.content_height + $height_connector
+        end
+
+        target.vertical_indent = vertical_indent
+      end
+
+      if target.children.empty?
+        target.height = target.content_height
+        vertical_end = target.vertical_indent + target.content_height
+      else
+        accum_array = []
+        target.children.each do |c|
+          accum_array << calculate_height(c)
+        end
+        target.height = accum_array.max - target.vertical_indent
+        vertical_end = accum_array.max
+      end
+      return vertical_end
+    end
+
+    def make_balance(id = 1)
+      target = @element_list.get_id(id)
+      if target.children.empty?
+        parent = @element_list.get_id(target.parent)
+        accum_array = []
+        parent.children.each do |c|
+          accum_array << @element_list.get_id(c).width
+        end
+        max = accum_array.max
+        parent.children.each do |c|
+          @element_list.get_id(c).width = max
+        end
+        return max
+      else
+        accum_array = []
+        target.children.each do |c|
+          accum_array << make_balance(c)
+        end
+        accum_width = accum_array.max
+        max = [accum_width, target.content_width].max
+        target.children.each do |c|
+          @element_list.get_id(c).width = max
+        end
+        return target.width
+      end
+    end
+
+    def calculate_indent
+      node_groups = @element_list.get_elements.group_by {|e| e.parent}
+      node_groups.each do |k, v|
+        next if k == 0
+        parent = @element_list.get_id(k)
+        if @symmetrize
+          num_leaves = v.size
+          partition_width = parent.width / num_leaves
+          left_offset = parent.horizontal_indent + parent.content_width / 2.0 - parent.width / 2.0
+          v.each do |e|
+            indent = left_offset + (partition_width - e.content_width) / 2.0
+            e.horizontal_indent = indent
+            left_offset += partition_width
+          end
+        else
+          left_offset = parent.horizontal_indent + parent.content_width / 2.0 - parent.width / 2.0
+          v.each do |e|
+            indent = left_offset + (e.width - e.content_width) / 2.0
+            e.horizontal_indent = indent
+            left_offset += e.width
+          end
+        end
+      end
+    end
+
+    def draw_elements
+      @element_list.get_elements.each do |element|
+        draw_element(element)
+      end
+    end
+
+    def draw_connector(id = 1)
+      parent = @element_list.get_id(id)
+      children = parent.children.map{|c| @element_list.get_id(c)}
+
+      if children.empty?
+        ;
+      elsif children.size == 1
+        child = children[0]
+        if (@leafstyle == "auto")
+          if child.contains_phrase || child.triangle
+            triangle_to_parent(parent, child)
+          else
+            line_to_parent(parent, child)
+          end
+        elsif(@leafstyle == "bar")
+          if child.triangle
+            triangle_to_parent(parent, child)
+          else
+            line_to_parent(parent, child)
+          end
+        elsif(@leafstyle == "nothing")
+          if child.triangle
+            triangle_to_parent(parent, child)
+          elsif ETYPE_LEAF == child.type
+            child.vertical_indent = parent.vertical_indent + parent.content_height + $height_connector / 2
+          end
+        end
+      else
+        children.each do |child|
+          line_to_parent(parent, child)
+        end
+      end
+
+      parent.children.each do |c|
+        draw_connector(c)
+      end
+    end
+
+    def get_leftmost(id = 1)
+      target = @element_list.get_id(id)
+      target_indent = target.horizontal_indent
+      children_indent = target.children.map{|c| get_leftmost(c)}
+      (children_indent << target_indent).min
+    end
+
+    def get_rightmost(id = 1)
+      target = @element_list.get_id(id)
+      target_right_end = target.horizontal_indent + target.content_width
+      children_right_end = target.children.map{|c| get_rightmost(c)}
+      (children_right_end << target_right_end).max
+    end
+
+    def node_centering
+      node_groups = @element_list.get_elements.group_by {|e| e.parent}
+      node_groups.sort_by{|k, v| -k}.each do |k, v|
+        next if k == 0
+        parent = @element_list.get_id(k)
+        child_positions =v.map {|child| child.horizontal_indent + child.content_width / 2 }
+        parent.horizontal_indent = child_positions.min + (child_positions.max - child_positions.min - parent.content_width) / 2
+      end
+    end
+
+    def parse_list
+      calculate_level
+      calculate_width
+      make_balance if @symmetrize
+      calculate_indent
+      node_centering
+
+      top = @element_list.get_id(1)
+      diff = top.horizontal_indent
+      @element_list.get_elements.each do |e|
+        e.horizontal_indent -= diff
+      end
+
+      offset_l = (top.horizontal_indent - get_leftmost) + $h_gap_between_nodes
+      offset_r = top.width / 2 - offset_l - $h_gap_between_nodes
+
+      @element_list.get_elements.each do |e|
+        e.horizontal_indent += offset_l
+      end
+
+      calculate_height
+      draw_elements
+      draw_connector
+      num_paths = draw_paths
+
+      # width = get_rightmost - get_leftmost + $h_gap_between_nodes * 2
+      width = get_rightmost - get_leftmost + $h_gap_between_nodes
+      height = @element_list.get_id(1).height
+      height = @height if @height > height
+
+      return {height: height, width: width}
+    end
+  end
+end

data/lib/rsyntaxtree/element.rb

@@ -6,43 +6,174 @@
 #==========================
 #
 # Aa class that represents a basic tree element, either node or leaf.
-#
-# This file is part of RSyntaxTree, which is a ruby port of Andre Eisenbach's
-# excellent program phpSyntaxTree.
-#
 # Copyright (c) 2007-2021 Yoichiro Hasebe <yohasebe@gmail.com>
-# Copyright (c) 2003-2004 Andre Eisenbach <andre@ironcreek.net>
 
-class Element
+require "markup_parser"
+require 'utils'
+
+module RSyntaxTree
+  class Element
 
-    attr_accessor :id, :parent, :type, :content, :level, :width, :indent, :triangle
-    def initialize(id = 0, parent = 0, content = "", level = 0, type = ETYPE_LEAF)
+    attr_accessor :id,
+      :parent, :type, :level,
+      :width, :height,
+      :content, :content_width, :content_height,
+      :horizontal_indent, :vertical_indent,
+      :triangle, :enclosure, :children, :parent,
+      :font, :fontsize, :contains_phrase,
+      :path
+
+    def initialize(id, parent, content, level, fontset, fontsize)
+      @type = ETYPE_LEAF
       @id = id                 # Unique element id
       @parent = parent         # Parent element id
+      @children = []           # Child element ids
       @type = type             # Element type
       @level = level           # Element level in the tree (0=top etc...)
-      @width = 0               # Width of the element in pixels
-      @indent = 0              # Drawing offset
+      @width = 0               # Width of the part of the tree including itself and it governs
+      @content_width = 0       # Width of the content
+      @horizontal_indent = 0   # Drawing offset
+      @vertical_indent = 0     # Drawing offset
       content = content.strip
-      if /\A.+\^\z/m =~ content
-        @content = content.gsub("^"){""} # The actual element content
-        @triangle = true # draw triangle instead of stright bar when in auto mode
+
+      if /.+?\^?((?:\+\>?\d+)+)\^?\z/m =~ content
+        @path = $1.sub(/\A\+/, "").split("+")
       else
-        @content = content.gsub("^"){""}.strip # The actual element content
-        @triangle = false # draw triangle instead of stright bar when in auto mode
+        @path = []
       end
-      # workaround to save "[A [B [C] [D] ] [E [F] [G [H] [J] ] ] ]"
+
+      @fontset = fontset
+      @fontsize = fontsize
+
+      parsed = Markup.parse(content)
+
+      if parsed[:status] == :success
+        results = parsed[:results]
+      else
+        error_text = "Error: input text contains an invalid string"
+        error_text += "\n > " + content
+        raise RSTError, error_text
+      end
+      @content = results[:contents]
+      @paths = results[:paths]
+      @enclosure = results[:enclosure]
+      @triangle = results[:triangle]
+
+      @contains_phrase = false
+      setup
     end
 
-    # Debug helper function
-    def dump
-      printf( "ID      : %d\n", @id );
-      printf( "Parent  : %d\n", @parent );
-      printf( "Level   : %d\n", @level );
-      printf( "Type    : %d\n", @type );
-      printf( "Width   : %d\n", @width );
-      printf( "Indent  : %d\n", @indent );
-      printf( "Content : %s\n\n", @content );
+    def setup
+      total_width = 0
+      total_height = 0
+      @content.each_with_index do |content, idx|
+        content_width = 0
+        case content[:type]
+        when :border, :bborder
+          height = $single_line_height / 2
+          content[:height] = height
+          total_height += height
+        when :text
+          row_width = 0
+          elements_height = []
+          content[:elements].each do |e|
+            text = e[:text]
+            e[:text] = text.gsub(" ", WHITESPACE_BLOCK).gsub(">", '&#62;').gsub("<", '&#60;')
+
+
+            @contains_phrase = true if text.include?(" ")
+            decoration = e[:decoration]
+            fontsize = decoration.include?(:small) || decoration.include?(:small) ? @fontsize * SUBSCRIPT_CONST : @fontsize
+            fontsize = decoration.include?(:subscript) || decoration.include?(:superscript)  ? fontsize * SUBSCRIPT_CONST : fontsize
+            style    = decoration.include?(:italic) || decoration.include?(:bolditalic) ? :italic : :normal
+            weight   = decoration.include?(:bold) || decoration.include?(:bolditalic) ? :bold : :normal
+
+            # e[:cjk] = false
+            # if e[:decoration].include?(:math)
+            #   font = @fontset[:math]
+            # elsif text.contains_cjk?
+            #   font = @fontset[:cjk]
+            #   e[:cjk] = true
+            # elsif decoration.include? :bolditalic
+            if decoration.include? :bolditalic
+              font = @fontset[:bolditalic]
+            elsif decoration.include? :bold
+              font = @fontset[:bold]
+            elsif decoration.include? :italic
+              font = @fontset[:italic]
+            else
+              font = @fontset[:normal]
+            end
+
+            standard_metrics = FontMetrics.get_metrics('X', @fontset[:normal], fontsize, :normal, :normal)
+            height = standard_metrics.height
+            if /\A[\<\>]+\z/ =~ text
+              width = standard_metrics.width * text.size / 2
+            elsif text.contains_emoji?
+              segments = text.split_by_emoji
+              width = 0
+              segments.each do |seg|
+                if /\s/ =~ seg[:char]
+                  ch = 't'
+                else
+                  ch = seg[:char]
+                end
+                if seg[:type] == :emoji
+                  this_font = @fontset[:emoji]
+                  metrics = FontMetrics.get_metrics(ch, this_font, fontsize, style, weight)
+                  width += metrics.width
+                else
+                  this_font = font
+                  metrics = FontMetrics.get_metrics(ch, this_font, fontsize, style, weight)
+                  width += metrics.width
+                end
+              end
+            else
+              text.gsub!("\\\\", 'i')
+              text.gsub!("\\", "")
+              text.gsub!(" ", "x")
+              text.gsub!("%", "X")
+              metrics = FontMetrics.get_metrics(text, font, fontsize, style, weight)
+              width = metrics.width
+            end
+
+            if e[:decoration].include?(:box) || e[:decoration].include?(:circle) || e[:decoration].include?(:bar)
+              if e[:text].size == 1
+                e[:content_width] = width
+                width += (height - width)
+              else
+                e[:content_width] = width
+                width += $width_half_X
+              end
+            end
+
+            if e[:decoration].include?(:whitespace)
+              width = $width_half_X / 2 * e[:text].size / 4
+              e[:text] = ""
+            end
+
+            e[:height] = height
+            elements_height << height + $box_vertical_margin / 2
+
+            e[:width] = width
+            row_width += width
+          end
+
+          if @enclosure != :none
+            total_height += (elements_height.max + $height_connector_to_text)
+          else
+            total_height += elements_height.max
+          end
+          content_width += row_width
+        end
+        total_width = content_width if total_width < content_width
+      end
+      @content_width = total_width
+      @content_height = total_height
     end
 
+    def add_child(child_id)
+      @children << child_id
+    end
+  end
 end

data/lib/rsyntaxtree/elementlist.rb

@@ -6,23 +6,25 @@
 #==========================
 #
 # Contains a list of unordered tree elements with a defined parent
-#
-# This file is part of RSyntaxTree, which is a ruby port of Andre Eisenbach's
-# excellent program phpSyntaxTree.
-#
 # Copyright (c) 2007-2021 Yoichiro Hasebe <yohasebe@gmail.com>
-# Copyright (c) 2003-2004 Andre Eisenbach <andre@ironcreek.net>
 
 require 'element'
 
-class ElementList
+module RSyntaxTree
+  class ElementList
 
-    attr_accessor :elements, :iterator
+    attr_accessor :elements, :iterator 
     def initialize
       @elements = Array.new # The element array
       @iterator = -1 # Iterator index (used for get_first / get_next)
     end
 
+    def set_hierarchy
+      @elements.each do |e|
+        get_id(e.parent).add_child(e.id) unless e.parent == 0
+      end
+    end
+
     def add(element)
       @elements << element
       if(element.parent != 0)
@@ -41,12 +43,12 @@ class ElementList
     end
 
     def get_next
-        @iterator += 1
-        if @elements[@iterator]
-          return @elements[@iterator]
-        else
-          return nil
-        end
+      @iterator += 1
+      if @elements[@iterator]
+        return @elements[@iterator]
+      else
+        return nil
+      end
     end
 
     def get_id(id)
@@ -118,4 +120,5 @@ class ElementList
       end
       return maxlevel + 1;
     end
+  end
 end