# Calculating axial R-factor

# Import addc from shared lib

acObj = libADDC.addc(fuetype)

AC = acObj.addc

# Define some matrices

nside = AC.shape[0] # Number of side pins of the assembly

dim = nside + 2 # Pin map storage dimension

# Calculate number of hot rods (POW[i,j] > 0)

Ntotrods = 100 # Total number of rods for ATRIUM 10XM

Nhotrods = sum(sum(POW>0)) # Number of hot rods

# Determine total power

FBUN = sum(sum(POW))

# Calculate local peaking distribution

POD = POW/FBUN * Nhotrods

# Calculate square root of power

RP = np.zeros((dim,dim))

RP[1:nside+1,1:nside+1] = np.sqrt(POD)

# Define Rod Weight factors

WP = np.zeros((dim,dim))

# Considered rod: wp = 1. wp(unheated rod) = 0.25 x wp(heated rod)

for i in range(1,nside+1):

for j in range(1,nside+1):

if POD[i-1,j-1] > 0.0001:

WP[i,j] = 1.0

else:

WP[i,j] = 0.25

# Calculate pinwise R-factors for fuel-rods where POW > 0

DOW = np.zeros((nside,nside))

# Side rods

WJ = 0.1 # Weighting factor for side neighboring rods

WK = WJ/2 # Weighting factor for diagonal neighboring rods

for i in range(1,nside+1):

for j in range(1,nside+1):

if POD[i-1,j-1] > 0.0001:

# Side rods

SJ1 = (RP[i-1,j]*WP[i-1,j] + RP[i+1,j]*WP[i+1,j] +

RP[i,j-1]*WP[i,j-1] + RP[i,j+1]*WP[i,j+1])*WJ

SJ2 = (WP[i-1,j] + WP[i+1,j] +

WP[i,j-1] + WP[i,j+1])*WJ*RP[i,j]

SJ = min([SJ1,SJ2])

# Diagonal rods

SK1 = (RP[i-1,j-1]*WP[i-1,j-1] + RP[i+1,j-1]*WP[i+1,j-1] +

RP[i-1,j+1]*WP[i-1,j+1] + RP[i+1,j+1]*WP[i+1,j+1])*WK

SK2 = (WP[i-1,j-1] + WP[i+1,j-1] +

WP[i-1,j+1] + WP[i+1,j+1])*WK*RP[i,j]

SK = min([SK1,SK2])

# Sum weighting factors

SWJ = (WP[i-1,j] + WP[i+1,j] + WP[i,j-1] + WP[i,j+1])*WJ # Side rods

SWK = (WP[i-1,j-1] + WP[i+1,j-1] + WP[i-1,j+1] + WP[i+1,j+1])*WK # Diagonal rods

DOW[i-1,j-1] = (RP[i,j] + SJ + SK)/(1.0 + SWJ + SWK)*np.sqrt(Ntotrods/float(Nhotrods)) + AC[i-1,j-1]

naxial_nodes = 25

DOW = np.zeros((naxial_nodes,POW3.shape[1],POW3.shape[2]))

# Calculate lattice K-factors for all nodes

for z in range(naxial_nodes):

DOW[z,:,:] = rfact_axial(fuetype,POW3[z,:,:])

# Integrate along z-direction to get pinwise R-factors

DOX = sum(DOW,0)/naxial_nodes