Render Layers

1 files changed, 5 insertions, 0 deletions

diff --git a/ui/disassemblyview.h b/ui/disassemblyview.h
index c7595f96..7c97576d 100644
--- a/ui/disassemblyview.h
+++ b/ui/disassemblyview.h
@@ -81,6 +81,9 @@ class BINARYNINJAUIAPI DisassemblyView : public FlowGraphWidget
 	void setDisplayedFileName();
 	void setAddressBaseOffset(bool toHere);
 
+	void toggleRenderLayer(const std::string& layer);
+	FlowGraphRef applyRenderLayers(FlowGraphRef graph);
+
 	virtual DisassemblySettingsRef getDisassemblySettings() override;
 	virtual void setDisassemblySettings(DisassemblySettingsRef settings) override;
 
@@ -89,6 +92,7 @@ class BINARYNINJAUIAPI DisassemblyView : public FlowGraphWidget
 	virtual void onHighlightChanged(const HighlightTokenState& highlight) override;
 
 	static void registerActions();
+	virtual void bindDynamicActions() override;
 
   private:
 	class DisassemblyViewOptionsWidget : public MenuHelper
@@ -138,6 +142,7 @@ class BINARYNINJAUIAPI DisassemblyView : public FlowGraphWidget
 	BNDisassemblyCallParameterHints m_callParamHints;
 	DisassemblyContainer* m_container;
 	SettingsRef m_settings;
+	std::set<std::string> m_layers;
 
   private Q_SLOTS:
 	void viewInHexEditor();