/*****************************************************************************
Sets up a column to be drawn with + - tree controls w/ indenting. Text
is automatically indented. Rows expand and collapse as appropriate
3/2001: Tree can be in fixed column, too
*****************************************************************************/
BOOL CTreeColumn::TreeSetup( CGridCtrl* apGridCtrl, // tree acts on a column in this grid
int aiColumn, // which column has tree
int aiTotalRows, // total number of rows in the
// tree if totally expanded
int aiFixedRowCount,// Set fixed row count now, too
// Grid total rows=aiTotalRows+aiFixedRowCount
const unsigned char* apucTreeLevelAry, // Tree Level data array --
// must have aiTotalRows of entries
BOOL abShowTreeLines, // T=show tree (not grid) lines; F=no tree lines
BOOL abUseImages, // T=use 1st 3 images from already set image list
// to display folder graphics
CRuntimeClass* apRuntimeClass) // can use your special
// CGridTreeCellBase-derived class
// returns: success / fail
{
ASSERT( apGridCtrl != NULL);
ASSERT( aiColumn < apGridCtrl->GetColumnCount());
m_pRuntimeClassTreeCell = apRuntimeClass;
if( m_pRuntimeClassTreeCell == NULL) // default can hold data
m_pRuntimeClassTreeCell = RUNTIME_CLASS(CGridTreeCell);
if (!m_pRuntimeClassTreeCell->IsDerivedFrom(RUNTIME_CLASS(CGridTreeCellBase)))
{
ASSERT( FALSE);
return FALSE;
}
m_pGrid = apGridCtrl;
m_bAllowDraw = FALSE; // prevents crash during reset
BOOL bRC = TRUE;
m_iFixedRowCount = aiFixedRowCount;
m_iRowCount = aiTotalRows + aiFixedRowCount;
m_iColumnWithTree = aiColumn;
m_pGrid->SetFixedRowCount( m_iFixedRowCount);
m_pGrid->SetRowCount( m_iRowCount);
if( aiTotalRows > 0)
{
ASSERT( apucTreeLevelAry != NULL);
// retain old cell properties: establishes size for
// tree drawing box based on current font
CGridTreeCell GridCellCopy;
GridCellCopy.SetTreeColumn( this);
CGridCellBase* pCurrCell = m_pGrid->GetCell( 0, m_iColumnWithTree);
if (pCurrCell)
GridCellCopy = *pCurrCell;
// copy all data while replacing cells to tree cell type
for( int i1=0; i1 < aiTotalRows; i1++)
{
int iCellRow = m_iFixedRowCount + i1;
if( !m_pGrid->SetCellType( iCellRow,
m_iColumnWithTree,
m_pRuntimeClassTreeCell ) )
{
bRC = FALSE;
break;
}
CGridTreeCell* pGridTreeCell = (CGridTreeCell*)m_pGrid->GetCell( iCellRow, m_iColumnWithTree);
if( pGridTreeCell != NULL)
{
pGridTreeCell->SetTreeColumn( this);
pGridTreeCell->SetLevelAndHide( *apucTreeLevelAry );
}
apucTreeLevelAry++;
}
SetTreeUsesImages( abUseImages); // T=use images
SetTreeLines( abShowTreeLines); // T=show tree lines
}
m_bAllowDraw = TRUE;
m_pGrid->Invalidate();
return bRC;
}
/*****************************************************************************
3/2001: Allows one to add to the existing tree structure. Repeatedly
calling TreeSetup() with different tree configurations was the only way to
change the tree structure with the original implementation, but had the
disadvantage of destroying cell contents. The original implementation
forced you to NOT store data directly in grid cells if you wished to
change the tree.
No validation of the tree branch "levels" done. Drawing code assumes
a parent / child relationship of tree nodes. Thus, if the lines of
the tree don't look right, make sure that the parent / child relationship
is not violated.
*****************************************************************************/
int CTreeColumn::InsertTreeBranch(
const unsigned char* apucTreeLevelAry, // Tree Level data array --
// must have aiNbrElements of entries
int aiNbrElements, // NUmber of tree elements to add
int aiRow, // insert tree elements beginning at this row
// If -1, append to end; If 0, Insert
// at top no matter how many fixed rows
BOOL abRedraw) // T=redraw; F=don't
// returns: row where elements inserted; -1 on error
{
ASSERT( apucTreeLevelAry != NULL);
ASSERT( aiNbrElements >= 0);
ASSERT( aiRow >= -1);
ASSERT( m_pGrid != NULL); // Must call TreeSetup(), first
ASSERT( m_iFixedRowCount >= 0);
ASSERT( m_pRuntimeClassTreeCell != NULL);
if( aiNbrElements <= 0)
return -1;
// if user specified 0 or -1, adjust
int iRowForAppend = 0;
if( m_iRowCount <= m_iFixedRowCount)
aiRow = -1; // if no non-fixed rows, then gotta append
if( aiRow <= -1)
{
iRowForAppend = -1;
aiRow = m_iRowCount;
}
else if( aiRow < m_iFixedRowCount)
aiRow = m_iFixedRowCount;
m_bAllowDraw = FALSE; // prevents crash during reset
// retain old cell properties: establishes size for
// tree drawing box based on current font
CGridTreeCell GridCellCopy;
GridCellCopy.SetTreeColumn( this);
CGridCellBase* pCurrCell = m_pGrid->GetCell( m_iRowCount-1, m_iColumnWithTree);
if (pCurrCell)
GridCellCopy = *pCurrCell;
// insert rows while replacing tree column cells to tree cell type
int iCellRow = aiRow;
int i1;
for( i1=0; i1 < aiNbrElements; i1++)
{
// CGridCtrl's InsertRow() requires a -1 to append values to
// the end of the grid
if( iRowForAppend != -1)
iRowForAppend = iCellRow;
if( m_pGrid->InsertRow( "", iRowForAppend) < 0)
{
aiRow = -1; // error
break;
}
if( !m_pGrid->SetCellType( iCellRow,
m_iColumnWithTree,
m_pRuntimeClassTreeCell ) )
{
aiRow = -1; // error
break;
}
CGridTreeCell* pGridTreeCell = (CGridTreeCell*)m_pGrid->GetCell( iCellRow, m_iColumnWithTree);
if( pGridTreeCell != NULL)
{
pGridTreeCell->SetTreeColumn( this);
pGridTreeCell->SetLevelAndHide( *apucTreeLevelAry );
pGridTreeCell->SetViewable( TRUE);
}
iCellRow++;
m_iRowCount++;
apucTreeLevelAry++;
}
ASSERT( m_iRowCount == m_pGrid->GetRowCount() );
// have to re-number all cells below insertion point
if( aiRow > 0)
{
int iRow;
int iColumnCount = m_pGrid->GetColumnCount();
for( iRow=iCellRow; iRow < m_iRowCount; iRow++)
{
int iCol;
for (iCol = 0; iCol < iColumnCount; iCol++)
{
CGridCellBase* pGridCellBase = m_pGrid->GetCell( iRow, iCol);
if( pGridCellBase != NULL)
{
pGridCellBase->SetCoords( iRow, iCol);
}
}
}
}
m_bAllowDraw = TRUE;
TreeRefreshRows();
if( abRedraw)
m_pGrid->Invalidate();
return aiRow;
}
