mitkNodeUidMapper.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkNodeUidMapper_h
#define mitkNodeUidMapper_h
 
#include <mitkDataStorage.h>
#include <mitkDataNode.h>
 
#include <unordered_map>
#include <mutex>
#include <string>
#include <optional>
#include <atomic>
 
#include <MitkRESTAPIExports.h>
 
namespace mitk
{
  class MITKRESTAPI_EXPORT NodeUidMapper
  {
  public:
    static constexpr const char* UID_PROPERTY_KEY = "restapi.uid";
 
    NodeUidMapper();
 
    ~NodeUidMapper();
 
    NodeUidMapper(const NodeUidMapper&) = delete;
    NodeUidMapper& operator=(const NodeUidMapper&) = delete;
    NodeUidMapper(NodeUidMapper&&) = delete;
    NodeUidMapper& operator=(NodeUidMapper&&) = delete;
 
    void SetDataStorage(DataStorage* dataStorage);
 
    void ClearDataStorage();
 
    std::string GetOrCreateUid(DataNode* node);
 
    DataNode::Pointer FindNodeByUid(const std::string& uid) const;
 
    std::optional<std::string> GetUid(const DataNode* node) const;
 
    bool HasUid(const std::string& uid) const;
 
    void RestoreUid(const DataNode* node, const std::string& uid);
 
  private:
    void OnNodeAdded(const DataNode* node);
    void OnNodeRemoved(const DataNode* node);
    std::string GenerateUid();
    void ClearRESTUIDProperty();
 
    mitk::WeakPointer<DataStorage> m_DataStorage;
 
    // Thread-safe UID mappings
    mutable std::mutex m_Mutex;
    std::unordered_map<std::string, mitk::WeakPointer<DataNode>> m_UidToNode;
    std::unordered_map<const DataNode*, std::string> m_NodeToUid;
 
    // Counter for unique ID generation (never reused in session)
    std::atomic<uint64_t> m_UidCounter{0};
 
    // Track if we're connected
    bool m_Connected = false;
  };
}
 
#endif