geotree 1.1.0

data/lib/geotree/pswriter.rb

@@ -0,0 +1,471 @@
+require_relative 'tools'
+
+module PS_Private
+  class SOper
+    attr_reader :type
+    def self.null
+      @@null_oper
+    end
+
+    def initialize(type, *args)
+      @type = type
+      @args = *args.dup
+    end
+
+    def arg(offset)
+      @args[offset]
+    end
+
+    def nargs
+      @args.size
+    end
+    @@null_oper = SOper.new(-1)
+
+  end
+end
+
+# A debugging / demonstration utility class that 
+# generates postscript images
+#
+class PSWriter
+  include PS_Private
+
+  LT_SOLID = 1900
+  LT_DASHED = 1901
+  LT_DOTTED = 1902
+
+  LINEWIDTH_FACTOR = 3.0
+  LINETYPE = 1
+  LINEWIDTH = 2
+  TRANS = 3
+  RGB = 4
+  FONTHEIGHT = 5
+  SCALE = 6
+
+  S_START = 0
+  S_OPEN = 1
+  S_CLOSED = 2
+  # @param path path of file to write (e.g. xxx.ps)
+  def initialize(path)
+    @path = path
+    @line_width = -1
+    @rgb = [-1,0,0]
+    @phys_size = [612,792]
+    @state = S_START
+    @stack = []
+
+    set_logical_page_size(1000,1200)
+
+    @line_type = 0
+    @scale_ = 0
+    @font_height = 0
+    @scaled_font_height = 0
+
+    @page_used = false
+    @s = ''
+  end
+
+  # Close document and write to disk
+  def close
+    if @state < S_CLOSED
+      if @stack.size != 0
+        warn("state stack nonempty for #{@path}")
+      end
+      flush_page
+      set_state(S_CLOSED)
+      write_text_file(@path, @s)
+    end
+  end
+
+  # Set logical page size.  Subsequent drawing operations will be scaled
+  # appropriately.  Default logical page size is 1000 x 1200 units.
+  def set_logical_page_size( width,  height)
+    raise IllegalStateException if @state != S_START
+    @document_size = [width,height]
+  end
+
+  def page_size
+    return @document_size
+  end
+
+  # Draw a rectangle
+  # @param inset distance to inset rectangle boundary (positive: shrink; negative:expand)
+  def draw_rect(x,y,w,h,inset = 0)
+    a(x + inset)
+    a(y + inset)
+    a(w - 2 * inset);
+    a(h - 2 * inset);
+    a(" RECT\n");
+  end
+
+  def set_font_size(height)
+    raise IllegalStateException if @state != S_OPEN
+
+    ret = SOper.null
+
+    if   @font_height != height
+      ret = SOper.new(FONTHEIGHT, @font_height)
+      @font_height = height;
+      @scaled_font_height = height / @scale;
+      a("/Monaco findfont ");
+      a(@scaled_font_height);
+      a("scalefont setfont\n");
+    end
+    ret
+  end
+
+  def draw_string(string, x, y)
+    a(x)
+    a(y - @scaled_font_height / 2);
+    a("M ")
+    work = make_eps_safe(string)
+    a(work)
+    a(" TEXTC\n")
+  end
+
+  def draw_disc(cx,   cy,    radius)
+    a("NP");
+    a(cx);
+    a(cy);
+    a(radius);
+    a("0 360 A CP F\n");
+  end
+
+  def draw_circle(cx, cy, radius)
+    a("NP");
+    a(cx);
+    a(cy);
+    a(radius);
+    a("0 360 A CP S\n");
+  end
+
+  def draw_line(x1,y1,x2,y2)
+    a("NP ");
+    a(x1);
+    a(y1);
+    a(" M ");
+    a(x2);
+    a(y2);
+    a(" L CP S\n");
+  end
+
+  def set_line_solid
+    set_line_type(LT_SOLID)
+  end
+
+  def set_line_dashed
+    set_line_type(LT_DASHED)
+  end
+
+  def set_line_dotted
+    set_line_type(LT_DOTTED)
+  end
+
+  def set_scale(f)
+    ret = SOper.null
+    if   f != 1
+      ret = SOper.new(SCALE, 1.0 / f)
+      a(f);
+      a(f);
+      a("scale\n");
+    end
+    ret
+  end
+
+  def set_line_type(type)
+    ret = SOper.null
+    if  @line_type != type
+      ret = SOper.new(LINETYPE, @line_type)
+      @line_type = type
+      case type
+      when LT_DASHED
+        n =    (@scale * 30).to_i
+        a("[");
+        a(n);
+        a(n);
+        a("] 0 setdash\n");
+      when LT_DOTTED
+        int n =    (@scale * 30).to_i
+        n2 = n / 4
+        a("[");
+        a(n2);
+        a(n);
+        a("] 0 setdash\n");
+      else # LT_SOLID
+        a("[] 0 setdash\n");
+      end
+    end
+    ret
+  end
+
+  # Draw a polygon
+  # @param polygon array of 2n coordinates, defining closed n-gon
+  # @param x translation to apply to coordinates
+  # @param y
+  def draw_polygon(polygon, x, y)
+    push(translate(x, y))
+    a("NP ")
+    i = 0
+    while i < polygon.size
+      a(polygon[i])
+      a(' ');
+      a((polygon[i + 1]))
+      if (i == 0)
+        a(" M ");
+      else
+        a(" L ");
+      end
+      i += 2
+    end
+
+    a(" CP S\n")
+    pop()
+  end
+
+  # Translate subsequent drawing operations
+  def translate(tx,ty,neg=false)
+    ret = SOper.null
+    if (neg)
+      tx = -tx;
+      ty = -ty;
+    end
+    if (tx != 0 || ty != 0)
+      ret = SOper.new(TRANS, -tx, -ty)
+
+      a(tx);
+      a(ty);
+      a("TR\n");
+    end
+    ret
+  end
+
+  def set_line_width(w)
+    ret = SOper.null
+    if @line_width != w
+      ret = SOper.new(LINEWIDTH, @line_width)
+      @line_width = w
+      a(LINEWIDTH_FACTOR * @scale   * @line_width)
+      a("SLW\n");
+    end
+    ret
+  end
+
+  def set_rgb(r,g,b)
+    ret = SOper.null
+    if (r != @rgb[0] || g != @rgb[1] || b != @rgb[2])
+      ret = SOper.new(RGB, @rgb[0], @rgb[1], @rgb[2])
+      a(r)
+      a(g)
+      a(b)
+      @rgb = [r,g,b]
+      a("SRGB\n");
+    end
+    ret
+  end
+
+  def set_gray(f)
+    set_rgb(f,f,f)
+  end
+
+  def set_state(  s)
+    if (@state != s)
+      case s
+      when S_OPEN
+        raise IllegalStateException if @state != S_START
+        @state = s
+        print_document_header
+      when S_START
+        raise IllegalStateException
+      end
+      @state = s
+    end
+  end
+
+  # Start a new page
+  # @param page_title title of new page
+  def new_page(page_title)
+    set_state(S_OPEN);
+    flush_page
+
+    print_page_header(page_title);
+    @page_used = true;
+  end
+
+  # Push previous state (returned by an operation) onto a stack to be
+  # restored later; call must be balanced by later call to pop()
+  # @param obj state object returned by operation such as setGray(), translate()
+  def push(obj)
+    @stack << obj
+  end
+
+  def pop(count = 1)
+    count.times do
+      n = @stack.size   - 1
+      op = @stack.pop
+      case op.type
+      when RGB
+        set_rgb(op.arg(0),op.arg(1),op.arg(2))
+      when LINETYPE
+        set_line_type(op.arg(0))
+      when LINEWIDTH
+        set_line_width(op.arg(0))
+      when TRANS
+        translate(op.arg(0),op.arg(1))
+      when FONTHEIGHT
+        set_font_size(op.arg(0))
+      when SCALE
+        set_scale(op.arg(0))
+      end
+    end
+  end
+
+  private
+
+  def a(obj)
+    if obj.is_a? Numeric
+      if obj.is_a? Float
+        w = sprintf(" %.2f",obj)
+        @s << w
+      else
+        @s << " #{obj}"
+      end
+    else
+      @s <<' ' << obj
+    end
+  end
+
+  def print_page_header(page_title)
+
+    # set up transformation
+    lMargin = @phys_size[0] * 0.07
+    lWorkX = @phys_size[0] - 2 * lMargin
+    lWorkY = @phys_size[1] - 2 * lMargin;
+
+    scl = [lWorkX / @document_size[0], lWorkY / @document_size[1]].min
+    @scale = scl
+    lcx = (@phys_size[0] - scl * @document_size[0]) / 2
+    lcy = (@phys_size[1] - scl * @document_size[1]) / 2
+
+    a(lcx)
+    a(lcy)
+    a("TR\n");
+    a(scl);
+    a(scl);
+    a("SCL\n\n");
+    set_line_width(1);
+    set_gray(0);
+
+    if false
+      warn("drawing doc rect");
+      draw_rect(0, 0, @document_size[0], @document_size[1], 0)
+    end
+    set_font_size(24);
+
+    if (page_title)
+      draw_string(page_title, (@document_size[0] / 2),     (@document_size[1] + @scaled_font_height * 2))
+    end
+  end
+
+  def print_document_header
+
+    a("%!PS\n");
+
+    a("/Monaco findfont 28 scalefont setfont\n")
+    a(<<"TXT"
+/A {arc} bind def
+/CP {closepath} bind def
+/NP {newpath} bind def
+/TR {translate} bind def
+/SCL {scale} bind def
+/SLW {setlinewidth} bind def
+/SRGB {setrgbcolor} bind def
+/S {stroke} bind def
+/F {fill} bind def
+/M {moveto} bind def
+/L {lineto} bind def
+/R {rmoveto} bind def
+/V {rlineto} bind def
+% Left-justified text
+/TEXTL { currentpoint S M 0 0 R show } def
+% Right-justified text
+/TEXTR { currentpoint S M dup stringwidth pop neg 0 R show } def
+% Centered text
+/TEXTC { currentpoint S M dup stringwidth pop -2 div 0 R show } def
+  
+% Plot rectangle at x,y,w,h
+/RECT {  3 index 3 index M
+ 1 index 0 V
+ 0 1 index V
+ 1 index neg 0 V
+ pop pop pop pop CP
+ S } def
+TXT
+    )
+  end
+
+  def flush_page
+    if @page_used
+      @s << "showpage\n\n"
+      @page_used = false
+    end
+  end
+
+  def make_eps_safe(str)
+    w = '('
+    str.length.times do |i|
+      c = str[i]
+      esc = false
+      if c.ord < 32
+        c = '_'
+      end
+      case c
+      when '(',')'
+        esc = true
+      end
+
+      w << '\\' if esc
+      w << c
+    end
+    w << ')'
+    w
+  end
+
+end
+
+if __FILE__ == $0
+
+  poly  = [ 0, 0, 50, 30, 30, 80]
+
+  w = PSWriter.new("_jeff_.ps")
+
+  w.set_logical_page_size(1000, 1000)
+  10.times do |i|
+
+    w.new_page("example page #{i}" )
+
+    w.draw_disc(200, 700, 200 - i * 18)
+    w.draw_circle(700, 220, 10 + i * 5);
+
+    w.push(w.set_gray(0.8));
+    w.draw_disc(500, 500, 30 + i * 10);
+    w.pop();
+    w.push(w.set_line_width(3));
+    w.push(w.set_line_dashed());
+    w.draw_circle(500, 500, 30 + i * 10);
+    w.pop(2);
+
+    w.draw_line(20 + i * 10, 20, 980, 500 + i * 48);
+    w.push(w.set_line_width(1.2   + 0.3   * i));
+    w.push(w.translate(500 - i * 40, 500 - i * 40));
+    w.push(w.set_scale(1 + i * 0.7));
+    w.draw_polygon(poly, 0, 0);
+    w.pop(3);
+
+    w.push(w.set_rgb(1, 0.2, 0.2));
+    w.push(w.set_font_size(10 + i * 3));
+    w.draw_string("Hello", 10 + i * 30, 900 - i * 40);
+    w.pop();
+    w.pop();
+  end
+  w.close();
+end