[bos #35159][EDF] (2023-T1) Following commands in Python console. Fixed intermediate...

[modules/gui.git] / src / SUIT / SUIT_Tools.cxx

diff --git a/src/SUIT/SUIT_Tools.cxx b/src/SUIT/SUIT_Tools.cxx
old mode 100755 (executable)
new mode 100644 (file)
index 11a8ddb..c5802ff
--- a/src/SUIT/SUIT_Tools.cxx
+++ b/src/SUIT/SUIT_Tools.cxx
@@ -1,4 +1,4 @@
-// Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
+// Copyright (C) 2007-2023  CEA, EDF, OPEN CASCADE
 //
 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
@@ -26,6 +26,7 @@
 
 #include <stdio.h>
 #include <stdarg.h>
+#include <iostream>
 
 /*!
   Traces output to log-file.
@@ -41,7 +42,7 @@ void SUIT_Tools::trace( const char* lpszLog, const char* lpszFormat, ... )
   tmpPath += QString( "Salome_trace" );
 
   FILE* pStream;
-  pStream = fopen( lpszLog ? lpszLog : (const char*)tmpPath.toLatin1(), "a" );
+  pStream = fopen( lpszLog ? lpszLog : (const char*)tmpPath.toUtf8(), "a" );
   if ( pStream ) 
   {     
     va_list argptr;
@@ -62,17 +63,6 @@ QRect SUIT_Tools::makeRect( const int x1, const int y1, const int x2, const int
   return QRect( qMin( x1, x2 ), qMin( y1, y2 ), qAbs( x2 - x1 ), qAbs( y2 - y1 ) );
 }
 
-/*!
-  Creates font from string description
-*/
-QFont SUIT_Tools::stringToFont( const QString& fontDescription )
-{
-  QFont font;
-  if ( fontDescription.trimmed().isEmpty() || !font.fromString( fontDescription ) )
-    font = QFont( "Courier", 11 );
-  return font;
-}
-
 /*!
   Creates font's string description
 */
@@ -88,3 +78,47 @@ void SUIT_Tools::centerWidget( QWidget* src, const QWidget* ref )
 {
   SUIT_Tools::alignWidget( src, ref, Qt::AlignCenter );
 }
+
+/*!
+  Add tracing code to given python command if it was activated by PYCONSOLE_TRACE env variable.
+  Immediately return if PYCONSOLE_TRACE wasn't set.
+*/
+void SUIT_Tools::addTraceToPythonCommand(const QString& fileName, QString& command)
+{
+  auto isPythonTraceEnabled = []() -> bool
+  {
+    const char* envVar = std::getenv("PYCONSOLE_TRACE");
+
+    if (envVar && (envVar[0] != '\0'))
+    {
+      try
+      {
+        const long long numValue = std::stoll(envVar);
+        return numValue > 0;
+      }
+      catch(const std::exception& e)
+      {
+        std::cerr << e.what() << '\n';
+      }
+    }
+
+    return false;
+  };
+
+  static const bool isActivated = isPythonTraceEnabled();
+  if (!isActivated)
+  {
+    return;
+  }
+
+  // Using sys.setprofile() instead of sys.settrace() because we don't need any other events except of 'call' and 'return'.
+  // Another reason: the trace function for sys.settrace() must return itself, so we can't use it properly with lambda.
+  command = QString("sys.setprofile(lambda frame, event, arg: "
+    "print('>>', frame.f_lineno, ': ', frame.f_code.co_name) if event == 'call' and frame.f_code.co_filename == '%1' and frame.f_code.co_name != '<module>' else "
+    "print('<<', frame.f_lineno, ': ', frame.f_code.co_name) if event == 'return' and frame.f_code.co_filename == '%1' and frame.f_code.co_name != '<module>' else "
+    "None); "
+    "%2; "
+    "sys.setprofile(None); ").
+    arg(fileName).
+    arg(command);
+}