import math

RTT1 = 0.01
RTT2 =0.0442857142857
RTT3 =0.0785714285714
RTT4 =0.112857142857
RTT5 =0.147142857143
RTT6 =0.181428571429
RTT7 =0.215714285714
RTT8 =0.25

drop_prob = 0.589985218787
W1=W2=W3=W4=W5=W6=W7=W8=0
simtime =0
delta = 0.01
drift1 = drift2 = drift3 = drift4 = drift5 = drift6 = drift7= drift8= 1
totWinSize = 0

congest_drift_outfile = open("drift_tcp.csv", "w")
congest_drift_outfile.write("simtime,w\n")

while (simtime <= 6):

  simtime+=delta

  drift1 = ( (1.0/RTT1) - ( drop_prob *  W1*W1 / (2.0*RTT1)  ) )
  W1 = W1 + delta * drift1

  drift2 = ( (1.0/RTT2) - ( drop_prob *  W2*W2 / (2.0*RTT2)  ) )
  W2 = W2 + delta * drift2

  drift3 = ( (1.0/RTT3) - ( drop_prob *  W3*W3 / (2.0*RTT3)  ) )
  W3 = W3 + delta * drift3

  drift4 = ( (1.0/RTT4) - ( drop_prob *  W4*W4 / (2.0*RTT4)  ) )
  W4 = W4 + delta * drift4

  drift5 = ( (1.0/RTT5) - ( drop_prob *  W5*W5 / (2.0*RTT5)  ) )
  W5 = W5 + delta * drift5

  drift6 = ( (1.0/RTT6) - ( drop_prob *  W6*W6 / (2.0*RTT6)  ) )
  W6 = W6 + delta * drift6

  drift7 = ( (1.0/RTT7) - ( drop_prob *  W7*W7 / (2.0*RTT7)  ) )
  W7 = W7 + delta * drift7

  drift8 = ( (1.0/RTT8) - ( drop_prob *  W8*W8 / (2.0*RTT8)  ) )
  W8 = W8 + delta * drift8

  totWinSize = (4*W1) + (4*W2) + (4*W3) + (4*W4) + (4*W5) +(4*W6) + (4*W7) +(4*W8)
  congest_drift_outfile.write(str(simtime) + "," + str(totWinSize) +"\n" )
  print totWinSize ,simtime