import flopy.utils.binaryfile as bf
import Base as base
import numpy as np

#给水度
water_u = 0.2
#研究区域面积平方米
water_F= float(5652 * 500 *500)
#有效计算单元格
water_invalid_cell =5652
#研究区域平方千米
water_F_KM= 680.250

pyq_cells_total =2721           


#地下水蓄变量
def get_grd_storage(model_name,per1,per2):       
    arr = np.loadtxt(base.xs_mp_path, dtype=int)
    pyq_cells=[]
    k=0
    for i in range(len(arr)):
           for j in range(len(arr[i])):  
               if arr[i][j] == 1:  
                   k+=1
                   pyq_cells.append((i,j))

    dir = base.model_dir + model_name  + "\\modflow.head" 

    if model_name=="202001_202212":
         dir = base.baseModel2  + "\\modflow.head" 
  
    head = bf.HeadFile(dir)
    alldata = head.get_alldata() 
    
    #初始水位数据
    z_start = alldata[int(per1+1)*3-3,0,:,:] 
    #结束水位数据
    
    z_end = alldata[int(per2+1)*3-1,0,:,:] 
    
    z_start[(z_start<=0)] = 0 
    z_end[(z_end<=0)] = 0 
    
    # z_start_avg = float(np.sum(z_start)/5652)
    # z_end_avg = float(np.sum(z_end)/5652)
    
    z_start_total = 0
    z_end_total = 0
    for item in pyq_cells:
        i = item[0]
        j = item[1]
        z_start_total  += z_start[i,j]
        z_end_total += z_end[i,j]
    
    z_start_avg = z_start_total/2721
    z_end_avg = z_end_total/2721
    
    #ΔW＝100·(h1-h2)·μ·F/t
    year = (per2+1-per1)/12
    # print(year)
    storage = 100 * (z_start_avg-z_end_avg) * water_u * water_F_KM /year
    return storage



#相对均衡差 Q1 
#Q总补－Q总排±ΔW＝Χ	
# Χ/Q总补 *100% 
def get_grd_relative_equ(Q1, Q2,w):
   x=  ( Q1-Q2) + w 
   y =  x/Q1
   return y