A court sentencing a person convicted of a criminal offense shall credit to the term of the person's sentence each of the following periods for which the person may be confined in a mental health facility, residential care facility, or jail:
(1) any period of confinement that occurs pending a determination under Subchapter C as to the defendant's competency to stand trial; and
(2) any period of confinement that occurs between the date of any initial determination of the defendant's incompetency under that subchapter and the date the person is transported to jail following a final judicial determination that the person has been restored to competency.