【一步步开发AI运动APP】十二、自定义扩展新运动项目2

/**
 * 双手并举运动分析器
 */
export class BothHandsUpSport {

	context = null;
	calculator = null; //ICalculator
	stateTran = -1;

	/**
	 * 初始化运动分析器
	 */
	constructor() {
		this.calculator = createCalculator();

		this.buildRules();
		const that = this;
		this.context = createExtendSportContext({
			key: 'both-hands-up',
			name: '自定义-双手并举',
			tickMode: true,
			view: 'front_back',
			start: () => that.start(),
			// reset: this.reset,
			// stop: this.stop,
			pushing: human => that.pushing(human)
		});
	}

	buildRules() {
		this.rules = {};
		this.rules.ups = {
			name: '双手上举',
			calc: '$and',
			rules: [{
				name: '左手上举',
				calc: 'position',
				referenceKey: 'left_eye',
				positionKey: 'left_wrist',
				position: 'top',
				relaxed: true
			}, {
				name: '右手上举',
				calc: 'position',
				referenceKey: 'right_eye',
				positionKey: 'right_wrist',
				position: 'top',
				relaxed: true
			}]
		};

		this.rules.downs = {
			name: '双手下垂',
			calc: '$and',
			rules: [{
				name: '站立姿态',
				calc: 'stand',
				offset: 25
			}, {
				name: '左腋夹角',
				calc: 'match-angle',
				angleKey: 'left_shoulder',
				secondKey: 'left_hip',
				thirdKey: 'left_wrist',
				angle: 30,
				offset: 30
			}, {
				name: '左手下垂',
				calc: 'vertical',
				upperKey: 'left_shoulder',
				centerKey: 'left_elbow',
				lowerKey: 'left_wrist',
				offset: 25
			}, {
				name: '右腋夹角',
				calc: 'match-angle',
				angleKey: 'right_shoulder',
				secondKey: 'right_hip',
				thirdKey: 'right_wrist',
				angle: 30,
				offset: 30
			}, {
				name: '右手下垂',
				calc: 'vertical',
				upperKey: 'right_shoulder',
				centerKey: 'right_elbow',
				lowerKey: 'right_wrist',
				offset: 25
			}]
		};
	}

	start() {
		this.stateTran = -1;
		console.log('运动启动', this);
	}

	pushing(fragment) {

		if (fragment.isNobody) {
			console.log('未识别到人体');
			return;
		}

		const human = fragment.human;
		if (this.stateTran != 1 && this.calculator.calculating(human, this.rules.downs)) {
			this.stateTran = 1;
			return;
		}

		if (this.stateTran == 1 && this.calculator.calculating(human, this.rules.ups)) {
			this.stateTran = 2;
			this.context.countTimes();
			this.context.emitTick(1);
		}
	}

	/**
	 * 获取当前扩展运动的原生分析器实例
	 * @returns 原生运动分析器实例
	 */
	getSportInstance() {
		return this.context.getSport();
	}

	/**
	 * 获取当前扩展运动描述
	 * 
	 * @returns  运动描述条目实例
	 */
	static getDescriptor() {
		let item = {
			key: 'both-hands-up',
			name: '自定义-双手并举'
		};

		return item;
	}
}