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     aSubShape->setImpl(new TopoDS_Shape(aShape));
 254     if (!aSubShape->isEqual(aContext)) {
 255       QString aTypeName;
 256       switch (aShape.ShapeType()) {
 257       case TopAbs_COMPOUND:
 258         aTypeName = "compound";
 259         break;
 260       case TopAbs_COMPSOLID:
 261         aTypeName = "compsolid";
 262         break;
 263       case TopAbs_SOLID:
 264         aTypeName = "solid";
 265         break;
 266       case TopAbs_SHELL:
 267         aTypeName = "shell";
 268         break;
 269       case TopAbs_FACE:
 270         aTypeName = "face";
 271         break;
 272       case TopAbs_WIRE:
 273         aTypeName = "wire";
 274         break;
 275       case TopAbs_EDGE:
 276         aTypeName = "edge";
 277         break;
 278       case TopAbs_VERTEX:
 279         aTypeName = "vertex";
 280         break;
 281       case TopAbs_SHAPE:
 282         aTypeName = "shape";
 283         break;
 284       }
 285       int aId = GeomAlgoAPI_CompoundBuilder::id(aContext, aSubShape);
 286       aName += QString("/%1_%2").arg(aTypeName).arg(aId);
 287     }
 288   }
 289   return aName;
 290 }
 291
 292 std::wstring strToWStr(const std::string& theStr) {
 293   size_t aLen = theStr.size();
 294   std::wstring aResult(aLen, L'#');
 295   mbstowcs(&aResult[0], theStr.c_str(), aLen);
 296   return aResult;
 297 }
 298
 299 std::string getTmpDirByPath( const std::string& theTmpPath)
 300 {
 301   std::string aTmpDir = theTmpPath;
 302   if (aTmpDir == "" || access(aTmpDir.c_str() , F_OK) != 0) {
 303 #ifdef WIN32
 304     char *Tmp_dir = getenv("TEMP");
 305     if (Tmp_dir == NULL) {
 306       Tmp_dir = getenv("TMP");
 307       if (Tmp_dir == NULL)
 308         aTmpDir = "C:\\";
 309       else
 310         aTmpDir = Tmp_dir;
 311     }
 312     else
 313       aTmpDir = Tmp_dir;
 314 #else
 315     aTmpDir = "/tmp/";
 316 #endif
 317   }
 318
 319   if (aTmpDir.back() != _separator_)
 320     aTmpDir += _separator_;
 321
 322   srand((unsigned int)time( NULL ));
 323   //Get a random number to present a name of a sub directory
 324   int aRND = 999 + (int)(100000.0*rand()/(RAND_MAX+1.0));
 325   char buffer[127];
 326   sprintf(buffer, "%d", aRND);
 327   std::string aSubDir(buffer);
 328   if (aSubDir.size() <= 1)
 329     aSubDir = "123049876";
 330
 331   aTmpDir += aSubDir; //Get RND sub directory
 332
 333   std::string aDir = aTmpDir;
 334
 335   for(aRND = 0; access(aDir.c_str() , F_OK) == 0; aRND++) {
 336     sprintf( buffer, "%d", aRND );
 337     aDir = aTmpDir + buffer;  //Build a unique directory name
 338   }
 339   if (aDir.back() != _separator_)
 340     aDir += _separator_;
 341
 342 #ifdef WIN32
 343   CreateDirectory(strToWStr(aDir).c_str(), NULL);
 344 #else
 345   mkdir( aDir.c_str(), 0x1ff );
 346 #endif
 347
 348   return aDir;
 349 }
 350
 351 std::string getTmpDirByEnv( const char* thePathEnv)
 352 {
 353   char* aVal = thePathEnv[0] == 0 ? 0 : getenv(thePathEnv);
 354   std::string dir = aVal ? aVal : "";
 355   return getTmpDirByPath(dir).c_str();
 356 }
 357
 358 void removeTemporaryFiles(const std::string& theDirectory,
 359   const std::list<std::string>& theFiles)
 360 {
 361   std::string aDirName = theDirectory;
 362
 363   std::list<std::string>::const_iterator aFilesIter = theFiles.cbegin();
 364   for(; aFilesIter != theFiles.cend(); aFilesIter++) {
 365     const std::string& aFile = *aFilesIter;
 366     if(access(aFile.c_str() , F_OK) != 0)
 367       continue;
 368
 369 #ifdef WIN32
 370     DeleteFile(strToWStr(aFile).c_str());
 371 #else
 372     unlink(aFile.c_str());
 373 #endif
 374   }
 375
 376   if(access(aDirName.c_str() , F_OK) == 0) {
 377 #ifdef WIN32
 378     RemoveDirectory(strToWStr(aDirName).c_str());
 379 #else
 380     rmdir(aDirName.c_str());
 381 #endif
 382   }
 383 }
 384
 385 }