UIView extension that returns a UIImage of the view.

UIView+Extension.swift

func getSnapshotImage() -> UIImage {
  UIGraphicsBeginImageContextWithOptions(self.bounds.size, self.opaque, 0)
  self.drawViewHierarchyInRect(self.bounds, afterScreenUpdates: false)
  let snapshotImage: UIImage = UIGraphicsGetImageFromCurrentImageContext()
  UIGraphicsEndImageContext()
  return snapshotImage
}