127 lines
3.2 KiB
JavaScript
127 lines
3.2 KiB
JavaScript
|
const utils = require("../../utils/utils");
|
|||
|
|
const module = require("../index.js");
|
|||
|
|
const SportHandLift = require("../sports/sport-hand-lift");
|
|||
|
|
const AiSport = requirePlugin("aiSport");
|
|||
|
|
const humanDetection = AiSport.humanDetection;
|
|||
|
|
const SportBase = AiSport.sports.SportBase;
|
|||
|
|
|
|||
|
|
//注册自定义运动
|
|||
|
|
SportBase.register(SportHandLift.KEY, SportHandLift.NAME, () => {
|
|||
|
|
return new SportHandLift();
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
Page({
|
|||
|
|
data: {
|
|||
|
|
version: AiSport.version.release,
|
|||
|
|
ve: null,
|
|||
|
|
enhanced: module.judgeEnhanced(),
|
|||
|
|
threshold: 0.25,
|
|||
|
|
sports: utils.clone(SportBase.SPORTS),
|
|||
|
|
sportKey: SportBase.SPORTS[1].key
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
async onLoad() {
|
|||
|
|
this.setData({
|
|||
|
|
enhanced: module.judgeEnhanced()
|
|||
|
|
});
|
|||
|
|
this.initializeVe();
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
initializeVe() {
|
|||
|
|
|
|||
|
|
wx.showLoading({
|
|||
|
|
title: '加载模型...'
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
const that = this;
|
|||
|
|
humanDetection.initialize({
|
|||
|
|
ve: that.data.ve,
|
|||
|
|
enhanced: that.data.enhanced,
|
|||
|
|
callback: (err) => {
|
|||
|
|
wx.hideLoading();
|
|||
|
|
const ve = humanDetection.getVe();
|
|||
|
|
let ehd = false;
|
|||
|
|
if (ve == 1)
|
|||
|
|
ehd = humanDetection.isEnhanced();
|
|||
|
|
|
|||
|
|
if (!err) {
|
|||
|
|
wx.showToast({
|
|||
|
|
icon: 'none',
|
|||
|
|
title: `初始成功，引擎ve${ve} ${ehd ? '增强' : ''}`
|
|||
|
|
})
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
wx.setClipboardData({
|
|||
|
|
data: err.message,
|
|||
|
|
success: () => {
|
|||
|
|
wx.showToast({
|
|||
|
|
title: '初始化失败',
|
|||
|
|
icon: 'none'
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
},
|
|||
|
|
onShowTips(e) {
|
|||
|
|
const msg = e.currentTarget.dataset.msg;
|
|||
|
|
wx.showModal({
|
|||
|
|
content: msg,
|
|||
|
|
showCancel: false
|
|||
|
|
});
|
|||
|
|
},
|
|||
|
|
onSelectVe(e) {
|
|||
|
|
const v = e.currentTarget.dataset.v;
|
|||
|
|
this.setData({ ve: v !== "null" ? Number(v) : null });
|
|||
|
|
this.initializeVe();
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
onToggleEnhanced() {
|
|||
|
|
this.setData({
|
|||
|
|
enhanced: !this.data.enhanced
|
|||
|
|
});
|
|||
|
|
this.initializeVe()
|
|||
|
|
},
|
|||
|
|
onThresholdChanged(e) {
|
|||
|
|
const reg = /^[0-1](\.\d{1,2})?$/;
|
|||
|
|
const val = e.detail.value;
|
|||
|
|
const th = parseFloat(val);
|
|||
|
|
|
|||
|
|
if (reg.test(val) && th <= 1) {
|
|||
|
|
module.initializePlugin(th, this.data.enhanced);
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
wx.showToast({
|
|||
|
|
icon: 'none',
|
|||
|
|
title: '阈值必须在0-1之间'
|
|||
|
|
});
|
|||
|
|
this.setData({
|
|||
|
|
threshold: 0.35
|
|||
|
|
});
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
onSelectSport(e) {
|
|||
|
|
const key = e.currentTarget.dataset.key;
|
|||
|
|
this.setData({
|
|||
|
|
sportKey: key
|
|||
|
|
});
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
onNavSport() {
|
|||
|
|
wx.navigateTo({
|
|||
|
|
url: `/module-sport/pages/sport?sportKey=${this.data.sportKey}`
|
|||
|
|
});
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
onNavHuman() {
|
|||
|
|
wx.navigateTo({ url: '/module-sport/pages/human' });
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
onShareAppMessage() {
|
|||
|
|
return {
|
|||
|
|
path: '/module-sport/pages/index',
|
|||
|
|
title: '跃动小程序'
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
});
|