[iOS/UI] Swift 앨범 사진 고른 갯수 차례로 번호 부여하기

우선 나는 아래 카톡 사진고르기와 같이 탭한 순서대로 번호가 부여되면서 사진배열에 사진이 들어가게 하고

다시 탭할 경우 배열의 사진을 삭제해주고 그보다 큰 번호들은 삭제한 갯수만큼 번호가 작아져야했다.

우선 2개의 배열이 필요한데 하나는 사진을 담을 배열과 하나는 몇 번째 인덱스가 선택되었는지 담을 배열이 필요하다.

    var photos = [UIImage]()
    var photoarray = [Int]()

그 다음 cellForItemAt에

포토어레이가 비어있지않다면 !photoarray.isEmpty{ for문을 이용해 photoarray안에 있는 숫자들을 차례로 indexPath와 비교해서 맞다면 해당 숫자라벨(cell.number)를 i로 바꿔준다. 그러면 photoarray안에 순서대로 텍스트가 표시된다. 그리고 cell.number.isHidden을 풀어주면 숫자가 보이게 된다. borderWidth와 borderColor는 해당 셀을 눌렀을 때 테두리의 색깔과 두께를 정해준 것이다,

  func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
          let cell = collection.dequeueReusableCell(withReuseIdentifier: "CustomPhotoCell", for: indexPath) as! CustomPhotoCell
        cell.image.image = fetchimages[indexPath.item]
        cell.image.contentMode = .scaleAspectFill
        cell.layer.borderWidth = 0
        cell.number.isHidden = true
         if !photoarray.isEmpty{
               for i in 0...photoarray.count-1 {
                   if indexPath.item == photoarray[i] {
                     cell.layer.borderColor = UIColor(displayP3Red: 162/255, green: 32/255, blue: 58/255, alpha: 1).cgColor
                        cell.layer.borderWidth = 3
                        cell.number.isHidden = false
                       cell.number.text = "\(i)"
                   }
               }
        }
        cell.number.layer.cornerRadius = 10
        cell.number.layer.masksToBounds = true
        return cell
    }

셀을 선택했을 때 didSelectItemAt에 

만약 해당 숫자라벨이 숨겨져 있다면 (나같은 경우는 최대 8대까지 고를 수 있게 해놨기 때문에 photos안에 있는 사진이 8개 미만이라면 넣었다.) photoarray에 해당 indexPath.item을 넣어주고 photos엔 해당 image를 넣어준다.

그렇지 않다면

photos에서 해당 cell이 어떤 숫자가 있는 지 확인하고 그 숫자에 맞게 삭제해준다. (photoarray도 마찬가지)

 func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
        let cell = collection.cellForItem(at: indexPath) as! CustomPhotoCell
        if cell.number.isHidden == true {
            if photos.count < 8 {
            photoarray.append(indexPath.item)
            photos.append(cell.image.image!)
            }
        } else{
            photos.remove(at:Int(cell.number.text!)!)
            photoarray.remove(at: Int(cell.number.text!)!)
        }
}

위와 같이 탭했을 경우 photoarray엔 다음과 같이 저장되고 삭제된다.