Support dot-broadcasting in ljl_propfunc

src/ljl_expressions.jl

@@ -97,13 +97,36 @@ function _process_ljlexpr_impl(@nospecialize(expr::Expr), @nospecialize(f_varsub
     _process_inner = Base.Fix2(_process_ljlexpr_impl, f_varsubst)
     if expr.head in ljl_expr_allowed_heads
         if expr.head == :.
-            obj = expr.args[begin]
-            propname = expr.args[begin+1]
-            return Expr(expr.head, _process_ljlexpr_impl(obj, f_varsubst), propname)
+            if length(expr.args) == 1
+                arg1 = only(expr.args)
+                if arg1 isa Symbol
+                    # Standalone dot-operator syntax:
+                    expr
+                else
+                    throw(ArgumentError("LEGEND Julia expressions don't support `$expr`"))
+                end
+            elseif length(expr.args) == 2
+                arg1 = expr.args[begin]
+                arg2 = expr.args[begin+1]
+                if arg2 isa Union{Symbol,QuoteNode}
+                    # Property access
+                    return Expr(expr.head, _process_ljlexpr_impl(arg1, f_varsubst), arg2)
+                elseif arg2 isa Expr && arg2.head == :tuple
+                    # Broadcast syntax
+                    return Expr(expr.head, arg1, Expr(:tuple, map(_process_inner, arg2.args)...))
+                else
+                    throw(ArgumentError("LEGEND Julia expressions don't support `$expr`"))
+                end
+            else
+                throw(ArgumentError("LEGEND Julia expressions don't support `$expr`"))
+            end
         elseif expr.head == :call
             funcname = expr.args[begin]
+            funcname_str = string(funcname)
+            # Handle constructs like `a .+ b`:
+            base_funcname = funcname_str[begin] == '.' ? Symbol(funcname_str[begin+1:end]) : funcname
             funcargs = expr.args[begin+1:end]
-            if funcname in ljl_expr_allowed_funcs
+            if base_funcname in ljl_expr_allowed_funcs
                 return Expr(expr.head, funcname, map(_process_inner, funcargs)...)
             else
                 throw(ArgumentError("Function \"$(funcname)\" not allowed in LEGEND Julia expression."))

test/test_ljl_expressions.jl

@@ -49,4 +49,7 @@ include("testing_utils.jl")
     multi_pf = ljl_propfunc(expr_map)
     @test multi_pf isa PropertyFunctions.PropertyFunction
     @test @inferred(broadcast(multi_pf, data)) == StructArray(e_cal = ref_numfunc.(data), e_flag = ref_boolfunc.(data))
+
+    @test @inferred(ljl_propfunc("a.+b")((a = [1,2,3], b = [5, 6, 7]))) == [6, 8, 10]
+    @test @inferred(ljl_propfunc("log.(a)")((a = [1,2,3],))) ≈ [0.0, 0.6931471805599453, 1.0986122886681098]
 end