The ballet, on now at New York City Center, imagines the renowned writer enclosed in a cell, tortured by images from his own works about the relationship between life and art.