From 7132f9d64a8621b257614cc9a5a39e3f8254a8bb Mon Sep 17 00:00:00 2001
From: Myk Taylor <myk002@yahoo.com>
Date: Sun, 10 Dec 2023 21:30:41 -0800
Subject: [PATCH] add identity traits for new types

---
 library/DataIdentity.cpp       | 4 ++++
 library/include/DataIdentity.h | 6 ++++++
 2 files changed, 10 insertions(+)

diff --git a/library/DataIdentity.cpp b/library/DataIdentity.cpp
index f8d52deb4..2e15e0d04 100644
--- a/library/DataIdentity.cpp
+++ b/library/DataIdentity.cpp
@@ -50,8 +50,12 @@ namespace df {
     OPAQUE_IDENTITY_TRAITS(std::mutex);
     OPAQUE_IDENTITY_TRAITS(std::future<void>);
     OPAQUE_IDENTITY_TRAITS(std::function<void()>);
+    OPAQUE_IDENTITY_TRAITS(std::function<bool()>);
+    OPAQUE_IDENTITY_TRAITS(std::function<int*()>);
+    OPAQUE_IDENTITY_TRAITS(std::function<std::string()>);
     OPAQUE_IDENTITY_TRAITS(std::optional<std::function<void()> >);
     OPAQUE_IDENTITY_TRAITS(std::variant<std::string, std::function<void()> >);
+    OPAQUE_IDENTITY_TRAITS(std::weak_ptr<df::widget_container>);
 
     buffer_container_identity buffer_container_identity::base_instance;
 }
diff --git a/library/include/DataIdentity.h b/library/include/DataIdentity.h
index 98f57368b..e9889b672 100644
--- a/library/include/DataIdentity.h
+++ b/library/include/DataIdentity.h
@@ -37,6 +37,8 @@ distribution.
 
 #include "DataDefs.h"
 
+#include <df/widget_container.h>
+
 namespace std {
     class condition_variable;
     class mutex;
@@ -578,8 +580,12 @@ namespace df
     OPAQUE_IDENTITY_TRAITS(std::mutex);
     OPAQUE_IDENTITY_TRAITS(std::future<void>);
     OPAQUE_IDENTITY_TRAITS(std::function<void()>);
+    OPAQUE_IDENTITY_TRAITS(std::function<bool()>);
+    OPAQUE_IDENTITY_TRAITS(std::function<int*()>);
+    OPAQUE_IDENTITY_TRAITS(std::function<std::string()>);
     OPAQUE_IDENTITY_TRAITS(std::optional<std::function<void()> >);
     OPAQUE_IDENTITY_TRAITS(std::variant<std::string, std::function<void()> >);
+    OPAQUE_IDENTITY_TRAITS(std::weak_ptr<df::widget_container>);
 
 #ifdef BUILD_DFHACK_LIB
     template<typename T>