codegen.py fix: bug in the communication of corner directions

In the function comm_directions the corner directions were treated wrongly.
This fix also generalized comm_directions so it will also work for higher dimensions