let simpleDataArray = ["Sachin", "Rahul", "Saurav", "Virat", "Suresh", "Ravindra", "Chris"]
var simpleSelectedArray = [String]()

let selectionMenu = RSSelectionMenu(dataSource: simpleDataArray, selectedItems:simpleSelectedArray) { (cell, object, indexPath) in
    cell.textLabel?.text = object
}
        
// title
selectionMenu.setNavigationBar(title: "Select Player")

selectionMenu.didSelectRow { (object, isSelected, selectedData) in
    
    // update existing array on select
    self.simpleSelectedArray = selectedData
}
selectionMenu.show(with: .push, from: self)

let dataArray = ["Sachin Tendulkar", "Rahul Dravid", "Saurav Ganguli", "Virat Kohli", "Suresh Raina", "Ravindra Jadeja", "Chris Gyle", "Steve Smith", "Anil Kumble"]
var selectedDataArray = [String]()

let selectionMenu = RSSelectionMenu(dataSource: dataArray, selectedItems: selectedDataArray, cellType: .rightDetail) { (cell, object, indexPath) in
            
	let firstName = object.components(separatedBy: " ").first
	let lastName = object.components(separatedBy: " ").last

	cell.textLabel?.text = firstName
	cell.detailTextLabel?.text = lastName
}
        
// add first row as None
selectionMenu.showFirstRowAs(type: .None, selected: firstRowSelected) { (text, selected) in
    self.firstRowSelected = selected
}

selectionMenu.didSelectRow { (object, isSelected, selectedArray) in
    self.selectedDataArray = selectedArray
}

selectionMenu.show(from: self)

// set type as Multiple
let selectionMenu = RSSelectionMenu(selectionType: .multiple, dataSource: simpleDataArray, selectedItems: simpleSelectedArray) { (cell, object, indexPath) in
            
    cell.textLabel?.text = object
}
        
// set navigation title and color
selectionMenu.setNavigationBar(title: "Select Player", attributes: [NSForegroundColorAttributeName: UIColor.white], barTintColor: UIColor.orange.withAlphaComponent(0.5))


// add first row as All selected
selectionMenu.showFirstRowAs(type: .All, selected: firstRowSelected) { (text, selected) in
    self.firstRowSelected = selected
}

selectionMenu.didSelectRow { (object, isSelected, selectedData) in
    self.simpleSelectedArray = selectedData
}

// add searchbar
selectionMenu.addSearchBar { (searchText) -> ([String]) in

	// return filtered array based on your criteria
    return self.simpleDataArray.filter({ $0.lowercased().hasPrefix(searchText.lowercased()) })
}

selectionMenu.show(from: self)

class Person: NSObject {
    
    let id: Int
    let firstName: String
    let lastName: String

    init(id: Int, firstName: String, lastName: String) {
        self.id = id
        self.firstName = firstName
        self.lastName = lastName
    }
}

var customDataArray = [Person]()
var customselectedDataArray = [Person]()

// prepare data array with your custom models
customDataArray.append(Person(id: 1, firstName: "Sachin", lastName: "Tendulkar"))
customDataArray.append(Person(id: 2, firstName: "Rahul", lastName: "Dravid"))
customDataArray.append(Person(id: 3, firstName: "Virat", lastName: "Kohli"))


// specify unique key for your model or dictionary that contains an unique value.
// This is required to identify each row uniquely.

let selectionMenu = RSSelectionMenu(selectionType: .multiple, dataSource: customDataArray, selectedItems: customselectedDataArray, uniqueKey: "id") { (cell, person, indexPath) in
            
    let customCell = cell as! CustomTableViewCell
    customCell.setData(person: person)
}


// register your xib for tableviewcell        
selectionMenu.registerNib(nibName: "CustomTableViewCell", forCellReuseIdentifier: "cell")

selectionMenu.didSelectRow { (object, isSelected, selectedData) in
    self.customselectedDataArray = selectedData
}

selectionMenu.addSearchBar { (text) -> ([Person]) in
    return self.customDataArray.filter({ $0.firstName.lowercased().hasPrefix(text.lowercased()) })
}
selectionMenu.show(with: .push, from: self)