Написать кастомный UICollectionViewLayout (плитка)

Нужно написать UICollectionViewLayout, который будет рассчитывать высоту ячеек, чтобы вместить их в заданную высоту.
Некоторые ячейки могут располагаться по 2 в ряд. Нужно написать простую структуру для описания паттерна.
Также этот класс лейаута должен принимать в качестве параметра расстояние между ячейками: по вертикали и по горизонтали (для тех "парных" ячеек)
swift!