summaryrefslogtreecommitdiff
path: root/benchmark/so_pidigits.yml
blob: 793745f5b5cd7cff3a0f8f98895913d2a3cc2686 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
prelude: |
  # The Great Computer Language Shootout
  # http://shootout.alioth.debian.org/
benchmark:
  so_pidigits: |
    #
    # contributed by Gabriele Renzi

    class PiDigitSpigot

        def initialize()
            @z = Transformation.new 1,0,0,1
            @x = Transformation.new 0,0,0,0
            @inverse = Transformation.new 0,0,0,0
        end

        def next!
            @y = @z.extract(3)
            if safe? @y
                @z = produce(@y)
                @y
            else
                @z = consume @x.next!()
                next!()
            end
        end

        def safe?(digit)
            digit == @z.extract(4)
        end

        def produce(i)
            @inverse.qrst(10,-10*i,0,1).compose(@z)
        end

        def consume(a)
            @z.compose(a)
        end
    end


    class Transformation
        attr_reader :q, :r, :s, :t
        def initialize(q, r, s, t)
            @q,@r,@s,@t,@k = q,r,s,t,0
        end

        def next!()
            @q = @k = @k + 1
            @r = 4 * @k + 2
            @s = 0
            @t = 2 * @k + 1
            self
        end

        def extract(j)
            (@q * j + @r) / (@s * j + @t)
        end

        def compose(a)
            self.class.new( @q * a.q,
                            @q * a.r + r * a.t,
                            @s * a.q + t * a.s,
                            @s * a.r + t * a.t
                        )
        end

        def qrst *args
            initialize *args
            self
        end


    end


    WIDTH = 10
    n = 2_500 # Integer(ARGV[0])
    j = 0

    digits = PiDigitSpigot.new

    while n > 0
        if n >= WIDTH
            WIDTH.times {print digits.next!}
            j += WIDTH
        else
            n.times {print digits.next!}
            (WIDTH-n).times {print " "}
            j += n
        end
        puts "\t:"+j.to_s
        n -= WIDTH
    end
loop_count: 1