Swiftgram/submodules/GraphCore/Sources/Charts/Renderes/VerticalScalesRenderer.swift

//
//  VerticalScalesRenderer.swift
//  GraphTest
//
//  Created by Andrei Salavei on 4/8/19.
//  Copyright © 2019 Andrei Salavei. All rights reserved.
//

import Foundation
#if os(macOS)
import Cocoa
public typealias UIColor = NSColor
#else
import UIKit
#endif

class VerticalScalesRenderer: BaseChartRenderer {
    private var verticalLabelsAndLines: [LinesChartLabel] = []
    private var animatedVerticalLabelsAndLines: [AnimatedLinesChartLabels] = []
    private lazy var horizontalLinesAlphaAnimator: AnimationController<CGFloat> = {
        return AnimationController(current: 1, refreshClosure: self.refreshClosure)
    }()
    
    var drawAxisX: Bool = true
    var axisXColor: GColor = .black
    var axisXWidth: CGFloat = GView.oneDevicePixel
    
    var isRightAligned: Bool = false
    var drawCurrency:((CGContext, UIColor, CGPoint)->Void)?
    
    var horizontalLinesColor: GColor = .black {
        didSet {
            setNeedsDisplay()
        }
    }
    var horizontalLinesWidth: CGFloat = GView.oneDevicePixel
    var labelsAxisOffset: CGFloat = 6
    var labelsColor: GColor = .black {
        didSet {
            setNeedsDisplay()
        }
    }
    var labelsFont: NSFont = .systemFont(ofSize: 11)

    func setHorizontalLinesVisible(_ visible: Bool, animated: Bool) {
        let destinationValue: CGFloat = visible ? 1 : 0
        guard self.horizontalLinesAlphaAnimator.end != destinationValue else { return }
        if animated {
            self.horizontalLinesAlphaAnimator.animate(to: destinationValue, duration: .defaultDuration)
        } else {
            self.horizontalLinesAlphaAnimator.set(current: destinationValue)
        }
    }

    func setup(verticalLimitsLabels: [LinesChartLabel], animated: Bool) {
        if animated {
            var labelsToKeepVisible: [LinesChartLabel] = []
            let labelsToHide: [LinesChartLabel]
            var labelsToShow: [LinesChartLabel] = []
            
            for label in verticalLimitsLabels {
                if verticalLabelsAndLines.contains(label) {
                    labelsToKeepVisible.append(label)
                } else {
                    labelsToShow.append(label)
                }
            }
            labelsToHide = verticalLabelsAndLines.filter { !verticalLimitsLabels.contains($0) }
            animatedVerticalLabelsAndLines.removeAll(where: { $0.isAppearing })
            verticalLabelsAndLines = labelsToKeepVisible
            
            let showAnimation = AnimatedLinesChartLabels(labels: labelsToShow, alphaAnimator: AnimationController(current: 1.0, refreshClosure: refreshClosure))
            showAnimation.isAppearing = true
            showAnimation.alphaAnimator.set(current: 0)
            showAnimation.alphaAnimator.animate(to: 1, duration: .defaultDuration)
            showAnimation.alphaAnimator.completionClosure = { [weak self, weak showAnimation] in
                guard let self = self, let showAnimation = showAnimation else { return }
                self.animatedVerticalLabelsAndLines.removeAll(where: { $0 === showAnimation })
                self.verticalLabelsAndLines = verticalLimitsLabels
            }
            
            let hideAnimation = AnimatedLinesChartLabels(labels: labelsToHide, alphaAnimator: AnimationController(current: 1.0, refreshClosure: refreshClosure))
            hideAnimation.isAppearing = false
            hideAnimation.alphaAnimator.set(current: 1)
            hideAnimation.alphaAnimator.animate(to: 0, duration: .defaultDuration)
            hideAnimation.alphaAnimator.completionClosure = { [weak self, weak hideAnimation] in
                guard let self = self, let hideAnimation = hideAnimation else { return }
                self.animatedVerticalLabelsAndLines.removeAll(where: { $0 === hideAnimation })
            }
            
            animatedVerticalLabelsAndLines.append(showAnimation)
            animatedVerticalLabelsAndLines.append(hideAnimation)
        } else {
            verticalLabelsAndLines = verticalLimitsLabels
            animatedVerticalLabelsAndLines = []
        }
    }
    
    override func render(context: CGContext, bounds: CGRect, chartFrame: CGRect) {
        guard isEnabled && verticalRange.current.distance > 0 && verticalRange.current.distance > 0 else { return }
        let generalAlpha = chartAlphaAnimator.current
        if generalAlpha == 0 { return }
        let labelColorAlpha = labelsColor.alphaValue
        
        let spacing: CGFloat = 1.0
        context.clip(to: CGRect(origin: CGPoint(x: 0.0, y: chartFrame.minY - spacing), size: CGSize(width: chartFrame.width + chartFrame.origin.x * 2.0, height: chartFrame.height + spacing * 2.0)))
        
        func drawLines(_ labels: [LinesChartLabel], alpha: CGFloat) {
            var lineSegments: [CGPoint] = []
            let x0 = chartFrame.minX
            let x1 = chartFrame.maxX
            
            context.setStrokeColor(horizontalLinesColor.withAlphaComponent(horizontalLinesColor.alphaValue * alpha).cgColor)
            
            for lineInfo in labels {
                let y = transform(toChartCoordinateVertical: lineInfo.value, chartFrame: chartFrame).roundedUpToPixelGrid()
                if y < chartFrame.maxY - 2.0 {
                    lineSegments.append(CGPoint(x: x0, y: y))
                    lineSegments.append(CGPoint(x: x1, y: y))
                }
            }
            context.strokeLineSegments(between: lineSegments)
        }
        
        func drawVerticalLabels(_ labels: [LinesChartLabel], attributes: [NSAttributedString.Key: Any]) {
            if isRightAligned {
                for label in labels {
                    let y = transform(toChartCoordinateVertical: label.value, chartFrame: chartFrame) - labelsFont.pointSize - labelsAxisOffset
                    let attributedString = NSAttributedString(string: label.text, attributes: attributes)
                    let textNode = LabelNode.layoutText(attributedString, bounds.size)
                    textNode.1.draw(CGRect(origin: CGPoint(x:chartFrame.maxX - textNode.0.size.width, y: y), size: textNode.0.size), in: context, backingScaleFactor: deviceScale)
                }
            } else {
                for label in labels {
                    let y = transform(toChartCoordinateVertical: label.value, chartFrame: chartFrame) - labelsFont.pointSize - labelsAxisOffset
                    let attributedString = NSAttributedString(string: label.text, attributes: attributes)
                    let textNode = LabelNode.layoutText(attributedString, bounds.size)
                    
                    var xOffset = 0.0
                    if let drawCurrency {
                        xOffset += 11.0
                        drawCurrency(context, attributes[.foregroundColor] as? UIColor ?? .black, CGPoint(x: chartFrame.minX, y: y + 4.0))
                    }
                    
                    textNode.1.draw(CGRect(origin: CGPoint(x: chartFrame.minX + xOffset, y: y), size: textNode.0.size), in: context, backingScaleFactor: deviceScale)
                }
            }
        }
        
        let horizontalLinesAlpha = horizontalLinesAlphaAnimator.current
        if horizontalLinesAlpha > 0 {
            context.setLineWidth(horizontalLinesWidth)
            
            drawLines(verticalLabelsAndLines, alpha: generalAlpha)
            for animatedLabesAndLines in animatedVerticalLabelsAndLines {
                drawLines(animatedLabesAndLines.labels, alpha: animatedLabesAndLines.alphaAnimator.current * generalAlpha * horizontalLinesAlpha)
            }
            
            if drawAxisX {
                context.setLineWidth(axisXWidth)
                context.setStrokeColor(axisXColor.withAlphaComponent(axisXColor.alphaValue * horizontalLinesAlpha * generalAlpha).cgColor)
                
                let lineSegments: [CGPoint] = [CGPoint(x: chartFrame.minX, y: chartFrame.maxY.roundedUpToPixelGrid()),
                                               CGPoint(x: chartFrame.maxX, y: chartFrame.maxY.roundedUpToPixelGrid())]
                
                context.strokeLineSegments(between: lineSegments)
            }
        }
        
        drawVerticalLabels(verticalLabelsAndLines, attributes: [.foregroundColor: labelsColor.withAlphaComponent(labelColorAlpha * generalAlpha),
                                                                .font: labelsFont])
        for animatedLabesAndLines in animatedVerticalLabelsAndLines {
            drawVerticalLabels(animatedLabesAndLines.labels,
                               attributes: [.foregroundColor: labelsColor.withAlphaComponent(animatedLabesAndLines.alphaAnimator.current * labelColorAlpha * generalAlpha),
                                            .font: labelsFont])
        }
        
        context.resetClip()
    }
}