Something went wrong. Try again.
Monorepo for Aesthetic.Computer aesthetic.computer
Something went wrong. Try again.
9.2 kB · 228 lines
Swift
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229// WaveformView.swift — an audio scrub player drawn as a waveform, with// timestamped comment markers. Click/drag anywhere to seek; option-click// (or the ＋ button in the controller) drops a comment at that instant.// Sibling in spirit to clip-wizard's ScrubPlayerView, but audio-first:// peaks are read off the file with AVAudioFile and the comment markers// are the whole point.import AppKitimport AVFoundation
protocol WaveformViewDelegate: AnyObject { func waveformRequestComment(at t: Double) // option-click → new comment func waveformDidFinish() // track ended → auto-advance func waveformTick() // ~30fps → update time label}
final class WaveformView: NSView { weak var delegate: WaveformViewDelegate? private let deck = DJDeckPlayer() private var peaks: [Float] = [] private var peaksToken = 0 private var timer: Timer? private var preferredVolume: Float = 0.8
var comments: [Comment] = [] { didSet { needsDisplay = true } } private var wasPlaying = false private var scratchOrigin = 0.0 private var lastScratchTime: TimeInterval = 0 private var isScratching = false var duration: Double { deck.duration } var currentTime: Double { deck.currentTime } var isPlaying: Bool { deck.isPlaying } var playbackRate: Double { get { deck.rate } set { deck.setRate(newValue) } } var volume: Float { get { preferredVolume } set { preferredVolume = max(0, min(1, newValue)) deck.setGain(preferredVolume) } }
override var isFlipped: Bool { true }
override init(frame: NSRect) { super.init(frame: frame) wantsLayer = true layer?.backgroundColor = NSColor(white: 0.03, alpha: 0.62).cgColor layer?.borderWidth = 1 layer?.borderColor = NSColor.white.withAlphaComponent(0.10).cgColor layer?.cornerRadius = 8 layer?.masksToBounds = true toolTip = "Click to seek · drag to scratch · Option-click to comment" setAccessibilityLabel("Track waveform and scratch surface") setAccessibilityHelp("Click to seek, drag left or right to scratch, or Option-click to add a comment") } required init?(coder: NSCoder) { fatalError() }
// ── load / transport ──────────────────────────────────────────────── func load(track: Track) { stop() peaks = [] deck.load(track) deck.setGain(preferredVolume) needsDisplay = true computePeaks(url: track.url) }
func play() { deck.play(); wasPlaying = true; startTimer() } func pause() { deck.pause(); wasPlaying = false; stopTimer(); needsDisplay = true } func togglePlay() { if isPlaying { pause() } else { play() } } func seek(to t: Double) { deck.seek(to: max(0, min(duration, t))) needsDisplay = true delegate?.waveformTick() } func stop() { stopTimer() deck.pause() wasPlaying = false needsDisplay = true }
private func startTimer() { stopTimer() timer = Timer.scheduledTimer(withTimeInterval: 1.0 / 60.0, repeats: true) { [weak self] _ in guard let self else { return } self.needsDisplay = true self.delegate?.waveformTick() if self.wasPlaying, !self.deck.isPlaying, self.duration > 0, self.currentTime >= self.duration - 0.05 { self.wasPlaying = false self.stopTimer() self.delegate?.waveformDidFinish() } } } private func stopTimer() { timer?.invalidate(); timer = nil }
// ── waveform peaks (read on a background queue) ────────────────────── private func computePeaks(url: URL) { peaksToken += 1 let token = peaksToken DispatchQueue.global(qos: .userInitiated).async { [weak self] in guard let file = try? AVAudioFile(forReading: url) else { return } let fmt = file.processingFormat let frames = AVAudioFrameCount(file.length) guard frames > 0, let buf = AVAudioPCMBuffer(pcmFormat: fmt, frameCapacity: frames), (try? file.read(into: buf)) != nil, let ch = buf.floatChannelData else { return } let n = Int(buf.frameLength) let chans = Int(fmt.channelCount) let bins = 1000 var out = [Float](repeating: 0, count: bins) let per = max(1, n / bins) for b in 0..<bins { let start = b * per let end = min(n, start + per) var peak: Float = 0 var i = start while i < end { var s: Float = 0 for c in 0..<chans { s += abs(ch[c][i]) } s /= Float(chans) if s > peak { peak = s } i += 1 } out[b] = peak } let mx = out.max() ?? 1 if mx > 0 { for i in 0..<bins { out[i] = out[i] / mx } } DispatchQueue.main.async { guard let self, token == self.peaksToken else { return } self.peaks = out self.needsDisplay = true } } }
// ── drawing ────────────────────────────────────────────────────────── override func draw(_ dirtyRect: NSRect) { let w = bounds.width, h = bounds.height, mid = h / 2 let frac = duration > 0 ? CGFloat(currentTime / duration) : 0 let progX = w * frac
if peaks.isEmpty { let s = "loading…" as NSString let attr: [NSAttributedString.Key: Any] = [ .foregroundColor: NSColor(white: 0.5, alpha: 1), .font: NSFont.systemFont(ofSize: 11)] s.draw(at: NSPoint(x: 10, y: mid - 7), withAttributes: attr) } else { let count = peaks.count let bw = w / CGFloat(count) for i in 0..<count { let x = CGFloat(i) * bw let amp = CGFloat(peaks[i]) * (mid - 4) let played = x <= progX (played ? NSColor.systemTeal : NSColor(white: 0.32, alpha: 1)).setFill() NSBezierPath(rect: NSRect(x: x, y: mid - amp, width: max(1, bw - 0.5), height: amp * 2)).fill() } }
// playhead NSColor.systemYellow.setStroke() let head = NSBezierPath() head.move(to: NSPoint(x: progX, y: 0)) head.line(to: NSPoint(x: progX, y: h)) head.lineWidth = 1.5 head.stroke()
// comment markers — a line + a top triangle flag for c in comments { guard duration > 0 else { break } let x = w * CGFloat(c.t / duration) NSColor.systemOrange.withAlphaComponent(0.85).setStroke() let p = NSBezierPath() p.move(to: NSPoint(x: x, y: 0)); p.line(to: NSPoint(x: x, y: h)); p.lineWidth = 1 p.stroke() NSColor.systemOrange.setFill() let tri = NSBezierPath() tri.move(to: NSPoint(x: x - 4, y: 0)) tri.line(to: NSPoint(x: x + 4, y: 0)) tri.line(to: NSPoint(x: x, y: 7)) tri.close(); tri.fill() } }
// ── interaction ────────────────────────────────────────────────────── private func timeAt(_ event: NSEvent) -> Double { let x = convert(event.locationInWindow, from: nil).x let frac = max(0, min(1, x / bounds.width)) return Double(frac) * duration }
override func mouseDown(with event: NSEvent) { guard duration > 0 else { return } let t = timeAt(event) if event.modifierFlags.contains(.option) { delegate?.waveformRequestComment(at: t) } else { seek(to: t) scratchOrigin = t lastScratchTime = event.timestamp isScratching = true deck.beginScratch() } } override func mouseDragged(with event: NSEvent) { guard duration > 0, isScratching else { return } let target = timeAt(event) let elapsed = max(1.0 / 240.0, event.timestamp - lastScratchTime) deck.scratch(to: target, movement: target - scratchOrigin, elapsed: elapsed) scratchOrigin = target lastScratchTime = event.timestamp needsDisplay = true delegate?.waveformTick() } override func mouseUp(with event: NSEvent) { guard isScratching else { return } isScratching = false deck.endScratch() wasPlaying = deck.isPlaying if wasPlaying { startTimer() } }}