src/XGUI/XGUI_Tools.cpp

   1 // Copyright (C) 2014-2019  CEA/DEN, EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 #include "XGUI_Tools.h"
  21
  22 #include "XGUI_ModuleConnector.h"
  23 #include "XGUI_Workshop.h"
  24
  25 #include "ModuleBase_IWorkshop.h"
  26 #include "ModuleBase_Tools.h"
  27
  28 #include <TopoDS_Shape.hxx>
  29 #include <ModelAPI_Object.h>
  30 #include <ModelAPI_Result.h>
  31 #include <ModelAPI_ResultParameter.h>
  32 #include <ModelAPI_Feature.h>
  33 #include <ModelAPI_Session.h>
  34 #include <ModelAPI_Document.h>
  35 #include <ModelAPI_ResultPart.h>
  36 #include <ModelAPI_CompositeFeature.h>
  37 #include <ModelAPI_Tools.h>
  38 #include <ModelAPI_ResultConstruction.h>
  39 #include <ModelAPI_ResultBody.h>
  40 #include <Events_InfoMessage.h>
  41
  42 #include <GeomAPI_Shape.h>
  43 #include <GeomAlgoAPI_CompoundBuilder.h>
  44
  45 #include <TopoDS_Shape.hxx>
  46
  47 #include <QDir>
  48 #include <QMessageBox>
  49
  50 #include <iostream>
  51 #include <sstream>
  52 #include <string>
  53
  54 #ifndef WIN32
  55 # include <sys/stat.h>
  56 # include <dirent.h>
  57 # include <unistd.h>
  58 # define _separator_ '/'
  59 #else
  60 #include <io.h>
  61 #define F_OK 0
  62 #define access _access
  63 # include <windows.h>
  64 # define _separator_ '\\'
  65 #endif
  66
  67 namespace XGUI_Tools {
  68 //******************************************************************
  69 QString dir(const QString& path, bool isAbs)
  70 {
  71   QDir aDir = QFileInfo(path).dir();
  72   QString dirPath = isAbs ? aDir.absolutePath() : aDir.path();
  73   if (dirPath == QString("."))
  74     dirPath = QString();
  75   return dirPath;
  76 }
  77
  78 //******************************************************************
  79 QString file(const QString& path, bool withExt)
  80 {
  81   QString fPath = path;
  82   while (!fPath.isEmpty() && (fPath[fPath.length() - 1] == '\\' ||
  83           fPath[fPath.length() - 1] == '/'))
  84     fPath.remove(fPath.length() - 1, 1);
  85
  86   if (withExt)
  87     return QFileInfo(fPath).fileName();
  88   else
  89     return QFileInfo(fPath).completeBaseName();
  90 }
  91
  92 //******************************************************************
  93 QString addSlash(const QString& path)
  94 {
  95   QString res = path;
  96   if (!res.isEmpty() && res.at(res.length() - 1) != QChar('/')
  97       && res.at(res.length() - 1) != QChar('\\'))
  98     res += QDir::separator();
  99   return res;
 100 }
 101
 102 //******************************************************************
 103 QString unionOfObjectNames(const QObjectPtrList& theObjects, const QString& theSeparator)
 104 {
 105   QStringList aObjectNames;
 106   foreach (ObjectPtr aObj, theObjects) {
 107     if (aObj->data()->isValid())
 108       aObjectNames << QString::fromStdString(aObj->data()->name());
 109   }
 110   if (aObjectNames.count() == 0)
 111     return QString();
 112   if (aObjectNames.count() == 1)
 113     return aObjectNames.first();
 114   return aObjectNames.join(theSeparator);
 115 }
 116
 117 //******************************************************************
 118 bool isModelObject(FeaturePtr theFeature)
 119 {
 120   return theFeature && !theFeature->data();
 121 }
 122
 123 //******************************************************************
 124 std::string featureInfo(FeaturePtr theFeature)
 125 {
 126   std::ostringstream aStream;
 127   if (theFeature)
 128     aStream << theFeature.get() << " " << theFeature->getKind();
 129   return QString(aStream.str().c_str()).toStdString();
 130 }
 131
 132 //******************************************************************
 133 /*FeaturePtr realFeature(const FeaturePtr theFeature)
 134  {
 135  if (theFeature->data()) {
 136  return theFeature;
 137  } else {
 138  ObjectPtr aObject = std::dynamic_pointer_cast<ModelAPI_Object>(theFeature);
 139  return aObject->featureRef();
 140  }
 141  }*/
 142
 143
 144 //******************************************************************
 145 bool canRemoveOrRename(QWidget* theParent, const std::set<FeaturePtr>& theFeatures)
 146 {
 147   bool aResult = true;
 148   std::string aNotActivatedNames;
 149   if (!ModelAPI_Tools::allDocumentsActivated(aNotActivatedNames)) {
 150     bool aFoundPartSetObject = ModuleBase_Tools::hasModuleDocumentFeature(theFeatures);
 151     if (aFoundPartSetObject) {
 152       const char* aKeyStr = "Selected objects can be used in Part documents which are not loaded: "
 153                             "%1. Whould you like to continue?";
 154       QMessageBox::StandardButton aRes = QMessageBox::warning(theParent, QObject::tr("Warning"),
 155                QObject::tr(aKeyStr).arg(aNotActivatedNames.c_str()),
 156                QMessageBox::No | QMessageBox::Yes, QMessageBox::No);
 157       aResult = aRes == QMessageBox::Yes;
 158     }
 159   }
 160   return aResult;
 161 }
 162
 163 //******************************************************************
 164 bool isAscii(const QString& theStr)
 165 {
 166   char aCh;
 167   for (int i = 0; i < theStr.size(); i++) {
 168     aCh = theStr[i].toLatin1();
 169     if (aCh == 0)
 170       return false;
 171     if ((aCh >= 0x30) && (aCh <= 0x39))
 172       continue;
 173     else if ((aCh >= 0x41) && (aCh <= 0x5A))
 174       continue;
 175     else if ((aCh >= 0x61) && (aCh <= 0x7A))
 176       continue;
 177     else if (aCh == 0x5f)
 178       continue;
 179     else
 180       return false;
 181   }
 182   return true;
 183 }
 184
 185 //******************************************************************
 186 bool canRename(const ObjectPtr& theObject, const QString& theName)
 187 {
 188   std::string aType = theObject->groupName();
 189   if (aType == ModelAPI_ResultParameter::group()) {
 190     // For parameters names only ASCII symbols have to be used
 191     if (!isAscii(theName))
 192       return false;
 193
 194     double aValue;
 195     ResultParameterPtr aParam;
 196     if (ModelAPI_Tools::findVariable(theObject->document(),
 197           FeaturePtr(), qPrintable(theName), aValue, aParam)) {
 198       const char* aKeyStr = "Selected parameter can not be renamed to: %1. "
 199                             "There is a parameter with the same name. Its value is: %2.";
 200       QString aErrMsg(QObject::tr(aKeyStr).arg(theName).arg(aValue));
 201       // We can not use here a dialog box for message -
 202       // it will crash editing process in ObjectBrowser
 203       Events_InfoMessage("XGUI_Tools", aErrMsg.toStdString()).send();
 204       return false;
 205     }
 206   }
 207   else {
 208     DocumentPtr aDoc = theObject->document();
 209     ObjectPtr aObj =
 210       aDoc->objectByName(aType, theName.toStdString());
 211
 212     if (aObj.get() && theObject != aObj) {
 213       QString aErrMsg(QObject::tr("Name %2 already exists in %1.").
 214         arg(aType.c_str()).arg(theName));
 215       // We can not use here a dialog box for message -
 216       // it will crash editing process in ObjectBrowser
 217       Events_InfoMessage("XGUI_Tools", aErrMsg.toStdString()).send();
 218       return false;
 219     }
 220   }
 221
 222   return true;
 223 }
 224
 225 //**************************************************************
 226
 227 XGUI_Workshop* workshop(ModuleBase_IWorkshop* theWorkshop)
 228 {
 229   XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(theWorkshop);
 230   return aConnector ? aConnector->workshop() : 0;
 231 }
 232
 233
 234 //********************************************************************
 235 QString generateName(const ModuleBase_ViewerPrsPtr& thePrs)
 236 {
 237   if (!thePrs.get() || !thePrs->object().get())
 238     return "Undefined";
 239
 240   GeomShapePtr aContext;
 241   ObjectPtr anObject = thePrs->object();
 242   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObject);
 243   if (aResult.get())
 244     aContext = aResult->shape();
 245   else {
 246     // TODO if there is this case
 247   }
 248
 249   QString aName = anObject->data()->name().c_str();
 250   if (aContext.get()) {
 251     GeomShapePtr aSubShape(new GeomAPI_Shape());
 252     TopoDS_Shape aShape = ModuleBase_Tools::getSelectedShape(thePrs);
 253     if (!aShape.IsNull()) {
 254       aSubShape->setImpl(new TopoDS_Shape(aShape));
 255       if (!aSubShape->isEqual(aContext)) {
 256         QString aTypeName;
 257         switch (aShape.ShapeType()) {
 258         case TopAbs_COMPOUND:
 259           aTypeName = "compound";
 260           break;
 261         case TopAbs_COMPSOLID:
 262           aTypeName = "compsolid";
 263           break;
 264         case TopAbs_SOLID:
 265           aTypeName = "solid";
 266           break;
 267         case TopAbs_SHELL:
 268           aTypeName = "shell";
 269           break;
 270         case TopAbs_FACE:
 271           aTypeName = "face";
 272           break;
 273         case TopAbs_WIRE:
 274           aTypeName = "wire";
 275           break;
 276         case TopAbs_EDGE:
 277           aTypeName = "edge";
 278           break;
 279         case TopAbs_VERTEX:
 280           aTypeName = "vertex";
 281           break;
 282         case TopAbs_SHAPE:
 283           aTypeName = "shape";
 284           break;
 285         }
 286         int aId = GeomAlgoAPI_CompoundBuilder::id(aContext, aSubShape);
 287         aName += QString("/%1_%2").arg(aTypeName).arg(aId);
 288       }
 289     }
 290   }
 291   return aName;
 292 }
 293
 294 std::wstring strToWStr(const std::string& theStr) {
 295   size_t aLen = theStr.size();
 296   std::wstring aResult(aLen, L'#');
 297   mbstowcs(&aResult[0], theStr.c_str(), aLen);
 298   return aResult;
 299 }
 300
 301 std::string getTmpDirByPath( const std::string& theTmpPath)
 302 {
 303   std::string aTmpDir = theTmpPath;
 304   if (aTmpDir == "" || access(aTmpDir.c_str() , F_OK) != 0) {
 305 #ifdef WIN32
 306     char *Tmp_dir = getenv("TEMP");
 307     if (Tmp_dir == NULL) {
 308       Tmp_dir = getenv("TMP");
 309       if (Tmp_dir == NULL)
 310         aTmpDir = "C:\\";
 311       else
 312         aTmpDir = Tmp_dir;
 313     }
 314     else
 315       aTmpDir = Tmp_dir;
 316 #else
 317     aTmpDir = "/tmp/";
 318 #endif
 319   }
 320
 321   if (aTmpDir.back() != _separator_)
 322     aTmpDir += _separator_;
 323
 324   srand((unsigned int)time( NULL ));
 325   //Get a random number to present a name of a sub directory
 326   int aRND = 999 + (int)(100000.0*rand()/(RAND_MAX+1.0));
 327   char buffer[127];
 328   sprintf(buffer, "%d", aRND);
 329   std::string aSubDir(buffer);
 330   if (aSubDir.size() <= 1)
 331     aSubDir = "123049876";
 332
 333   aTmpDir += aSubDir; //Get RND sub directory
 334
 335   std::string aDir = aTmpDir;
 336
 337   for(aRND = 0; access(aDir.c_str() , F_OK) == 0; aRND++) {
 338     sprintf( buffer, "%d", aRND );
 339     aDir = aTmpDir + buffer;  //Build a unique directory name
 340   }
 341   if (aDir.back() != _separator_)
 342     aDir += _separator_;
 343
 344 #ifdef WIN32
 345   CreateDirectory(strToWStr(aDir).c_str(), NULL);
 346 #else
 347   mkdir( aDir.c_str(), 0x1ff );
 348 #endif
 349
 350   return aDir;
 351 }
 352
 353 std::string getTmpDirByEnv( const char* thePathEnv)
 354 {
 355   char* aVal = thePathEnv[0] == 0 ? 0 : getenv(thePathEnv);
 356   std::string dir = aVal ? aVal : "";
 357   return getTmpDirByPath(dir).c_str();
 358 }
 359
 360 void removeTemporaryFiles(const std::string& theDirectory,
 361   const std::list<std::string>& theFiles)
 362 {
 363   std::string aDirName = theDirectory;
 364
 365   std::list<std::string>::const_iterator aFilesIter = theFiles.cbegin();
 366   for(; aFilesIter != theFiles.cend(); aFilesIter++) {
 367     const std::string& aFile = *aFilesIter;
 368     if(access(aFile.c_str() , F_OK) != 0)
 369       continue;
 370
 371 #ifdef WIN32
 372     DeleteFile(strToWStr(aFile).c_str());
 373 #else
 374     unlink(aFile.c_str());
 375 #endif
 376   }
 377
 378   if(access(aDirName.c_str() , F_OK) == 0) {
 379 #ifdef WIN32
 380     RemoveDirectory(strToWStr(aDirName).c_str());
 381 #else
 382     rmdir(aDirName.c_str());
 383 #endif
 384   }
 385 }
 386
 387 }