From b6d1375c7bdf0f06eb32f1273ad5fd4f7c0e5673 Mon Sep 17 00:00:00 2001
From: JP Appel <jeanpierre.appel01@gmail.com>
Date: Sat, 28 Jun 2025 01:15:44 -0400
Subject: Add leveld optimization wrapper

---
 pkg/query/optimizer.go | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

(limited to 'pkg/query/optimizer.go')

diff --git a/pkg/query/optimizer.go b/pkg/query/optimizer.go
index c203b9c..cdb2455 100644
--- a/pkg/query/optimizer.go
+++ b/pkg/query/optimizer.go
@@ -47,6 +47,38 @@ func NewOptimizer(root *Clause, workers uint) Optimizer {
 	}
 }
 
+// Optimize clause according to level.
+// level 0 is automatic and levels < 0 do nothing.
+func (o Optimizer) Optimize(level int) {
+	o.Simplify()
+	if level < 0 {
+		return
+	} else if level == 0 {
+		// TODO: determine smarter level determination strategy
+		level = o.root.Depth()
+	}
+
+	oldDepth := o.root.Depth()
+	for range level {
+		// clause level parallel
+		o.Compact()
+		o.StrictEquality()
+		o.Tighten()
+		o.Contradictions()
+		// parallel + serial
+		o.Tidy()
+		// purely serial
+		o.Flatten()
+
+		depth := o.root.Depth()
+		if depth == oldDepth {
+			break
+		} else {
+			oldDepth = depth
+		}
+	}
+}
+
 // Perform optimizations in parallel. They should **NOT** mutate the tree
 func (o Optimizer) parallel(optimize func(*Clause)) {
 	jobs := make(chan *Clause, o.workers)
@@ -131,6 +163,12 @@ func (o *Optimizer) Flatten() {
 	})
 }
 
+// Remove multiples of equivalent statements within the same clause
+//
+// Examples
+//
+//	(and a="Fred Flinstone" a="Fred Flinstone") --> (and a="Fred Flinstone")
+//	(or a=Shaggy -a!=Shaggy) --> (or a=Shaggy)
 func (o *Optimizer) Compact() {
 	o.parallel(func(c *Clause) {
 		c.Statements = slices.CompactFunc(c.Statements, StatementEq)
-- 
cgit v1.2.3