From b06e9fcbee295e757b647589cb607233158a0a40 Mon Sep 17 00:00:00 2001
From: Danila Fedorin <danila.fedorin@gmail.com>
Date: Thu, 3 Aug 2017 22:16:08 -0700
Subject: [PATCH] Add attempt to find the number implementation from the class.

---
 .../abacus/plugin/NumberImplementation.java   | 12 +++++++++++
 .../nwapw/abacus/plugin/PluginManager.java    | 21 +++++++++++++++++++
 2 files changed, 33 insertions(+)

diff --git a/src/main/java/org/nwapw/abacus/plugin/NumberImplementation.java b/src/main/java/org/nwapw/abacus/plugin/NumberImplementation.java
index c3827a5..60e02c6 100644
--- a/src/main/java/org/nwapw/abacus/plugin/NumberImplementation.java
+++ b/src/main/java/org/nwapw/abacus/plugin/NumberImplementation.java
@@ -18,6 +18,18 @@ public abstract class NumberImplementation {
         promotionPaths = new HashMap<>();
     }
 
+    public final Map<Class<? extends NumberInterface>, Function<NumberInterface, NumberInterface>> getPromotionPaths(){
+        return promotionPaths;
+    }
+
+    public final Class<? extends NumberInterface> getImplementation(){
+        return implementation;
+    }
+
+    public final int getPriority(){
+        return priority;
+    }
+
     public abstract NumberInterface instanceForString(String string);
     public abstract NumberInterface instanceForPi();
 
diff --git a/src/main/java/org/nwapw/abacus/plugin/PluginManager.java b/src/main/java/org/nwapw/abacus/plugin/PluginManager.java
index 7d837cb..9c98dfd 100644
--- a/src/main/java/org/nwapw/abacus/plugin/PluginManager.java
+++ b/src/main/java/org/nwapw/abacus/plugin/PluginManager.java
@@ -2,6 +2,7 @@ package org.nwapw.abacus.plugin;
 
 import org.nwapw.abacus.function.Function;
 import org.nwapw.abacus.function.Operator;
+import org.nwapw.abacus.number.NumberInterface;
 
 import java.lang.reflect.InvocationTargetException;
 import java.util.*;
@@ -31,6 +32,7 @@ public class PluginManager {
      */
     private Map<String, Operator> cachedOperators;
     private Map<String, NumberImplementation> cachedNumberImplementations;
+    private Map<Class<? extends NumberInterface>, NumberImplementation> cachedInterfaceImplementations;
     /**
      * List of all functions loaded by the plugins.
      */
@@ -54,6 +56,7 @@ public class PluginManager {
         cachedFunctions = new HashMap<>();
         cachedOperators = new HashMap<>();
         cachedNumberImplementations = new HashMap<>();
+        cachedInterfaceImplementations = new HashMap<>();
         allFunctions = new HashSet<>();
         allOperators = new HashSet<>();
         allNumberImplementations = new HashSet<>();
@@ -117,6 +120,24 @@ public class PluginManager {
         return searchCached(plugins, cachedNumberImplementations, Plugin::providedNumberImplementations,
                 Plugin::getNumberImplementation, name);
     }
+
+    public NumberImplementation interfaceImplementationFor(Class<? extends NumberInterface> name){
+        if(cachedInterfaceImplementations.containsKey(name)) return cachedInterfaceImplementations.get(name);
+        NumberImplementation toReturn = null;
+        outside:
+        for(Plugin plugin : plugins){
+            for(String implementationName : plugin.providedNumberImplementations()){
+                NumberImplementation implementation = plugin.getNumberImplementation(implementationName);
+                if(implementation.getImplementation().equals(name)) {
+                    toReturn = implementation;
+                    break outside;
+                }
+            }
+        }
+        cachedInterfaceImplementations.put(name, toReturn);
+        return toReturn;
+    }
+
     /**
      * Adds an instance of Plugin that already has been instantiated.
      *