更新对齐新sdk接口
This commit is contained in:
@@ -32,6 +32,7 @@ low-level checks, override the default with --power-factor 1.
|
||||
|
||||
import argparse
|
||||
import json
|
||||
import os
|
||||
import signal
|
||||
import subprocess
|
||||
import sys
|
||||
@@ -43,6 +44,31 @@ from pathlib import Path
|
||||
|
||||
import numpy as np
|
||||
|
||||
HERE = Path(__file__).parent.resolve()
|
||||
DEPLOY_ROOT = HERE.parents[1]
|
||||
|
||||
|
||||
def _add_sdk_paths():
|
||||
candidates = []
|
||||
env_root = os.environ.get("GO1_PRO_SDK_ROOT")
|
||||
if env_root:
|
||||
candidates.append(Path(env_root).expanduser())
|
||||
for parent in (HERE, *HERE.parents):
|
||||
candidates.append(parent / "go1_pro_sdk")
|
||||
|
||||
for sdk_root in candidates:
|
||||
fast_root = sdk_root / "fast_lowcmd_cpp"
|
||||
if (sdk_root / "go1_pro_sdk").exists():
|
||||
sys.path.insert(0, str(sdk_root))
|
||||
if fast_root.exists():
|
||||
sys.path.insert(0, str(fast_root))
|
||||
return sdk_root
|
||||
return None
|
||||
|
||||
|
||||
SDK_ROOT = _add_sdk_paths()
|
||||
SDK_FAST_LOW_CMD = SDK_ROOT / "fast_lowcmd_cpp" if SDK_ROOT is not None else None
|
||||
|
||||
from go1_pro_sdk import (
|
||||
MCUClient, MotorMode, PowerProtectViolation, JOINT_NAMES,
|
||||
)
|
||||
@@ -50,12 +76,6 @@ from go1_pro_sdk import (
|
||||
from bpu_policy import BpuInferLibPolicy
|
||||
|
||||
|
||||
HERE = Path(__file__).parent.resolve()
|
||||
DEPLOY_ROOT = HERE.parents[1]
|
||||
WORKSPACE_ROOT = HERE.parents[2]
|
||||
SDK_FAST_LOW_CMD = WORKSPACE_ROOT / "go1_pro_sdk" / "fast_lowcmd_cpp"
|
||||
if SDK_FAST_LOW_CMD.exists():
|
||||
sys.path.insert(0, str(SDK_FAST_LOW_CMD))
|
||||
try:
|
||||
from fast_lowcmd import FastLowCmdBuilder
|
||||
from fast_mcu import FastMCUClient
|
||||
|
||||
Reference in New Issue
Block a user