NX二次开发——树列表使用

//基本概念：一个Node节点代表一行
//***部分命令请注意顺序关系***
 
//插入列，注意插入列仅initialize_cb()/dialogShown_cb()设置一次，update()时会报错
for (int i = 0; i < columnCount; ++i)
{
    tree->InsertColumn(i, "", columnWidth);
    tree->SetColumnResizePolicy(i, BlockStyler::Tree::ColumnResizePolicyConstantWidth);
    tree->SetColumnSortable(i, false);
}
 
//创建    
Node *node = tree->CreateNode("label");
tree->InsertNode(node, NULL, NULL, Tree::NodeInsertOptionLast);
tree->InsertNode(node, parentNode, NULL, Tree::NodeInsertOptionLast);//父节点
 
//遍历子节点
void GetAllNode(Node* inputNode, vector<Node*> &nodes)
{
    if (inputNode == NULL)
    {
        return;
    }
 
    Node* node = inputNode->FirstChildNode();
    while (node != NULL)
    {
        nodes.push_back(node);
        NXCommon::GetAllNode(node, nodes);
        node = node->NextSiblingNode();
    }
}
 
//得到树控件所有节点
vector<Node*> GetAllNodes(Tree* tree)
{
    vector<Node*> res;
    Node *nd = tree->RootNode();
    while (nd != NULL)
    {
        res.push_back(nd);
        nd = nd->NextSiblingNode();
    }
    return res;
}
 
/得到树控件所有节点 含子节点
std::vector<Node*> GetAllNodesContainSon(Tree* tree)
{
    Node *nd = tree->RootNode();
 
    vector<Node*> res;
    while (nd != NULL)
    {
        res.push_back(nd);
        GetAllNode(nd, res);
        nd = nd->NextSiblingNode();
    }
 
    return res;
}
 
//清空树节点
void ClearTree(Tree* tree)
{
    vector<BlockStyler::Node*> nodes = GetAllNodes(tree);
    for (int i = 0; i < nodes.size(); i++)
    {
        tree->DeleteNode(nodes[i]);
    }
}
 
//节点选择状态
tree->SelectNode(node, false, false); //不选中
node->IsSelected();  //节点是否选中
 
//显示文本
node->SetColumnDisplayText(columnID, "Text");
 
//节点添加属性
node->GetNodeData()->AddInteger("XX",int);
node->GetNodeData()->SetInteger("XX",int); //先调用AddInteger才有效，否则报错
 
//展开节点
node->Expand(NXOpen::BlockStyler::Node::ExpandOption::ExpandOptionExpand);
 
//注意：OnEditOptionSelectedCallback()不支持节点增删改
NXOpen::BlockStyler::Tree::EditControlOption OnEditOptionSelectedCallback(NXOpen::BlockStyler::Tree *tree, NXOpen::BlockStyler::Node *node, int columnID, int selectedOptionID, NXString selectedOptionText, NXOpen::BlockStyler::Tree::ControlType type)
{
    if (tree == NULL || node == NULL)
        return Tree::EditControlOptionReject;
 
    //如果要检查输入值是否为double，必须强制检查
    std::string originText = selectedOptionText.GetLocaleText(); //重要 得到输入的值
    std::string text = NXCommon::DoubleToStr(NXCommon::StrToDouble(originText));
    if (originText != text)
    {
        return Tree::EditControlOptionReject;
    }
 
    return Tree::EditControlOptionAccept;
}
 
//编辑Node某列
NXOpen::BlockStyler::Tree::ControlType AskEditControlCallback(NXOpen::BlockStyler::Tree *tree, NXOpen::BlockStyler::Node *node, int columnID)
{
    if (tree == NULL || node == NULL)
    {
        return Tree::ControlTypeNone;
    }
    
    //只有枚举样式，设置一个值时，效果等同字符串控件    
    std::vector<NXString> tmpContents;
    tmpContents.push_back("A");
    tmpContents.push_back("B");
    tree->SetEditOptions(tmpContents, 0);
    return Tree::ControlTypeComboBox;
}
 
//Node 右键菜单
void OnMenuCallback(NXOpen::BlockStyler::Tree *tree, NXOpen::BlockStyler::Node *node, int columnID )
{
    TreeListMenu *menu = tree->CreateMenu();
    menu->AddMenuItem(0, "创建");
    menu->AddMenuItem(1, "删除");
    tree->SetMenu(menu);
    delete menu;
    menu = NULL;
}
 
//Node 拖动
//更新树控件时，Tree最好只更新一次，
//否则代码中未执行完的代码可能报错，如：node[i]->GetColumnDisplayText(0).GetLocaleText();
目前NX树控件不支持直接拖动，可以通过
(1) 重绘树控件
(2) 将拖动节点备份，重新插入目标节点（个人觉得比较麻烦）
(3) 其他....
 
NXOpen::BlockStyler::Node::DragType IsDragAllowedCallback(NXOpen::BlockStyler::Tree *tree, NXOpen::BlockStyler::Node *node, int columnID)
{
    if (tree == NULL || node == NULL)
        return BlockStyler::Node::DragTypeNone;
 
    return BlockStyler::Node::DragTypeAll;//允许节点拖动，但没有拖动效果
}
 
//Node 某列排序  注意:代码放置在数据填充之后生效
tree->SetSortRootNodes(true);
tree->SetColumnSortable(columnID, true);
tree->SetColumnSortOption(columnID, Tree::ColumnSortOption::ColumnSortOptionDescending);
//排序回调函数处理 
int ColumnSortCallback(NXOpen::BlockStyler::Tree *tree, int columnID, NXOpen::BlockStyler::Node *node1, NXOpen::BlockStyler::Node *node2)
{
    //返回值为0、正负值；分别表示两个节点相同，第一个节点大于第二个，第一个节点小于第二个
    int res = 0;
 
    
    //注意：node1是树控件上后一个节点，node2是前一个节点 !!!
    std::string content1 = node1->GetColumnDisplayText(columnID).GetLocaleText();
    std::string content2 = node2->GetColumnDisplayText(columnID).GetLocaleText();
    if (IsDoubleString(content1))
    {
        if (IsGT(StrToDouble(content1), StrToDouble(content2)))
        {
            res = 1;
        }
        else
        {
            res = -1;
        }
    }
    else
    {
        if (content1 > content2)
        {
            res = 1;
        }
        else
        {
            res = -1;
        }
    }
 
    return res;
}
 
 
//Node 某列显示/隐藏
tree->SetColumnVisible(columnID,false);
 
//Node 第一列前：打勾+图片
node->SetState(1); //1 = 不打勾,2 = 打勾
node->SetDisplayIcon(bmpPath);
node->SetSelectedIcon(bmpPath);
 
//Node 设置显示文本颜色
node->SetForegroundColor(186);