using PyPlot
using Distributions
using QuickTypes

const 抽象浮動小数点型 = AbstractFloat
const もしもまたはそれ以外 = ifelse
const ゼロ = zero
const 長さ = length
const 乱数 = rand
const 平均 = mean
const ガンマ分布 = Gamma
const 指数分布 = Exponential

rc("font", family="IPAPGothic")

function 資産分配してからプロット!(asset, taxfunc, Dist; 回数=10^7, bins=floor(Int,sqrt(endof(asset)/2)))
    figure(figsize=(8,3))

    subplot(121)
    plt[:hist](asset, normed=true, bins=bins, label="資産分布",
        range=(0, max(2.0, maximum(asset))))
    grid(ls=":")
    legend()
    title("分配前")

    @time 資産をランダムに分配!(asset, taxfunc, 回数)
    @show fitdist = fit(Dist, asset)
    sleep(0.1)

    subplot(122)
    h = plt[:hist](asset, normed=true, bins=bins, label="資産分布")
    x = linspace(0.0, max(2.0, maximum(asset)), 201)
    #Dist_str = replace("$Dist", "Distributions.", "")
    #plot(x, pdf.(fitdist, x), label="$Dist_str dist")
    plot(x, pdf.(fitdist, x), label="指数分布")
    grid(ls=":")
    legend()
    title("分配後")

    tight_layout()    
end

@qmutable 常に同額を徴収(; 税額 = 0.02)
function (徴収::常に同額を徴収)(保有資産)
    もしもまたはそれ以外(保有資産 ≥ 徴収.税額, 徴収.税額, ゼロ(徴収.税額))
end

function 資産をランダムに分配!(資産, 徴収額, 回数)
    人数 = 長さ(資産)
    for iter in 1:回数
        徴収先, 分配先 = 乱数(1:人数), 乱数(1:人数)
        額 = 徴収額(資産[徴収先])
        資産[徴収先] -= 額
        資産[分配先] += 額
    end
end

人数 = 2^11
資産 = 乱数(ガンマ分布(5.0, 0.2), 人数)

回数 = 10^7
税額 = 平均(資産)/50
徴収額 = 常に同額を徴収(税額=税額)
資産分配してからプロット!(資産, 徴収額, 指数分布, 回数=回数)