Computer simulation of the heat treatment process applied to a forged shaft