Get either columns or rows based on axis.
axis
ViewportAxisContext getAxisContextFor({required Axis axis});