renderer and vis readthedocs #1363

Summary: Autogenerate docs for the renderer too. This will be helpful but make a slightly ugly TOC

Reviewed By: kjchalup

Differential Revision: D40977315

fbshipit-source-id: 10831de3ced68080cb5671c5dc31d4da8500f761

docs/generate_stubs.py

@@ -12,7 +12,6 @@ This script makes the stubs for implicitron in docs/modules.
 from pathlib import Path
 
 ROOT_DIR = Path(__file__).resolve().parent.parent
-DEST_DIR = Path(__file__).resolve().parent / "modules/implicitron"
 
 
 def paths_to_modules(paths):
@@ -95,54 +94,69 @@ def make_directory_index(title: str, directory_path: Path):
             print(f"    {rst.stem}", file=f)
 
 
-iterate_directory(ROOT_DIR / "pytorch3d/implicitron/models", DEST_DIR / "models")
+def do_implicitron():
+    DEST_DIR = Path(__file__).resolve().parent / "modules/implicitron"
 
-unwanted_tools = ["configurable", "depth_cleanup", "utils"]
-tools_sources = sorted(ROOT_DIR.glob("pytorch3d/implicitron/tools/*.py"))
-tools_modules = [
-    str(i.relative_to(ROOT_DIR))[:-3].replace("/", ".")
-    for i in tools_sources
-    if i.stem not in unwanted_tools
-]
-create_one_file(
-    "pytorch3d.implicitron.tools",
-    "Tools for implicitron",
-    tools_modules,
-    DEST_DIR / "tools.rst",
-)
+    iterate_directory(ROOT_DIR / "pytorch3d/implicitron/models", DEST_DIR / "models")
 
-dataset_files = sorted(ROOT_DIR.glob("pytorch3d/implicitron/dataset/*.py"))
-basic_dataset = [
-    "dataset_base",
-    "dataset_map_provider",
-    "data_loader_map_provider",
-    "data_source",
-    "scene_batch_sampler",
-]
-basic_dataset_modules = [f"pytorch3d.implicitron.dataset.{i}" for i in basic_dataset]
-create_one_file(
-    "pytorch3d.implicitron.dataset in general",
-    "Basics of data for implicitron",
-    basic_dataset_modules,
-    DEST_DIR / "data_basics.rst",
-)
+    unwanted_tools = ["configurable", "depth_cleanup", "utils"]
+    tools_sources = sorted(ROOT_DIR.glob("pytorch3d/implicitron/tools/*.py"))
+    tools_modules = [
+        str(i.relative_to(ROOT_DIR))[:-3].replace("/", ".")
+        for i in tools_sources
+        if i.stem not in unwanted_tools
+    ]
+    create_one_file(
+        "pytorch3d.implicitron.tools",
+        "Tools for implicitron",
+        tools_modules,
+        DEST_DIR / "tools.rst",
+    )
 
-specific_dataset_files = [
-    i for i in dataset_files if i.stem.find("_dataset_map_provider") != -1
-]
-create_one_file(
-    "pytorch3d.implicitron.dataset specific datasets",
-    "specific datasets",
-    paths_to_modules(specific_dataset_files),
-    DEST_DIR / "datasets.rst",
-)
+    dataset_files = sorted(ROOT_DIR.glob("pytorch3d/implicitron/dataset/*.py"))
+    basic_dataset = [
+        "dataset_base",
+        "dataset_map_provider",
+        "data_loader_map_provider",
+        "data_source",
+        "scene_batch_sampler",
+    ]
+    basic_dataset_modules = [
+        f"pytorch3d.implicitron.dataset.{i}" for i in basic_dataset
+    ]
+    create_one_file(
+        "pytorch3d.implicitron.dataset in general",
+        "Basics of data for implicitron",
+        basic_dataset_modules,
+        DEST_DIR / "data_basics.rst",
+    )
 
-evaluation_files = sorted(ROOT_DIR.glob("pytorch3d/implicitron/evaluation/*.py"))
-create_one_file(
-    "pytorch3d.implicitron.evaluation",
-    "evaluation",
-    paths_to_modules(evaluation_files),
-    DEST_DIR / "evaluation.rst",
-)
+    specific_dataset_files = [
+        i for i in dataset_files if i.stem.find("_dataset_map_provider") != -1
+    ]
+    create_one_file(
+        "pytorch3d.implicitron.dataset specific datasets",
+        "specific datasets",
+        paths_to_modules(specific_dataset_files),
+        DEST_DIR / "datasets.rst",
+    )
 
-make_directory_index("pytorch3d.implicitron", DEST_DIR)
+    evaluation_files = sorted(ROOT_DIR.glob("pytorch3d/implicitron/evaluation/*.py"))
+    create_one_file(
+        "pytorch3d.implicitron.evaluation",
+        "evaluation",
+        paths_to_modules(evaluation_files),
+        DEST_DIR / "evaluation.rst",
+    )
+
+    make_directory_index("pytorch3d.implicitron", DEST_DIR)
+
+
+def iterate_toplevel_module(name: str) -> None:
+    dest_dir = Path(__file__).resolve().parent / "modules" / name
+    iterate_directory(ROOT_DIR / "pytorch3d" / name, dest_dir)
+
+
+do_implicitron()
+iterate_toplevel_module("renderer")
+iterate_toplevel_module("vis")

docs/modules/index.rst

@@ -12,5 +12,5 @@ API Documentation
     utils
     datasets
     common
-    vis
+    vis/index
     implicitron/index

docs/modules/renderer/blending.rst

@@ -1,7 +1,9 @@
-blending
+pytorch3d.renderer.blending
 ===========================
 
+blending
+
 .. automodule:: pytorch3d.renderer.blending
     :members:
     :undoc-members:
-    :show-inheritance:
+    :show-inheritance:

docs/modules/renderer/camera_conversions.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.camera_conversions
+=====================================
+
+camera_conversions
+
+.. automodule:: pytorch3d.renderer.camera_conversions
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/camera_utils.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.camera_utils
+===============================
+
+camera_utils
+
+.. automodule:: pytorch3d.renderer.camera_utils
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/cameras.rst

@@ -1,7 +1,9 @@
+pytorch3d.renderer.cameras
+==========================
+
 cameras
-===========================
 
 .. automodule:: pytorch3d.renderer.cameras
     :members:
     :undoc-members:
-    :show-inheritance:
+    :show-inheritance:

docs/modules/renderer/compositing.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.compositing
+==============================
+
+compositing
+
+.. automodule:: pytorch3d.renderer.compositing
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/fisheyecameras.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.fisheyecameras
+=================================
+
+fisheyecameras
+
+.. automodule:: pytorch3d.renderer.fisheyecameras
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/implicit/harmonic_embedding.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.implicit.harmonic_embedding
+==============================================
+
+harmonic_embedding
+
+.. automodule:: pytorch3d.renderer.implicit.harmonic_embedding
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/implicit/index.rst

@@ -0,0 +1,11 @@
+pytorch3d.renderer.implicit
+===========================
+
+.. toctree::
+
+    harmonic_embedding
+    raymarching
+    raysampling
+    renderer
+    sample_pdf
+    utils

docs/modules/renderer/implicit/raymarching.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.implicit.raymarching
+=======================================
+
+raymarching
+
+.. automodule:: pytorch3d.renderer.implicit.raymarching
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/implicit/raysampling.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.implicit.raysampling
+=======================================
+
+raysampling
+
+.. automodule:: pytorch3d.renderer.implicit.raysampling
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/implicit/renderer.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.implicit.renderer
+====================================
+
+renderer
+
+.. automodule:: pytorch3d.renderer.implicit.renderer
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/implicit/sample_pdf.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.implicit.sample_pdf
+======================================
+
+sample_pdf
+
+.. automodule:: pytorch3d.renderer.implicit.sample_pdf
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/implicit/utils.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.implicit.utils
+=================================
+
+utils
+
+.. automodule:: pytorch3d.renderer.implicit.utils
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/index.rst

@@ -1,15 +1,19 @@
-pytorch3d.renderer 
-===========================
+pytorch3d.renderer
+==================
 
 .. toctree::
 
-    rasterizer
+    blending
+    camera_conversions
+    camera_utils
     cameras
+    compositing
+    fisheyecameras
     lighting
     materials
-    texturing
-    blending
-    shading
-    shader
-    renderer
-    utils
+    splatter_blend
+    utils
+    implicit/index
+    mesh/index
+    opengl/index
+    points/index

docs/modules/renderer/lighting.rst

@@ -1,6 +1,9 @@
-lighting
+pytorch3d.renderer.lighting
 ===========================
 
+lighting
+
 .. automodule:: pytorch3d.renderer.lighting
     :members:
-    :undoc-members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/materials.rst

@@ -1,7 +1,9 @@
+pytorch3d.renderer.materials
+============================
+
 materials
-===========================
 
 .. automodule:: pytorch3d.renderer.materials
     :members:
     :undoc-members:
-    :show-inheritance:
+    :show-inheritance:

docs/modules/renderer/mesh/clip.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.mesh.clip
+============================
+
+clip
+
+.. automodule:: pytorch3d.renderer.mesh.clip
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/mesh/index.rst

@@ -0,0 +1,13 @@
+pytorch3d.renderer.mesh
+=======================
+
+.. toctree::
+
+    clip
+    rasterize_meshes
+    rasterizer
+    renderer
+    shader
+    shading
+    textures
+    utils

docs/modules/renderer/mesh/rasterize_meshes.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.mesh.rasterize_meshes
+========================================
+
+rasterize_meshes
+
+.. automodule:: pytorch3d.renderer.mesh.rasterize_meshes
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/mesh/rasterizer.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.mesh.rasterizer
+==================================
+
+rasterizer
+
+.. automodule:: pytorch3d.renderer.mesh.rasterizer
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/mesh/renderer.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.mesh.renderer
+================================
+
+renderer
+
+.. automodule:: pytorch3d.renderer.mesh.renderer
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/mesh/shader.rst

@@ -1,7 +1,9 @@
+pytorch3d.renderer.mesh.shader
+==============================
+
 shader
-===========================
 
 .. automodule:: pytorch3d.renderer.mesh.shader
     :members:
     :undoc-members:
-    
+    :show-inheritance:

docs/modules/renderer/mesh/shading.rst

@@ -1,7 +1,9 @@
+pytorch3d.renderer.mesh.shading
+===============================
+
 shading
-===========================
 
 .. automodule:: pytorch3d.renderer.mesh.shading
     :members:
     :undoc-members:
-    
+    :show-inheritance:

docs/modules/renderer/mesh/textures.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.mesh.textures
+================================
+
+textures
+
+.. automodule:: pytorch3d.renderer.mesh.textures
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/mesh/utils.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.mesh.utils
+=============================
+
+utils
+
+.. automodule:: pytorch3d.renderer.mesh.utils
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/opengl/index.rst

@@ -0,0 +1,7 @@
+pytorch3d.renderer.opengl
+=========================
+
+.. toctree::
+
+    opengl_utils
+    rasterizer_opengl

docs/modules/renderer/opengl/opengl_utils.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.opengl.opengl_utils
+======================================
+
+opengl_utils
+
+.. automodule:: pytorch3d.renderer.opengl.opengl_utils
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/opengl/rasterizer_opengl.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.opengl.rasterizer_opengl
+===========================================
+
+rasterizer_opengl
+
+.. automodule:: pytorch3d.renderer.opengl.rasterizer_opengl
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/points/compositor.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.points.compositor
+====================================
+
+compositor
+
+.. automodule:: pytorch3d.renderer.points.compositor
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/points/index.rst

@@ -0,0 +1,10 @@
+pytorch3d.renderer.points
+=========================
+
+.. toctree::
+
+    compositor
+    rasterize_points
+    rasterizer
+    renderer
+    pulsar/index

docs/modules/renderer/points/pulsar/index.rst

@@ -0,0 +1,7 @@
+pytorch3d.renderer.points.pulsar
+================================
+
+.. toctree::
+
+    renderer
+    unified

docs/modules/renderer/points/pulsar/renderer.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.points.pulsar.renderer
+=========================================
+
+renderer
+
+.. automodule:: pytorch3d.renderer.points.pulsar.renderer
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/points/pulsar/unified.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.points.pulsar.unified
+========================================
+
+unified
+
+.. automodule:: pytorch3d.renderer.points.pulsar.unified
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/points/rasterize_points.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.points.rasterize_points
+==========================================
+
+rasterize_points
+
+.. automodule:: pytorch3d.renderer.points.rasterize_points
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/points/rasterizer.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.points.rasterizer
+====================================
+
+rasterizer
+
+.. automodule:: pytorch3d.renderer.points.rasterizer
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/points/renderer.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.points.renderer
+==================================
+
+renderer
+
+.. automodule:: pytorch3d.renderer.points.renderer
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/rasterizer.rst

@@ -1,10 +0,0 @@
-rasterizer
-===========================
-
-.. automodule:: pytorch3d.renderer.mesh.rasterize_meshes
-    :members:
-    :undoc-members:
-    
-.. automodule:: pytorch3d.renderer.mesh.rasterizer
-    :members:
-    :undoc-members:

docs/modules/renderer/splatter_blend.rst

@@ -0,0 +1,9 @@
+pytorch3d.renderer.splatter_blend
+=================================
+
+splatter_blend
+
+.. automodule:: pytorch3d.renderer.splatter_blend
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/renderer/texturing.rst

@@ -1,7 +0,0 @@
-texturing
-===========================
-
-.. automodule:: pytorch3d.renderer.mesh.texturing
-    :members:
-    :undoc-members:
-    

docs/modules/renderer/utils.rst

@@ -1,6 +1,9 @@
+pytorch3d.renderer.utils
+========================
+
 utils
-===========================
 
 .. automodule:: pytorch3d.renderer.utils
     :members:
-    :undoc-members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/vis.rst

@@ -1,6 +0,0 @@
-pytorch3d.vis
-===========================
-
-.. automodule:: pytorch3d.vis
-    :members:
-    :undoc-members:

docs/modules/vis/index.rst

@@ -0,0 +1,7 @@
+pytorch3d.vis
+=============
+
+.. toctree::
+
+    plotly_vis
+    texture_vis

docs/modules/vis/plotly_vis.rst

@@ -0,0 +1,9 @@
+pytorch3d.vis.plotly_vis
+========================
+
+plotly_vis
+
+.. automodule:: pytorch3d.vis.plotly_vis
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/modules/vis/texture_vis.rst

@@ -0,0 +1,9 @@
+pytorch3d.vis.texture_vis
+=========================
+
+texture_vis
+
+.. automodule:: pytorch3d.vis.texture_vis
+    :members:
+    :undoc-members:
+    :show-inheritance: