Skip to content

Commit

Permalink
move manager_live_visualizer to ui and rename ui_visualizer_mixin to …
Browse files Browse the repository at this point in the history
…base
  • Loading branch information
jtigue-bdai committed Oct 8, 2024
1 parent 3987e4b commit 8d8fdfb
Show file tree
Hide file tree
Showing 7 changed files with 11 additions and 10 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,11 @@
import carb
import omni.isaac.core.utils.torch as torch_utils

from omni.isaac.lab.managers import ActionManager, EventManager, ManagerLiveVisualizer, ObservationManager
from omni.isaac.lab.managers import ActionManager, EventManager, ObservationManager
from omni.isaac.lab.scene import InteractiveScene
from omni.isaac.lab.sim import SimulationContext
from omni.isaac.lab.utils.timer import Timer
from omni.isaac.lab.ui.widgets import ManagerLiveVisualizer

from .common import VecEnvObs
from .manager_based_env_cfg import ManagerBasedEnvCfg
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,11 +18,10 @@
from omni.isaac.lab.managers import (
CommandManager,
CurriculumManager,
ManagerLiveVisualizer,
RewardManager,
TerminationManager,
)

from omni.isaac.lab.ui.widgets import ManagerLiveVisualizer
from .common import VecEnvStepReturn
from .manager_based_env import ManagerBasedEnv
from .manager_based_rl_env_cfg import ManagerBasedRLEnvCfg
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
import omni.usd
from pxr import PhysxSchema, Sdf, Usd, UsdGeom, UsdPhysics

from omni.isaac.lab.managers import ManagerLiveVisualizer
from omni.isaac.lab.ui.widgets import ManagerLiveVisualizer

if TYPE_CHECKING:
import omni.ui
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@
from .curriculum_manager import CurriculumManager
from .event_manager import EventManager
from .manager_base import ManagerBase, ManagerTermBase
from .manager_live_visualizer import ManagerLiveVisualizer
from .manager_term_cfg import (
ActionTermCfg,
CommandTermCfg,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,4 +5,5 @@

from .image_plot import ImagePlot
from .line_plot import LiveLinePlot
from .ui_visualizer_mixin import UiVisualizerMixin
from .ui_visualizer_base import UiVisualizerBase
from .manager_live_visualizer import ManagerLiveVisualizer
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from omni.ui import CollapsableFrame, Frame, VStack, Window

from omni.isaac.lab.managers import ManagerBase
from omni.isaac.lab.ui.widgets import ImagePlot, LiveLinePlot, UiVisualizerMixin
from omni.isaac.lab.ui.widgets import ImagePlot, LiveLinePlot, UiVisualizerBase
from omni.isaac.lab.utils import configclass


Expand All @@ -35,7 +35,7 @@ class ManagerLiveVisualizerCfg:
"""


class ManagerLiveVisualizer(UiVisualizerMixin):
class ManagerLiveVisualizer(UiVisualizerBase):
"""A interface object used to transfer data from a manager to a UI widget. This class handles the creation of UI
Widgets for selected terms given a ManagerLiveVisualizerCfg.
"""
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,9 @@
from omni.ui import Frame, Window


class UiVisualizerMixin:
"""Mixin Class for components that support debug visualizations that requires access to some UI elements.
class UiVisualizerBase:
"""Base Class for components that support debug visualizations that requires access to some UI elements.
This class provides a set of functions that can be used to assign ui interfaces.
The following functions are provided:
* :func:`set_debug_vis`: Assigns a debug visualization interface. This function is called by the main UI
Expand Down

0 comments on commit 8d8fdfb

Please sign in to comment.