// This is based on the following Apple example:
// https://developer.apple.com/documentation/metal/achieving-smooth-frame-rates-with-a-metal-display-link?language=objc
// https://developer.apple.com/wwdc23/10123/

#import "MetalView.h"
#import "MetalConfig.h"

@implementation MetalView {
#if !RENDER_ON_MAIN_THREAD
    // The secondary thread containing the render loop.
    NSThread *_renderThread;

    // The flag to indicate that rendering needs to cease on the main thread.
    BOOL _continueRunLoop;
#endif
}

#pragma mark - Initialization and Setup.

- (instancetype)initWithFrame:(CGRect)frame {
    self = [super initWithFrame:frame];
    if (self) {
        [self initCommon];
    }
    return self;
}

- (instancetype)initWithCoder:(NSCoder *)aDecoder {
    self = [super initWithCoder:aDecoder];
    if (self) {
        [self initCommon];
    }
    return self;
}

- (void)initCommon {
#if TARGET_OS_OSX
    self.wantsLayer = YES;

    self.layerContentsRedrawPolicy = NSViewLayerContentsRedrawDuringViewResize;
#endif

    _metalLayer = (CAMetalLayer *)self.layer;

    self.layer.delegate = self;
}

#if TARGET_OS_IOS || TARGET_OS_TV
+ (Class)layerClass {
    return [CAMetalLayer class];
}

- (void)didMoveToWindow {
    [self movedToWindow];
}
#else
- (CALayer *)makeBackingLayer {
    return [CAMetalLayer layer];
}

- (void)viewDidMoveToWindow {
    [self movedToWindow];
}
#endif  // END TARGET_OS_IOS || TARGET_OS_TV

- (void)movedToWindow {
#if !RENDER_ON_MAIN_THREAD
    // Protect _continueRunLoop with a `@synchronized` block because it's accessed by the separate
    // animation thread.
    @synchronized(self) {
        // Stop the animation loop, allowing it to complete if it's in progress.
        _continueRunLoop = NO;
    }

    // Create and start a secondary NSThread that has another run runloop. The NSThread
    // class calls the 'runThread' method at the start of the secondary thread's execution.
    _renderThread = [[NSThread alloc] initWithTarget:self selector:@selector(runThread) object:nil];
    _continueRunLoop = YES;
    _renderThread.qualityOfService = NSQualityOfServiceUserInteractive;
    [_renderThread start];
#endif  // END !RENDER_ON_MAIN_THREAD

    // Perform any actions that need to know the size and scale of the drawable. When UIKit calls
    // didMoveToWindow after the view initialization, this is the first opportunity to notify
    // components of the drawable's size.
#if AUTOMATICALLY_RESIZE
#if TARGET_OS_IOS || TARGET_OS_TV
    [self resizeDrawable:self.window.screen.nativeScale];
#else
    [self resizeDrawable:self.window.screen.backingScaleFactor];
#endif
#else
    // Notify the delegate of the default drawable size when the system can calculate it.
    CGSize defaultDrawableSize = self.bounds.size;
    defaultDrawableSize.width *= self.layer.contentsScale;
    defaultDrawableSize.height *= self.layer.contentsScale;
    [self.delegate drawableResize:defaultDrawableSize];
#endif
}

#if !RENDER_ON_MAIN_THREAD
- (void)runThread {
    // The system sets the '_continueRunLoop' ivar outside this thread, so it needs to synchronize. Create a
    // 'continueRunLoop' local var that the system can set from the _continueRunLoop ivar in a @synchronized block.
    BOOL continueRunLoop = YES;

    // Begin the run loop.
    while (continueRunLoop) {
        @autoreleasepool {
            [_delegate waitToRenderTo:_metalLayer];

            @synchronized(self) {
                continueRunLoop = _continueRunLoop;
            }
            if (!continueRunLoop) {
                break;
            }

            [_delegate renderTo:_metalLayer];

            @synchronized(self) {
                continueRunLoop = _continueRunLoop;
            }
        }
    }
}
#endif  // END !RENDER_ON_MAIN_THREAD

#pragma mark - Resizing

#if AUTOMATICALLY_RESIZE

// Override all methods that indicate the view's size has changed.

#if TARGET_OS_IOS || TARGET_OS_TV
- (void)setContentScaleFactor:(CGFloat)contentScaleFactor {
    [super setContentScaleFactor:contentScaleFactor];
    [self resizeDrawable:self.window.screen.nativeScale];
}

- (void)layoutSubviews {
    [super layoutSubviews];
    [self resizeDrawable:self.window.screen.nativeScale];
}

- (void)setFrame:(CGRect)frame {
    [super setFrame:frame];
    [self resizeDrawable:self.window.screen.nativeScale];
}

- (void)setBounds:(CGRect)bounds {
    [super setBounds:bounds];
    [self resizeDrawable:self.window.screen.nativeScale];
}
#else
- (void)viewDidChangeBackingProperties {
    [super viewDidChangeBackingProperties];
    [self resizeDrawable:self.window.screen.backingScaleFactor];
}

- (void)setFrameSize:(NSSize)size {
    [super setFrameSize:size];
    [self resizeDrawable:self.window.screen.backingScaleFactor];
}

- (void)setBoundsSize:(NSSize)size {
    [super setBoundsSize:size];
    [self resizeDrawable:self.window.screen.backingScaleFactor];
}
#endif

- (void)resizeDrawable:(CGFloat)scaleFactor {
    CGSize newSize = self.bounds.size;
    newSize.width *= scaleFactor;
    newSize.height *= scaleFactor;

    if (newSize.width <= 0 || newSize.width <= 0) {
        return;
    }

#if RENDER_ON_MAIN_THREAD
    if (newSize.width == _metalLayer.drawableSize.width && newSize.height == _metalLayer.drawableSize.height) {
        return;
    }

    _metalLayer.drawableSize = newSize;

    [_delegate drawableResize:newSize];
#else
    // The system calls all AppKit and UIKit calls that notify of a resize on the main thread. Use
    // a synchronized block to ensure that resize notifications on the delegate are atomic.
    @synchronized(_metalLayer) {
        if (newSize.width == _metalLayer.drawableSize.width && newSize.height == _metalLayer.drawableSize.height) {
            return;
        }

        _metalLayer.drawableSize = newSize;

        [_delegate drawableResize:newSize];
    }
#endif
}
#endif  // END AUTOMATICALLY_RESIZE

@end