Drift-flux model for rod bundle geometry