From 085dc1d333ba4b76c89e362cbc6b61673736e06d Mon Sep 17 00:00:00 2001
From: Nathan Marz <nathan@nathanmarz.com>
Date: Mon, 11 May 2015 15:00:21 -0400
Subject: [PATCH] optimize lookup of structurepath implementation for regular
 functions (about a 3x speedup in non-compiled execution when functions are
 involved

---
 src/clj/com/rpl/specter/impl.clj | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/clj/com/rpl/specter/impl.clj b/src/clj/com/rpl/specter/impl.clj
index d1c44c7..9ba248d 100644
--- a/src/clj/com/rpl/specter/impl.clj
+++ b/src/clj/com/rpl/specter/impl.clj
@@ -83,8 +83,14 @@
               )]
     (->TransformFunctions StructureValsPathExecutor afn afn)))
 
+
+(defn structure-path-impl [this]
+  (if (fn? this) 
+    (-> StructurePath :impls (get clojure.lang.AFn))
+    (find-protocol-impl! StructurePath this)))
+
 (defn coerce-structure-path [this]
-  (let [pimpl (find-protocol-impl! StructurePath this)
+  (let [pimpl (structure-path-impl this)
         selector (:select* pimpl)
         updater (:update* pimpl)]
     (->TransformFunctions
@@ -96,7 +102,7 @@
     )))
 
 (defn coerce-structure-path-direct [this]
-  (let [pimpl (find-protocol-impl! StructurePath this)
+  (let [pimpl (structure-path-impl this)
         selector (:select* pimpl)
         updater (:update* pimpl)]
     (->TransformFunctions
@@ -110,6 +116,9 @@
 (defn obj-extends? [prot obj]
   (->> obj (find-protocol-impl prot) nil? not))
 
+(defn structure-path? [obj]
+  (or (fn? obj) (obj-extends? StructurePath obj)))
+
 (extend-protocol CoerceTransformFunctions
 
   TransformFunctions
@@ -122,7 +131,7 @@
 
   Object
   (coerce-path [this]
-    (cond (obj-extends? StructurePath this) (coerce-structure-path this)
+    (cond (structure-path? this) (coerce-structure-path this)
           (obj-extends? Collector this) (coerce-collector this)
           (obj-extends? StructureValsPath this) (coerce-structure-vals-path this)
           :else (throw-illegal (no-prot-error-str this))
@@ -192,7 +201,7 @@
         ))))
 
 (defn coerce-structure-vals-direct [this]
-  (cond (obj-extends? StructurePath this) (coerce-structure-path-direct this)
+  (cond (structure-path? this) (coerce-structure-path-direct this)
         (obj-extends? Collector this) (coerce-collector this)
         (obj-extends? StructureValsPath this) (coerce-structure-vals-path this)
         (instance? TransformFunctions this) this