Kevin Pietersen is set to join England men's cricket white-ball teams in a coaching role ahead of next year's World Cup in Africa.