0025812: Replace dynamic loading mechanism of OCAF persistence with dynamic-link one
[occt.git] / src / XDEDRAW / XDEDRAW_Common.cxx
1 // Created on: 2003-08-15
2 // Created by: Sergey ZARITCHNY
3 // Copyright (c) 2003-2014 OPEN CASCADE SAS
4 //
5 // This file is part of Open CASCADE Technology software library.
6 //
7 // This library is free software; you can redistribute it and/or modify it under
8 // the terms of the GNU Lesser General Public License version 2.1 as published
9 // by the Free Software Foundation, with special exception defined in the file
10 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
11 // distribution for complete text of the license and disclaimer of any warranty.
12 //
13 // Alternatively, this file may be used under the terms of Open CASCADE
14 // commercial license or contractual agreement.
15
16
17 #include <DDocStd.hxx>
18 #include <DDocStd_DrawDocument.hxx>
19 #include <Dico_DictionaryOfTransient.hxx>
20 #include <Dico_IteratorOfDictionaryOfTransient.hxx>
21 #include <Draw.hxx>
22 #include <Draw_Interpretor.hxx>
23 #include <IFSelect_SessionPilot.hxx>
24 #include <IGESCAFControl_Reader.hxx>
25 #include <IGESCAFControl_Writer.hxx>
26 #include <IGESControl_Controller.hxx>
27 #include <Interface_Macros.hxx>
28 #include <STEPCAFControl_DictionaryOfExternFile.hxx>
29 #include <STEPCAFControl_ExternFile.hxx>
30 #include <STEPCAFControl_IteratorOfDictionaryOfExternFile.hxx>
31 #include <STEPCAFControl_Reader.hxx>
32 #include <STEPCAFControl_Writer.hxx>
33 #include <STEPControl_Controller.hxx>
34 #include <TCollection_ExtendedString.hxx>
35 #include <TDataStd_Name.hxx>
36 #include <TDF_Data.hxx>
37 #include <TDocStd_Application.hxx>
38 #include <TDocStd_Document.hxx>
39 #include <XDEDRAW.hxx>
40 #include <XDEDRAW_Common.hxx>
41 #include <XSControl_WorkSession.hxx>
42 #include <XSDRAW.hxx>
43 #include <XSDRAW_Vars.hxx>
44 #include <XSDRAWIGES.hxx>
45 #include <XSDRAWSTEP.hxx>
46 #include <DDF.hxx>
47
48 #include <DBRep.hxx>
49 #include <XCAFDoc_DocumentTool.hxx>
50 #include <XCAFDoc_ShapeTool.hxx>
51 #include <XCAFDoc_Editor.hxx>
52 #include <TDF_Tool.hxx>
53 #include <TopoDS_Shape.hxx>
54
55 #include <stdio.h>
56 //============================================================
57 // Support for several models in DRAW
58 //============================================================
59 static Handle(Dico_DictionaryOfTransient) thedictws = new Dico_DictionaryOfTransient;
60
61 static Standard_Boolean ClearDicWS()
62 {
63   thedictws->Clear();
64   return Standard_True;
65 }
66
67 static void AddWS(TCollection_AsciiString filename,
68                   const Handle(XSControl_WorkSession)& WS)
69 {
70   WS->SetVars ( new XSDRAW_Vars ); // support of DRAW variables
71   thedictws->SetItem( filename, WS );
72 }
73
74
75 static Standard_Boolean FillDicWS(Handle(STEPCAFControl_DictionaryOfExternFile)& dicFile)
76 {
77   ClearDicWS();
78   if ( dicFile->IsEmpty() ) {
79     return Standard_False;
80   }
81   Handle(STEPCAFControl_ExternFile) EF;
82   STEPCAFControl_IteratorOfDictionaryOfExternFile DicEFIt ( dicFile );
83   for (; DicEFIt.More(); DicEFIt.Next() ) {
84     TCollection_AsciiString filename = DicEFIt.Name();
85     EF = DicEFIt.Value();
86     AddWS ( filename, EF->GetWS() );
87   }
88   return Standard_True;  
89 }
90
91 static Standard_Boolean SetCurrentWS (TCollection_AsciiString filename)
92 {
93   if ( !thedictws->HasItem(filename) ) return Standard_False;
94   Handle(XSControl_WorkSession) CurrentWS = 
95     Handle(XSControl_WorkSession)::DownCast( thedictws->Item(filename) );
96   XSDRAW::Pilot()->SetSession( CurrentWS );
97   
98   return Standard_True;
99 }
100
101
102 //=======================================================================
103 //function : SetCurWS
104 //purpose  : Set current file if many files are read
105 //=======================================================================
106
107 static Standard_Integer SetCurWS (Draw_Interpretor& di , Standard_Integer argc, const char** argv)
108 {
109   if (argc <2) {
110     di<<"Use: "<<argv[0]<<" filename \n";
111     return 1;
112   }
113   TCollection_AsciiString filename (argv[1]);
114   SetCurrentWS( filename );
115   return 0;
116 }
117
118
119 //=======================================================================
120 //function : GetDicWSList
121 //purpose  : List all files recorded after translation
122 //=======================================================================
123
124 static Standard_Integer GetDicWSList (Draw_Interpretor& di, Standard_Integer /*argc*/, const char** /*argv*/)
125 {
126   Handle(Dico_DictionaryOfTransient) DictWS = thedictws;
127   if ( DictWS->IsEmpty() ) return 1;
128   Dico_IteratorOfDictionaryOfTransient DicIt ( DictWS );
129   di << " The list of last translated files:\n";
130   Standard_Integer num = 0;
131   for (; DicIt.More() ; DicIt.Next(), num++ ) {
132     TCollection_AsciiString strng ( DicIt.Name() );
133     if ( num ) di << "\n";
134     di << "\"" << strng.ToCString() << "\"";
135   }
136   return 0;
137 }
138
139 //=======================================================================
140 //function : GetCurWS
141 //purpose  : Return name of file which is current
142 //=======================================================================
143
144 static Standard_Integer GetCurWS (Draw_Interpretor& di, Standard_Integer /*argc*/, const char** /*argv*/)
145 {
146   Handle(XSControl_WorkSession) WS = XSDRAW::Session();
147   di << "\"" << WS->LoadedFile() << "\"";
148   return 0;
149 }
150
151 //=======================================================================
152 //function : FromShape
153 //purpose  : Apply fromshape command to all the loaded WSs
154 //=======================================================================
155
156 static Standard_Integer FromShape (Draw_Interpretor& di, Standard_Integer argc, const char** argv)
157 {
158   if ( argc <2 ) {
159     di << argv[0] << " shape: search for shape origin among all last tranalated files\n";
160     return 0;
161   }
162   
163   char command[256];
164   Sprintf ( command, "fromshape %.200s -1", argv[1] );
165   Handle(Dico_DictionaryOfTransient) DictWS = thedictws;
166   if ( DictWS->IsEmpty() ) return di.Eval ( command );
167   
168   Handle(XSControl_WorkSession) WS = XSDRAW::Session();
169
170   Dico_IteratorOfDictionaryOfTransient DicIt ( DictWS );
171 //  di << "Searching for shape among all the loaded files:\n";
172   Standard_Integer num = 0;
173   for (; DicIt.More() ; DicIt.Next(), num++ ) {
174     Handle(XSControl_WorkSession) CurrentWS = 
175       Handle(XSControl_WorkSession)::DownCast( DicIt.Value() );
176     XSDRAW::Pilot()->SetSession( CurrentWS );
177     di.Eval ( command );
178   }
179
180   XSDRAW::Pilot()->SetSession( WS );
181   return 0;
182 }
183
184 //=======================================================================
185 //function : ReadIges
186 //purpose  : Read IGES to DECAF document
187 //=======================================================================
188
189 static Standard_Integer ReadIges (Draw_Interpretor& di, Standard_Integer argc, const char** argv)
190 {
191   if ( argc <3 ) {
192     di << "Use: " << argv[0] << " Doc filename [mode]: read IGES file to a document\n";
193     return 0;
194   }
195   
196   DeclareAndCast(IGESControl_Controller,ctl,XSDRAW::Controller());
197   if (ctl.IsNull()) XSDRAW::SetNorm("IGES");
198
199   IGESCAFControl_Reader reader ( XSDRAW::Session(),Standard_True);
200   
201   if (argc == 4) {
202     Standard_Boolean mode = Standard_True;
203     for ( Standard_Integer i = 0; argv[3][i] ; i++ ) 
204       switch (argv[3][i]) {
205       case '-' : mode = Standard_False; break;
206       case '+' : mode = Standard_True; break;
207       case 'c' : reader.SetColorMode (mode); break;
208       case 'n' : reader.SetNameMode (mode); break;
209       case 'l' : reader.SetLayerMode (mode); break;
210       }
211   }
212   TCollection_AsciiString fnom,rnom;
213   Standard_Boolean modfic = XSDRAW::FileAndVar (argv[2],argv[1],"IGES",fnom,rnom);
214   if (modfic) di<<" File IGES to read : "<<fnom.ToCString()<<"\n";
215   else        di<<" Model taken from the session : "<<fnom.ToCString()<<"\n";
216 //  di<<" -- Names of variables BREP-DRAW prefixed by : "<<rnom<<"\n";
217   IFSelect_ReturnStatus readstat = IFSelect_RetVoid;
218   if (modfic) readstat = reader.ReadFile (fnom.ToCString());
219   else  if (XSDRAW::Session()->NbStartingEntities() > 0) readstat = IFSelect_RetDone;
220   if (readstat != IFSelect_RetDone) {
221     if (modfic) di<<"Could not read file "<<fnom.ToCString()<<" , abandon\n";
222     else di<<"No model loaded\n";
223     return 1;
224   }
225
226   Handle(TDocStd_Document) doc;
227   if (!DDocStd::GetDocument(argv[1],doc,Standard_False)) {  
228     Handle(TDocStd_Application) A = DDocStd::GetApplication();
229     A->NewDocument("BinXCAF",doc);  
230     TDataStd_Name::Set(doc->GetData()->Root(),argv[1]);  
231     Handle(DDocStd_DrawDocument) DD = new DDocStd_DrawDocument(doc);  
232     Draw::Set(argv[1],DD);       
233 //     di << "Document saved with name " << argv[1];
234   }
235   if ( ! reader.Transfer ( doc ) ) {
236     di << "Cannot read any relevant data from the IGES file\n";
237     return 1;
238   }
239   
240 //  Handle(DDocStd_DrawDocument) DD = new DDocStd_DrawDocument(doc);  
241 //  Draw::Set(argv[1],DD);       
242   di << "Document saved with name " << argv[1];
243   
244   return 0;
245 }
246
247 //=======================================================================
248 //function : WriteIges
249 //purpose  : Write DECAF document to IGES
250 //=======================================================================
251
252 static Standard_Integer WriteIges (Draw_Interpretor& di, Standard_Integer argc, const char** argv) 
253 {
254   if ( argc <3 ) {
255     di << "Use: " << argv[0] << " Doc filename [mode]: write document to IGES file\n";
256     return 0;
257   }
258   
259   Handle(TDocStd_Document) Doc;
260   DDocStd::GetDocument(argv[1], Doc);
261   if ( Doc.IsNull() ) {
262     di << argv[1] << " is not a document\n";
263     return 1;
264   }
265   
266   XSDRAW::SetNorm ("IGES");
267   
268 //  IGESControl_Writer ICW (Interface_Static::CVal("write.iges.unit"),
269 //                        Interface_Static::IVal("write.iges.brep.mode"));
270     
271   IGESCAFControl_Writer writer ( XSDRAW::Session(), Standard_True );
272   if (argc == 4) {
273     Standard_Boolean mode = Standard_True;
274     for ( Standard_Integer i = 0; argv[3][i] ; i++ ) 
275       switch (argv[3][i]) {
276       case '-' : mode = Standard_False; break;
277       case '+' : mode = Standard_True; break;
278       case 'c' : writer.SetColorMode (mode); break;
279       case 'n' : writer.SetNameMode (mode); break;
280       case 'l' : writer.SetLayerMode (mode); break;
281       }
282   }
283   writer.Transfer ( Doc );
284
285   di << "Writig IGES model to file " << argv[2] << "\n";
286   if ( writer.Write ( argv[2] ) ) di<<" Write OK\n";
287   else di<<" Write failed\n";
288
289   return 0;
290 }
291
292 //=======================================================================
293 //function : ReadStep
294 //purpose  : Read STEP file to DECAF document 
295 //=======================================================================
296
297 static Standard_Integer ReadStep (Draw_Interpretor& di, Standard_Integer argc, const char** argv)
298 {
299   if ( argc <3 ) {
300     di << "Use: " << argv[0] << " Doc filename [mode]: read STEP file to a document\n";
301     return 0;
302   }
303   
304   DeclareAndCast(STEPControl_Controller,ctl,XSDRAW::Controller());
305   if (ctl.IsNull()) XSDRAW::SetNorm("STEP");
306
307   STEPCAFControl_Reader reader ( XSDRAW::Session(),Standard_True);
308   
309   if (argc == 4) {
310     Standard_Boolean mode = Standard_True;
311     for ( Standard_Integer i = 0; argv[3][i] ; i++ ) 
312       switch (argv[3][i]) {
313       case '-' : mode = Standard_False; break;
314       case '+' : mode = Standard_True; break;
315       case 'c' : reader.SetColorMode (mode); break;
316       case 'n' : reader.SetNameMode (mode); break;
317       case 'l' : reader.SetLayerMode (mode); break;
318       case 'v' : reader.SetPropsMode (mode); break;
319       }
320   }
321   
322   TCollection_AsciiString fnom,rnom;
323   Standard_Boolean modfic = XSDRAW::FileAndVar (argv[2],argv[1],"STEP",fnom,rnom);
324   if (modfic) di<<" File STEP to read : "<<fnom.ToCString()<<"\n";
325   else        di<<" Model taken from the session : "<<fnom.ToCString()<<"\n";
326 //  di<<" -- Names of variables BREP-DRAW prefixed by : "<<rnom<<"\n";
327   IFSelect_ReturnStatus readstat = IFSelect_RetVoid;
328   if (modfic) readstat = reader.ReadFile (fnom.ToCString());
329   else  if (XSDRAW::Session()->NbStartingEntities() > 0) readstat = IFSelect_RetDone;
330   if (readstat != IFSelect_RetDone) {
331     if (modfic) di<<"Could not read file "<<fnom.ToCString()<<" , abandon\n";
332     else di<<"No model loaded\n";
333     return 1;
334   }
335
336   Handle(TDocStd_Document) doc;
337   if (!DDocStd::GetDocument(argv[1],doc,Standard_False)) {  
338     Handle(TDocStd_Application) A = DDocStd::GetApplication();
339     A->NewDocument("BinXCAF",doc);  
340     TDataStd_Name::Set(doc->GetData()->Root(),argv[1]);  
341     Handle(DDocStd_DrawDocument) DD = new DDocStd_DrawDocument(doc);  
342     Draw::Set(argv[1],DD);       
343 //     di << "Document saved with name " << argv[1];
344   }
345   if ( ! reader.Transfer ( doc ) ) {
346     di << "Cannot read any relevant data from the STEP file\n";
347     return 1;
348   }
349   
350   Handle(DDocStd_DrawDocument) DD = new DDocStd_DrawDocument(doc);  
351   Draw::Set(argv[1],DD);       
352   di << "Document saved with name " << argv[1];
353
354   Handle(STEPCAFControl_DictionaryOfExternFile) DicFile = reader.ExternFiles();
355   FillDicWS( DicFile );
356   AddWS ( fnom , XSDRAW::Session() );
357   
358   return 0;
359 }
360
361 //=======================================================================
362 //function : WriteStep
363 //purpose  : Write DECAF document to STEP
364 //=======================================================================
365
366 static Standard_Integer WriteStep (Draw_Interpretor& di, Standard_Integer argc, const char** argv)
367 {
368   if ( argc <3 ) {
369     di << "Use: " << argv[0] << " Doc filename [mode [multifile_prefix [label]]]: write document to the STEP file\n";
370     di << "mode can be: a or 0 : AsIs (default)\n";
371     di << "             f or 1 : FacettedBRep        s or 2 : ShellBasedSurfaceModel\n";
372     di << "             m or 3 : ManifoldSolidBrep   w or 4 : GeometricCurveSet/WireFrame\n";
373     di << "multifile_prefix: triggers writing assembly components as separate files,\n";
374     di << "                  and defines common prefix for their names\n";
375     di << "label: tag of the sub-assembly label to save only that sub-assembly\n";
376     return 0;
377   }
378   
379   Handle(TDocStd_Document) Doc;   
380   DDocStd::GetDocument(argv[1], Doc);
381   if ( Doc.IsNull() ) {
382     di << argv[1] << " is not a document\n";
383     return 1;
384   }
385   Standard_CString multifile = 0;
386   
387   Standard_Integer k = 3;
388   DeclareAndCast(STEPControl_Controller,ctl,XSDRAW::Controller());
389   if (ctl.IsNull()) XSDRAW::SetNorm("STEP");
390   STEPCAFControl_Writer writer ( XSDRAW::Session(), Standard_True );
391    
392   STEPControl_StepModelType mode = STEPControl_AsIs;
393   if ( argc > k ) {
394     switch (argv[k][0]) {
395     case 'a' :
396     case '0' : mode = STEPControl_AsIs;                    break;
397     case 'f' :
398     case '1' : mode = STEPControl_FacetedBrep;             break;
399     case 's' :
400     case '2' : mode = STEPControl_ShellBasedSurfaceModel;  break;
401     case 'm' :
402     case '3' : mode = STEPControl_ManifoldSolidBrep;       break;
403     case 'w' :
404     case '4' : mode = STEPControl_GeometricCurveSet;       break;
405     default :  di<<"3st arg = mode, incorrect [give fsmw]\n"; return 1;
406     }
407     Standard_Boolean wrmode = Standard_True;
408     for ( Standard_Integer i = 0; argv[k][i] ; i++ ) 
409       switch (argv[3][i]) {
410       case '-' : wrmode = Standard_False; break;
411       case '+' : wrmode = Standard_True; break;
412       case 'c' : writer.SetColorMode (wrmode); break;
413       case 'n' : writer.SetNameMode (wrmode); break;
414       case 'l' : writer.SetLayerMode (wrmode); break;
415       case 'v' : writer.SetPropsMode (wrmode); break;
416       }
417     k++;
418   }
419
420   TDF_Label label;
421   if( argc > k)
422   {
423     TCollection_AsciiString aStr(argv[k]);
424     if( aStr.Search(":") ==-1)
425       multifile = argv[k++];
426     
427   }
428   if( argc > k)
429   {
430       
431     if( !DDF::FindLabel(Doc->Main().Data(), argv[k], label) || label.IsNull()) {  
432       di << "No label for entry"  << "\n";
433       return 1; 
434       
435     }
436   }
437   if( !label.IsNull())
438   {  
439     di << "Translating label "<< argv[k]<<" of document " << argv[1] << " to STEP\n";
440     if(!writer.Transfer ( label, mode, multifile )) 
441     {
442       di << "The label of document cannot be translated or gives no result\n";
443       return 1;
444     }
445
446   }
447   else
448   {
449     di << "Translating document " << argv[1] << " to STEP\n";
450     if ( ! writer.Transfer ( Doc, mode, multifile ) ) {
451       di << "The document cannot be translated or gives no result\n";
452     }
453   }
454   
455
456   di << "Writing STEP file " << argv[2] << "\n";
457   IFSelect_ReturnStatus stat = writer.Write(argv[2]);
458   switch (stat) {
459     case IFSelect_RetVoid : di<<"No file written\n"; break;
460     case IFSelect_RetDone : {
461       di<<"File "<<argv[2]<<" written\n";
462
463       Handle(STEPCAFControl_DictionaryOfExternFile) DicFile = writer.ExternFiles();
464       FillDicWS( DicFile );
465       AddWS( argv[2], XSDRAW::Session() );
466       break;
467     }
468     default : di<<"Error on writing file\n"; break;
469   }
470   return 0;
471 }
472
473 static Standard_Integer Expand (Draw_Interpretor& di, Standard_Integer argc, const char** argv)
474 {
475   if (argc < 3) {
476     di<<"Use: "<<argv[0]<<" Doc recurs(0/1) or Doc recurs(0/1) label1 label2 ... or Doc recurs(0/1 shape1 shape2 ...\n";
477     return 1;
478   }
479   Handle(TDocStd_Document) Doc;   
480   DDocStd::GetDocument(argv[1], Doc);
481   if ( Doc.IsNull() ) { di << argv[1] << " is not a document\n"; return 1; }
482
483   Handle(XCAFDoc_ShapeTool) aShapeTool = XCAFDoc_DocumentTool::ShapeTool(Doc->Main());
484   Standard_Boolean recurs = Standard_False;
485   if(atoi(argv[2]) != 0)
486     recurs = Standard_True;
487
488   if (argc == 3)
489   {
490     if(!XCAFDoc_Editor::Expand(Doc->Main(), recurs)){
491       di << "The shape is assembly or not compaund\n";
492       return 1;
493     }
494   }
495   else 
496   {
497     for (Standard_Integer i = 3; i < argc; i++)
498     {
499       TDF_Label aLabel;
500       TDF_Tool::Label(Doc->GetData(), argv[i], aLabel);
501       if(aLabel.IsNull()){
502         TopoDS_Shape aShape;
503         aShape = DBRep::Get(argv[i]);
504         aLabel = aShapeTool->FindShape(aShape);
505       }
506
507       if (!aLabel.IsNull()){
508         if(!XCAFDoc_Editor::Expand(Doc->Main(), aLabel, recurs)){
509           di << "The shape is assembly or not compaund\n";
510           return 1;
511         }
512       }
513       else
514       { di << argv[i] << " is not a shape\n"; return 1; }
515     }
516   }
517   return 0;
518 }
519
520 void XDEDRAW_Common::InitCommands(Draw_Interpretor& di) {
521
522   static Standard_Boolean initactor = Standard_False;
523   if (initactor) return;  initactor = Standard_True;
524
525   Standard_CString g = "XDE translation commands";
526
527   di.Add("ReadIges" , "Doc filename: Read IGES file to DECAF document" ,__FILE__, ReadIges, g);
528   di.Add("WriteIges" , "Doc filename: Write DECAF document to IGES file" ,__FILE__, WriteIges, g);
529   di.Add("ReadStep" , "Doc filename: Read STEP file to DECAF document" ,__FILE__, ReadStep, g);
530   di.Add("WriteStep" , "Doc filename [mode=a [multifile_prefix] [label]]: Write DECAF document to STEP file" ,__FILE__, WriteStep, g);  
531   
532   di.Add("XFileList","Print list of files that was transfered by the last transfer" ,__FILE__, GetDicWSList , g);
533   di.Add("XFileCur", ": returns name of file which is set as current",__FILE__, GetCurWS, g);
534   di.Add("XFileSet", "filename: Set the specified file to be the current one",__FILE__, SetCurWS, g);
535   di.Add("XFromShape", "shape: do fromshape command for all the files",__FILE__, FromShape, g);
536
537   di.Add("XExpand", "XExpand Doc recursively(0/1) or XExpand Doc recursively(0/1) label1 abel2 ..."  
538           "or XExpand Doc recursively(0/1) shape1 shape2 ...",__FILE__, Expand, g);
539
540 }