src/GEOM/GEOM_PythonDump.cxx

   1 //  Copyright (C) 2007-2008  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 #include "GEOM_PythonDump.hxx"
  23
  24 #include <TDF_Tool.hxx>
  25
  26 #include <TopAbs.hxx>
  27
  28 #include <TColStd_ListOfInteger.hxx>
  29 #include <TColStd_ListIteratorOfListOfInteger.hxx>
  30
  31 namespace GEOM
  32 {
  33   size_t TPythonDump::myCounter = 0;
  34
  35   TPythonDump::TPythonDump (Handle(GEOM_Function)& theFunction, bool theAppend)
  36   {
  37     myFunction = theFunction;
  38     myCounter++;
  39     myAppend = theAppend;
  40   }
  41
  42   TPythonDump::~TPythonDump()
  43   {
  44     if (--myCounter == 0) {
  45       TCollection_AsciiString aDescr;
  46       if ( myAppend )
  47         aDescr = myFunction->GetDescription() + "\n\t";
  48       std::string aString = myStream.str();
  49       aDescr += (char *)aString.c_str();
  50       myFunction->SetDescription( aDescr );
  51     }
  52   }
  53
  54   TPythonDump& TPythonDump::operator<< (long int theArg)
  55   {
  56     myStream<<theArg;
  57     return *this;
  58   }
  59
  60   TPythonDump& TPythonDump::operator<< (int theArg)
  61   {
  62     myStream<<theArg;
  63     return *this;
  64   }
  65
  66   TPythonDump& TPythonDump::operator<< (double theArg)
  67   {
  68     myStream.precision(16);
  69     myStream<<theArg;
  70     return *this;
  71   }
  72
  73   TPythonDump& TPythonDump::operator<< (float theArg)
  74   {
  75     myStream.precision(8);
  76     myStream<<theArg;
  77     return *this;
  78   }
  79
  80   TPythonDump& TPythonDump::operator<< (const void* theArg)
  81   {
  82     myStream<<theArg;
  83     return *this;
  84   }
  85
  86   TPythonDump& TPythonDump::operator<< (const char* theArg)
  87   {
  88     myStream<<theArg;
  89     return *this;
  90   }
  91
  92   TPythonDump& TPythonDump::operator<< (const TopAbs_ShapeEnum theArg)
  93   {
  94     myStream<<"geompy.ShapeType[\"";
  95     TopAbs::Print(theArg, myStream);
  96     myStream<<"\"]";
  97     return *this;
  98   }
  99
 100   TPythonDump& TPythonDump::operator<< (const Handle(GEOM_Object)& theObject)
 101   {
 102     if (theObject.IsNull()) {
 103       myStream << "None";
 104     } else {
 105       TCollection_AsciiString anEntry;
 106       TDF_Tool::Entry(theObject->GetEntry(), anEntry);
 107       myStream << anEntry.ToCString();
 108     }
 109     return *this;
 110   }
 111
 112   Handle(GEOM_Object) GetCreatedLast(const Handle(GEOM_Object)& theObj1,
 113                                      const Handle(GEOM_Object)& theObj2)
 114   {
 115     if (theObj1.IsNull()) return theObj2;
 116     if (theObj2.IsNull()) return theObj1;
 117
 118     TColStd_ListOfInteger aTags1, aTags2;
 119     TDF_Tool::TagList(theObj1->GetEntry(), aTags1);
 120     TDF_Tool::TagList(theObj2->GetEntry(), aTags2);
 121     TColStd_ListIteratorOfListOfInteger aListIter1(aTags1), aListIter2(aTags2);
 122     for (; aListIter1.More(); aListIter1.Next(), aListIter2.Next()) {
 123       if (!aListIter2.More())
 124         return theObj1; // anObj1 is stored under anObj2
 125
 126       if (aListIter1.Value() > aListIter2.Value())
 127         return theObj1;
 128       else if (aListIter1.Value() < aListIter2.Value())
 129         return theObj2;
 130     }
 131     return theObj1;
 132   }
 133
 134   Handle(GEOM_Object) GetCreatedLast(const Handle(TColStd_HSequenceOfTransient)& theObjects)
 135   {
 136     Handle(GEOM_Object) anObject, aLatest;
 137     int i, aLen = theObjects->Length();
 138     if (aLen < 1)
 139       return aLatest;
 140
 141     for (i = 1; i <= aLen; i++) {
 142       anObject = Handle(GEOM_Object)::DownCast(theObjects->Value(i));
 143       aLatest = GetCreatedLast(aLatest, anObject);
 144     }
 145     return aLatest;
 146   }
 147 }