yayacal/scripts/profile.sh

#!/bin/bash
#
# YaYa 性能追踪脚本
# 使用 Perfetto 抓取应用 trace，保存到 logs/ 目录
#
# 用法:
#   ./scripts/profile.sh              # 默认抓取 8 秒
#   ./scripts/profile.sh 15           # 抓取 15 秒
#   ./scripts/profile.sh --no-launch  # 不自动启动应用（应用已在运行）
#

set -euo pipefail

PACKAGE="plus.rua.project"
ACTIVITY="plus.rua.project.MainActivity"
PROJECT_ROOT="$(cd "$(dirname "$0")/.." && pwd)"
LOGS_DIR="${PROJECT_ROOT}/logs"

# 解析参数
DURATION_SEC=8
NO_LAUNCH=false

for arg in "$@"; do
  case "$arg" in
    --no-launch)
      NO_LAUNCH=true
      ;;
    --help|-h)
      echo "用法: $0 [秒数] [--no-launch]"
      echo ""
      echo "选项:"
      echo "  秒数         抓取时长（默认 8 秒）"
      echo "  --no-launch  不自动启动应用"
      echo "  --help       显示此帮助"
      exit 0
      ;;
    [0-9]*)
      DURATION_SEC="$arg"
      ;;
  esac
done

DURATION_MS=$((DURATION_SEC * 1000))
TIMESTAMP=$(date +%Y%m%d_%H%M%S)

echo "========================================"
echo "  YaYa 性能追踪"
echo "  包名: ${PACKAGE}"
echo "  时长: ${DURATION_SEC}s"
echo "  输出: ${LOGS_DIR}/"
echo "========================================"

# 创建 logs 目录
mkdir -p "${LOGS_DIR}"

# 检查 adb
if ! command -v adb &>/dev/null; then
  echo "错误: adb 未找到。请确保 Android SDK 的 platform-tools 在 PATH 中。"
  exit 1
fi

# 检查设备连接
DEVICE_COUNT=$(adb devices | grep -c "device$" || true)
if [ "$DEVICE_COUNT" -eq 0 ]; then
  echo "错误: 没有已连接的 Android 设备。"
  exit 1
fi
if [ "$DEVICE_COUNT" -gt 1 ]; then
  echo "警告: 检测到多个设备，将使用默认设备。"
fi

# 检查应用是否已安装
if ! adb shell pm list packages | grep -q "${PACKAGE}"; then
  echo "错误: 应用 ${PACKAGE} 未安装。请先运行 ./gradlew :app:installDebug"
  exit 1
fi

# 启动应用（如果未禁用）
if [ "$NO_LAUNCH" = false ]; then
  echo ""
  echo "[1/5] 启动应用..."
  adb shell am start -n "${PACKAGE}/${ACTIVITY}" >/dev/null 2>&1 || true
  sleep 2
else
  echo ""
  echo "[1/5] 跳过启动 (--no-launch)"
fi

# 抓取 Perfetto trace
echo ""
echo "[2/5] 抓取 Perfetto trace (${DURATION_SEC}s)..."
echo "      请在设备上操作应用，trace 正在记录..."

TRACE_FILE="/data/misc/perfetto-traces/yaya_${TIMESTAMP}.perfetto-trace"
LOCAL_TRACE="${LOGS_DIR}/trace_${TIMESTAMP}.perfetto-trace"
LOCAL_CONFIG="${LOGS_DIR}/.perfetto_config_${TIMESTAMP}.txt"
DEVICE_CONFIG="/data/misc/perfetto-configs/yaya_config_${TIMESTAMP}.txt"

# 生成本地配置文件，然后 push 到设备
cat > "${LOCAL_CONFIG}" <<EOF
buffers {
  size_kb: 131072
  fill_policy: RING_BUFFER
}
buffers {
  size_kb: 4096
  fill_policy: RING_BUFFER
}
data_sources {
  config {
    name: "linux.ftrace"
    ftrace_config {
      ftrace_events: "sched/sched_switch"
      ftrace_events: "sched/sched_wakeup"
      ftrace_events: "power/cpu_frequency"
      ftrace_events: "power/cpu_idle"
      atrace_categories: "gfx"
      atrace_categories: "view"
      atrace_categories: "wm"
      atrace_categories: "am"
      atrace_categories: "input"
      atrace_categories: "sched"
      atrace_categories: "freq"
      atrace_categories: "idle"
      atrace_apps: "${PACKAGE}"
    }
  }
}
data_sources {
  config {
    name: "linux.process_stats"
    target_buffer: 1
    process_stats_config {
      scan_all_processes_on_start: true
    }
  }
}
duration_ms: ${DURATION_MS}
EOF

adb push "${LOCAL_CONFIG}" "${DEVICE_CONFIG}" > /dev/null
rm -f "${LOCAL_CONFIG}"

# 运行 perfetto（前台阻塞，直到 duration_ms 结束）
adb shell "perfetto --txt -c ${DEVICE_CONFIG} -o ${TRACE_FILE}"

# 清理设备上的临时配置文件
adb shell "rm -f ${DEVICE_CONFIG}"

# 拉取 trace 文件
echo "      拉取 trace 文件..."
adb pull "${TRACE_FILE}" "${LOCAL_TRACE}"
adb shell "rm -f ${TRACE_FILE}" || true

# 抓取帧统计
echo ""
echo "[3/5] 抓取帧统计..."
FRAMESTATS_FILE="${LOGS_DIR}/framestats_${TIMESTAMP}.txt"
adb shell dumpsys gfxinfo "${PACKAGE}" framestats > "${FRAMESTATS_FILE}"

# 抓取内存信息
echo ""
echo "[4/5] 抓取内存信息..."
MEMINFO_FILE="${LOGS_DIR}/meminfo_${TIMESTAMP}.txt"
adb shell dumpsys meminfo "${PACKAGE}" > "${MEMINFO_FILE}"

# 生成报告摘要
echo ""
echo "[5/5] 生成摘要..."
REPORT_FILE="${LOGS_DIR}/report_${TIMESTAMP}.md"

# 计算帧率相关数据
FRAME_COUNT=$(grep -c "FrameTimeline" "${FRAMESTATS_FILE}" 2>/dev/null || echo "0")
JANK_COUNT=$(grep -c "jank" "${FRAMESTATS_FILE}" 2>/dev/null || echo "0")

# 获取应用版本
APP_VERSION=$(adb shell dumpsys package "${PACKAGE}" | grep versionName | head -1 | awk '{print $1}' | cut -d= -f2 2>/dev/null || echo "unknown")

# 获取设备信息
DEVICE_MODEL=$(adb shell getprop ro.product.model 2>/dev/null | tr -d '\r')
ANDROID_VERSION=$(adb shell getprop ro.build.version.release 2>/dev/null | tr -d '\r')

cat > "${REPORT_FILE}" <<EOF
# YaYa 性能追踪报告

**时间:** $(date '+%Y-%m-%d %H:%M:%S')
**设备:** ${DEVICE_MODEL} (Android ${ANDROID_VERSION})
**应用版本:** ${APP_VERSION}
**追踪时长:** ${DURATION_SEC}s

## 文件清单

| 文件 | 说明 |
|------|------|
| \`trace_${TIMESTAMP}.perfetto-trace\` | Perfetto trace（在 https://ui.perfetto.dev 中打开） |
| \`framestats_${TIMESTAMP}.txt\` | GPU 帧统计 |
| \`meminfo_${TIMESTAMP}.txt\` | 内存快照 |
| \`report_${TIMESTAMP}.md\` | 本报告 |

## 快速分析指南

### 1. 查看 Compose 渲染耗时
打开 [Perfetto UI](https://ui.perfetto.dev)，上传 trace 文件，搜索：
- \`MonthView:Compose\` — 月视图重组耗时
- \`YearView:Compose\` — 年视图重组耗时
- \`VM:collapseProgress\` — 折叠动画状态更新
- \`getMonthDays\` — 月份网格计算

### 2. 分析帧率
在 \`framestats_${TIMESTAMP}.txt\` 中查看：
- \`FrameTimeline\` 行 — 每帧的 CPU/GPU 耗时
- \`jank\` 标记 — 掉帧情况

### 3. 内存分析
在 \`meminfo_${TIMESTAMP}.txt\` 中关注：
- \`TOTAL\` 行 — 应用总内存占用
- \`Graphics\` 行 — GPU 内存使用
- \`Native Heap\` 行 — 原生堆内存

## 帧统计摘要

- FrameTimeline 条目数: ${FRAME_COUNT}
- Jank 标记数: ${JANK_COUNT}

EOF

echo ""
echo "========================================"
echo "  完成！"
echo "========================================"
echo ""
echo "输出文件:"
echo "  trace:     ${LOCAL_TRACE}"
echo "  framestats: ${FRAMESTATS_FILE}"
echo "  meminfo:   ${MEMINFO_FILE}"
echo "  report:    ${REPORT_FILE}"
echo ""
echo "下一步:"
echo "  1. 打开 https://ui.perfetto.dev"
echo "  2. 上传 trace_${TIMESTAMP}.perfetto-trace"
echo "  3. 搜索 'MonthView:Compose' 查看自定义标记"
echo ""