Pro/TOOLKIT示例程序(十一)获取组立档的层次结构

在Pro/TOOLKIT中，类型ProAssembly表示组立档，而组立档下的元件，则用ProAsmcomp表示。

typedef struct pro_model_item
{
  ProType  type;
  int      id;
  ProMdl owner;
} ProAsmcomp;

ProAsmcomp和ProModelitem共用相同的数据结构，更确切地说，ProAsmcomp是特征(ProFeature)的一种。

• type：值为PRO_FEATURE。
• id：ID。
• owner：拥有元件特征的组立档。

我们对元件特征ProAsmcomp调用ProFeatureTypeGet，获取的特征类型为PRO_FEAT_COMPONENT。
要想访问组立档下的所有元件特征，可以调用ProSolidFeatVisit函数来遍历组立档下的所有特征，通过判断特征类型是否为PRO_FEAT_COMPONENT来确定是否为元件。
函数ProAsmcompMdlGet用于获取元件特征关联的实体档案。举个例子来说明元件特征和关联的实体档案的关系：

如上图所示：A.ASM下包含了两个不同的元件特征，而这两个元件特征关联的模型均为B.PRT。
事实上，元件特征只是记录了将哪个模型（模型的路径信息）通过怎样的方式组装到组立档案中，并不包含关联的档案本身。

代码

struct AsmTreeNode
{
    wstring           strName;   // 节点名称
    bool              bIsAsm;    // true:组立档; false:零件档
    AsmTreeNode       *pParent;  // 父节点
    list<AsmTreeNode> lstChilds; // 子节点的集合

    AsmTreeNode()
    {
        strName = L"";
        pParent = NULL;
    }
};

// 获取组立档的层次结构并输出到对话框
int OutputAsmTree(uiCmdCmdId  command,
                  uiCmdValue *p_value,
                  void       *p_push_command_data)
{
    AFX_MANAGE_STATE(AfxGetStaticModuleState()); 

    ProError err;

    // 获取当前模型
    ProMdl mdlCurr;
    err = ProMdlCurrentGet(&mdlCurr);
    ProMdlType mdlType;
    err = ProMdlTypeGet(mdlCurr, &mdlType);
    if (PRO_MDL_ASSEMBLY != mdlType)
    {
        return -1;
    }
    ProName name;
    err = ProMdlNameGet(mdlCurr, name);

    // 构造节点
    AsmTreeNode head;
    head.strName = wstring(name) + wstring(L".ASM");
    head.bIsAsm = true;
    head.pParent = NULL;

    // 遍历访问组立档下的元件特征
    err = ProSolidFeatVisit((ProSolid)mdlCurr, AsmCompVisitFunc, NULL, &head);

    // 将模型树的层次结构输出到对话框中
    CAsmTreeDlg dlg(&head);
    dlg.DoModal();

    return 0;
}

ProError AsmCompVisitFunc(ProFeature* p_feature,
                          ProError status,
                          ProAppData app_data)
{
    AsmTreeNode *pParent = (AsmTreeNode*)app_data;

    // 判断是不是元件特征
    ProError err;
    ProFeattype featType;
    ProFeatureTypeGet(p_feature, &featType);
    if (PRO_FEAT_COMPONENT != featType)
    {
        return PRO_TK_NO_ERROR;
    }

    // 获取元件特征关联的模型的信息
    ProMdl mdl;
    ProAsmcompMdlGet(p_feature, &mdl);
    ProName mdlName;
    ProMdlNameGet(mdl, mdlName);
    ProMdlType mdlType;
    ProMdlTypeGet(mdl, &mdlType);

    // 构造节点
    AsmTreeNode node;
    if (mdlType == PRO_MDL_ASSEMBLY)
    {
        node.bIsAsm = true;
        node.strName = wstring(mdlName) + wstring(L".ASM");
    }
    else
    {
        node.strName = wstring(mdlName) + wstring(L".PRT");
        node.bIsAsm = false;
    }

    // 递归遍历子节点
    if (mdlType == PRO_MDL_ASSEMBLY)
    {
        err = ProSolidFeatVisit((ProSolid)mdl, AsmCompVisitFunc, NULL, &node);
    }

    // 和父节点关联
    node.pParent = pParent;
    pParent->lstChilds.push_back(node);

    return PRO_TK_NO_ERROR;
}