For BF537_FAMILY, when offset != 1, the mask is 1.
Thus add proper mask for comparing pfunc with function.

Also has small refactor for better readability.
In portmux_setup(), it looks odd having "pmux &= ~(3 << 1);"
while in current code we do pmux |= (function << offset);.
Signed-off-by: NAxel Lin <axel.lin@ingics.com>
Signed-off-by: NSonic Zhang <sonic.zhang@analog.com>