Internal change

PiperOrigin-RevId: 530806084
This commit is contained in:
Mark McDonald 2023-05-09 22:31:46 -07:00 committed by Copybara-Service
parent ea1643f7f8
commit a7ede9235c

View File

@ -14,6 +14,7 @@
# ============================================================================== # ==============================================================================
"""Generate Java reference docs for MediaPipe.""" """Generate Java reference docs for MediaPipe."""
import pathlib import pathlib
import shutil
from absl import app from absl import app
from absl import flags from absl import flags
@ -41,7 +42,9 @@ def main(_) -> None:
mp_root = pathlib.Path(__file__) mp_root = pathlib.Path(__file__)
while (mp_root := mp_root.parent).name != 'mediapipe': while (mp_root := mp_root.parent).name != 'mediapipe':
# Find the nearest `mediapipe` dir. # Find the nearest `mediapipe` dir.
pass if not mp_root.name:
# We've hit the filesystem root - abort.
raise FileNotFoundError('"mediapipe" root not found')
# Find the root from which all packages are relative. # Find the root from which all packages are relative.
root = mp_root.parent root = mp_root.parent
@ -51,6 +54,14 @@ def main(_) -> None:
if (mp_root / 'mediapipe').exists(): if (mp_root / 'mediapipe').exists():
mp_root = mp_root / 'mediapipe' mp_root = mp_root / 'mediapipe'
# We need to copy this into the tasks dir to ensure we don't leave broken
# links in the generated docs.
old_api_dir = 'java/com/google/mediapipe/framework/image'
shutil.copytree(
mp_root / old_api_dir,
mp_root / 'tasks' / old_api_dir,
dirs_exist_ok=True)
gen_java.gen_java_docs( gen_java.gen_java_docs(
package='com.google.mediapipe', package='com.google.mediapipe',
source_path=mp_root / 'tasks/java', source_path=mp_root / 'tasks/java',