Изучение TTreeView часть 2 стр.6

стр. 6

Изучение компонента TTreeView, часть II

Автор Билл Уитни перевод Виктор Алексеев aka prom

Модуль формы

#include <vcl.h>
#pragma hdrstop

#include "Unit1.h"
//---------------------------------------------------------------------------
#pragma package(smart_init)
#pragma resource "*.dfm"
TForm1 *Form1;
// Секция 1 ------------------------------------------------------------------------
__fastcall TForm1::TForm1(TComponent* Owner)
        : TForm(Owner)
{
}
//Секция 2---------------------------------------------------------------------------
void __fastcall TForm1::QuitButtonClick(TObject *Sender)
{
     Close();
}
// Секция 3 -------------------------------------------------------------------------
void __fastcall TForm1::LeagueButtonClick(TObject *Sender)
{
     AddLeague("(Новая лига)");
}
//Секция 4 --------------------------------------------------------------------------
void __fastcall TForm1::NewLeague1Click(TObject *Sender)
{
     LeagueButtonClick(Sender);
}

// Секция 5 ----------------------------------------------------------------
 void TForm1::AddLeague(char *name)
 {
    ActiveControl = TreeView1;
    TTreeNode *n = TreeView1->Items->Add(NULL,name);
    n->EditText();
 }

// Секция 6 -------------------------------------------------------------------
void __fastcall TForm1::TreeView1Edited(TObject *Sender, TTreeNode *Node,
      AnsiString &S)
{
      Node->Selected = true;
}
// Секция 7 -----------------------------------------------------------------------
void __fastcall TForm1::AddDivision(char* name, TTreeNode* LeagueNode)
{
  ActiveControl = TreeView1;
  TTreeNode* n = TreeView1->Items->AddChild(LeagueNode, name);
  LeagueNode->Expand(true);
  n->EditText();

}
// Секция 8 ----------------------------------------------------------
void __fastcall TForm1::DivisionButtonClick(TObject *Sender)
{
     if ((TreeView1->Selected != NULL) &&
        (TreeView1->Selected->Level == ev_league))
        {
          AddDivision("(Новый дивизион)", TreeView1->Selected);
        }
     else
        {
           Application->MessageBox("Дивизион добавляется только к лиге!",
                            "OOPS!", MB_OK);
        }
 }
// Секция 9 ------------------       -------------------------------------------
void __fastcall TForm1::Division1Click(TObject *Sender)
{
        DivisionButtonClick(Sender);
}
// Секция 10 -----------------------------------------------------------------
void __fastcall TForm1::TeamButtonClick(TObject *Sender)
{
  if  ( (TreeView1->Selected != NULL) &&
      (TreeView1->Selected->Level == ev_division) )
        {
           AddTeam("(Новая команда)", TreeView1->Selected);
        }
  else
        {
           Application->MessageBox("Команда добавляется только в дивизион!",
                       "OOPS!", MB_OK);
        }
}
// Секция 11 ------------------------------------------------------------------

void __fastcall TForm1::Team1Click(TObject *Sender)
{
   TeamButtonClick(Sender);
}
// Секция 12 -------------------------------------------------------------
 void TForm1::AddTeam(char* name, TTreeNode* divisionNode)
{
  ActiveControl = TreeView1;
  TTreeNode* n = TreeView1->Items->AddChild(divisionNode, name);
  divisionNode->Expand(true);

TreeView1->Items->AddChild(n, "Нападающий");
  TreeView1->Items->AddChild(n, "Защитник");
  TreeView1->Items->AddChild(n, "Полевой игрок");
  TreeView1->Items->AddChild(n, "Запасной игрок");
  n->EditText();
}
// Секция 13
void __fastcall TForm1::PlayerButtonClick(TObject *Sender)
{
     if  ( (TreeView1->Selected != NULL) &&
         (TreeView1->Selected->Level == ev_categories) )
           {
              AddPlayer("(Новый игрок)", TreeView1->Selected);
           }
     else
           {
              Application->MessageBox("Игрок добавляется только в категорию!",
                       "OOPS!", MB_OK);
           }
}
// Секция 14 ------------------------------------------------------------

void __fastcall TForm1::Player1Click(TObject *Sender)
{
    PlayerButtonClick(Sender);
}
// Секция 15 ------------------------------------------------------------

void TForm1:: AddPlayer(char* name, TTreeNode* categoryNode)
{
  ActiveControl = TreeView1;
  TTreeNode *n = TreeView1->Items->AddChild(categoryNode, name);
  // Add a structure to the Data (void*) 
  // attribute of this node
  PlayerStr *p = new PlayerStr;
  p->number = 20;
  p->age = 31;
  p->weight = 210.0;
  p->height = 74;
  n->Data = (void*) p;
  categoryNode->Expand(true);
  n->EditText();
 }
 // Секция 16 -----------------------------------------------
void __fastcall TForm1::DeleteButtonClick(TObject *Sender)
{
  // Delete a selected node.
  ActiveControl = TreeView1;
  if ( TreeView1->Selected == NULL )
       {
        Application->MessageBox("Вы должны выделить удаляемый узел",
                            "OOPS!", MB_OK);
       }
  else
       {
    switch ( TreeView1->Selected->Level )
           {
             case ev_categories :
             Application->MessageBox("Вы не можете удалить категорию!",

"OOPS!", MB_OK);
             break;
             default :
             TreeView1->Items->Delete(TreeView1->Selected);
           }
       }
}
// Секция 17---------------------------------------------------------------------------
// Удаление через всплывающее меню -----------------------------------
void __fastcall TForm1::Delete1Click(TObject *Sender)
{
      DeleteButtonClick(Sender);
}
// Секция 18--------------------------------------------------------------------
void __fastcall TForm1::TreeView1DragOver(TObject *Sender, TObject *Source,
      int X, int Y, TDragState State, bool &Accept)
{
     if ( (Source->InheritsFrom(__classid(TShape))) ||
       (Source->InheritsFrom(__classid(TTreeView))))
     Accept = true;
   else
     Accept = false;
}
// Секция 19 -------------------------------------------------------------------
void __fastcall TForm1::TreeView1DragDrop(TObject *Sender, TObject *Source,
      int X, int Y)
{
   if (Source->InheritsFrom(__classid(TShape)))
  {
    TTreeNode *target = TreeView1->GetNodeAt(X, Y);
    if ( Source == LeagueShape)
    {
      if ( target == NULL )
        AddLeague("(Новая лига)");
      else
        Application->MessageBox(
        "Лига может быть перетащена только корень", "OOPS!", MB_OK);
    }
    else if ( Source == DivisionShape )
    {
      if ( (target != NULL) && (target->Level == ev_league) )
        AddDivision("(Новый дивизион)", target);
      else
        Application->MessageBox(
        "Дивизион может быть перетащен только в лигу","OOPS!", MB_OK);
    }
    else if ( Source == TeamShape )
    {
      if ( (target != NULL) && (target->Level == ev_division) )

AddTeam("(Новая команда)", target);
      else
        Application->MessageBox(
        "Команды могут быть перетащены только в дивизионы", "OOPS!", MB_OK);
    }
    else if ( Source == PlayerShape )
    {
      if ( (target != NULL) && (target->Level == ev_categories) )
        AddPlayer("(Новый игрок)", target);
      else
        Application->MessageBox(
        "Игроки могут быть перетащены только в команды", "OOPS!", MB_OK);
    }
  }
  else if (Source->InheritsFrom(__classid(TTreeView)))
  {
    TTreeNode *target = TreeView1->GetNodeAt(X, Y);
    TTreeNode *source = (TTreeNode*) TreeView1->Selected;

if ( source->Level != ev_player )
    {
      Application->MessageBox(
     "Только игрок может быть подцеплен и перетащен","OOPS!", MB_OK);
    }
    else
    {
      if ( target->Level == ev_categories )
      {
        if ( (target != NULL) )
          MovePlayer(source, target);
      }
      else
      {
        Application->MessageBox(
        "Игрок может быть перемещен только в категорию",
        "OOPS!", MB_OK);
      }
    }
  }
}
// Секция 20 ----------------------------------------------------------------
  void TForm1::MovePlayer(TTreeNode* player,TTreeNode* newTeam)
{
  TTreeNode* n = TreeView1->Items->AddChild(newTeam, "x");
  n->Assign(player);
  player->Delete();
}
// Секция 21 ----------------------------------------------------------------
void __fastcall TForm1::SaveClick(TObject *Sender)
{
  ofstream   nodeFile;
  TTreeNode* curNode;
  SaveNode   savNode;
  PlayerStr  plyStruct;

// nodeFile.open("C:\saved.ttv", ios::binary | ios::trunc);
  nodeFile.open("C:\saved.txt");
  for ( int a = 0; a < TreeView1->Items->Count; a++ )
  {
    curNode = TreeView1->Items->Item[a];
    strcpy(savNode.name, curNode->Text.c_str());
    savNode.type = curNode->Level;
    if ( curNode->Level == ev_player )
      savNode.hasStruct = 1;
    else
      savNode.hasStruct = 0;
    nodeFile.write((unsigned char*)&savNode, sizeof(SaveNode));
    if ( curNode->Level == ev_player )
    {

memcpy(&plyStruct, (PlayerStr*)curNode->Data, 
             sizeof(PlayerStr));
      nodeFile.write((unsigned char*) 
                      &plyStruct, sizeof(PlayerStr));
    }
  }
}
// Секция 22---------------------------------------------------------------
void __fastcall TForm1::LoadClick(TObject *Sender)
{
  ifstream   nodeFile;
  TTreeNode* curNode;
  TTreeNode* nxtNode;
  SaveNode   savNode;
  PlayerStr  plyStruct;
  PlayerStr* pStr;
  entryValue curLvl;

//  nodeFile.open("C:\saved.ttv", ios::binary);
  nodeFile.open("C:\saved.txt");
  curNode = (TTreeNode*) NULL;
  while (nodeFile.read((unsigned char*) 
         &savNode, sizeof(SaveNode)))
  {
    curLvl = savNode.type;
    if ( curNode == NULL )
    {
      curNode = TreeView1->Items->AddChild(NULL, savNode.name);
    }
    else if ( curNode->Level == curLvl )
    {
      curNode = TreeView1->Items->AddChild(curNode->Parent,
                savNode.name);
    }
    else if ( curNode->Level == (curLvl - 1) )
    {
      curNode = TreeView1->Items->AddChild(curNode,
                savNode.name);
    }
    else if ( curNode->Level > curLvl )
    {
      nxtNode = curNode->Parent;
      while ( nxtNode->Level > curLvl )
      {
        nxtNode = nxtNode->Parent;
      }
      curNode = TreeView1->Items->AddChild(nxtNode->Parent,
                savNode.name);

}
    if ( savNode.type == ev_player )
    {
      nodeFile.read((unsigned char*) 
                     &plyStruct, sizeof(PlayerStr));
      pStr = new PlayerStr;
      pStr->number = plyStruct.number;
      pStr->age = plyStruct.age;
      pStr->weight = plyStruct.weight;
      pStr->height = plyStruct.height;
      curNode->Data = (void*) pStr;
    }
  }
  TreeView1->FullExpand();
}
//---------------------------------------------------------------------------
void __fastcall TForm1::ClearClick(TObject *Sender)
{
      TreeView1->Items->Clear();
     
}

Ниже по ссылке можно скачать полностью файл с проектом:

Download: TTreeView_sample_Part_II.rar 473 kb

А т.к. не у всех установлен движок билдера, то по ссылке можно скачать исполняемый файл.

Download: исполняемый файл TTreeView_sample_PartII.exe 816 kb.