synchronized with date2 3.5.

git-svn-id: http://svn.ruby-lang.org/repos/ruby/branches/ruby_1_8@5495 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 134 insertions, 91 deletions

diff --git a/sample/cal.rb b/sample/cal.rb
index e17c5e2a3..fa20352f7 100644
--- a/sample/cal.rb
+++ b/sample/cal.rb
@@ -1,111 +1,154 @@
 #! /usr/bin/env ruby
 
-# cal.rb: Written by Tadayoshi Funaba 1998-2002
-# $Id: cal.rb,v 2.4 2002-06-08 00:40:29+09 tadf Exp $
+# cal.rb: Written by Tadayoshi Funaba 1998-2004
+# $Id: cal.rb,v 2.7 2004-01-10 23:52:51+09 tadf Exp $
 
 require 'date'
-require 'getopts'
-
-$tab =
-{
-  'cn' => true,    # China
-  'de' => 2342032, # Germany (protestant states)
-  'dk' => 2342032, # Denmark
-  'es' => 2299161, # Spain
-  'fi' => 2361390, # Finland
-  'fr' => 2299227, # France
-  'gb' => 2361222, # United Kingdom
-  'gr' => 2423868, # Greece
-  'hu' => 2301004, # Hungary
-  'it' => 2299161, # Italy
-  'jp' => true,    # Japan
-  'no' => 2342032, # Norway
-  'pl' => 2299161, # Poland
-  'pt' => 2299161, # Portugal
-  'ru' => 2421639, # Russia
-  'se' => 2361390, # Sweden
-  'us' => 2361222, # United States
-  'os' => false,   # (old style)
-  'ns' => true     # (new style)
-}
-
-$cc = 'gb'
-
-def usage
-  warn 'usage: cal [-c iso3166] [-jmty] [[month] year]'
-  exit 1
-end
-
-def pict(y, m, sg)
-  d = (1..31).detect{|d| Date.valid_date?(y, m, d, sg)}
-  fi = Date.new(y, m, d, sg)
-  fi -= (fi.jd - $k + 1) % 7
 
-  ve  = (fi..fi +  6).collect{|cu|
-    %w(S M Tu W Th F S)[cu.wday]
-  }
-  ve += (fi..fi + 41).collect{|cu|
-    if cu.mon == m then cu.send($da) end.to_s
+class Cal
+
+  START =
+    {
+    'cn' => true,    # China
+    'de' => 2342032, # Germany (protestant states)
+    'dk' => 2342032, # Denmark
+    'es' => 2299161, # Spain
+    'fi' => 2361390, # Finland
+    'fr' => 2299227, # France
+    'gb' => 2361222, # United Kingdom
+    'gr' => 2423868, # Greece
+    'hu' => 2301004, # Hungary
+    'it' => 2299161, # Italy
+    'jp' => true,    # Japan
+    'no' => 2342032, # Norway
+    'pl' => 2299161, # Poland
+    'pt' => 2299161, # Portugal
+    'ru' => 2421639, # Russia
+    'se' => 2361390, # Sweden
+    'us' => 2361222, # United States
+    'os' => false,   # (old style)
+    'ns' => true     # (new style)
   }
 
-  ve = ve.collect{|e| e.rjust($dw)}
-
-  gr = group(ve, 7)
-  gr = trans(gr) if $OPT_t
-  ta = gr.collect{|xs| xs.join(' ')}
-
-  ca = %w(January February March April May June July
-	  August September October November December)[m - 1]
-  ca = ca + ' ' + y.to_s if not $OPT_y
-  ca = ca.center($mw)
-
-  ta.unshift(ca)
-end
+  DEFAULT_START = 'gb'
+
+  def initialize
+    opt_j; opt_m; opt_t; opt_y; opt_c
+  end
+
+  def opt_j(flag=false) @opt_j = flag end
+  def opt_m(flag=false) @opt_m = flag end
+  def opt_t(flag=false) @opt_t = flag end
+  def opt_y(flag=false) @opt_y = flag end
+
+  def opt_c(arg=DEFAULT_START) @start = START[arg] end
+
+  def set_params
+    @dw = if @opt_j then 3 else 2 end
+    @mw = (@dw + 1) * 7 - 1
+    @mn = if @opt_j then 2 else 3 end
+    @tw = (@mw + 2) * @mn - 2
+    @k  = if @opt_m then 1 else 0 end
+    @da = if @opt_j then :yday else :mday end
+  end
+
+  def pict(y, m)
+    d = (1..31).detect{|d| Date.valid_date?(y, m, d, @start)}
+    fi = Date.new(y, m, d, @start)
+    fi -= (fi.jd - @k + 1) % 7
+
+    ve  = (fi..fi +  6).collect{|cu|
+      %w(S M Tu W Th F S)[cu.wday]
+    }
+    ve += (fi..fi + 41).collect{|cu|
+      if cu.mon == m then cu.send(@da) end.to_s
+    }
+
+    ve = ve.collect{|e| e.rjust(@dw)}
+
+    gr = group(ve, 7)
+    gr = trans(gr) if @opt_t
+    ta = gr.collect{|xs| xs.join(' ')}
+
+    ca = %w(January February March April May June July
+	    August September October November December)[m - 1]
+    ca = ca + ' ' + y.to_s if not @opt_y
+    ca = ca.center(@mw)
+
+    ta.unshift(ca)
+  end
+
+  def group(xs, n)
+    (0..xs.size / n - 1).collect{|i| xs[i * n, n]}
+  end
+
+  def trans(xs)
+    (0..xs[0].size - 1).collect{|i| xs.collect{|x| x[i]}}
+  end
+
+  def stack(xs)
+    if xs.empty? then [] else xs[0] + stack(xs[1..-1]) end
+  end
+
+  def block(xs, n)
+    stack(group(xs, n).collect{|ys| trans(ys).collect{|zs| zs.join('  ')}})
+  end
+
+  def unlines(xs)
+    xs.collect{|x| x + "\n"}.join
+  end
+
+  def monthly(y, m)
+    unlines(pict(y, m))
+  end
+
+  def addmon(y, m, n)
+    y, m = (y * 12 + (m - 1) + n).divmod(12)
+    return y, m + 1
+  end
+
+  def yearly(y)
+    y.to_s.center(@tw) + "\n\n" +
+      unlines(block((0..11).collect{|n| pict(*addmon(y, 1, n))}, @mn)) + "\n"
+  end
+
+  def print(y, m)
+    set_params
+    if @opt_y then yearly(y) else monthly(y, m) end
+  end
 
-def group(xs, n)
-  (0..xs.size / n - 1).collect{|i| xs[i * n, n]}
 end
 
-def trans(xs)
-  (0..xs[0].size - 1).collect{|i| xs.collect{|x| x[i]}}
-end
+if __FILE__ == $0
 
-def unite(xs)
-  if xs.empty? then [] else xs[0] + unite(xs[1..-1]) end
-end
+  require 'getopts'
 
-def block(xs, n)
-  unite(group(xs, n).collect{|ys| trans(ys).collect{|zs| zs.join('  ')}})
-end
+  def usage
+    warn 'usage: cal [-c iso3166] [-jmty] [[month] year]'
+    exit 1
+  end
 
-def unlines(xs)
-  xs.collect{|x| x + "\n"}.join
-end
+  usage unless getopts('jmty', "c:#{Cal::DEFAULT_START}")
 
-usage unless getopts('jmty', "c:#{$cc}")
+  y, m = ARGV.values_at(1, 0).compact.collect{|x| x.to_i}
+  $OPT_y ||= (y and not m)
 
-y, m = ARGV.values_at(1, 0).compact.collect{|x| x.to_i}
-$OPT_y ||= (y and not m)
+  to = Date.today
+  y ||= to.year
+  m ||= to.mon
 
-to = Date.today
-y ||= to.year
-m ||= to.mon
+  usage unless m >= 1 and m <= 12
+  usage unless y >= -4712
+  usage if Cal::START[$OPT_c].nil?
 
-usage unless m >= 1 and m <= 12
-usage unless y >= -4712
-usage if (sg = $tab[$OPT_c]).nil?
+  cal = Cal.new
 
-$dw = if $OPT_j then 3 else 2 end
-$mw = ($dw + 1) * 7 - 1
-$mn = if $OPT_j then 2 else 3 end
-$tw = ($mw + 2) * $mn - 2
+  cal.opt_j($OPT_j)
+  cal.opt_m($OPT_m)
+  cal.opt_t($OPT_t)
+  cal.opt_y($OPT_y)
+  cal.opt_c($OPT_c)
 
-$k  = if $OPT_m then 1 else 0 end
-$da = if $OPT_j then :yday else :mday end
+  print cal.print(y, m)
 
-print(if not $OPT_y
-	unlines(pict(y, m, sg))
-      else
-	y.to_s.center($tw) + "\n\n" +
-	  unlines(block((1..12).collect{|m| pict(y, m, sg)}, $mn)) + "\n"
-      end)
+end