Geometry optimization for supercritical water heat transfer enhancement in non-uniformly heated rifled tubes