128 lines
3.3 KiB
JavaScript
128 lines
3.3 KiB
JavaScript
|
/**
|
|||
|
|
* 手平举运动分析器
|
|||
|
|
*
|
|||
|
|
* @alphaair
|
|||
|
|
* 20220303 created.
|
|||
|
|
**/
|
|||
|
|
|
|||
|
|
const utils = require("../../utils/utils");
|
|||
|
|
const AiSport = requirePlugin("aiSport");
|
|||
|
|
const SportBase = AiSport.sports.SportBase;
|
|||
|
|
const Calculator = AiSport.calc.Calculator;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 手平举运动分析器
|
|||
|
|
*/
|
|||
|
|
class SportHandLift extends SportBase {
|
|||
|
|
_calc = null
|
|||
|
|
_calculator = null
|
|||
|
|
|
|||
|
|
stateTran = -1
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 初始化分析器
|
|||
|
|
*
|
|||
|
|
* @param {int?} pointThreshold 计算器取点评分阈值
|
|||
|
|
*/
|
|||
|
|
constructor(pointThreshold) {
|
|||
|
|
super(pointThreshold);
|
|||
|
|
|
|||
|
|
this.tickMode = true; //计次模式
|
|||
|
|
this._calc = new AiSport.calc.CalcBase();
|
|||
|
|
this._calculator = new Calculator();
|
|||
|
|
this._calculator.pointThreshold = this.pointThreshold;
|
|||
|
|
this.buildRules();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
buildRules() {
|
|||
|
|
this.basePose = {
|
|||
|
|
name: '基本姿态',
|
|||
|
|
calc: '$and',
|
|||
|
|
rules: [{
|
|||
|
|
name: '全身需入镜',
|
|||
|
|
calc: 'whole'
|
|||
|
|
}, {
|
|||
|
|
name: '站立姿态',
|
|||
|
|
calc: 'stand'
|
|||
|
|
}, {
|
|||
|
|
name: '站立姿态',
|
|||
|
|
calc: 'stand'
|
|||
|
|
}, {
|
|||
|
|
name: '任一手平直',
|
|||
|
|
calc: '$or',
|
|||
|
|
rules: [{
|
|||
|
|
name: '左平直状态',
|
|||
|
|
calc: 'match-angle',
|
|||
|
|
angleKey: 'left_shoulder',
|
|||
|
|
secondKey: 'left_hip',
|
|||
|
|
thirdKey: 'left_wrist',
|
|||
|
|
angle: 90,
|
|||
|
|
offset: 20
|
|||
|
|
}, {
|
|||
|
|
name: '右平直状态',
|
|||
|
|
calc: 'match-angle',
|
|||
|
|
angleKey: 'right_shoulder',
|
|||
|
|
secondKey: 'right_hip',
|
|||
|
|
thirdKey: 'right_wrist',
|
|||
|
|
angle: 90,
|
|||
|
|
offset: 20
|
|||
|
|
}]
|
|||
|
|
}]
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
pushing(body) {
|
|||
|
|
|
|||
|
|
if (utils.isNone(body))
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
//基本姿态不符合要求
|
|||
|
|
if (!this._calculator.calculating(body, {
|
|||
|
|
name: '站立姿态',
|
|||
|
|
calc: 'stand'
|
|||
|
|
})) {
|
|||
|
|
let f = this.invalid ? null : -2;
|
|||
|
|
this.invalid = true;
|
|||
|
|
return f;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!this._calculator.calculating(body, this.basePose))
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
this.invalid = false;
|
|||
|
|
let nose = this._calc.findPoint(body.keypoints, 'nose');
|
|||
|
|
if (!nose)
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
let wrist = null;
|
|||
|
|
let leftWrist = this._calc.findPoint(body.keypoints, 'left_wrist');
|
|||
|
|
let rightWrist = this._calc.findPoint(body.keypoints, 'right_wrist');
|
|||
|
|
if (!leftWrist || !rightWrist)
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
if (leftWrist && rightWrist)
|
|||
|
|
wrist = leftWrist.score > rightWrist.score ? leftWrist : rightWrist;
|
|||
|
|
else
|
|||
|
|
wrist = leftWrist || rightWrist;
|
|||
|
|
|
|||
|
|
//前伸算起始动作
|
|||
|
|
if (wrist.x > nose.x) {
|
|||
|
|
this.stateTran = 1;
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//到达结束动作，计数1
|
|||
|
|
if (wrist.x >= nose.x || this.stateTran !== 1)
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
this.stateTran = -1;
|
|||
|
|
this.countTimes();
|
|||
|
|
this.emitTick();
|
|||
|
|
console.log('on-tick');
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
SportHandLift.KEY = "Hand-Lift";
|
|||
|
|
SportHandLift.NAME = "自定义-手臂平伸运动";
|
|||
|
|
|
|||
|
|
module.exports = SportHandLift;
|