nHydrate.Dsl.nHydrateDiagram.CreateChildShape C# (CSharp) Method

        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;
        }