Custom Drawing

procedure TCustomDrawForm.Tree1BeforeDraw(Sender: TObject);
var
  t: Integer;
begin
  // Draw custom background graphics before nodes are rendered
  with Tree1.Canvas do
  begin
    Pen.Style := psSolid;
    Pen.Color := clRed;
    Pen.Width := 2;

    // Draw random lines from bottom-right corner
    for t := 1 to 10 do
      LineWithZ(Tree1.Width, Tree1.Height,
                Tree1.Width - (t * Random(40)),
                Tree1.Height - (t * Random(30)), 0);
  end;
end;

procedure TCustomDrawForm.Tree1AfterDraw(Sender: TObject);
var
  t: Integer;
begin
  // Draw custom graphics after nodes are rendered
  with Tree1.Canvas do
  begin
    Pen.Style := psSolid;
    Pen.Color := clBlack;
    Pen.Width := 2;

    // Draw random lines from top-left corner
    for t := 1 to 10 do
      LineWithZ(0, 0, t * Random(40), t * Random(30), 0);
  end;
end;

type
  // Global formatting options for custom shapes
  TSpecialShapeFormat = class
  public
    MaxLevel: Integer;
    VertOffset: Integer;
    Brush: TTeeBrush;
    Border: TTeePen;

    constructor Create;
    destructor Destroy; override;
  end;

  // Custom shape with special drawing behavior
  TSpecialShape = class(TTreeNodeShape)
  public
    constructor Create(AOwner: TComponent); override;
    procedure Draw; override;
    class function Format: TSpecialShapeFormat;
  end;

constructor TSpecialShape.Create(AOwner: TComponent);
begin
  inherited;
  Transparent := True;
  Border.Visible := False;
end;

// Custom drawing logic
procedure TSpecialShape.Draw;
var
  R: TRect;
begin
  // Only draw for nodes at specified level
  if Level <= Format.MaxLevel then
  with Tree.Canvas do
  begin
    AssignBrush(Format.Brush);
    AssignVisiblePen(Format.Border);

    R := Bounds;
    // Draw a full-width rectangle behind the node
    Rectangle(0, R.Top, Tree.ClientWidth, R.Bottom + Format.VertOffset);
  end;

  // Draw node text, image, etc. as usual
  inherited;
end;

procedure TCustomShapeForm.FormCreate(Sender: TObject);
begin
  // Set custom shape as default for new nodes
  Tree1.GlobalFormat.NodeClass := TSpecialShape;

  // Add sample nodes (will use TSpecialShape)
  Tree1.Add('Africa').Add('Kenya');
  Tree1.Add('America').Add('USA').Add('Los Angeles');
  Tree1.Add('Asia').Add('Japan').Add('Tokyo');

  // Explicitly create a node with custom shape class
  Tree1.AddShapeClass(0, 0, 'Universe', nil, TSpecialShape);
end;

uses TeeEdiGrad;

procedure TNodeGradientForm.FormCreate(Sender: TObject);
begin
  // Show selected nodes with their original gradient
  Tree1.Selected.Color := clNone;
end;

procedure TNodeGradientForm.Button1Click(Sender: TObject);
begin
  // Edit gradient for selected node
  if Tree1.Selected.Count > 0 then
    TTeeGradientEditor.Edit(Self, Tree1.Selected.First.Gradient);
end;

procedure TNodeGradientForm.CheckBox1Click(Sender: TObject);
var
  t: Integer;
begin
  // Enable/disable gradient clipping for all nodes
  for t := 0 to Tree1.Shapes.Count - 1 do
    Tree1[t].GradientClip := CheckBox1.Checked;
end;

type
  TCustomPolygonShape = class(TTreeCustomPolygonShape)
  protected
    function GetShapePoints(const R: TRect; var P: TShapePoints): Integer; override;
  end;

function TCustomPolygonShape.GetShapePoints(const R: TRect;
  var P: TShapePoints): Integer;
begin
  // Define custom polygon points
  SetLength(P, 5);
  P[0] := Point(R.Left, R.Top + (R.Bottom - R.Top) div 2);
  P[1] := Point(R.Left + (R.Right - R.Left) div 2, R.Top);
  P[2] := Point(R.Right, R.Top + (R.Bottom - R.Top) div 2);
  P[3] := Point(R.Left + (R.Right - R.Left) div 2, R.Bottom);
  P[4] := P[0]; // Close the polygon
  Result := Length(P);
end;

// Draw lines with Z-order
Tree1.Canvas.LineWithZ(X1, Y1, X2, Y2, ZDepth);

// Control node drawing order
Node1.ZOrder := 100; // Draw on top
Node2.ZOrder := 0;   // Draw behind

procedure TSpecialShape.Draw;
begin
  // Different appearance for different tree levels
  case Level of
    0: Format.Brush.Color := clLime;
    1: Format.Brush.Color := clAqua;
    2: Format.Brush.Color := clYellow;
  end;
  
  inherited;
end;

procedure DrawCustomBackground(const Canvas: TCanvas3D);
begin
  // Draw a gradient background
  Canvas.GradientFill(
    Rect(0, 0, Tree1.Width, Tree1.Height),
    clWhite, clSilver, gdVertical
  );
end;

procedure TCustomShape.Draw;
begin
  if Selected then
  begin
    // Draw glow effect for selected nodes
    Canvas.Pen.Color := clYellow;
    Canvas.Pen.Width := 3;
  end;
  
  inherited;
end;