Fix conditional style management for custom widgets.

Change-Id: I252869f88a7ae149cf3c6a62934c9b93655ad87f

2 files changed, 12 insertions, 0 deletions

diff --git a/plugins/org.eclipse.eef/src/org/eclipse/eef/util/EEFConditionalStyleToWidgetStyleSwitch.java b/plugins/org.eclipse.eef/src/org/eclipse/eef/util/EEFConditionalStyleToWidgetStyleSwitch.java
index 411768586..a6b195c89 100644
--- a/plugins/org.eclipse.eef/src/org/eclipse/eef/util/EEFConditionalStyleToWidgetStyleSwitch.java
+++ b/plugins/org.eclipse.eef/src/org/eclipse/eef/util/EEFConditionalStyleToWidgetStyleSwitch.java
@@ -13,6 +13,7 @@ package org.eclipse.eef.util;
 
 import org.eclipse.eef.EEFButtonConditionalStyle;
 import org.eclipse.eef.EEFCheckboxConditionalStyle;
+import org.eclipse.eef.EEFCustomWidgetConditionalStyle;
 import org.eclipse.eef.EEFHyperlinkConditionalStyle;
 import org.eclipse.eef.EEFLabelConditionalStyle;
 import org.eclipse.eef.EEFListConditionalStyle;
@@ -106,4 +107,9 @@ public class EEFConditionalStyleToWidgetStyleSwitch extends EefSwitch<EEFWidgetS
 	public EEFWidgetStyle caseEEFListConditionalStyle(EEFListConditionalStyle object) {
 		return object.getStyle();
 	}
+
+	@Override
+	public EEFWidgetStyle caseEEFCustomWidgetConditionalStyle(EEFCustomWidgetConditionalStyle object) {
+		return object.getStyle();
+	}
 }
diff --git a/plugins/org.eclipse.eef/src/org/eclipse/eef/util/EEFDescriptionToConditionalStylesSwitch.java b/plugins/org.eclipse.eef/src/org/eclipse/eef/util/EEFDescriptionToConditionalStylesSwitch.java
index 8f2edd0c1..71e419c19 100644
--- a/plugins/org.eclipse.eef/src/org/eclipse/eef/util/EEFDescriptionToConditionalStylesSwitch.java
+++ b/plugins/org.eclipse.eef/src/org/eclipse/eef/util/EEFDescriptionToConditionalStylesSwitch.java
@@ -17,6 +17,7 @@ import java.util.List;
 import org.eclipse.eef.EEFButtonDescription;
 import org.eclipse.eef.EEFCheckboxDescription;
 import org.eclipse.eef.EEFConditionalStyle;
+import org.eclipse.eef.EEFCustomWidgetDescription;
 import org.eclipse.eef.EEFHyperlinkDescription;
 import org.eclipse.eef.EEFLabelDescription;
 import org.eclipse.eef.EEFListDescription;
@@ -134,4 +135,9 @@ public class EEFDescriptionToConditionalStylesSwitch extends EefSwitch<List<EEFC
 	public List<EEFConditionalStyle> caseEEFListDescription(EEFListDescription object) {
 		return this.asConditionalStyleList(object.getConditionalStyles());
 	}
+
+	@Override
+	public List<EEFConditionalStyle> caseEEFCustomWidgetDescription(EEFCustomWidgetDescription object) {
+		return this.asConditionalStyleList(object.getConditionalStyles());
+	}
 }