ResultController.swift

import UIKit
import NerauModel

class ResultHoverView: UIView {
    public var trainResult: TrainResult? {
        didSet {
            setNeedsDisplay()
        }
    }
    
    private var tileSize: (x: CGFloat, y: CGFloat) = (0, 0)
    
    override func draw(_ rect: CGRect) {
        super.draw(rect)
        guard let strengthMap = trainResult?.strengthMap else { return }
        let size = rect.size
        let ctx = UIGraphicsGetCurrentContext()!
        ctx.setFillColor(UIColor.white.cgColor)
        ctx.fill(CGRect(x: 0, y: 0, width: size.width, height: size.height))
        tileSize = (x: size.width / CGFloat(TrainResult.mapSize.0), y: size.height / CGFloat(TrainResult.mapSize.1))
        for (index, value) in strengthMap.enumerated() {
            let row = CGFloat(index / TrainResult.mapSize.0)
            let col = CGFloat(index % TrainResult.mapSize.1)
            let rect = CGRect(x: col * tileSize.x, y: row * tileSize.y, width: tileSize.x, height: tileSize.y)
            let color = UIColor(white: CGFloat(1.0 - value), alpha: 1.0)
            ctx.setFillColor(color.cgColor)
            ctx.fill(rect)
            
        }
        ctx.setFillColor(UIColor.red.cgColor)
        ctx.fill(CGRect(x: size.width / 2 - 1, y: 0, width: 2, height: size.height))
        ctx.fill(CGRect(x: 0, y: size.height / 2 - 1, width: size.width, height: 2))
        ctx.setStrokeColor(UIColor.darkGray.cgColor)
        ctx.stroke(CGRect(origin: CGPoint.zero, size: size))

    }
    
    public func value(at point: CGPoint) -> Double {
        guard let r = trainResult else { return 0.0 }
        let x = Int(point.x / tileSize.x)
        let y = Int(point.y / tileSize.y)
        let index = (y * TrainResult.mapSize.0) + x
        return r.strengthMap[index]
    }
}

final class ResultWrappingController: UIViewController {
    public var result: TrainResult?

    @IBAction func doClose(sender: UIBarButtonItem) {
        self.dismiss(animated: true, completion: nil)
    }
    
    override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
        switch segue.identifier {
        case "embedResult"?:
            (segue.destination as? ResultController)?.result = result
        default: fatalError("Unknown Segue \(segue.identifier ?? "Anon")")
        }
    }
}

final class ResultController: UIViewController {
    
    #if targetEnvironment(UIKitForMac)
    static let SomethingTouchBarIdentifier = NSTouchBarItem.Identifier(rawValue: "ToggleStar")
    #endif
    
    public var result: TrainResult? {
        didSet {
            listContainer?.dataPoints = (result?.durationPoints.map { $0.1 }) ?? []
            hoverView?.trainResult = result
        }
    }
    
    @IBOutlet weak var containerViewBottom: NSLayoutConstraint!
    @IBOutlet weak var bottomToolbar: UIToolbar!
    
    @IBOutlet var imageView: UIImageView!
    @IBOutlet var durationField: UITextField!
    
    @IBOutlet weak var hoverView: ResultHoverView!
    @IBOutlet weak var valueLabel: UILabel!
    
    private weak var listContainer: ResultTimeIntervalListController?
    
    override func viewDidLoad() {
        super.viewDidLoad()
        #if targetEnvironment(UIKitForMac)
        navigationController?.setNavigationBarHidden(false, animated: true)
        bottomToolbar.removeFromSuperview()
        containerViewBottom.constant = 0
        #endif
        
        valueLabel.text = "0.0"
        
        hoverView.trainResult = self.result
        let hover = UIHoverGestureRecognizer(target: self, action: #selector(hovering(_:)))
        hoverView.addGestureRecognizer(hover)
    }

    #if targetEnvironment(UIKitForMac)
    override func viewDidAppear(_ animated: Bool) {
        super.viewDidAppear(animated)
    }
    #endif
    
    override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
        guard let points = result?.durationPoints else { return }
        if segue.identifier == "resultList" {
            listContainer = segue.destination as? ResultTimeIntervalListController
            (segue.destination as? ResultTimeIntervalListController)?.dataPoints = points.map { $0.1 }
        }
    }
    
    @objc func toggleStar(sender: Any) {
        result?.stared.toggle()
        guard let r = result else { return }
        Database.shared.toggleStarResult(result: r)
        #if targetEnvironment(UIKitForMac)
        self.touchBar = self.makeTouchBar()
        #endif
    }
    
    @IBAction func storeResult(sender: Any?) {
        guard let result = result else { return }
        let data = result.exportData
        let filename = DateFormatter.localizedString(from: result.date, dateStyle: .medium, timeStyle: .medium)
        let paths = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)
        let filePath = paths[0].appendingPathComponent("\(filename).json")
        try! data.write(to: filePath)
        let controller = UIDocumentPickerViewController(url: filePath, in: UIDocumentPickerMode.exportToService)
        (view.window?.rootViewController)?.present(controller, animated: true, completion: nil)
    }
    
    @objc
    func hovering(_ recognizer: UIHoverGestureRecognizer) {
        guard let view = recognizer.view else { return }
        let loc = recognizer.location(in: view)
        let value = hoverView.value(at: loc)
        valueLabel.text = NumberFormatter.localizedString(from: NSNumber(value: value), number: .decimal)
    }
}

#if targetEnvironment(UIKitForMac)
extension ResultController: NSTouchBarDelegate {
    override func makeTouchBar() -> NSTouchBar? {
        let touchBar = NSTouchBar()
        touchBar.delegate = self
        touchBar.defaultItemIdentifiers = [ResultController.SomethingTouchBarIdentifier]
        return touchBar
    }
    
    func touchBar(_ touchBar: NSTouchBar, makeItemForIdentifier identifier: NSTouchBarItem.Identifier) -> NSTouchBarItem? {
        guard let r = result else { return nil }
        switch identifier {
        case ResultController.SomethingTouchBarIdentifier:
            return NSButtonTouchBarItem.init(identifier: identifier,
                                             title: r.stared ? "Unstar" : "Star",
                                             target: self,
                                             action: #selector(self.toggleStar(sender:)))
        default:
            return nil
        }
    }
}

#endif