NX二次开发源码分享：NXOpen C++创建UDO的过程

admin · 发表于 2015-2-6 12:43:29

请使用QQ关联注册PLM之家，学习更多关于内容，更多精彩原创视频供你学习！

您需要登录才可以下载或查看，没有账号？注册

x

NX二次开发源码分享：NXOpen C++创建UDO的过程

UDO的创建是NX二次开发中最高深的一部分内容，通过这个实例，你可以看到如何注册UDO的回调函数，如何创建UDO，定义UDO的编辑，UDO的信息显示等内容！

#include <stdio.h>
#include <string.h>
#include <stdarg.h>
#include <uf.h>
#include <uf_ui.h>
#include <uf_exit.h>
#include <uf_disp.h>

static void ECHO(char *format, ...)
{
char msg[UF_UI_MAX_STRING_LEN+1];
va_list args;
va_start(args, format);
vsprintf(msg, format, args);
va_end(args);
UF_UI_open_listing_window();
UF_UI_write_listing_window(msg);
UF_print_syslog(msg, FALSE);
}

#define UF_CALL(X) (report_error( __FILE__, __LINE__, #X, (X)))

static int report_error( char *file, int line, char *call, int irc)
{
if (irc)
{
      char err[133];

      UF_get_fail_message(irc, err);
      ECHO("*** ERROR code %d at line %d in %s:\n",
         irc, line, file);
      ECHO("+++ %s\n", err);
      ECHO("%s;\n", call);
}

return(irc);
}

#include <NXOpen/Session.hxx>
#include <NXOpen/Part.hxx>
#include <NXOpen/ParTCollection.hxx>
#include <NXOpen/Callback.hxx>
#include <NXOpen/NXException.hxx>
#include <NXOpen/UI.hxx>
#include <NXOpen/Selection.hxx>
#include <NXOpen/LogFile.hxx>
#include <NXOpen/NXObjectManager.hxx>
#include <NXOpen/ListingWindow.hxx>
#include <NXOpen/View.hxx>

#include <NXOpen/UserDefinedObjects_UserDefinedClass.hxx>
#include <NXOpen/UserDefinedObjects_UserDefinedClassManager.hxx>
#include <NXOpen/UserDefinedObjects_UserDefinedObject.hxx>
#include <NXOpen/UserDefinedObjects_UserDefinedObjectManager.hxx>
#include <NXOpen/UserDefinedObjects_UserDefinedEvent.hxx>
#include <NXOpen/UserDefinedObjects_UserDefinedDisplayEvent.hxx>
#include <NXOpen/UserDefinedObjects_UserDefinedLinkEvent.hxx>
#include <NXOpen/UserDefinedObjects_UserDefinedObjectDisplayContext.hxx>

using namespace NXOpen;
using namespace NXOpen::UserDefinedObjects;

//static variables
static NXOpen::Session* theSession = NULL;
static UI* theUI = NULL;
static UserDefinedClass* myUDOclass = NULL;

//------------------------------------------------------------------------------
// Callback Name: myDisplayCB
// This is a callback method associated with displaying a UDO.
// This same callback is registered for display, select, fit, and attention point
//------------------------------------------------------------------------------
extern int myDisplayCB(UserDefinedDisplayEvent*  displayEvent)
{
try
{
      // Get the doubles used to define the selected screen position for this UDO.
      std::vector<double> myUDOdoubles = displayEvent->UserDefinedObject()->GetDoubles();

      // Use the doubles to define points of a triangle
      std::vector<Point3d> myPoints(4);

      myPoints[0].X = myUDOdoubles[0] + 0;
      myPoints[0].Y = myUDOdoubles[1] + 0;
      myPoints[0].Z = myUDOdoubles[2] + 0;

      myPoints[1].X = myUDOdoubles[0] + 100;
      myPoints[1].Y = myUDOdoubles[1] + 0;
      myPoints[1].Z = myUDOdoubles[2] + 0;

      myPoints[2].X = myUDOdoubles[0] + 0;
      myPoints[2].Y = myUDOdoubles[1] + 100;
      myPoints[2].Z = myUDOdoubles[2] + 0;

      myPoints[3].X = myUDOdoubles[0] + 0;
      myPoints[3].Y = myUDOdoubles[1] + 0;
      myPoints[3].Z = myUDOdoubles[2] + 0;

      // Display the triangle
      displayEvent->DisplayContext()->DisplayPolyline(myPoints);

      // Display the text next to the triangle
      Point3d myPt = Point3d(myUDOdoubles[0] + 100, myUDOdoubles[1], myUDOdoubles[2]);
      displayEvent->DisplayContext()->DisplayText("C++ UDO", myPt, UserDefinedObjectDisplayContext::TextRefBottomLeft);

      // UserDefinedObjectDisplayContext::DisplayFacets fails - see PR 6731653
      // This demonstrates using UF_DISP_display_facets as a work around
      double normals[9] = { 0,0,1, 0,0,1, 0,0,1 };
      double vertices[9] = { myUDOdoubles[0], myUDOdoubles[1], myUDOdoubles[2],
                           myUDOdoubles[0] + 100, myUDOdoubles[1], myUDOdoubles[2],
                           myUDOdoubles[0], myUDOdoubles[1] + 100, myUDOdoubles[2] };
      UF_DISP_facet_t facets[1] = { vertices, normals };
      void *context = displayEvent->DisplayContext()->GetHandle();

      UF_initialize();
      UF_CALL(UF_DISP_display_facets(facets, 3, 1, UF_DISP_TRIANGLE, context));
      UF_terminate();

}
catch (NXException ex)
{
      ECHO("CaUGht exception: %s\n", ex.Message());
}
return 0;
}
//------------------------------------------------------------------------------
// Callback Name: myEditCB
// This is a callback method associated with editing a UDO.
//------------------------------------------------------------------------------
extern int myEditCB(UserDefinedEvent* editEvent)
{
try
{
      // required for calls to legacy UF routines
      // such as UF_DISP_add_item_to_display
      UF_initialize();

      View* myView = NULL;
      Point3d myCursor(0,0,0);

      // highlight the current udo we are about to edit
      // this is helpful if multiple udo's were on the selection
      // list when the user decided to edit them
      editEvent->UserDefinedObject()->Highlight();

      // ask the user to select a new origin for this UDO
      Selection::DialogResponse myResponse = theUI->SelectionManager()->SelectScreenPosition("Select New Origin for C++ UDO", &myView, &myCursor);
      // we are done asking the user for input... unhighlight the udo
      editEvent->UserDefinedObject()->Unhighlight();

      // use the new screen position (if the user picked one)
      if( myResponse == Selection::DialogResponsePick )
      {
         std::vector<double> myUDOdoubles(3);
         myUDOdoubles[0] = myCursor.X;
         myUDOdoubles[1] = myCursor.Y;
         myUDOdoubles[2] = myCursor.Z;
         // store the newly selected origin with the udo
         editEvent->UserDefinedObject()->SetDoubles(myUDOdoubles);
         // add the udo to the display list manually
         // this will force the udo display to regenerate
         // immediately and show the changes we just made
         UF_DISP_add_item_to_display(editEvent->UserDefinedObject()->GetTag());
      }
      UF_terminate();
} catch (NXException ex)
{
      ECHO("Caught exception: %s\n", ex.Message());
} return 0;
}
//------------------------------------------------------------------------------
// Callback Name: myInfoCB
// This is a callback method associated with querying information for a UDO.
// The information is printed in the listing window.
//------------------------------------------------------------------------------
extern int myInfoCB(UserDefinedEvent* infoEvent)
{
try
{
      ListingWindow* theLW = theSession->ListingWindow();
      char msg[256];
      theLW->Open();
      theLW->WriteLine(" ");
      theLW->WriteLine("------------------------------------------------------------");
      theLW->WriteLine("Begin Custom Information");
      theLW->WriteLine(" ");
      sprintf( msg, "UDO Class Name: '%s'", infoEvent->UserDefinedObject()->UserDefinedClass()->ClassName().GetLocaleText() );
      theLW->WriteLine(msg);
      sprintf( msg, "UDO Friendly Name: '%s'", infoEvent->UserDefinedObject()->UserDefinedClass()->FriendlyName().GetLocaleText() );
      theLW->WriteLine(msg);
      std::vector<double> myUDOdoubles = infoEvent->UserDefinedObject()->GetDoubles();
      sprintf( msg, "myUDOdoubles(0) = %f", myUDOdoubles[0] );
      theLW->WriteLine(msg);
      sprintf( msg, "myUDOdoubles(1) = %f", myUDOdoubles[1] );
      theLW->WriteLine(msg); sprintf( msg, "myUDOdoubles(2) = %f", myUDOdoubles[2] );
      theLW->WriteLine(msg);
      theLW->WriteLine(" ");
      theLW->WriteLine("End Custom Information");
}
catch (NXException ex)
{
      ECHO("Caught exception: %s\n", ex.Message());
}
return 0;
}

//------------------------------------------------------------------------------
// initUDO
// Checks to see which (if any) of the application's static variables are
// uninitialized, and sets them accordingly.
// Initializes the UDO class and registers all of its callback methods.
//------------------------------------------------------------------------------
static int initUDO( bool alertUser)
{
try
{
      if (theSession == NULL)
      {
         theSession = Session::GetSession();
      }
      if( theUI == NULL )
      {
         theUI = UI::GetUI();
      }
      if (myUDOclass == NULL)
      {
         if (alertUser)
         {
            ListingWindow*
                  theLW = theSession->ListingWindow();
            theLW->Open();
            theLW->WriteLine("Registering C++ UDO Class");
         }
         // Define your custom UDO class
         myUDOclass = theSession->UserDefinedClassManager()->CreateUserDefinedObjectClass("Sample_Cpp_UDO", "Sample C++ UDO");
         // Setup properties on the custom UDO class
         myUDOclass->SetAllowQueryClassFromName(UserDefinedClass::AllowQueryClassOn);
         // Register callbacks for the UDO class
         myUDOclass->AddDisplayHandler(make_callback(&myDisplayCB));
         myUDOclass->AddAttentionPointHandler(make_callback(&myDisplayCB));
         myUDOclass->AddFitHandler(make_callback(&myDisplayCB));
         myUDOclass->AddSelectionHandler(make_callback(&myDisplayCB));
         myUDOclass->AddEditHandler(make_callback(&myEditCB));
         myUDOclass->AddInformationHandler(make_callback(&myInfoCB));
         // Add this class to the list of object types available for selection in NX.
         // If you skip this step you won't be able to select UDO's of this class,
         // even though you registered a selection callback.
         theUI->SelectionManager()->SetSelectionStatusOfUserDefinedClass(myUDOclass, true);
      }
}
catch (NXException ex)
{
      ECHO("Caught exception: %s\n", ex.Message());
}
return 0;
}

//------------------------------------------------------------------------------
// ufusr (Explicit Activation)
// This entry point is used to activate the application explicitly, as in
// "File->Execute UG/Open->NX Open..."
//------------------------------------------------------------------------------
extern void ufusr( char *parm, int *returnCode, int rlen )
{
try
{
      // required for calls to legacy UF routines
      // such as UF_DISP_add_item_to_display
      UF_initialize();

      // initialize the UDO - if we didn't load this library at
      // startup, here is our second chance to load it
      initUDO(true);

      // if we don't have any parts open create one
      BasePart* myBasePart = theSession->Parts()->BaseDisplay();
      if( myBasePart == NULL)
      {
         myBasePart = theSession->Parts()->NewBaseDisplay("test_cpp_udo.prt", BasePart::UnitsMillimeters);
      }

      View* myView = NULL;
      Point3d myCursor(0,0,0);

      // ask the user to select an origin for this UDO
      Selection::DialogResponse myResponse = theUI->SelectionManager()->SelectScreenPosition("Select Origin of C++ UDO", &myView, &myCursor);
      if( myResponse == Selection::DialogResponsePick )
      {
         // The user selected a point - go ahead and create the udo
         UserDefinedObjectManager* myUDOmanager = myBasePart->UserDefinedObjectManager();
         UserDefinedObject* firstUDO = myUDOmanager->CreateUserDefinedObject(myUDOclass);
         // set the color property of the udo - just for fun :)
         firstUDO->SetColor(36);
         // store the origin selected by the user with the udo
         std::vector<double> myUDOdoubles(3);
         myUDOdoubles[0] = myCursor.X;
         myUDOdoubles[1] = myCursor.Y;
         myUDOdoubles[2] = myCursor.Z;
         firstUDO->SetDoubles(myUDOdoubles);
         // add the udo to the display list manually
         // this will force the udo to display immediately
         UF_DISP_add_item_to_display(firstUDO->GetTag());
      }
      UF_terminate();
}
catch (const NXOpen::NXException& ex)
{
      ECHO("Caught exception: %s\n", ex.Message());
}
}

//------------------------------------------------------------------------------
// ufsta
// Entrypoint used when program is loaded automatically
// as NX starts up. Note this application must be placed in a
// special folder for NX to find and load it during startup.
// Refer to the NX Open documentation for more details on how
// NX finds and loads applications during startup.
//------------------------------------------------------------------------------
extern void ufsta( char *param, int *returnCode, int rlen )
{
try
{
      initUDO(false);
}
catch (const NXOpen::NXException& ex)
{
      ECHO("Caught exception: %s\n", ex.Message());
}
}

//------------------------------------------------------------------------------
// ufusr_ask_unload
// Make sure you specify AtTermination for the unload option.
// If you unload the library before the NX Session Terminates
// bad things could happen when we try to execute a udo
// callback that no longer exists in the session.
//------------------------------------------------------------------------------
extern int ufusr_ask_unload( void )
{
return (int)Session::LibraryUnloadOptionAtTermination;
}

PLM之家PLMHome-工业软件践行者

热图推荐

[二次开发源码] NX二次开发源码分享：NXOpen C++创建UDO的过程

admin 楼主

请使用QQ关联注册PLM之家，学习更多关于内容，更多精彩原创视频供你学习！

相关帖子

发表回复

浏览过的版块