添加语音websocket等,还没测试
This commit is contained in:
163
keyBoard/Class/AiTalk/V/KBAiWaveformView.m
Normal file
163
keyBoard/Class/AiTalk/V/KBAiWaveformView.m
Normal file
@@ -0,0 +1,163 @@
|
||||
//
|
||||
// KBAiWaveformView.m
|
||||
// keyBoard
|
||||
//
|
||||
// Created by Mac on 2026/1/15.
|
||||
//
|
||||
|
||||
#import "KBAiWaveformView.h"
|
||||
|
||||
@interface KBAiWaveformView ()
|
||||
@property(nonatomic, strong) NSMutableArray<CAShapeLayer *> *barLayers;
|
||||
@property(nonatomic, strong) NSMutableArray<NSNumber *> *barHeights;
|
||||
@property(nonatomic, strong) CADisplayLink *displayLink;
|
||||
@property(nonatomic, assign) float currentRMS;
|
||||
@property(nonatomic, assign) float targetRMS;
|
||||
@property(nonatomic, assign) BOOL isAnimating;
|
||||
@end
|
||||
|
||||
@implementation KBAiWaveformView
|
||||
|
||||
- (instancetype)initWithFrame:(CGRect)frame {
|
||||
self = [super initWithFrame:frame];
|
||||
if (self) {
|
||||
[self setup];
|
||||
}
|
||||
return self;
|
||||
}
|
||||
|
||||
- (instancetype)initWithCoder:(NSCoder *)coder {
|
||||
self = [super initWithCoder:coder];
|
||||
if (self) {
|
||||
[self setup];
|
||||
}
|
||||
return self;
|
||||
}
|
||||
|
||||
- (void)setup {
|
||||
_waveColor = [UIColor systemBlueColor];
|
||||
_barCount = 5;
|
||||
_barWidth = 4;
|
||||
_barSpacing = 3;
|
||||
_barLayers = [[NSMutableArray alloc] init];
|
||||
_barHeights = [[NSMutableArray alloc] init];
|
||||
_currentRMS = 0;
|
||||
_targetRMS = 0;
|
||||
|
||||
self.backgroundColor = [UIColor clearColor];
|
||||
}
|
||||
|
||||
- (void)layoutSubviews {
|
||||
[super layoutSubviews];
|
||||
[self setupBars];
|
||||
}
|
||||
|
||||
- (void)setupBars {
|
||||
// 移除旧的图层
|
||||
for (CAShapeLayer *layer in self.barLayers) {
|
||||
[layer removeFromSuperlayer];
|
||||
}
|
||||
[self.barLayers removeAllObjects];
|
||||
[self.barHeights removeAllObjects];
|
||||
|
||||
// 计算总宽度
|
||||
CGFloat totalWidth =
|
||||
self.barCount * self.barWidth + (self.barCount - 1) * self.barSpacing;
|
||||
CGFloat startX = (self.bounds.size.width - totalWidth) / 2;
|
||||
CGFloat maxHeight = self.bounds.size.height;
|
||||
CGFloat minHeight = maxHeight * 0.2;
|
||||
|
||||
for (NSInteger i = 0; i < self.barCount; i++) {
|
||||
CAShapeLayer *barLayer = [CAShapeLayer layer];
|
||||
barLayer.fillColor = self.waveColor.CGColor;
|
||||
barLayer.cornerRadius = self.barWidth / 2;
|
||||
|
||||
CGFloat x = startX + i * (self.barWidth + self.barSpacing);
|
||||
CGFloat height = minHeight;
|
||||
CGFloat y = (maxHeight - height) / 2;
|
||||
|
||||
barLayer.frame = CGRectMake(x, y, self.barWidth, height);
|
||||
barLayer.backgroundColor = self.waveColor.CGColor;
|
||||
|
||||
[self.layer addSublayer:barLayer];
|
||||
[self.barLayers addObject:barLayer];
|
||||
[self.barHeights addObject:@(height)];
|
||||
}
|
||||
}
|
||||
|
||||
#pragma mark - Public Methods
|
||||
|
||||
- (void)updateWithRMS:(float)rms {
|
||||
self.targetRMS = MIN(MAX(rms, 0), 1);
|
||||
}
|
||||
|
||||
- (void)startIdleAnimation {
|
||||
if (self.isAnimating)
|
||||
return;
|
||||
|
||||
self.isAnimating = YES;
|
||||
self.displayLink =
|
||||
[CADisplayLink displayLinkWithTarget:self
|
||||
selector:@selector(updateAnimation)];
|
||||
[self.displayLink addToRunLoop:[NSRunLoop mainRunLoop]
|
||||
forMode:NSRunLoopCommonModes];
|
||||
}
|
||||
|
||||
- (void)stopAnimation {
|
||||
self.isAnimating = NO;
|
||||
[self.displayLink invalidate];
|
||||
self.displayLink = nil;
|
||||
}
|
||||
|
||||
- (void)reset {
|
||||
self.currentRMS = 0;
|
||||
self.targetRMS = 0;
|
||||
[self updateBarsWithRMS:0];
|
||||
}
|
||||
|
||||
#pragma mark - Animation
|
||||
|
||||
- (void)updateAnimation {
|
||||
// 平滑过渡到目标 RMS
|
||||
CGFloat smoothing = 0.3;
|
||||
self.currentRMS =
|
||||
self.currentRMS + (self.targetRMS - self.currentRMS) * smoothing;
|
||||
|
||||
[self updateBarsWithRMS:self.currentRMS];
|
||||
}
|
||||
|
||||
- (void)updateBarsWithRMS:(float)rms {
|
||||
CGFloat maxHeight = self.bounds.size.height;
|
||||
CGFloat minHeight = maxHeight * 0.2;
|
||||
CGFloat range = maxHeight - minHeight;
|
||||
|
||||
// 为每个条添加略微不同的高度和相位
|
||||
NSTimeInterval time = CACurrentMediaTime();
|
||||
|
||||
for (NSInteger i = 0; i < self.barLayers.count; i++) {
|
||||
CAShapeLayer *layer = self.barLayers[i];
|
||||
|
||||
// 添加基于时间的波动效果
|
||||
CGFloat phase = (CGFloat)i / self.barLayers.count * M_PI * 2;
|
||||
CGFloat wave = sin(time * 3 + phase) * 0.3 + 0.7; // 0.4 - 1.0
|
||||
|
||||
// 计算高度
|
||||
CGFloat heightFactor = rms * wave;
|
||||
CGFloat height = minHeight + range * heightFactor;
|
||||
height = MAX(minHeight, MIN(maxHeight, height));
|
||||
|
||||
// 更新位置
|
||||
CGFloat y = (maxHeight - height) / 2;
|
||||
|
||||
[CATransaction begin];
|
||||
[CATransaction setDisableActions:YES];
|
||||
layer.frame = CGRectMake(layer.frame.origin.x, y, self.barWidth, height);
|
||||
[CATransaction commit];
|
||||
}
|
||||
}
|
||||
|
||||
- (void)dealloc {
|
||||
[self stopAnimation];
|
||||
}
|
||||
|
||||
@end
|
||||
Reference in New Issue
Block a user