[jboss-svn-commits] JBL Code SVN: r33040 - labs/jbossrules/trunk/drools-planner/drools-planner-examples/src/main/java/org/drools/planner/examples/nurserostering/domain.

[jboss-svn-commits at lists.jboss.org]

Author: ge0ffrey
Date: 2010-05-24 09:10:09 -0400 (Mon, 24 May 2010)
New Revision: 33040

Modified:
   labs/jbossrules/trunk/drools-planner/drools-planner-examples/src/main/java/org/drools/planner/examples/nurserostering/domain/WeekendDefinition.java
Log:
dayOfWeekSet

Modified: labs/jbossrules/trunk/drools-planner/drools-planner-examples/src/main/java/org/drools/planner/examples/nurserostering/domain/WeekendDefinition.java
===================================================================
--- labs/jbossrules/trunk/drools-planner/drools-planner-examples/src/main/java/org/drools/planner/examples/nurserostering/domain/WeekendDefinition.java	2010-05-24 13:04:33 UTC (rev 33039)
+++ labs/jbossrules/trunk/drools-planner/drools-planner-examples/src/main/java/org/drools/planner/examples/nurserostering/domain/WeekendDefinition.java	2010-05-24 13:10:09 UTC (rev 33040)
@@ -1,14 +1,21 @@
 package org.drools.planner.examples.nurserostering.domain;
 
+import java.util.EnumSet;
+
 /**
  * @author Geoffrey De Smet
  */
 public enum WeekendDefinition {
-    SATURDAY_SUNDAY("SaturdaySunday"),
-    FRIDAY_SATURDAY_SUNDAY("FridaySaturdaySunday"),
-    FRIDAY_SATURDAY_SUNDAY_MONDAY("FridaySaturdaySundayMonday"),
-    SATURDAY_SUNDAY_MONDAY("SaturdaySundayMonday");
+    SATURDAY_SUNDAY("SaturdaySunday",
+            EnumSet.of(DayOfWeek.SATURDAY, DayOfWeek.SUNDAY)),
+    FRIDAY_SATURDAY_SUNDAY("FridaySaturdaySunday",
+            EnumSet.of(DayOfWeek.FRIDAY, DayOfWeek.SATURDAY, DayOfWeek.SUNDAY)),
+    FRIDAY_SATURDAY_SUNDAY_MONDAY("FridaySaturdaySundayMonday",
+            EnumSet.of(DayOfWeek.FRIDAY, DayOfWeek.SATURDAY, DayOfWeek.SUNDAY, DayOfWeek.MONDAY)),
+    SATURDAY_SUNDAY_MONDAY("SaturdaySundayMonday",
+            EnumSet.of(DayOfWeek.SATURDAY, DayOfWeek.SUNDAY, DayOfWeek.MONDAY));
 
+    private EnumSet<DayOfWeek> dayOfWeekSet;
 
     public static WeekendDefinition valueOfCode(String code) {
         for (WeekendDefinition weekendDefinition : values()) {
@@ -21,15 +28,21 @@
 
     private String code;
 
-    private WeekendDefinition(String code) {
+    private WeekendDefinition(String code, EnumSet<DayOfWeek> dayOfWeekSet) {
         this.code = code;
+        this.dayOfWeekSet = dayOfWeekSet;
     }
 
     public String getCode() {
         return code;
     }
 
+    private boolean isWeekend(DayOfWeek dayOfWeek) {
+        return dayOfWeekSet.contains(dayOfWeek);
+    }
+
     public String toString() {
         return code;
     }
+
 }