3-state sorting on Grid

I am looking for something similar to what was asked in this post https://vaadin.com/forum#!/thread/4387769, but for Grid instead of Table. As I understand it currently the sorting of a grid changes between ascending and descending when clicking a column header. Is there any way to get it to change between ascending, descending and the original sorting?

I know I could make a separate button that restores the ordering, but I would really like it to work by clicking the column headers…