CollectionView does not display the correct Images about collection-view-in-a-table-view-cell HOT 4 CLOSED

    override func viewDidLoad() {
        super.viewDidLoad()


        let query = MPMediaQuery.audiobooksQuery()
        query.groupingType = MPMediaGrouping.Artist
        collection = query.collections!
        query.groupingType = MPMediaGrouping.Artist

        print(books)

        }

        // Do any additional setup after loading the view.

    override func didReceiveMemoryWarning() {
        super.didReceiveMemoryWarning()
        // Dispose of any resources that can be recreated.
    }





    // MARK - TABLE VIEW HEADER

    func numberOfSectionsInTableView(tableView: UITableView) -> Int {
        return collection.count
    }

    func tableView(tableView: UITableView, heightForHeaderInSection section: Int) -> CGFloat {
        return 80
    }

    func tableView(tableView: UITableView, viewForHeaderInSection section: Int) -> UIView? {
        let headerCell = tableView.dequeueReusableCellWithIdentifier("HeaderCell") as! CustomHeaderCell

        let rowItem = collection[section].representativeItem

        headerCell.artistLabel.text = rowItem?.artist

        let headerView = HeaderView(tableView: self.tableView, section: section)

        headerView.addSubview(headerCell)

        return headerView
    }




    // MARK - TABLE VIEW CELL

    func tableView(tableView: UITableView, numberOfRowsInSection section: Int) -> Int {

        if (!collection.isEmpty) {
            if (self.tableView.sectionOpen != NSNotFound && section == self.tableView.sectionOpen) {
                return 1
            }
        }
        return 0
    }

    func tableView(tableView: UITableView, cellForRowAtIndexPath indexPath: NSIndexPath) -> UITableViewCell {
        let cell = tableView.dequeueReusableCellWithIdentifier("AlbumCell", forIndexPath: indexPath)

//        let rowItem = collection[indexPath.row].representativeItem
//        
//        let artwork = rowItem!.valueForProperty(MPMediaItemPropertyArtwork) as! MPMediaItemArtwork
//        albumImage = artwork.imageWithSize(cell.imageView!.bounds.size)
//        
//        cell.imageView?.image = albumImage


        return cell
    }

    func tableView(tableView: UITableView, willDisplayCell cell: UITableViewCell, forRowAtIndexPath indexPath: NSIndexPath) {

        guard let tableViewCell = cell as? CustomAlbumCell else { return }

        tableViewCell.setCollectionViewDataSourceDelegate(self, forRow: indexPath.row)
        tableViewCell.collectionViewOffset = storedOffsets[indexPath.row] ?? 0

    }

    func tableView(tableView: UITableView, didEndDisplayingCell cell: UITableViewCell, forRowAtIndexPath indexPath: NSIndexPath) {

        guard let tableViewCell = cell as? CustomAlbumCell else { return }

        storedOffsets[indexPath.row] = tableViewCell.collectionViewOffset
    }


    func tableView(tableView: UITableView, didSelectRowAtIndexPath indexPath: NSIndexPath) {
        tableView.deselectRowAtIndexPath(indexPath, animated: true)

    }



    /*
    // MARK: - Navigation

    // In a storyboard-based application, you will often want to do a little preparation before navigation
    override func prepareForSegue(segue: UIStoryboardSegue, sender: AnyObject?) {
        // Get the new view controller using segue.destinationViewController.
        // Pass the selected object to the new view controller.
    }
    */
}

extension MainController: UICollectionViewDelegate, UICollectionViewDataSource {

    func collectionView(collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return books.count
    }


    func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell {
        let cell = collectionView.dequeueReusableCellWithReuseIdentifier("ArtworkCell", forIndexPath: indexPath) as! AlbumCoverCell

        let rowItem = collection[collectionView.tag][indexPath.item]

        var artist = (rowItem?.artist)! as String

        let query = MPMediaQuery.audiobooksQuery()
        let predicate = MPMediaPropertyPredicate(value: artist, forProperty: MPMediaItemPropertyArtist, comparisonType: .EqualTo)
        query.addFilterPredicate(predicate)
        query.groupingType = MPMediaGrouping.Album

        books = query.collections!

        let rowItem1 = books[indexPath.section].representativeItem

        let artwork = rowItem1!.valueForProperty(MPMediaItemPropertyArtwork) as! MPMediaItemArtwork
        albumImage = artwork.imageWithSize(cell.albumCover.bounds.size)

        cell.albumCover.image = albumImage

        return cell
    }

    func collectionView(collectionView: UICollectionView, didSelectItemAtIndexPath indexPath: NSIndexPath) {
        print("Collection view at row \(collectionView.tag) selected index path \(indexPath)")
    }
}

let rowItem1 = books[indexPath.section].representativeItem