1 files changed, 22 insertions, 0 deletions
diff --git a/src/curveeditor/curveeditorstyle.h b/src/curveeditor/curveeditorstyle.h
index 23fdaf8..f60948a 100644
--- a/src/curveeditor/curveeditorstyle.h
+++ b/src/curveeditor/curveeditorstyle.h
@@ -25,12 +25,23 @@
 
 #pragma once
 
+#include <QBitmap>
 #include <QBrush>
 #include <QColor>
 #include <QDialog>
+#include <QIcon>
 
 namespace DesignTools {
 
+struct TreeItemStyleOption
+{
+    double margins;
+    QIcon pinnedIcon = QIcon(":/ICON_PINNED");
+    QIcon unpinnedIcon = QIcon(":/ICON_UNPINNED");
+    QIcon lockedIcon = QIcon(":/ICON_LOCKED");
+    QIcon unlockedIcon = QIcon(":/ICON_UNLOCKED");
+};
+
 struct HandleItemStyleOption
 {
     double size = 10.0;
@@ -92,6 +103,17 @@ struct CurveEditorStyle
     KeyframeItemStyleOption keyframeStyle;
 
     CurveItemStyleOption curveStyle;
+
+    TreeItemStyleOption treeItemStyle;
 };
 
+inline QPixmap pixmapFromIcon(const QIcon &icon, const QSize &size, const QColor &color)
+{
+    QPixmap pixmap = icon.pixmap(size);
+    QPixmap mask(pixmap.size());
+    mask.fill(color);
+    mask.setMask(pixmap.createMaskFromColor(Qt::transparent));
+    return mask;
+}
+
 } // End namespace DesignTools.