A thermal modelling of displacement cascades in uranium dioxide