From fefa99193b3e197107be082954e55f2aeeda6b85 Mon Sep 17 00:00:00 2001
From: Frank Dellaert <dellaert@gmail.com>
Date: Thu, 16 Dec 2021 13:52:08 -0500
Subject: [PATCH] Add operators

---
 gtsam/discrete/DiscreteBayesNet.h  | 5 +++++
 gtsam/discrete/DiscreteBayesTree.h | 6 ++++++
 2 files changed, 11 insertions(+)

diff --git a/gtsam/discrete/DiscreteBayesNet.h b/gtsam/discrete/DiscreteBayesNet.h
index 4ffac95ed..2d92b72e8 100644
--- a/gtsam/discrete/DiscreteBayesNet.h
+++ b/gtsam/discrete/DiscreteBayesNet.h
@@ -83,6 +83,11 @@ namespace gtsam {
     //** evaluate for given DiscreteValues */
     double evaluate(const DiscreteValues & values) const;
 
+    //** (Preferred) sugar for the above for given DiscreteValues */
+    double operator()(const DiscreteValues & values) const {
+      return evaluate(values);
+    }
+
     /**
     * Solve the DiscreteBayesNet by back-substitution
     */
diff --git a/gtsam/discrete/DiscreteBayesTree.h b/gtsam/discrete/DiscreteBayesTree.h
index 655bcb9ee..42ec7d417 100644
--- a/gtsam/discrete/DiscreteBayesTree.h
+++ b/gtsam/discrete/DiscreteBayesTree.h
@@ -80,6 +80,12 @@ class GTSAM_EXPORT DiscreteBayesTree
 
   //** evaluate probability for given DiscreteValues */
   double evaluate(const DiscreteValues& values) const;
+
+  //** (Preferred) sugar for the above for given DiscreteValues */
+  double operator()(const DiscreteValues & values) const {
+    return evaluate(values);
+  }
+
 };
 
 }  // namespace gtsam