import numpy as np, pandas as pd, array as a
from scipy.stats import norm

params=pd.read_csv("./scen3_params.csv", sep=",", names=['alpha', 'beta', 'p1', 'n1', 'n2'])
alpha=float(params.alpha[params.alpha.index[0]])
beta=float(params.beta[params.beta.index[0]])
p1=float(params.p1[params.p1.index[0]])
n1=int(params.n1[params.n1.index[0]])
n2=int(params.n2[params.n2.index[0]])
print 'alpha', alpha, 'beta', beta, 'p1', p1, 'n1', n1, 'n2', n2
z_alpha_div_2=norm.ppf(alpha/2)
z_1_minus_beta=norm.ppf(1-beta)
c=float((z_1_minus_beta-z_alpha_div_2)**2)
_1c=1/c
_1n2=1/float(n2)
print _1c, _1n2
# if 1==1:
A=float(1/c+1/float(n2))
B=float(-2*p1/c-1/float(n2))
C=float((1/c+1/float(n1))*(p1**2)-(1/float(n1))*p1)
print 'A', A, 'B', B, 'C', C, 'c', c
p2=float((-1*B+(B**2-4*A*C)**0.5)/(2*A))
td="%.6f" % p2
print td