Skip to content

Instantly share code, notes, and snippets.

@warrenm

warrenm/ARQLThumbnailGenerator.swift

Created June 12, 2018 06:31
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save warrenm/f4f1c7f7e71bd88fc3d3df95b60d5f04 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save warrenm/f4f1c7f7e71bd88fc3d3df95b60d5f04 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. warrenm created this gist Jun 12, 2018.
    31 changes: 31 additions & 0 deletions ARQLThumbnailGenerator.swift
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,31 @@

    import Foundation
    import SceneKit

    class ARQLThumbnailGenerator {
    private let device = MTLCreateSystemDefaultDevice()!

    /// Create a thumbnail image of the asset with the specified URL at the specified
    /// animation time. Supports loading of .scn, .usd, .usdz, .obj, and .abc files,
    /// and other formats supported by ModelIO.
    /// - Parameters:
    /// - url: The file URL of the asset.
    /// - size: The size (in points) at which to render the asset.
    /// - time: The animation time to which the asset should be advanced before snapshotting.
    func thumbnail(for url: URL, size: CGSize, time: TimeInterval = 0) -> UIImage? {
    let renderer = SCNRenderer(device: device, options: [:])
    renderer.autoenablesDefaultLighting = true

    if (url.pathExtension == "scn") {
    let scene = try? SCNScene(url: url, options: nil)
    renderer.scene = scene
    } else {
    let asset = MDLAsset(url: url)
    let scene = SCNScene(mdlAsset: asset)
    renderer.scene = scene
    }

    let image = renderer.snapshot(atTime: time, with: size, antialiasingMode: .multisampling4X)
    return image
    }
    }