Multiple inputs & results

example/.gitignore

@@ -0,0 +1,9 @@
+# Files and directories created by pub
+.dart_tool/
+.packages
+
+# Conventional directory for build outputs
+build/
+
+# Directory created by dartdoc
+doc/api/

example/README.md

@@ -0,0 +1 @@
+A simple example showcase of memoize library

example/bin/example.dart

@@ -0,0 +1,32 @@
+import 'dart:math' as math;
+
+import 'package:memoize/memoize.dart';
+
+num heavyCalc(num number) {
+  num sum = 0;
+  for (int x = 0; x < 999999; x++) {
+    sum += math.sqrt(number);
+  }
+  return sum;
+}
+
+void main(List<String> arguments) {
+  print('Some heavy calculations done 99 times with 6 different inputs:');
+  List<double> numbers = [10000, 4206969, 777, 21372137, 99999999, 10000];
+  var w = Stopwatch()..start();
+  for (int x = 0; x < 99; x++) {
+    for (var n in numbers) {
+      heavyCalc(n);
+    }
+  }
+  print('\tNo cache: ${w.elapsedMilliseconds}ms');
+
+  w.reset();
+  var cacheSqrt = memo1(heavyCalc);
+  for (int x = 0; x < 99; x++) {
+    for (var n in numbers) {
+      cacheSqrt(n);
+    }
+  }
+  print('\tWith cache: ${w.elapsedMilliseconds}ms');
+}

example/pubspec.yaml

@@ -0,0 +1,9 @@
+name: example
+description: Example showcase of memoize library
+
+environment:
+  sdk: '>=2.0.0-dev <3.0.0'
+
+dependencies:
+  memoize:
+    path: ../

lib/memoize.dart

@@ -1,3 +1,5 @@
+import 'dart:collection';
+
 import 'function_defs.dart';
 
 /// Lazy evaluates function and returns cached result on each call.
@@ -20,19 +22,14 @@ Func0<R> memo0<R>(Func0<R> func) {
 /// Checks 1 argument for equality with [==] operator and returns the cached
 /// result if it was not changed.
 Func1<A, R> memo1<A, R>(Func1<A, R> func) {
-  A prevArg;
-  R prevResult;
-  bool isInitial = true;
+  HashMap<A, R> argsToOutput = HashMap();
 
   return ((A arg) {
-    if (!isInitial && arg == prevArg) {
-      return prevResult;
+    if (argsToOutput.containsKey(arg)) {
+      return argsToOutput[arg];
     } else {
-      prevArg = arg;
-      prevResult = func(arg);
-      isInitial = false;
-
-      return prevResult;
+      argsToOutput[arg] = func(arg);
+      return argsToOutput[arg];
     }
   });
 }