/**
  * Set the gradient data to the correct figure
  *
  * <p>{@inheritDoc}
  */
 @Override
 protected void setGradient(GradientData gradient) {
   IPapyrusNodeFigure fig = getPrimaryShape();
   FillStyle style = (FillStyle) getPrimaryView().getStyle(NotationPackage.Literals.FILL_STYLE);
   if (gradient != null) {
     fig.setIsUsingGradient(true);
     ;
     fig.setGradientData(
         style.getFillColor(), gradient.getGradientColor1(), gradient.getGradientStyle());
   } else {
     fig.setIsUsingGradient(false);
   }
 }