uicollectionview select an item immediately after reloaddata?

Question

After calling -[UICollectionView reloadData] it takes some time for cells to be displayed, so selecting an item immediately after calling reloadData do

Answer 1

This is what worked for me:

I kept a reference of the selected index path and overide the reloadData function:

override func reloadData() {
    super.reloadData()
    self.selectItem(at: self.selectedIndexPath, animated: false, scrollPosition: UICollectionViewScrollPosition())
}

I tried doing it using indexPathForSelectedItems, but it was creating an infinite loop on collection view load.