Catia二次开发源码分享： 对面进行颜色修改和添加属性

admin

Catia二次开发源码分享： 对面进行颜色修改和添加属性

* e" j4 n- T, `4 }3 D- n% y( t1 w! z0 O 6 w! L9 J1 L; O( X+ K [mw_shl_code=c,true]// ! [( |8 C9 Q' o5 n/ ^1 }2 |! W // 3- Loads the input document // , z4 e9 i$ b& @ CATDocument *pDoc = NULL; , e0 P0 s! t* O9 S7 I rc= CATDocumentServices::OpenDocument(iArgv[1], pDoc) ; 1 [. O+ D" t4 {; b+ \- |# p3 j if( FAILED(rc) ) { : y) k3 E7 J; h, c cout <<"Error in opening the document: " << iArgv[1] << endl ; return 1; } . K. N! e) q1 r cout <<" " << iArgv[1] << " is opened" << endl; 4 F) @) Z) e/ g 8 Z# A0 S) H, _5 ]8 s CATInit *pDocAsInit = NULL; $ ~6 ~+ P9 w! `8 @0 S5 d rc= pDoc->QueryInterface(IID_CATInit, (void**)&pDocAsInit) ; if( FAILED(rc) ) + A: P) \/ N4 v/ L5 i6 t { cout << "Error, the document does not implement CATInit"<< endl; return 1; } . d f5 p) S' f0 d+ i // // 4- Gets root container of the document // CATIPrTContainer *pSpecContainer = NULL ; - V B* d0 {* t) ]7 { pSpecContainer = (CATIPrtContainer*)pDocAsInit->GetRootContainer("CATIPrtContainer"); pDocAsInit->Release(); # @2 b6 o$ \ ^7 R$ J2 u f' ^- j pDocAsInit = NULL ; 8 L2 c, ]& x- p+ E) ~6 m& \ if( NULL == pSpecContainer ) { 5 `) ?7 t# s. `* { cout <<"Error, the root container is NULL" << endl; return 1; 9 t" e1 [7 m" |: j% a6 F } // // 5- Retrieves the MechanicalPart of the document ; I- n0 n% B1 S& [! v; l/ f // CATIPrtPart_var spPart ( pSpecContainer->GetPart() ); 6 Q: b4 \' t5 L% G6 D5 v _5 K if ( NULL_var == spPart ) { cout <<"Error, the MechanicalPart is NULL" << endl; return 1; } 1 n! j& O8 ?2 }/ C7 {0 R pSpecContainer->Release(); pSpecContainer = NULL ; 8 ?$ [* e) C R+ L+ c // 0 h; @& W* {7 a" R A |* U$ @ //6 - Retrieves BRepAccess of all faces // + Z4 ~' m c& C8 N6 J$ n% U //6 - 1 Retrieving the feature holding the result of the main body 1 V% m' V) Q9 w, \% q6 p" [ // //get the part & j' n- T. z. K; u" c CATBaseUnknown_var spMainPartBody ; 4 \2 ~: H0 F: Q, u( E7 U R1 }* A CATLISTV(CATBaseUnknown_var) ListResult ; . G0 x& o; r. ? CATIPartRequest_var spPartRequest = spPart ; if ( NULL_var == spPartRequest ) 1 w& {8 M$ e' ~. }" A { cout <<"Error on CATIPartRequest" << endl; return 1; } 4 ]& c1 F; H4 Q //get the main tool rc = spPartRequest->GetMainBody("",spMainPartBody); if ( FAILED(rc) || ( NULL_var == spMainPartBody) ) { cout <<"Error with GetMainBody" << endl; return 1; % ~( Z) x" L- v7 |' M+ D } 2 O4 J( G( H# w //get its associated body : e. K3 Y3 t) }) c" r/ A/ L CATIBodyRequest_var spMainPartBodyRequest = spMainPartBody; if ( NULL_var == spMainPartBodyRequest ) { % _2 P+ _$ Y7 D4 g W+ u cout <<"Error, spMainPartBodyRequest is NULL" << endl; return 1; } ' @" U' m6 w" I //Retrieves the feature holding the result of the main body //It's the first element of the list returned by GetResults (CAA documentation) . g. G! L# m: Y1 o* H: e rc = spMainPartBodyRequest->GetResults( "", ListResult) ; if (!SUCCEEDED(rc) || 0>= ListResult.Size()) ; {/ \, E1 g# L8 K" o { % H2 c% H; v' ~ cout <<"Error with GetResults" << endl; - c' m/ E% W- D7 v+ ~9 U$ K return 1; } CATIGeometricalElement_var spFeatureResultGeomElem = ListResult[1]; if ( NULL_var == spFeatureResultGeomElem ) { cout <<"Error, spFeatureResultGeomElem is NULL" << endl; return 1; } . l) I$ Z) s$ E" C% { //retrieves its shape CATIShapeFeatureBody_var shapefeat = spFeatureResultGeomElem; if ( NULL_var == shapefeat ) " M4 q u# K4 }# Q/ Y { cout <<"Error, the Shape Feature is NULL" << endl; % L' \6 D% P0 b% B0 I. m% D return 1; } & m3 O7 h9 E, ~" u# x . Y9 E" ]& K7 Y" e //retrieves the feature associated to the BodyOUT ) V. ]. ~/ Q4 R+ v+ Z) s2 e% O) P$ ^ CATISpecObject_var FeatureSolid = shapefeat->GetResultOUT(); - o; g: D. L2 T$ l if ( NULL_var == FeatureSolid ) { ) \ L0 i5 m' _. m1 _) h cout <<"Error, the Feature Solid is NULL" << endl; ' D6 S0 K; t1 Q' H return 1; } 9 Y1 ~3 {" v- G) p+ E, P1 c# q" ^ 4 U3 o7 R0 h% _5 v* l+ x+ T 8 d! z; n- w9 L; N' y //6 - 2 retrieving its associated geometry // CATIMfGeometryAccess_var geoAccess = FeatureSolid; if (NULL_var == geoAccess) ( o# r& i4 _, |: h% Q$ g { . Q5 P2 C; u. d: E y8 q3 g cout <<"Error, the geometry access is NULL" << endl; & ~% s6 i6 k5 q5 u& s0 J return 1; ' z1 Z) M8 t. Y) w! s } T+ {2 O+ \8 k% [ 7 \5 U+ w2 t1 x" g. n //6 - 3 retrieving BRepAccess from geometry " }) v0 ~3 I' D$ ?5 v0 o6 r( n // CATLISTV(CATBaseUnknown_var) breps; ) F5 L* o: x( E6 q4 f5 g- k% B int nbfaces = geoAccess -> GetBReps(breps); if (0 == nbfaces) ! @/ F( ~, D' L' o3 B, I { - i3 q& _* g; z3 o* O0 x- G cout<<"Error, there is no face associated to the geometry"<

admin

[mw_shl_code=c,true]
8 N1 I& d( b; ?  //
  B/ ~- ], g) T0 `. J% g& V  //7 - retrieving and displaying current faces colors
/ H  s( Z; J4 C  //
$ R2 a) T' v# @* w7 d2 _. b% U  cout<<"--------------------------"<<endl;
  cout<<"Retrieving current colors"<<endl;
: ~4 a" p$ X& |0 K- d) r  CATListOfInt retrieveRed,retrieveGreen,retrieveBlue;
   
. b4 ]3 k; g" i6 j$ Z  rc = CATMmrApplicativeAttributes::GetColorsOnFaces( ListBreps, retrieveRed,retrieveGreen,retrieveBlue);
5 |1 S" R4 o) _/ F  if (E_FAIL == rc )
7 ^, Y/ b2 F9 v' _( j  {
1 b9 H1 _) Z% x5 O    cout<<"Error, an error occured while retrieving current part colors"<<endl;
0 @( @: n) O6 E/ S9 j    return 1;
  }

( b/ p* C  p4 e# g  for (compt =1;compt<=nbfaces;compt++)
- U* u/ A" X( n$ Y  U  {
/ p! w3 m: W! U- N: R, I    cout<<"for the "<<compt<<" face, associated color components are : R "<<retrieveRed[compt]<<" G "<<retrieveGreen[compt]<<" B "<<retrieveBlue[compt]<<endl;
$ S; Z/ D4 E$ w7 J4 B  }

4 }( \/ W4 s7 x& }0 F  //
  //8 - Setting a new Color on all faces
  //

7 e, y2 X' p+ h5 `& X/ p6 T  //we are to color the faces with different color according their position : three by three will have the same color
  cout<<"--------------------------"<<endl;
  cout<<"Setting new colors"<<endl;
% b9 g5 c! ?" x) n5 |# f( v  
  CATListOfInt newRed,newGreen,newBlue,FailedIndex;
  int x=0;
  for (compt = 1;compt<= nbfaces;compt++)
  {
    x = compt%3;
9 z; X2 w6 J0 f$ x7 K7 m    int red(0),green(0),blue(0);
    if (1 == x)
      red = 255;
- y1 t7 x6 \) Z- b; b2 k" A+ Y5 J    if (2 == x)
  A% E  y- y+ ~* N6 O      green = 255;
    if (0 == x)
* R) I% ^* l( g3 s; U      blue = 255;
. J8 Y. [' S" r0 L3 F    newRed.Append(red);
    newGreen.Append(green);
    newBlue.Append (blue);
    cout<<"Setting R "<<red<<" G "<<green<<" B "<<blue<<" on the "<<compt<<" face"<<endl;
  }

  rc = CATMmrApplicativeAttributes::SetColorsOnFaces(ListBreps,newRed,newGreen,newBlue,FailedIndex);
  if (E_FAIL == rc)
  {
    cout<<"Error, setting new color on faces failed for all faces"<<endl;
    return 1;
6 y1 ]* V7 `' U# j0 T8 g; O  }
  else if (S_FALSE == rc )
  {
4 }$ h" K% `! g/ R    int nberrors = FailedIndex.Size();
    cout<<"Error, setting new color on faces fails for faces :";
7 E" y! `; t/ B6 A9 D3 A# T    for (compt = 1;compt<= FailedIndex.Size(); compt++)
% y9 Y; i" S" v' x# L* V, @    {
      cout<<" "<<FailedIndex[compt];
$ k3 N* i: g% N! {    }
    cout<<endl;
  }
  else
    cout<<"Colors successfully set on every faces"<<endl;
  //
  //9 - retrieving colors we have just set
  //
8 P1 D( e9 t& Q+ y$ Z  cout<<"--------------------------"<<endl;
  cout<<"Retrieving new colors"<<endl;
  rc = CATMmrApplicativeAttributes::GetColorsOnFaces( ListBreps, retrieveRed,retrieveGreen,retrieveBlue);
  if (E_FAIL == rc )
8 I( F8 e, n( D, _. y  {
+ H- O7 s- T0 U! H    cout<<"Error, an error occured while retrieving current part colors"<<endl;
    return 1;
  }
. u7 u2 A. b2 B4 g+ a
" W" L( Y' h! n  for (compt =1;compt<=nbfaces;compt++)
  {
    cout<<"for the "<<compt<<" face, associated color components are : R "<<retrieveRed[compt]<<" G "<<retrieveGreen[compt]<<" B "<<retrieveBlue[compt]<<endl;
  }
; P) \- C; v/ x
  //
  //10 - Setting Applicative Attributes
  //
) d$ n: }7 o5 k9 f
  //
  //10 - 1 creating new attributes
  cout<<"--------------------------"<<endl;
6 l% ~2 \$ w9 }4 `3 x! Z  cout<<"Creating applicative attributes for different faces"<<endl;
7 o, Z( t; l  i  CATListOfCATUnicodeString **AttrNameListToSet = new CATListOfCATUnicodeString*[nbfaces];
        CATCkeListOfParm *AttrValueListToSet = new CATCkeListOfParm[nbfaces];
7 J4 q9 Y2 S# b% v( h% d% ]  CATICkeParmFactory_var factory = CATCkeGlobalFunctions::GetVolatileFactory();
  for (compt = 1;compt<= nbfaces;compt++)
  {
    cout<<"Creating ";
# n. p' R! X* T/ L/ m    x = compt%3;
! r" |/ S. y+ E8 e! E" X    //for each face we create a new list of apllicative attributes
6 Y' K4 z  i. ~2 I" |) |    //an applicative attribute is made of a Name and a value
    if (1 == x)
+ @. P4 N1 ~. X6 V1 N    {
" U8 [$ Y6 g) d4 S2 e1 y; N- U      AttrNameListToSet [compt-1] = new CATListOfCATUnicodeString (3);
      AttrValueListToSet[compt-1]   = new CATLISTV(CATBaseUnknown_var)(3);

      //The first attribute will be the number of the face
      AttrNameListToSet [compt-1]->Append("FACE_NUMBER");
. P4 M0 p0 l& {% l( t8 A' y- F      CATICkeParm_var tempcke1 = factory -> CreateInteger ("FACE_NUMBER",compt);
0 e/ ^* f: ?# W7 j2 ~      AttrValueListToSet[compt-1]->Append(tempcke1);
      
' ], m2 I/ L4 a+ Y7 X7 G      cout<<" FACE_NUMBER with value "<<compt;
  @7 s% h3 z$ n
) E' p* z0 M  b7 B$ [      //the second will be the color
8 x3 w. p% M4 L: C% t      AttrNameListToSet [compt-1]->Append("COLOR");
      CATICkeParm_var tempcke2 = factory -> CreateString ("COLOR","Red");
& v6 @6 e7 B' f% V4 B& D8 u; @      AttrValueListToSet[compt-1]->Append(tempcke2);
" L( Q7 ~, ^9 S! {! X2 B5 {      
% w" Z/ z$ I. D      cout<<" COLOR with value Red";
      
      //the third is x
      AttrNameListToSet [compt-1]->Append("X");
8 D6 q% p) j% d0 ~' S& m' o      CATICkeParm_var tempcke3 = factory -> CreateReal ("X",x);
      AttrValueListToSet[compt-1]->Append(tempcke3);
  d" c. \" l  U" H      
+ o" h  ^5 g% |7 I& O      cout<<" X with value "<<x;
. _7 O$ X" d, y7 |& e: {2 Y6 v, C      cout<<" for face nb "<<compt<<endl;
. q6 m* T. R  D3 o2 x% x& `    }
    else
. O' r3 p: E; D1 f0 w/ Q    //just to show you you can have different size of list
1 f! {& Z  f( y* g6 q( u    {
1 y% o3 ~" v; d" i      AttrNameListToSet [compt-1] = new CATListOfCATUnicodeString (2);
      AttrValueListToSet[compt-1]   = new CATLISTV(CATBaseUnknown_var)(2);
) \7 X0 w* X) H6 L6 ^8 J9 L
      //The first attribute will be the number of the face
* F8 g+ j5 \5 z! z9 E, y      AttrNameListToSet [compt-1]->Append("FACE_NUMBER");
      CATICkeParm_var tempcke1 = factory -> CreateInteger ("FACE_NUMBER",compt);
  D9 C4 O1 ~& W- }      AttrValueListToSet[compt-1]->Append(tempcke1);
  N7 M1 {8 z0 S( [
8 @! Z+ X% s$ J  @- q" X      cout<<" FACE_NUMBER with value "<<compt;

      //the second will be the color
' k0 F1 Q' Y2 H/ d! J+ m      AttrNameListToSet [compt-1]->Append("COLOR");
$ l- S6 f3 ~5 U  T8 y: W+ B! m  c, j      if (2 == x)
      {
        CATICkeParm_var tempcke2 = factory -> CreateString ("COLOR","Green");
        AttrValueListToSet[compt-1]->Append(tempcke2);
( p- Q* s( {% E8 P( o' B        
; P9 r2 ]* x! B3 w2 ^: H        cout<<" COLOR with value Green";
- X1 Q; B1 K+ y; I      }
      if (0 == x)
7 ^9 j% F; y& Q. M0 W2 F+ j      {
        CATICkeParm_var tempcke2 = factory -> CreateString ("COLOR","Blue");
        AttrValueListToSet[compt-1]->Append(tempcke2);
        
, \0 o$ L% w3 |* r; N% D, O        cout<<" COLOR with value Blue";
      }

" o1 K' W7 w- F3 ]' l2 j8 _. k      cout<<" on face nb "<<compt<<endl;

4 c' z: ^# r. ~5 e" n' X/ g& D    }
  }
" c3 b4 P! d9 f! ~* v
( y7 B" H# y* v' \  //10 - 2 Setting the applicative attributes
2 {; x+ ?6 H  k2 ]  cout<<endl<<"Setting applicative attributes on different faces"<<endl;
  FailedIndex.RemoveAll();
  rc=CATMmrApplicativeAttributes::SetAttributesOnFaces(ListBreps, AttrNameListToSet,AttrValueListToSet,FailedIndex);
  if (E_FAIL == rc)
  {
    cout<<"Error, we failed to set applicative attributes on all faces"<<endl;
  }
/ t( c; K: @* V8 Q5 B2 M* H5 m9 ]+ ^. X  else if (S_FALSE == rc)
  {
* n+ U! u; w$ U( g4 M, j    int nberrors = FailedIndex.Size();
    cout<<"Error, setting applicative attriutes on faces fails for faces :";
    for (compt = 1;compt<= FailedIndex.Size(); compt++)
' s. Q" @# b, _/ ~    {
      cout<<" "<<FailedIndex[compt];
    }
# z) [7 D# {& q/ C) P    cout<<endl;
1 q# f; k$ e: T/ L  }
  else
  {
    cout<<"Applicative attributes successfully set"<<endl;
  }
3 Y( B) C, \. F% N8 A
  //10 - 3 cleaning
  for  (compt=1; compt<=nbfaces;compt++)
0 g# D5 g: O" X3 b8 l4 x! g        {
                AttrNameListToSet[compt-1]->RemoveAll(); delete AttrNameListToSet[compt-1];  AttrNameListToSet[compt-1] = NULL;
                AttrValueListToSet[compt-1]->RemoveAll();delete AttrValueListToSet[compt-1]; AttrValueListToSet[compt-1] = NULL;
        }
. d  k" U: J/ t" I
3 U0 r9 m5 d% D) `: n% Y6 V, ]        delete [] AttrNameListToSet; AttrNameListToSet = NULL;
        delete [] AttrValueListToSet;AttrValueListToSet = NULL;

& i4 x& l' K" [# ~  //
  //11 - Retrieving the applicative attributes we have just set
3 r% n$ h9 ^; p. C  //
  cout<<"--------------------------"<<endl;
8 M5 q' ?& ?, a, z2 F  cout<<"Getting applicative attributes on faces"<<endl;
  //
  //11 - 1 Getting attributes
, h; R3 F7 I$ [( E  CATListOfCATUnicodeString * AttributNameList = new CATListOfCATUnicodeString[nbfaces];
$ C" Z+ k# i( E) M2 c  CATCkeListOfParm  AttrValList = new CATListValCATBaseUnknown_var[nbfaces];
  rc = CATMmrApplicativeAttributes::GetAttributesOnFaces(ListBreps, AttributNameList, AttrValList );
, r( S: }/ R# T. T* D& V  if (SUCCEEDED(rc))
  {
    CATListOfCATUnicodeString templiststring;
. @* i2 m  L8 j5 Q1 g! U2 f    CATListValCATBaseUnknown_var tempCke;
0 X+ ^2 h* E: `2 S0 G    for (compt=1;compt<=nbfaces;compt++)
3 n; b; o  B. h* g# G( j) U    {
      //retrieve a list a position k (arrays start at position zero).
      templiststring =  (AttributNameList)[compt-1];
      tempCke = (AttrValList)[compt-1];
& `* W" Y4 I# i
      int nbAttributes = templiststring.Size();
/ g# _' ]1 B' r  b      int nbValues     = tempCke.Size();
! N: t' i1 V+ n( T- Q& Z
      //the list must have the same size! If not, we jump to the next face
, W% j5 i1 f* \8 k) J# C      if (nbAttributes != nbValues)
% ]7 h1 d4 E/ m) B0 A2 Z      {
        cout<<"Error on face "<<compt<<" : nbAttributes and nbValues are different"<<endl;
% A( I/ u% e5 B* x        continue;
      }
      else
; n( P$ }) z, M6 U. E      {
        cout<<"For face "<<compt<<" there is(are) "<<nbAttributes<<" attribute(s) that is(are) : "<<endl;
$ M5 O4 f9 B. U( h' ?- _      }
' M, n' \- r3 j# s% d      for (int i=1;i<=nbAttributes;i++)
      {
8 V" [- ^) i1 W6 c2 n, A* i" _        cout<<"    - "<<templiststring.ConvertToChar()<<" = "<<((CATICkeParm_var)( tempCke ))->Content().ConvertToChar()<<endl;
. H: w( {; ^5 u9 Y7 y      }
      templiststring.RemoveAll();
      tempCke.RemoveAll();

    }
4 j5 {1 d! b! c2 O* V+ G  }
  else if (E_FAIL == rc)
4 l9 }8 S9 S5 {! `2 ^9 @  {
    cout<<"Error, applicative attributes couldn't be retrieved"<<endl;
: `- A( e. K1 u$ f$ A5 ^    return 1;
  }
  //
  //11 - 2 Cleaning List:
  int i = 0;
" m' q0 f7 ^, g0 x% X$ t  for  (i=0; i < nbfaces;i++)
5 k8 u; i4 i( ?* ?  {
    AttributNameList.RemoveAll();
) C4 a6 J& r8 m; `& T3 A5 Y    AttrValList.RemoveAll();
  }
  delete [] AttributNameList;        AttributNameList = NULL;
  delete [] AttrValList;            AttrValList = NULL;

4 s9 {9 u$ z5 S/ M8 K7 H) H0 K  for (i = 1;i<= nbfaces;i++)
% F$ C7 B2 \" n- V/ @' Q' }/ K  {
    Brep = ListBreps ;
    if (NULL != Brep)
( |4 C2 J5 P5 `& M7 \    {
      Brep->Release();
& h5 \$ J, T8 L5 n. L2 p2 V    }
  }
  ListBreps.RemoveAll();[/mw_shl_code]