src/GEOM/GEOM_PythonDump.cxx

   1 //  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 //  This library is free software; you can redistribute it and/or
   7 //  modify it under the terms of the GNU Lesser General Public
   8 //  License as published by the Free Software Foundation; either
   9 //  version 2.1 of the License.
  10 //
  11 //  This library is distributed in the hope that it will be useful,
  12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 //  Lesser General Public License for more details.
  15 //
  16 //  You should have received a copy of the GNU Lesser General Public
  17 //  License along with this library; if not, write to the Free Software
  18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 #include "GEOM_PythonDump.hxx"
  24
  25 #include <TDF_Tool.hxx>
  26
  27 #include <TopAbs.hxx>
  28
  29 #include <TColStd_ListOfInteger.hxx>
  30 #include <TColStd_ListIteratorOfListOfInteger.hxx>
  31
  32 namespace GEOM
  33 {
  34   size_t TPythonDump::myCounter = 0;
  35
  36   TPythonDump::TPythonDump (Handle(GEOM_Function)& theFunction, bool theAppend)
  37   {
  38     myFunction = theFunction;
  39     myCounter++;
  40     myAppend = theAppend;
  41   }
  42
  43   TPythonDump::~TPythonDump()
  44   {
  45     if (--myCounter == 0) {
  46       TCollection_AsciiString aDescr;
  47       if ( myAppend )
  48         aDescr = myFunction->GetDescription() + "\n\t";
  49       std::string aString = myStream.str();
  50       aDescr += (char *)aString.c_str();
  51       myFunction->SetDescription( aDescr );
  52     }
  53   }
  54
  55   TPythonDump& TPythonDump::operator<< (long int theArg)
  56   {
  57     myStream<<theArg;
  58     return *this;
  59   }
  60
  61   TPythonDump& TPythonDump::operator<< (int theArg)
  62   {
  63     myStream<<theArg;
  64     return *this;
  65   }
  66
  67   TPythonDump& TPythonDump::operator<< (double theArg)
  68   {
  69     myStream.precision(16);
  70     myStream<<theArg;
  71     return *this;
  72   }
  73
  74   TPythonDump& TPythonDump::operator<< (float theArg)
  75   {
  76     myStream.precision(8);
  77     myStream<<theArg;
  78     return *this;
  79   }
  80
  81   TPythonDump& TPythonDump::operator<< (const void* theArg)
  82   {
  83     myStream<<theArg;
  84     return *this;
  85   }
  86
  87   TPythonDump& TPythonDump::operator<< (const char* theArg)
  88   {
  89     myStream<<theArg;
  90     return *this;
  91   }
  92
  93   TPythonDump& TPythonDump::operator<< (const TopAbs_ShapeEnum theArg)
  94   {
  95     myStream<<"geompy.ShapeType[\"";
  96     TopAbs::Print(theArg, myStream);
  97     myStream<<"\"]";
  98     return *this;
  99   }
 100
 101   TPythonDump& TPythonDump::operator<< (const Handle(GEOM_Object)& theObject)
 102   {
 103     if (theObject.IsNull()) {
 104       myStream << "None";
 105     } else {
 106       TCollection_AsciiString anEntry;
 107       TDF_Tool::Entry(theObject->GetEntry(), anEntry);
 108       myStream << anEntry.ToCString();
 109     }
 110     return *this;
 111   }
 112
 113   Handle(GEOM_Object) GetCreatedLast(const Handle(GEOM_Object)& theObj1,
 114                                      const Handle(GEOM_Object)& theObj2)
 115   {
 116     if (theObj1.IsNull()) return theObj2;
 117     if (theObj2.IsNull()) return theObj1;
 118
 119     TColStd_ListOfInteger aTags1, aTags2;
 120     TDF_Tool::TagList(theObj1->GetEntry(), aTags1);
 121     TDF_Tool::TagList(theObj2->GetEntry(), aTags2);
 122     TColStd_ListIteratorOfListOfInteger aListIter1(aTags1), aListIter2(aTags2);
 123     for (; aListIter1.More(); aListIter1.Next(), aListIter2.Next()) {
 124       if (!aListIter2.More())
 125         return theObj1; // anObj1 is stored under anObj2
 126
 127       if (aListIter1.Value() > aListIter2.Value())
 128         return theObj1;
 129       else if (aListIter1.Value() < aListIter2.Value())
 130         return theObj2;
 131     }
 132     return theObj1;
 133   }
 134
 135   Handle(GEOM_Object) GetCreatedLast(const Handle(TColStd_HSequenceOfTransient)& theObjects)
 136   {
 137     Handle(GEOM_Object) anObject, aLatest;
 138     int i, aLen = theObjects->Length();
 139     if (aLen < 1)
 140       return aLatest;
 141
 142     for (i = 1; i <= aLen; i++) {
 143       anObject = Handle(GEOM_Object)::DownCast(theObjects->Value(i));
 144       aLatest = GetCreatedLast(aLatest, anObject);
 145     }
 146     return aLatest;
 147   }
 148 }