Obtener imagen de CALayer o NSView (swift 3)

Estaba buscando una forma de renderizar CALayer o NSView y recuperar NSImage.

Tengo una clase personalizada que es una subclase de NSView. Al principio, simplemente hago una capa de degradado para cubrir todo NSView.

class ContentView: NSView {

   override func draw(_ dirtyRect: NSRect) {
      fillGradientLayer()
   }

   func fillGradientLayer() {
      gradientLayer = CAGradientLayer()
      gradientLayer.colors = [#colorLiteral(red: 0, green: 0.9909763549, blue: 0.7570167824, alpha: 1),#colorLiteral(red: 0, green: 0.4772562545, blue: 1, alpha: 1)].map({return $0.cgColor})
      gradientLayer.startPoint = CGPoint(x: 0, y: 0.5)
      gradientLayer.endPoint = CGPoint(x: 1, y: 0.5)
      gradientLayer.frame = self.frame.insetBy(dx: margin, dy: margin)
      gradientLayer.zPosition = 2
      gradientLayer.name = "gradientLayer"
      gradientLayer.contentsScale = (NSScreen.main()?.backingScaleFactor)!
      self.layer?.addSublayer(gradientLayer)
   }
}

En algún momento quiero obtener NSImage (o CGImage) deCALayers. En algunos temas encontré el siguiente enfoque y lo convertí en esto:

extension CALayer {

func getBitmapImage() -> NSImage {

    let btmpImgRep = NSBitmapImageRep(bitmapDataPlanes: nil, pixelsWide: Int(self.frame.width), pixelsHigh: Int(self.frame.height), bitsPerSample: 8, samplesPerPixel: 4, hasAlpha: true, isPlanar: false, colorSpaceName: NSDeviceRGBColorSpace, bytesPerRow: 0, bitsPerPixel: 0)

    let  image: NSImage = NSImage(size: self.frame.size)
    image.lockFocus()
    let ctx = NSGraphicsContext(bitmapImageRep: btmpImgRep!)
    let cgCtxt = ctx!.cgContext

    self.render(in: cgCtxt)
    cgCtxt.draw(layer: CGLayer, in: CGRect)

    image.unlockFocus()
    return image
  }
}

El primer problema es quedibujar (capa: CGLayer, en: CGRect) El método quiere CGLayer pero no CALayer.

En segundo lugar, no sé si el código con el que terminé finalmente generará mi capa o si estoy haciendo algo mal.

Gracias por cualquier ayuda.

Editado: Solución para obtener una CGImage de CALayer.

extension CALayer {

func getBitmapImage() -> NSImage {

    let btmpImgRep =
        NSBitmapImageRep(bitmapDataPlanes: nil, pixelsWide: Int(self.frame.width), pixelsHigh: Int(self.frame.height), bitsPerSample: 8, samplesPerPixel: 4, hasAlpha: true, isPlanar: false, colorSpaceName: NSDeviceRGBColorSpace, bytesPerRow: 0, bitsPerPixel: 32)

    let ctx = NSGraphicsContext(bitmapImageRep: btmpImgRep!)
    let cgContext = ctx!.cgContext

    self.render(in: cgContext)

    let cgImage = cgContext.makeImage()

    let nsimage = NSImage(cgImage: cgImage!, size: CGSize(width: self.frame.width, height: self.frame.height))

        return nsimage    
  }
}

Cómo obtener NSImage de NSView se encuentra en la respuesta a continuación.

PD No sé acerca de las propiedades NSBitmapImageRep, por lo que esta parte podría estar equivocada.

Respuestas a la pregunta(1)

Su respuesta a la pregunta