【转载】在ARX中通过COM在ACAD中添加菜单和工具条

admin1

[code]代码如下：

extern "C" AcRx::AppRetCode
acrxEntryPoint(AcRx::AppMsgCode msg, void* pkt)
{undefined
switch (msg) {undefined
case AcRx::kInitAppMsg:
  // Comment out the following line if your
  // application should be locked into memory
  acrxDynamicLinker->unlockApplication(pkt);
  acrxDynamicLinker->registerAppMDIAware(pkt);
  InitApplication();
  addMenuThroughMfcCom();
  break;
case AcRx::kUnloadAppMsg:
  UnloadApplication();
  break;
}
return AcRx::kRetOK;
}

void addMenuThroughMfcCom()
{undefined
TRY
{undefined
  IAcadApplication IAcad(acedGetAcadWinApp()->GetIDispatch(TRUE));
  IAcadMenuBar IMenuBar(IAcad.GetMenuBar());

  long numberOfMenus;
  numberOfMenus = IMenuBar.GetCount();

  IAcadMenuGroups IMenuGroups(IAcad.GetMenuGroups());

  VARIANT index;
  VariantInit(&index);
  V_VT(&index) = VT_I4;
  V_I4(&index) = 0;
  IAcadMenuGroup IMenuGroup(IMenuGroups.Item(index));
  IAcadPopupMenus IPopUpMenus(IMenuGroup.GetMenus());

  CString cstrMenuName = "灵宇断面(&L)";

  VariantInit(&index);
  V_VT(&index) = VT_BSTR;
  V_BSTR(&index) = cstrMenuName.AllocSysString();

  IDispatch* pDisp=NULL;

  //see if the menu is already there
  TRY{pDisp = IPopUpMenus.Item(index); pDisp->AddRef();} CATCH(COleDispatchException,e){}END_CATCH;
  VariantClear(&index);

  if (pDisp==NULL) {undefined
   //create it
   IAcadPopupMenu IPopUpMenu(IPopUpMenus.Add(cstrMenuName));
   
   VariantInit(&index);
   V_VT(&index) = VT_I4;
   V_I4(&index) = 0;
   IPopUpMenu.AddMenuItem(index, "横断面成图(&H)", "HDM ");

   VariantInit(&index);
   V_VT(&index) = VT_I4;
   V_I4(&index) = 1;
   IPopUpMenu.AddMenuItem(index, "纵断面成图(&Z)", "ZDM ");

   VariantInit(&index);
   V_VT(&index) = VT_I4;
   V_I4(&index) = 2;
   IPopUpMenu.AddSeparator(index);

   VariantInit(&index);
   V_VT(&index) = VT_I4;
   V_I4(&index) = 3;
   IPopUpMenu.AddMenuItem(index, "灵宇断面帮助(&B)", "DMHELP ");

   pDisp = IPopUpMenu.m_lpDispatch;
   pDisp->AddRef();
  }

  IAcadPopupMenu IPopUpMenu(pDisp);
  if (!IPopUpMenu.GetOnMenuBar())
  {undefined
   VariantInit(&index);
   V_VT(&index) = VT_I4;
   V_I4(&index) = numberOfMenus - 2;;
   IPopUpMenu.InsertInMenuBar(index);
  }
  else
  {undefined
   VariantInit(&index);
   V_VT(&index) = VT_BSTR;
   V_BSTR(&index) = cstrMenuName.AllocSysString();
   IPopUpMenus.RemoveMenuFromMenuBar(index);
   VariantClear(&index);
  }
  pDisp->Release();
}
CATCH(COleDispatchException,e)
{undefined
  e->ReportError();
  e->Delete();
}
END_CATCH;
}

也可以用相同的方法添加工具条，这样加入的工具条和CAD本身的完全相同。代码如下：

void AddDMToolbar()
{undefined
IAcadApplication acad;
IDispatch *pDisp = acedGetAcadWinApp()->
        GetIDispatch(TRUE); //AddRef is called on the pointer
    acad.AttachDispatch(pDisp); // does not call AddRef()
IAcadMenuGroups acMenuGroups ( acad.GetMenuGroups() );

IAcadMenuGroup  acMenu;
CComVariant vt;
long cnt = acMenuGroups.GetCount();
for ( long i = 0; i < cnt; i++ )
{undefined
  vt = i;
  acMenu = acMenuGroups.Item( vt );

  CString cgrpName = acMenu.GetName();
  if ( cgrpName.CompareNoCase( "Acad" ) == 0 )
  {undefined
   break;
  }
}

IAcadToolbars acToolbars (acMenu.GetToolbars());

CString sToolbarName = "断面";
IAcadToolbar acToolbar;
cnt = acToolbars.GetCount();
for ( i = 0; i < cnt; i++ )
{undefined
  vt = i;
  acToolbar = acToolbars.Item( vt );
  if( acToolbar.GetName() == sToolbarName )
  {undefined
   return;
  }
}

CString appFileName = acedGetAppName(); //取出完整的应用程序名称，含路径
char dir[_MAX_DIR], drive[_MAX_DRIVE], path[_MAX_PATH];
_splitpath(appFileName, drive, dir, NULL, NULL);
_makepath(path, drive, dir, NULL, NULL);

try
{undefined
  acToolbar = acToolbars.Add ( sToolbarName );
  acToolbar.SetVisible( true );
  acToolbar.Dock( 0 );

  COleVariant flyOutButton;
  IAcadToolbarItem acToolbarItem;
  CString bmpFile;

  vt = 0;
  acToolbarItem = acToolbar.AddToolbarButton( vt, "横断面","绘制横断面图", "hdm ", flyOutButton );
  bmpFile.Format( "%s%s", path, "hdm.bmp" );
  acToolbarItem.SetBitmaps ( bmpFile, bmpFile );

  vt = 1;
  acToolbarItem = acToolbar.AddToolbarButton( vt, "纵断面","绘制纵断面图", "zdm ", flyOutButton );
  bmpFile.Format( "%s%s", path, "zdm.bmp" );
  acToolbarItem.SetBitmaps ( bmpFile, bmpFile );

  vt = 2;
  acToolbarItem = acToolbar.AddToolbarButton( vt, "帮助","显示软件帮助信息", "dmhelp ", flyOutButton );
  bmpFile.Format( "%s%s", path, "help.bmp" );
  acToolbarItem.SetBitmaps ( bmpFile, bmpFile );
}
catch(...)
{undefined
  //acutPrintf("/n创建断面工具条失败!");
  //acutPrintf("/n请与作者联系");
  return;
}
}
[/code]