Add localshapes method to distribution
get_localshapes -> localshapes, get_shape_from... -> shapes_from...

get_shape_from_dim_data_per_rank -> shapes_from_dim_data_per_rank