unit Main;

{$mode objfpc}{$H+}

interface

uses
  Classes, SysUtils, BufDataset, db, Forms, Controls, Graphics, Dialogs,
  DBGrids, DbCtrls, StdCtrls, ExtCtrls, ComCtrls,
  JvDBTreeView;

type

  { TForm1 }

  TForm1 = class(TForm)
    Bevel1: TBevel;
    BufDataset1: TBufDataset;
    Button1: TButton;
    Button2: TButton;
    DataSource1: TDataSource;
    DBGrid1: TDBGrid;
    ImageList1: TImageList;
    JvDBTreeView1: TJvDBTreeView;
    Panel1: TPanel;
    StatusBar1: TStatusBar;
    procedure Button1Click(Sender: TObject);
    procedure Button2Click(Sender: TObject);
    procedure FormCreate(Sender: TObject);
    procedure JvDBTreeView1CustomDrawItem(Sender: TCustomTreeView;
      Node: TTreeNode; State: TCustomDrawState; var DefaultDraw: Boolean);
    procedure JvDBTreeView1GetSelectedIndex(Sender: TObject; Node: TTreeNode);
  private

  public

  end;

var
  Form1: TForm1;

implementation

{$R *.lfm}

const
  ICON_MALE = 0;
  ICON_FEMALE = 1;
  ICON_DE = 2;
  ICON_UK = 3;
  ICON_ES = 4;
  ICON_FR = 5;
  ICON_IT = 6;
  ICON_LANDSCAPE = 7;
  ICON_CITY = 8;
  ICON_PEOPLE = 9;

{ TForm1 }

procedure TForm1.Button1Click(Sender: TObject);
begin
  JvDBTreeView1.FullExpand;
end;

procedure TForm1.Button2Click(Sender: TObject);
begin
  JvDBTreeView1.FullCollapse;
end;

{ Assignment of table fields to tree

   Table                 Tree
   ID         -->    Tree.MasterField
   ParentID   -->    Tree.DetailField
   Name       -->    Tree.itemField
   Icon       -->    Tree.IconField
}
procedure TForm1.FormCreate(Sender: TObject);

  procedure AddRecord(ID, ParentID: Integer; AName: String; AIcon: Integer = -1);
  begin
    //with Dbf1 do begin
    with BufDataset1 do begin
      Insert;
      Fields[0].AsInteger := ID;
      Fields[1].AsInteger := ParentID;
      Fields[2].AsString := AName;
      Fields[3].AsInteger := AIcon;
      Post;
    end;
  end;

begin
  BufDataset1.Filename := Application.Location + 'JvDBGridDemoData.dat';

  if not FileExists(BufDataset1.FileName) then begin
    if BufDataset1.FieldDefs.IndexOf('ID') = -1 then;
      BufDataset1.FieldDefs.Add('ID', ftInteger);
    if BufDataset1.FieldDefs.IndexOf('ParentID') = -1 then
      BufDataset1.FieldDefs.Add('ParentID', ftInteger);
    if BufDataset1.FieldDefs.IndexOf('Name') = -1 then
      BufDataset1.FieldDefs.Add('Name', ftString, 20);
    if BufDataset1.FieldDefs.IndexOf('Icon') = -1 then
      BufDataset1.FieldDefs.Add('Icon', ftInteger);
    BufDataset1.CreateDataset;

    BufDataset1.Open;

    AddRecord( 1, 0, 'Politicians', ICON_PEOPLE);
    AddRecord( 2, 0, 'Cities', ICON_CITY);
    AddRecord( 3, 0, 'Rivers', ICON_LANDSCAPE);
    AddRecord( 4, 1, 'France', ICON_FR);
    AddRecord( 5, 1, 'Germany', ICON_DE);
    AddRecord( 6, 1, 'Great Britain', ICON_UK);
    AddRecord( 7, 1, 'Italy', ICON_IT);
    AddRecord( 8, 4, 'Charles de Gaulles', ICON_MALE);
    AddRecord( 9, 4, 'Emmanuel Macron', ICON_MALE);
    AddRecord(10, 4, 'François Mitterrand', ICON_MALE);
    AddRecord(11, 5, 'Angela Merkel', ICON_FEMALE);
    AddRecord(12, 6, 'Tony Blair', ICON_MALE);
    AddRecord(13, 6, 'Theresa May', ICON_FEMALE);
    AddRecord(14, 5, 'Konrad Adenauer', ICON_MALE);
    AddRecord(15, 5, 'Willy Brandt', ICON_MALE);
    AddRecord(16, 7, 'Matteo Renzi', ICON_MALE);
    AddRecord(20, 3, 'France');
    AddRecord(21, 20, 'Seine');
    AddRecord(22, 20, 'Rhône');
    AddRecord(23, 3, 'England');
    AddRecord(24, 23, 'Thames');
//    AddRecord(25, 2, 'France');
    AddRecord(26, 2, 'Paris', ICON_FR);
    AddRecord(27, 2, 'Marseilles', ICON_FR);
    AddRecord(29, 2, 'London', ICON_UK);
    AddRecord(30, 2, 'Oxford', ICON_UK);
    AddRecord(31, 2, 'Lyon', ICON_FR);
    AddRecord(33, 2, 'Berlin', ICON_DE);
    AddRecord(34, 2, 'Hamburg', ICON_DE);
    AddRecord(35, 2, 'Munich', ICON_DE);
    AddRecord(36, 2, 'Frankfurt', ICON_DE);
    AddRecord(38, 2, 'Rome', ICON_IT);
    AddRecord(39, 2, 'Venice', ICON_IT);
    AddRecord(40, 2, 'Madrid', ICON_ES);
    AddRecord(41, 2, 'Barcelona', ICON_ES);

    BufDataset1.SaveToFile;
    BufDataset1.Close;
  end;

  BufDataset1.Open;
  BufDataset1.IndexFieldNames := 'ParentID;Name';
end;

procedure TForm1.JvDBTreeView1CustomDrawItem(Sender: TCustomTreeView;
  Node: TTreeNode; State: TCustomDrawState; var DefaultDraw: Boolean);
begin
  if Node.Level = 0 then
    Sender.Canvas.Font.Style := [fsBold]
  else
    Sender.Canvas.Font.Style := [];
end;

procedure TForm1.JvDBTreeView1GetSelectedIndex(Sender: TObject; Node: TTreeNode
  );
begin
  Node.SelectedIndex := Node.ImageIndex;
end;

end.