Updated

Source/FLESHEditor/Public/FLESHViewportClient.h

@@ -28,6 +28,18 @@ public:
     /** Handle mouse clicks - new API */
     virtual bool InputKey(const FInputKeyEventArgs& EventArgs) override;
     
+    /** Override mouse movement handling to provide camera controls similar to asset editor */
+    virtual bool InputAxis(FViewport* InViewport, int32 ControllerId, FKey Key, float Delta, float DeltaTime, int32 NumSamples = 1, bool bGamepad = false) override;
+    
+    /** Load and display objects from NodeTree */
+    void LoadNodesFromNodeTree();
+    
+    /** Update objects displayed in the viewport */
+    void UpdateVisibleNodes();
+    
+    /** Focus on the selected object */
+    void FocusOnSelectedNode();
+    
     /** Reset camera */
     void ResetCamera();
     
@@ -52,4 +64,7 @@ private:
     
     /** Preview scene for the viewport */
     TSharedPtr<FPreviewScene> PreviewScene;
+    
+    /** Recursively load node and its children */
+    void LoadNodeRecursive(TSharedPtr<FVisceraNodeItem> Node, USceneComponent* ParentComponent);
 };