From b5f30aaee9b0569e7d1925b77d4bc8996f4a8040 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lars=20G=C3=B6ttgens?= <lars.goettgens@rwth-aachen.de>
Date: Wed, 25 Oct 2023 11:00:49 +0200
Subject: [PATCH] Add function to print operators

---
 .../src/UserFunctions.jl                      | 23 +++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/experimental/BasisLieHighestWeight/src/UserFunctions.jl b/experimental/BasisLieHighestWeight/src/UserFunctions.jl
index 7332d48a7202..2b2ef898beb1 100644
--- a/experimental/BasisLieHighestWeight/src/UserFunctions.jl
+++ b/experimental/BasisLieHighestWeight/src/UserFunctions.jl
@@ -1,3 +1,26 @@
+@doc """
+```jldoctest
+julia> BasisLieHighestWeight.basis_lie_highest_weight_operators(:B, 2)
+4-element Vector{Tuple{Int64, Vector{QQFieldElem}}}:
+ (1, [1, 0])
+ (2, [0, 1])
+ (3, [1, 1])
+ (4, [1, 2])
+```
+"""
+function basis_lie_highest_weight_operators(type::Symbol, rank::Int)
+  lie_algebra, chevalley_basis = lie_algebra_with_basis(type, rank)
+  operators = chevalley_basis[1]  # TODO: change to [2]
+  weights_w = weights_for_operators(
+    lie_algebra.lie_algebra_gap, chevalley_basis[3], operators
+  )
+  weights_alpha = [
+    w_to_alpha(lie_algebra, convert(Vector{QQFieldElem}, weight_w)) for
+    weight_w in weights_w
+  ]
+  return collect(enumerate(weights_alpha))
+end
+
 @doc """
 basis_lie_highest_weight(
     type::Symbol,