Author: srowen
Date: Thu Apr 16 09:11:46 2009
New Revision: 765527

URL: http://svn.apache.org/viewvc?rev=765527&view=rev
Log:
Partial fix for last regression involving change to FileDataModel

Modified:
    
lucene/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/impl/model/BooleanPrefUser.java

Modified: 
lucene/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/impl/model/BooleanPrefUser.java
URL: 
http://svn.apache.org/viewvc/lucene/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/impl/model/BooleanPrefUser.java?rev=765527&r1=765526&r2=765527&view=diff
==============================================================================
--- 
lucene/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/impl/model/BooleanPrefUser.java
 (original)
+++ 
lucene/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/impl/model/BooleanPrefUser.java
 Thu Apr 16 09:11:46 2009
@@ -20,6 +20,7 @@
 import org.apache.mahout.cf.taste.model.Preference;
 import org.apache.mahout.cf.taste.model.User;
 import org.apache.mahout.cf.taste.impl.common.FastSet;
+import org.apache.mahout.cf.taste.impl.common.ArrayIterator;
 
 import java.io.Serializable;
 
@@ -47,24 +48,27 @@
 
   @Override
   public Preference getPreferenceFor(Object itemID) {
-    return itemIDs.contains(itemID) ?
-        new GenericPreference(this, new 
GenericItem<String>(itemID.toString()), 1.0) : null;
+    return itemIDs.contains(itemID) ? buildPreference(itemID) : null;
   }
 
-  /**
-   * @throws UnsupportedOperationException
-   */
   @Override
   public Iterable<Preference> getPreferences() {
-    throw new UnsupportedOperationException();
+    return new ArrayIterator<Preference>(getPreferencesAsArray());
   }
 
-  /**
-   * @throws UnsupportedOperationException
-   */
   @Override
   public Preference[] getPreferencesAsArray() {
-    throw new UnsupportedOperationException();
+    Preference[] result = new Preference[itemIDs.size()];
+    int i = 0;
+    for (Object itemID : itemIDs) {
+      result[i] = buildPreference(itemID);
+      i++;
+    }
+    return result;
+  }
+
+  private Preference buildPreference(Object itemID) {
+    return new BooleanPreference(this, new 
GenericItem<String>(itemID.toString()));
   }
 
   /**


Reply via email to