IMSS TRAMITE ST4 INVALIDEZ URGENTE

Tuve una operación por enfermedad general pero no quede bien de la rodilla y soy chofer. En el Hospital de Rehabilitación me dieron una hoja para tramite de invalidez ST-4 para mi UMF. Aun no me ha dado de baja el patrón. ¿Es necesario que este dado de baja en el IMSS con derechos vigentes se tienen 56 días después de la baja? o ¿Puedo estar dado de alta y así empezar la evaluación de SALUD DEL TRABAJO? ¿Cuál es el fundamento legal del IMSS?