protected override ShapeElement CreateChildShape(Microsoft.VisualStudio.Modeling.ModelElement element)
{
var shape = base.CreateChildShape(element);
var model = this.ModelElement as nHydrate.Dsl.nHydrateModel;
if (model != null)
{
if (shape is StoredProcedureShape)
{
if ((model.DiagramVisibility & VisibilityTypeConstants.StoredProcedure) == VisibilityTypeConstants.StoredProcedure) shape.Show();
else shape.Hide();
}
else if (shape is ViewShape)
{
if ((model.DiagramVisibility & VisibilityTypeConstants.View) == VisibilityTypeConstants.View) shape.Show();
else shape.Hide();
}
else if (shape is FunctionShape)
{
if ((model.DiagramVisibility & VisibilityTypeConstants.Function) == VisibilityTypeConstants.Function) shape.Show();
else shape.Hide();
}
}
return shape;
}