public override Object GetAspect(Object rowObject)
{
LayerTreeNode layerTreeNode = rowObject as LayerTreeNode;
if (layerTreeNode != null)
{
IILayer layer = MaxAnimatable.GetAnimByHandle(layerTreeNode.Handle) as IILayer;
if (layer != null)
{
IILayerProperties layerProps = MaxLayers.GetLayerProperties(layer);
return(layerProps.Current);
}
}
FolderTreeNode folderTreeNode = rowObject as FolderTreeNode;
if (folderTreeNode != null)
{
// Default behaviour of getFolderCheckState returns true.
// Only get folder check state if necessary as it is
// more expensive than returning false.
if (folderTreeNode.Children.Count > 0)
{
Boolean?checkState = getFolderCheckState(folderTreeNode);
if (checkState == true || checkState == null)
{
return(null);
}
}
}
return(false);
}
private static void onCreateLayer(Object sender, ClickEventArgs e, Boolean addSelection)
{
NlmTreeListView listView = e.ListView;
listView.NodeControl.MaxEvents.LayerEvents.LayerCreated.UnregisterNotification();
IILayer layer = MaxLayers.CreateLayer(true, addSelection);
UIntPtr handle = MaxAnimatable.GetHandleByAnim(layer);
LayerTreeNode layerTreeNode = new LayerTreeNode(handle, listView.NodeControl.HandleMap);
// Get parent node.
BaseTreeNode parentTreeNode = null;
if (listView.SelectedObjects.Count > 0)
{
BaseTreeNode treeNode = listView.SelectedObjects[0] as BaseTreeNode;
if (treeNode is FolderTreeNode)
{
parentTreeNode = treeNode;
}
if (treeNode is LayerTreeNode)
{
parentTreeNode = treeNode.Parent;
}
}
// Add folder to listview, ensure is visible by expanding parents.
listView.AddObject(layerTreeNode, parentTreeNode);
if (parentTreeNode != null)
{
listView.Expand(parentTreeNode);
}
// Scroll to new item.
Int32 parentIndex = listView.IndexOf(layerTreeNode);
if (parentIndex != -1)
{
listView.EnsureVisible(parentIndex);
}
// Sort, select
listView.Sort(listView.NlmColumns.NameColumn, SortOrder.Ascending);
listView.SelectedObjects = new List <Object> {
layerTreeNode
};
// Focus on the listview to ensure text entry goes to the edit box, and begin edit.
listView.Focus();
listView.EditModel(layerTreeNode);
// Register notification.
listView.NodeControl.MaxEvents.LayerEvents.LayerCreated.RegisterNotification();
}
private void BuildChildObjects(LayerTreeNode layerTreeNode, IILayer maxLayer)
{
IEnumerable <IINode> layerNodes = MaxLayers.GetChildNodes(maxLayer);
foreach (IINode maxNode in layerNodes)
{
UIntPtr iNodeHandle = MaxAnimatable.GetHandleByAnim(maxNode);
ObjectTreeNode objectTreeNode = new ObjectTreeNode(MaxNodes.GetObjectClass(maxNode), iNodeHandle, HandleMap);
ListView.NodeControl.Parent.AddChild(objectTreeNode, layerTreeNode, false);
}
}
// If the cell being edited is layer 0, cancel the cell edit.
private void onBeforeLabelEdit(Object sender, CellEditEventArgs e)
{
if (e.RowObject is LayerTreeNode)
{
IILayer layer = MaxLayers.GetLayer(0);
UIntPtr handle = MaxAnimatable.GetHandleByAnim(layer);
if ((e.RowObject as LayerTreeNode).Handle == handle)
{
e.Cancel = true;
}
}
}
public override void PutAspect(Object rowObject, Object newValue)
{
LayerTreeNode layerTreeNode = rowObject as LayerTreeNode;
if (layerTreeNode != null)
{
IILayer layer = MaxAnimatable.GetAnimByHandle(layerTreeNode.Handle) as IILayer;
if (layer != null)
{
IILayerProperties layerProps = MaxLayers.GetLayerProperties(layer);
layerProps.Current = (Boolean)newValue;
}
}
}
private void onNodeCreated(object sender, SystemNotificationAnimEventArgs e)
{
#if DEBUG
MaxListener.PrintToListener("onNodeCreated");
#endif
try
{
ListView.BeginUpdate();
List <BaseTreeNode> refreshNodes = new List <BaseTreeNode>();
foreach (UIntPtr handle in e.Handles)
{
IAnimatable anim = MaxAnimatable.GetAnimByHandle(handle);
if (anim == null)
{
return;
}
IINode node = anim as IINode;
IILayer layer = MaxLayers.GetLayer(node);
UIntPtr layerHandle = MaxAnimatable.GetHandleByAnim(layer);
ObjectTreeNode.ObjectClass objectClass = MaxNodes.GetObjectClass(node);
List <BaseTreeNode> layerTreeNodes = NodeControl.HandleMap.GetTreeNodesByHandle(layerHandle);
foreach (BaseTreeNode layerTreeNode in layerTreeNodes)
{
ObjectTreeNode objectTreeNode = new ObjectTreeNode(objectClass, handle, NodeControl.HandleMap);
NodeControl.Parent.AddChild(objectTreeNode, layerTreeNode, false);
if (!refreshNodes.Contains(layerTreeNode))
{
refreshNodes.Add(layerTreeNode);
}
}
}
MaxListener.PrintToListener(refreshNodes.Count.ToString());
ListView.RefreshObjects(refreshNodes);
ListView.Sort(ListView.NlmColumns.NameColumn, SortOrder.Ascending);
}
catch
{
throw new Exception();
}
finally
{
ListView.EndUpdate();
}
}
public static void onFreezeUnfreezeAll(Object sender, ClickEventArgs e)
{
try
{
e.ListView.BeginUpdate();
MaxLayers.FreezeUnfreezeAll();
}
catch
{
throw new Exception();
}
finally
{
e.ListView.EndUpdate();
}
}
private void ValidateCellEdit(Object o, CellEditEventArgs e)
{
String oldValue = e.Value as String;
String newValue = e.NewValue as String;
if (oldValue != newValue)
{
IILayer layer = MaxLayers.GetLayer(newValue);
if (layer != null)
{
e.Cancel = true;
MessageBox.Show("A Layer with that name already Exists.");
}
}
}
public override void LayerChanged(ITab <UIntPtr> nodes)
{
#if DEBUG
MaxListener.PrintToListener("NodeEventCallback > LayerChanged");
#endif
List <BaseTreeNode> deleteNodes = new List <BaseTreeNode>();
List <Tuple <BaseTreeNode, BaseTreeNode> > addNodes = new List <Tuple <BaseTreeNode, BaseTreeNode> >();
for (int i = 0; i < nodes.Count; i++)
{
UIntPtr maxNodeHandle = nodes[(IntPtr)i];
IINode maxNode = MaxAnimatable.GetAnimByHandle(maxNodeHandle) as IINode;
IILayer maxLayer = MaxLayers.GetLayer(maxNode);
UIntPtr layerHandle = MaxAnimatable.GetHandleByAnim(maxLayer);
// We need to handle the following scenarios:
// An object being moved to another layer.
// Objects on instances layers moving to uninstanced layers.
// Objects on uninstanced layers moving to instanced layers.
// The easiest way to do this is to remove old object nodes and create new ones.
// This should be pretty fast, and this event should fire relatively rarely,
// but it may have to be rethought if it's too slow.
// First we remove the old nodes.
List <BaseTreeNode> objectTreeNodes = NodeControl.HandleMap.GetTreeNodesByHandle(maxNodeHandle);
deleteNodes.AddRange(objectTreeNodes);
// Then we add the object node to the new layer.
List <BaseTreeNode> layerTreeNodes = NodeControl.HandleMap.GetTreeNodesByHandle(layerHandle);
foreach (BaseTreeNode layerTreeNode in layerTreeNodes)
{
ObjectTreeNode newObjectTreeNode = new ObjectTreeNode(MaxNodes.GetObjectClass(maxNode), maxNodeHandle, NodeControl.HandleMap);
addNodes.Add(new Tuple <BaseTreeNode, BaseTreeNode> (newObjectTreeNode, layerTreeNode));
}
}
// And finally we actually do the update all at once.
NodeControl.Destroy.DeleteTreeNodes(deleteNodes);
foreach (Tuple <BaseTreeNode, BaseTreeNode> tuple in addNodes)
{
NodeControl.Parent.AddChild(tuple.Item1, tuple.Item2, false);
}
// And sort :)
ListView.Sort(ListView.NlmColumns.NameColumn, SortOrder.Ascending);
}
public void AddMissingChildObjects(IEnumerable <LayerTreeNode> layerNodes)
{
foreach (LayerTreeNode layerNode in layerNodes)
{
IILayer layer = MaxAnimatable.GetAnimByHandle(layerNode.Handle) as IILayer;
IEnumerable <IINode> maxNodes = MaxLayers.GetChildNodes(layer);
foreach (IINode maxNode in maxNodes)
{
UIntPtr iNodeHandle = MaxAnimatable.GetHandleByAnim(maxNode);
// TODO: This needs to work with instanced layers.
if (!HandleMap.ContainsHandle(iNodeHandle))
{
ObjectTreeNode objectTreeNode = new ObjectTreeNode(MaxNodes.GetObjectClass(maxNode), iNodeHandle, HandleMap);
objectTreeNode.Parent = layerNode;
layerNode.Children.Add(objectTreeNode);
}
}
}
}
//TODO:
// Perhaps move this to a new class / node control. It's pretty huge.
// Use MaxLayers static class to manipulate max objects instead of in code here.
/// <summary>
/// Isolate selected TreeNodes in Nested Layer Manager.
/// Any TreeNode that is not isolated is disabled, and the respective node hidden in Max.
/// </summary>
/// <param name="sender">An NLMCheckButton that contains a checked property.</param>
/// <param name="e">NLM Button Event Args, that contain a pointer to the listview.</param>
public static void onIsolateSelection(Object sender, ClickEventArgs e)
{
// If the sender is not an NLMCheckButton, the request has come from somewhere else (right click or shortcut).
// For this reason, change the check state of the button UI which will raise this event again.
if (!(sender is NlmCheckButton))
{
e.ListView.ButtonPanelLeft.IsolateSelectedButton.Checked = !e.ListView.ButtonPanelLeft.IsolateSelectedButton.Checked;
return;
}
NlmTreeListView listView = e.ListView;
NlmCheckButton button = sender as NlmCheckButton;
// Pause UI from refreshing.
try
{
listView.BeginUpdate();
// Isolate selected if button checked.
if (button.Checked)
{
// This method has some serious processing going on, so hashsets are used to maximise performance.
// Calculate all enabled nodes and disabled nodes.
HashSet <BaseTreeNode> enabledNodes = new HashSet <BaseTreeNode>(listView.NodeControl.Query.SelectionAndAllChildNodes);
HashSet <BaseTreeNode> disabledNodes = new HashSet <BaseTreeNode>(listView.NodeControl.Query.AllNodes.Where(x => !enabledNodes.Contains(x)));
// Disable disabled nodes.
listView.DisableObjects(disabledNodes);
// Now we need to work out what maxNodes and layers to temporarily hide.
// Firstly, get the highest selected objects.
IEnumerable <ObjectTreeNode> selectedNodes = listView.NodeControl.Query.SelectedAncestors
.Where(x => (x is ObjectTreeNode)).Cast <ObjectTreeNode>();
HashSet <IILayer> onLayers = new HashSet <IILayer>();
// If objects are selected, we need to ensure that the layer is not hidden.
// If any objects are selected, append the layer to the onLayers hashset.
foreach (ObjectTreeNode objectTreeNode in selectedNodes)
{
IINode maxNode = MaxAnimatable.GetAnimByHandle(objectTreeNode.Handle) as IINode;
if (maxNode == null)
{
continue;
}
IILayer layer = MaxLayers.GetLayer(maxNode);
if (!onLayers.Contains(layer))
{
onLayers.Add(layer);
}
}
// Ensure that the layer is on so that objects are visible.
// The layer is only turned on if an object is selected, and no ancestor is selected.
// For any object that is not selected, the object needs to be hidden.
foreach (IILayer layer in onLayers)
{
if (layer.IsHidden)
{
#if Max2013 || Max2014
layer.IsHidden = false;
#endif
#if Max2015
layer.Hide(false, false);
#endif
}
foreach (IINode maxNode in MaxLayers.GetChildNodes(layer))
{
UIntPtr maxNodeHandle = MaxAnimatable.GetHandleByAnim(maxNode);
if (!selectedNodes.Any(x => (x).Handle == maxNodeHandle) && !maxNode.IsObjectHidden)
{
listView.DisabledHandles.Add(maxNodeHandle);
maxNode.Hide(true);
}
}
}
// Loop through all layers that should be turned off.
// If the layer should be hidden (because it is not in onLayers or in enabledLayers), hide it.
IEnumerable <LayerTreeNode> enabledLayers = enabledNodes.Where(x => (x is LayerTreeNode)).Cast <LayerTreeNode>();
IEnumerable <IILayer> offLayers = MaxLayers.Layers.Where(x => !onLayers.Contains(x));
foreach (IILayer layer in offLayers)
{
UIntPtr layerHandle = MaxAnimatable.GetHandleByAnim(layer);
if (!layer.IsHidden && !enabledLayers.Any(x => (x).Handle == layerHandle))
{
#if Max2013 || Max2014
layer.IsHidden = true;
#endif
#if Max2015
layer.Hide(true, false);
#endif
listView.DisabledHandles.Add(layerHandle);
}
}
}
// Remove isolation and restore state if unchecked.
else
{
// Enable all objects.
listView.EnableObjects(listView.DisabledObjects);
// Turn on all handles that were turned off.
foreach (UIntPtr handle in listView.DisabledHandles)
{
IAnimatable maxAnim = MaxAnimatable.GetAnimByHandle(handle);
IILayer maxLayer = maxAnim as IILayer;
if (maxLayer != null)
{
#if Max2013 || Max2014
maxLayer.IsHidden = false;
#endif
#if Max2015
maxLayer.Hide(false, false);
#endif
continue;
}
IINode maxNode = maxAnim as IINode;
if (maxNode != null)
{
maxNode.Hide(false);
continue;
}
}
// Clear list of handles.
listView.DisabledHandles.Clear();
}
}
catch
{
throw new Exception();
}
finally
{
listView.EndUpdate();
}
MaxUI.RedrawViewportsNow();
}
/// <summary>
/// Deletes the selected treenodes, including associated max nodes.
/// All children of selection are also deleted.
/// Max nodes will not be deleted if not all instanced nodes are selected.
/// If every layer 0 instance is selected, nothing will happen.
/// </summary>
public void DeleteSelection()
{
#if DEBUG
System.Diagnostics.Stopwatch stopwatch = new System.Diagnostics.Stopwatch();
stopwatch.Start();
#endif
ListView.NodeControl.MaxEvents.NodeEvents.Unregister();
ListView.NodeControl.MaxEvents.LayerEvents.LayerDeleted.UnregisterNotification();
// Don't do anything if every single layer 0 is selected.
IILayer layer0 = MaxLayers.GetLayer(0);
UIntPtr layer0handle = MaxAnimatable.GetHandleByAnim(layer0);
List <BaseTreeNode> layerTreeNodes = HandleMap.GetTreeNodesByHandle(layer0handle);
if (layerTreeNodes.All(x => NodeQuery.SelectionAndAllChildNodes.Contains(x)))
{
return;
}
// Collect the selection IEnumerables to use.
HashSet <BaseTreeNode> selectionAndAllChildren = new HashSet <BaseTreeNode>
(NodeQuery.SelectionAndAllChildNodes);
// Calculate layer and object handles.
IEnumerable <UIntPtr> objectHandles = selectionAndAllChildren.Where
(x => x is ObjectTreeNode).Cast <ObjectTreeNode>().Select(x => x.Handle);
IEnumerable <UIntPtr> layerHandles = selectionAndAllChildren.Where
(x => x is LayerTreeNode).Cast <LayerTreeNode>().Select(x => x.Handle);
// Delete handles from max.
foreach (UIntPtr handle in objectHandles)
{
List <BaseTreeNode> instances = HandleMap.GetTreeNodesByHandle(handle);
if (instances.Count() > 1)
{
if (!instances.All(x => selectionAndAllChildren.Contains(x)))
{
continue;
}
}
MaxNodes.DeleteNode(handle);
}
foreach (UIntPtr handle in layerHandles)
{
List <BaseTreeNode> instances = HandleMap.GetTreeNodesByHandle(handle);
if (instances.Count() > 1)
{
if (!instances.All(x => selectionAndAllChildren.Contains(x)))
{
continue;
}
}
MaxLayers.DeleteLayer(handle);
}
// And now to delete the tree nodes now there are no max nodes.
DeleteTreeNodes(selectionAndAllChildren);
// The default behaviour of the listview is to maintain selection based on item index.
// This is not very desirable, as the selection should be nothing.
ListView.SelectedObjects = new Object[] {};
ListView.NodeControl.MaxEvents.LayerEvents.LayerDeleted.RegisterNotification();
ListView.NodeControl.MaxEvents.NodeEvents.Register();
#if DEBUG
stopwatch.Stop();
MaxListener.PrintToListener("DeleteSelection completed in: " + stopwatch.ElapsedMilliseconds);
#endif
}
