0032527: Data Exchange, RWGltf_CafWriter - make name format configurable

[occt.git] / src / RWMesh / RWMesh.cxx

diff --git a/src/RWMesh/RWMesh.cxx b/src/RWMesh/RWMesh.cxx
new file mode 100644 (file)
index 0000000..19a94ab
--- /dev/null
+++ b/src/RWMesh/RWMesh.cxx
@@ -0,0 +1,112 @@
+// Copyright (c) 2021 OPEN CASCADE SAS
+//
+// This file is part of Open CASCADE Technology software library.
+//
+// This library is free software; you can redistribute it and/or modify it under
+// the terms of the GNU Lesser General Public License version 2.1 as published
+// by the Free Software Foundation, with special exception defined in the file
+// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
+// distribution for complete text of the license and disclaimer of any warranty.
+//
+// Alternatively, this file may be used under the terms of Open CASCADE
+// commercial license or contractual agreement.
+
+#include <RWMesh.hxx>
+
+#include <TDataStd_Name.hxx>
+#include <TDF_Tool.hxx>
+
+// ================================================================
+// Function : ReadNameAttribute
+// Purpose  :
+// ================================================================
+TCollection_AsciiString RWMesh::ReadNameAttribute (const TDF_Label& theLabel)
+{
+  Handle(TDataStd_Name) aNodeName;
+  return theLabel.FindAttribute (TDataStd_Name::GetID(), aNodeName)
+       ? TCollection_AsciiString (aNodeName->Get())
+       : TCollection_AsciiString();
+}
+
+// ================================================================
+// Function : FormatName
+// Purpose  :
+// ================================================================
+TCollection_AsciiString RWMesh::FormatName (RWMesh_NameFormat theFormat,
+                                            const TDF_Label& theLabel,
+                                            const TDF_Label& theRefLabel)
+{
+  switch (theFormat)
+  {
+    case RWMesh_NameFormat_Empty:
+    {
+      return TCollection_AsciiString();
+    }
+    case RWMesh_NameFormat_Product:
+    {
+      Handle(TDataStd_Name) aRefNodeName;
+      return theRefLabel.FindAttribute (TDataStd_Name::GetID(), aRefNodeName)
+           ? TCollection_AsciiString (aRefNodeName->Get())
+           : TCollection_AsciiString();
+    }
+    case RWMesh_NameFormat_Instance:
+    {
+      Handle(TDataStd_Name) aNodeName;
+      return theLabel.FindAttribute (TDataStd_Name::GetID(), aNodeName)
+           ? TCollection_AsciiString (aNodeName->Get())
+           : TCollection_AsciiString();
+    }
+    case RWMesh_NameFormat_InstanceOrProduct:
+    {
+      Handle(TDataStd_Name) aNodeName;
+      if (theLabel.FindAttribute (TDataStd_Name::GetID(), aNodeName)
+      && !aNodeName->Get().IsEmpty())
+      {
+        return TCollection_AsciiString (aNodeName->Get());
+      }
+
+      Handle(TDataStd_Name) aRefNodeName;
+      return theRefLabel.FindAttribute (TDataStd_Name::GetID(), aRefNodeName)
+           ? TCollection_AsciiString (aRefNodeName->Get())
+           : TCollection_AsciiString();
+    }
+    case RWMesh_NameFormat_ProductOrInstance:
+    {
+      Handle(TDataStd_Name) aRefNodeName;
+      if (theRefLabel.FindAttribute (TDataStd_Name::GetID(), aRefNodeName)
+      && !aRefNodeName->Get().IsEmpty())
+      {
+        return TCollection_AsciiString (aRefNodeName->Get());
+      }
+
+      Handle(TDataStd_Name) aNodeName;
+      return theLabel.FindAttribute (TDataStd_Name::GetID(), aNodeName)
+           ? TCollection_AsciiString (aNodeName->Get())
+           : TCollection_AsciiString();
+    }
+    case RWMesh_NameFormat_ProductAndInstance:
+    {
+      const TCollection_AsciiString anInstName = ReadNameAttribute (theLabel);
+      const TCollection_AsciiString aProdName  = ReadNameAttribute (theRefLabel);
+      return !anInstName.IsEmpty()
+           && aProdName != anInstName
+            ? aProdName + " [" + anInstName + "]"
+            : (!aProdName.IsEmpty()
+              ? aProdName
+              : TCollection_AsciiString(""));
+    }
+    case RWMesh_NameFormat_ProductAndInstanceAndOcaf:
+    {
+      const TCollection_AsciiString anInstName = ReadNameAttribute (theLabel);
+      const TCollection_AsciiString aProdName  = ReadNameAttribute (theRefLabel);
+      TCollection_AsciiString anEntryId;
+      TDF_Tool::Entry (theLabel, anEntryId);
+      return !anInstName.IsEmpty()
+           && aProdName != anInstName
+            ? aProdName + " [" + anInstName + "]" + " [" + anEntryId + "]"
+            : aProdName + " [" + anEntryId + "]";
+    }
+  }
+
+  return TCollection_AsciiString();
+}