Greynet

README.md

@@ -46,11 +46,11 @@ maturin develop --release
 # RoadMap
 - Types, arguments validation
 - Write docs
-- Tests, tests, tests...
+- Tests, tests, tests... + integration wtih CI/CD
 - Composite termination criterion (for example: solving limit minutes N AND score not improving M seconds)
 - Multi-level score
 - Custom moves support
-- Try to impove incremental (pseudo-incremental) score calculation mechanism (caching, no clonning, etc)
 - Website
 - Useful text materials, guides, presentations
-- Score explainer / interpreter for OOP API
+- Score explainer / interpreter for OOP API
+- Reimplement GreyNet in Rust

examples/greynet/greynet_example_aggregations.py

@@ -0,0 +1,75 @@
+# In main_example_fixed.py, after the existing code
+
+# --- 4. New Usage Example for Aggregations ---
+from dataclasses import dataclass
+from greyjack.score_calculation.greynet.greynet_fact import *
+from greyjack.score_calculation.greynet.builder import ConstraintBuilder, Collectors
+from greyjack.score_calculation.scores.SimpleScore import SimpleScore
+
+@greynet_fact
+@dataclass()
+class SalesTransaction:
+    region: str
+    amount: float
+
+builder = ConstraintBuilder(name="temporal-security", score_class=SimpleScore)
+# Define a constraint to analyze sales data per region.
+# We use a penalty of 0 because the goal is data extraction, not scoring.
+@builder.constraint("Sales Regional Analysis")
+def sales_analysis():
+    return (
+        builder.for_each(SalesTransaction)
+            .group_by(
+                lambda tx: tx.region,
+                Collectors.compose({
+                    "min_sale":    Collectors.min(lambda tx: tx.amount),
+                    "max_sale":    Collectors.max(lambda tx: tx.amount),
+                    "avg_sale":    Collectors.avg(lambda tx: tx.amount),
+                    "stddev_sale": Collectors.stddev(lambda tx: tx.amount),
+                    "total_sales": Collectors.sum(lambda tx: tx.amount),
+                    "num_sales":   Collectors.count()
+                })
+            )
+            .penalize_simple(0)
+    )
+
+# --- Execute the new analysis ---
+print("\n" + "="*50)
+print("--- Starting Sales Aggregation Example ---")
+print("="*50)
+
+# Re-use the same builder to create a new session if needed, or add to the existing one.
+# For a clean test, we'll build it again.
+sales_session = builder.build()
+
+transactions = [
+    SalesTransaction("North", 110.0),
+    SalesTransaction("North", 150.0),
+    SalesTransaction("North", 195.5),
+    SalesTransaction("South", 500.0),
+    SalesTransaction("South", 600.0),
+    SalesTransaction("West", 300.0),
+]
+
+sales_session.insert_batch(transactions)
+sales_session.flush()
+
+print("\n--- Sales Aggregation Results ---")
+matches = sales_session.get_constraint_matches()
+for constraint_id, violations in matches.items():
+    if constraint_id == "Sales Regional Analysis":
+        print(f"\nAnalysis: '{constraint_id}'")
+        # Sort results by region for consistent output
+        sorted_violations = sorted(violations, key=lambda v: v[1].fact_a)
+        for _, facts_tuple in sorted_violations:
+            region = facts_tuple.fact_a
+            stats = facts_tuple.fact_b
+            print(f"  - Region: {region}")
+            print(f"    - Count:       {stats['num_sales']}")
+            print(f"    - Total Sales: ${stats['total_sales']:.2f}")
+            print(f"    - Average Sale:  ${stats['avg_sale']:.2f}")
+            print(f"    - Min Sale:      ${stats['min_sale']:.2f}")
+            print(f"    - Max Sale:      ${stats['max_sale']:.2f}")
+            print(f"    - Std Dev:     ${stats['stddev_sale']:.2f}")
+
+print("\n--- Sales Example Complete ---")

examples/greynet/greynet_example_all_collectors.py

@@ -0,0 +1,253 @@
+# main_example.py
+
+from __future__ import annotations
+from dataclasses import dataclass
+from typing import Type, Callable, List
+from datetime import datetime, timedelta, timezone
+from greyjack.score_calculation.greynet.greynet_fact import *
+from greyjack.score_calculation.greynet.builder import ConstraintBuilder, Collectors
+from greyjack.score_calculation.scores.SimpleScore import SimpleScore
+
+
+# 1. Data Class Definitions (Facts)
+# =================================
+
+@greynet_fact
+@dataclass()
+class Sale:
+    sale_id: str
+    product_id: str
+    customer_id: str
+    price: float
+    quantity: int
+    timestamp: datetime
+
+@greynet_fact
+@dataclass()
+class Shipment:
+    order_id: str
+    shipment_id: str
+    shipment_no: int
+
+@greynet_fact
+@dataclass()
+class Maintenance:
+    machine_id: str
+    start_time: datetime
+    end_time: datetime
+
+@greynet_fact
+@dataclass()
+class UserEvent:
+    user_id: str
+    event_type: str
+    value: float  # e.g., transaction amount
+    timestamp: datetime
+
+
+# 2. Constraint and Collector Definitions
+# =======================================
+
+# Initialize the constraint builder
+cb = ConstraintBuilder(name="collector_showcase", score_class=SimpleScore)
+
+@cb.constraint("count_total_sales_per_product")
+def count_collector_example():
+    """Demonstrates: CountCollector
+    Counts the number of sales transactions for each product. Penalizes if a product has more than 3 sales.
+    """
+    return (cb.for_each(Sale)
+            .group_by(lambda s: s.product_id, Collectors.count())
+            .filter(lambda product_id, count: count > 3)
+            .penalize_simple(lambda product_id, count: count)
+           )
+
+@cb.constraint("sum_revenue_per_product")
+def sum_collector_example():
+    """Demonstrates: SumCollector
+    Calculates the total revenue (price * quantity) for each product.
+    """
+    return (cb.for_each(Sale)
+            .group_by(lambda s: s.product_id, Collectors.sum(lambda s: s.price * s.quantity))
+            .filter(lambda product_id, total_revenue: total_revenue > 0)
+            .penalize_simple(lambda product_id, total_revenue: 0) # Use penalty 0 to just report
+           )
+
+@cb.constraint("basic_price_stats_per_product")
+def min_max_avg_collectors_example():
+    """Demonstrates: MinCollector, MaxCollector, AvgCollector
+    Finds the minimum, maximum, and average sale price for each product.
+    """
+    return (cb.for_each(Sale)
+            .group_by(lambda s: s.product_id, Collectors.compose({
+                "min_price": Collectors.min(lambda s: s.price),
+                "max_price": Collectors.max(lambda s: s.price),
+                "avg_price": Collectors.avg(lambda s: s.price)
+            }))
+            .filter(lambda product_id, stats: stats["max_price"] > 1.0)
+            .penalize_simple(lambda product_id, stats: 0) # Reporting only
+           )
+
+@cb.constraint("advanced_price_stats_per_product")
+def stddev_variance_collectors_example():
+    """Demonstrates: StdDevCollector, VarianceCollector
+    Calculates the standard deviation and variance of prices for each product.
+    """
+    return (cb.for_each(Sale)
+            .group_by(lambda s: s.product_id, Collectors.compose({
+                "price_stddev": Collectors.stddev(lambda s: s.price),
+                "price_variance": Collectors.variance(lambda s: s.price)
+            }))
+            .filter(lambda product_id, stats: stats["price_stddev"] > 0)
+            .penalize_simple(lambda product_id, stats: 0) # Reporting only
+           )
+
+@cb.constraint("list_of_sales_per_product")
+def list_collector_example():
+    """Demonstrates: ListCollector
+    Collects all `Sale` objects for each product into a list.
+    """
+    return (cb.for_each(Sale)
+            .group_by(lambda s: s.product_id, Collectors.to_list())
+            .filter(lambda product_id, sales_list: len(sales_list) > 0)
+            .penalize_simple(lambda product_id, sales_list: 0) # Reporting only
+           )
+
+@cb.constraint("set_of_customers_per_product")
+def set_collector_example():
+    """Demonstrates: SetCollector and MappingCollector
+    Collects the unique set of customer IDs for each product.
+    """
+    return (cb.for_each(Sale)
+            .group_by(
+                lambda s: s.product_id,
+                Collectors.mapping(
+                    lambda s: s.customer_id, 
+                    Collectors.to_set()
+                )
+            )
+            .filter(lambda product_id, customer_set: len(customer_set) > 0)
+            .penalize_simple(lambda product_id, customer_set: 0) # Reporting only
+           )
+
+@cb.constraint("distinct_list_of_customers_per_product")
+def distinct_collector_example():
+    """Demonstrates: DistinctCollector
+    Collects a list of unique customer IDs for each product, preserving insertion order.
+    """
+    return (cb.for_each(Sale)
+            .group_by(
+                lambda s: s.product_id,
+                Collectors.mapping(
+                    lambda s: s.customer_id,
+                    Collectors.distinct()
+                )
+            )
+            .filter(lambda product_id, customer_list: len(customer_list) > 0)
+            .penalize_simple(lambda product_id, customer_list: 0) # Reporting only
+           )
+
+@cb.constraint("count_high_quantity_sales")
+def filtering_collector_example():
+    """Demonstrates: FilteringCollector
+    Counts only the sales where the quantity is greater than 2.
+    """
+    return (cb.for_each(Sale)
+            .group_by(
+                lambda s: s.product_id,
+                Collectors.filtering(
+                    lambda s: s.quantity > 2,
+                    Collectors.count()
+                )
+            )
+            .filter(lambda product_id, count: count > 0)
+            .penalize_simple(lambda product_id, count: 0) # Reporting only
+           )
+
+# 3. Main Execution Block
+# =======================
+
+def run_demonstration():
+    """Builds the session, inserts data, and prints the results."""
+
+    # --- Sample Data ---
+    now = datetime.now(timezone.utc)
+    sales_data = [
+        Sale("s1", "prod-a", "cust-1", 10.0, 1, now),
+        Sale("s2", "prod-b", "cust-1", 25.5, 2, now + timedelta(seconds=1)),
+        Sale("s3", "prod-a", "cust-2", 12.0, 5, now + timedelta(seconds=2)),
+        Sale("s4", "prod-a", "cust-1", 11.5, 2, now + timedelta(seconds=3)),
+        Sale("s5", "prod-b", "cust-3", 24.0, 1, now + timedelta(seconds=4)),
+        Sale("s6", "prod-a", "cust-3", 12.5, 3, now + timedelta(seconds=5)),
+    ]
+
+    shipments_data = [
+        Shipment("order-1", "sh-101", 1),
+        Shipment("order-1", "sh-102", 2),
+        Shipment("order-2", "sh-201", 1),
+        Shipment("order-1", "sh-104", 4), # Gap in sequence
+        Shipment("order-1", "sh-103", 3),
+    ]
+
+    maintenance_data = [
+        Maintenance("m1", now, now + timedelta(hours=2)),
+        Maintenance("m2", now, now + timedelta(hours=1)),
+        Maintenance("m1", now + timedelta(hours=1), now + timedelta(hours=3)), # Overlaps with the first
+        Maintenance("m1", now + timedelta(hours=4), now + timedelta(hours=5)), # Adjacent
+    ]
+
+    user_events_data = [
+        UserEvent("u1", "tx", 100, now),
+        UserEvent("u2", "tx", 150, now + timedelta(seconds=2)),
+        UserEvent("u1", "tx", 50,  now + timedelta(seconds=8)),
+        UserEvent("u3", "tx", 200, now + timedelta(seconds=11)), # New window
+        UserEvent("u2", "tx", 300, now + timedelta(seconds=15)),
+    ]
+
+    # --- Build and Run Session ---
+    session = cb.build()
+
+    print("## [INITIAL STATE] Inserting all facts...")
+    session.insert_batch(sales_data)
+    session.insert_batch(shipments_data)
+    session.insert_batch(maintenance_data)
+    session.insert_batch(user_events_data)
+
+    matches = session.get_constraint_matches()
+    print_results(matches)
+
+    # --- Demonstrate Retraction ---
+    print("\n\n## [RETRACTION] Retracting one sale (s6) and one shipment (sh-103)...")
+    sale_to_retract = sales_data[-1]  # Sale("s6", "prod-a", ...)
+    shipment_to_retract = shipments_data[-1] # Shipment("order-1", "sh-103", 3)
+
+    session.retract(sale_to_retract)
+    session.retract(shipment_to_retract)
+
+    matches_after_retract = session.get_constraint_matches()
+    print("## Results after retraction:")
+    print_results(matches_after_retract)
+
+def print_results(matches):
+    """Helper function to print constraint matches in a structured way."""
+    if not matches:
+        print("  No constraint matches found.")
+        return
+
+    for constraint_id, match_list in matches.items():
+        print(f"\n### Constraint: `{constraint_id}`")
+        print("-" * (len(constraint_id) + 18))
+        for score_obj, match_tuple in match_list:
+            facts = [f for f in [
+                getattr(match_tuple, 'fact_a', None),
+                getattr(match_tuple, 'fact_b', None),
+            ] if f is not None]
+
+            print(f"  - Match: {facts}")
+            print(f"    Score: {score_obj}")
+        print("-" * (len(constraint_id) + 18))
+
+
+if __name__ == "__main__":
+    run_demonstration() 
+