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,6 +94,9 @@ def make_directory_index(title: str, directory_path: Path):
             print(f"    {rst.stem}", file=f)
 
 
+def do_implicitron():
+    DEST_DIR = Path(__file__).resolve().parent / "modules/implicitron"
+
     iterate_directory(ROOT_DIR / "pytorch3d/implicitron/models", DEST_DIR / "models")
 
     unwanted_tools = ["configurable", "depth_cleanup", "utils"]
@@ -119,7 +121,9 @@ basic_dataset = [
         "data_source",
         "scene_batch_sampler",
     ]
-basic_dataset_modules = [f"pytorch3d.implicitron.dataset.{i}" for i in basic_dataset]
+    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",
@@ -146,3 +150,13 @@ create_one_file(
     )
 
     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,6 +1,8 @@
-blending
+pytorch3d.renderer.blending
 ===========================
 
+blending
+
 .. automodule:: pytorch3d.renderer.blending
     :members:
     :undoc-members:

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,5 +1,7 @@
+pytorch3d.renderer.cameras
+==========================
+
 cameras
-===========================
 
 .. automodule:: pytorch3d.renderer.cameras
     :members:

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
-===========================
+==================
 
 .. toctree::
 
-    rasterizer
+    blending
+    camera_conversions
+    camera_utils
     cameras
+    compositing
+    fisheyecameras
     lighting
     materials
-    texturing
+    splatter_blend
-    blending
-    shading
-    shader
-    renderer
     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:
+    :show-inheritance:

docs/modules/renderer/materials.rst

@@ -1,5 +1,7 @@
+pytorch3d.renderer.materials
+============================
+
 materials
-===========================
 
 .. automodule:: pytorch3d.renderer.materials
     :members:

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:
+    :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: