DiagramTraitButton.java

package jetbrains.mps.lang.editor.diagram.runtime.jetpad.palette.impl;

/*Generated by MPS */

import jetbrains.mps.lang.editor.diagram.runtime.jetpad.palette.openapi.PaletteSimpleAction;
import jetbrains.mps.nodeEditor.cells.jetpad.DiagramCell;
import javax.swing.Icon;
import com.intellij.icons.AllIcons;

public class DiagramTraitButton implements PaletteSimpleAction {
  private DiagramCell myDiagramCell;
  public DiagramTraitButton(DiagramCell cell) {
    myDiagramCell = cell;
  }
  public void onClick() {
    myDiagramCell.setExternalTrait(null);
  }
  public Icon getIcon() {
    return AllIcons.Actions.Close;
  }
  public String getText() {
    return "Set diagram initial trait";
  }
}