using Plots

plotly()

using Random, StatsBase

Random.seed!(123456)

begin

    PRODUCTOS_POR_LOTE = 120

    MIN_DEFECTUOSOS_POR_LOTE = 3

    MAX_DEFECTUOSOS_POR_LOTE = 8

    N_ITERACIONES = 100

    N_PRODUCTOS_EXTRAIDOS = 20

end

function simulacion(

  N::Int;

  prod_por_lote::Int = PRODUCTOS_POR_LOTE,

  rango_defectuosos::UnitRange = MIN_DEFECTUOSOS_POR_LOTE:MAX_DEFECTUOSOS_POR_LOTE,

  n_prod_extraidos::Int = N_PRODUCTOS_EXTRAIDOS

  )::Float64

  saldo_inicial::Float64 = 0.0

  lote::BitArray{1} = BitArray(undef, prod_por_lote)

  num_defectuosos::Int = 0

  for i in 1:N

    num_defectuosos = rand(rango_defectuosos)

    fill!(lote, false)

    lote[sample(1:prod_por_lote, num_defectuosos, replace = false)] .= true

    if (sum(sample(lote, n_prod_extraidos, replace = false)) ≥ 2)

      saldo_inicial -= 25.0

    else

      saldo_inicial += 100.0

    end

  end

  saldo_inicial / N

end

function f(x::Int,y::Int)

    if y < x

        return NaN

    end

    simulacion(10000; rango_defectuosos=x:y)

end

plot(0:10, 0:10, f, st=:scatter, xlabel="Minimo de defectos", ylabel="Máximo de defectos")

surface(0:10, 0:10, f, xlabel="Minimo de defectos", ylabel="Máximo de defectos", camera=(-20,160))