/*
    Copyright (C) 2017 Apple Inc. All Rights Reserved.
    See LICENSE.txt for this sample’s licensing information
    
    Abstract:
    An extension of `DataItem` that provides string properties for a `DataItem`'s image name.
*/

import Foundation

extension DataItem {
    var imageName: String {
        switch group {
            case .Scenery:
                return "\(group.rawValue) \(number)"
                
            default:
                return "\(group.rawValue) \(number).jpg"
        }
    }
    
    var largeImageName: String {
        switch group {
            case .Scenery:
                return "\(group.rawValue) \(number) Large"
                
            default:
                return "\(group.rawValue) \(number) Large.jpg"
        }
    }
}
