1 files changed, 65 insertions, 0 deletions
diff --git a/plugins/org.eclipse.emf.cdo.dawn.examples.acore.graphiti/src/org/eclipse/emf/cdo/dawn/examples/acore/graphiti/features/AcoreAddAClassFeature.java b/plugins/org.eclipse.emf.cdo.dawn.examples.acore.graphiti/src/org/eclipse/emf/cdo/dawn/examples/acore/graphiti/features/AcoreAddAClassFeature.java
new file mode 100644
index 0000000000..c5c114281b
--- /dev/null
+++ b/plugins/org.eclipse.emf.cdo.dawn.examples.acore.graphiti/src/org/eclipse/emf/cdo/dawn/examples/acore/graphiti/features/AcoreAddAClassFeature.java
@@ -0,0 +1,65 @@
+/*
+ * Copyright (c) 2004 - 2011 Eike Stepper (Berlin, Germany) and others.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *  
+ * Contributors:
+ *     Martin Fluegge - initial API and implementation
+ * 
+ */
+package org.eclipse.emf.cdo.dawn.examples.acore.graphiti.features;
+
+import org.eclipse.emf.cdo.dawn.examples.acore.AClass;
+
+import org.eclipse.graphiti.features.IFeatureProvider;
+import org.eclipse.graphiti.features.context.IAddContext;
+import org.eclipse.graphiti.mm.algorithms.styles.Color;
+import org.eclipse.graphiti.mm.pictograms.Diagram;
+import org.eclipse.graphiti.util.ColorConstant;
+import org.eclipse.graphiti.util.IColorConstant;
+
+/**
+ * @author Martin Fluegge
+ */
+public class AcoreAddAClassFeature extends AcoreBasicAddElementFeature
+{
+  private static final IColorConstant CLASS_TEXT_FOREGROUND = new ColorConstant(51, 51, 153);
+
+  private static final IColorConstant CLASS_FOREGROUND = new ColorConstant(150, 150, 255);
+
+  private static final IColorConstant CLASS_BACKGROUND = new ColorConstant(230, 230, 255);
+
+  public AcoreAddAClassFeature(IFeatureProvider fp)
+  {
+    super(fp);
+  }
+
+  @Override
+  public boolean canAdd(IAddContext context)
+  {
+    // check if user wants to add a AClass
+    if (context.getNewObject() instanceof AClass)
+    {
+      // check if user wants to add to a diagram
+      if (context.getTargetContainer() instanceof Diagram)
+      {
+        return true;
+      }
+    }
+    return false;
+  }
+
+  @Override
+  protected Color getBackgroundColor()
+  {
+    return manageColor(CLASS_BACKGROUND);
+  }
+
+  @Override
+  protected Color getForegroundColor()
+  {
+    return manageColor(CLASS_FOREGROUND);
+  }
+}