1 ///////////////////////////////////////////////////////////////////////////////
3 // Purpose: Displays a wxDbTable in a wxGrid.
4 // Author: Roger Gammans, Paul Gammans
8 // Copyright: (c) 1999 The Computer Surgery (roger@computer-surgery.co.uk)
9 // Licence: wxWindows licence
10 ///////////////////////////////////////////////////////////////////////////////
11 // Branched From : dbgrid.cpp,v 1.18 2000/12/19 13:00:58
12 ///////////////////////////////////////////////////////////////////////////////
15 #pragma implementation "dbgrid.h"
18 #include "wx/wxprec.h"
29 #include "wx/textctrl.h"
33 #include "wx/generic/gridctrl.h"
34 #include "wx/dbgrid.h"
37 wxDbGridCellAttrProvider::wxDbGridCellAttrProvider()
43 wxDbGridCellAttrProvider::wxDbGridCellAttrProvider(wxDbTable
*tab
, wxDbGridColInfoBase
* ColInfo
)
49 wxDbGridCellAttrProvider::~wxDbGridCellAttrProvider()
53 wxGridCellAttr
*wxDbGridCellAttrProvider::GetAttr(int row
, int col
,
54 wxGridCellAttr::wxAttrKind kind
) const
56 wxGridCellAttr
*attr
= wxGridCellAttrProvider::GetAttr(row
,col
,kind
);
58 if (m_data
&& m_ColInfo
&& (m_data
->GetNumberOfColumns() > m_ColInfo
[col
].DbCol
))
60 //FIXME: this test could.
61 // ??::InsertPending == m_data->get_ModifiedStatus()
62 // and if InsertPending use colDef[].InsertAllowed
63 if (!(m_data
->GetColDefs()[(m_ColInfo
[col
].DbCol
)].Updateable
))
67 case (wxGridCellAttr::Any
):
70 attr
= new wxGridCellAttr
;
71 // Store so we don't keep creating / deleting this...
72 wxDbGridCellAttrProvider
* self
= wxConstCast(this, wxDbGridCellAttrProvider
) ;
74 self
->SetColAttr(attr
, col
);
79 //We now must check what we were returned. and do the right thing (tm)
80 wxGridCellAttr::wxAttrKind attrkind
= attr
->GetKind();
81 if ((attrkind
== (wxGridCellAttr::Default
)) || (attrkind
== (wxGridCellAttr::Cell
)) ||
82 (attrkind
== (wxGridCellAttr::Col
)))
84 wxGridCellAttr
*attrtomerge
= attr
;
85 attr
= new wxGridCellAttr
;
86 attr
->SetKind(wxGridCellAttr::Merged
);
87 attr
->MergeWith(attrtomerge
);
89 attrtomerge
->DecRef();
94 case (wxGridCellAttr::Col
):
95 //As we must have a Coll, and were setting Coll attributes
96 // we can based on wxdbTable's so just set RO if attr valid
99 attr
= new wxGridCellAttr
;
100 wxDbGridCellAttrProvider
* self
= wxConstCast(this, wxDbGridCellAttrProvider
) ;
102 self
->SetColAttr(attr
, col
);
108 // wxGridCellAttr::Cell - Not required, will inherit on merge from row.
109 // wxGridCellAttr::Row - If wxDbtable ever supports row locking could add
110 // support to make RO on a row basis also.
111 // wxGridCellAttr::Default - Don't edit this ! or all cell with a attr will become readonly
112 // wxGridCellAttr::Merged - This should never be asked for.
121 void wxDbGridCellAttrProvider::AssignDbTable(wxDbTable
*tab
)
126 wxDbGridTableBase::wxDbGridTableBase(wxDbTable
*tab
, wxDbGridColInfo
* ColInfo
,
127 int count
, bool takeOwnership
) :
130 m_dbowner(takeOwnership
),
134 if (count
== wxUSE_QUERY
)
136 m_rowtotal
= m_data
? m_data
->Count() : 0;
142 // m_keys.Size(m_rowtotal);
146 m_nocols
= ColInfo
->Length();
147 m_ColInfo
= new wxDbGridColInfoBase
[m_nocols
];
149 wxDbGridColInfo
*ptr
= ColInfo
;
151 while (ptr
&& i
< m_nocols
)
153 m_ColInfo
[i
] = ptr
->m_data
;
160 wxLogDebug(wxT("NoCols over length after traversing %i items"),i
);
164 wxLogDebug(wxT("NoCols under length after traversing %i items"),i
);
170 wxDbGridTableBase::~wxDbGridTableBase()
172 wxDbGridCellAttrProvider
*provider
;
174 //Can't check for update here as
176 //FIXME: should i remove m_ColInfo and m_data from m_attrProvider if a wxDbGridAttrProvider
177 // if ((provider = dynamic_cast<wxDbGridCellAttrProvider *>(GetAttrProvider())))
178 // Using C casting for now until we can support dynamic_cast with wxWindows
179 provider
= (wxDbGridCellAttrProvider
*)(GetAttrProvider());
182 provider
->AssignDbTable(NULL
);
193 bool wxDbGridTableBase::CanHaveAttributes()
195 if (!GetAttrProvider())
197 // use the default attr provider by default
198 SetAttrProvider(new wxDbGridCellAttrProvider(m_data
, m_ColInfo
));
204 bool wxDbGridTableBase::AssignDbTable(wxDbTable
*tab
, int count
, bool takeOwnership
)
206 wxDbGridCellAttrProvider
*provider
;
208 //Remove Information from grid about old data
211 wxGrid
*grid
= GetView();
213 grid
->ClearSelection();
214 if (grid
->IsCellEditControlEnabled())
216 grid
->DisableCellEditControl();
218 wxGridTableMessage
msg(this, wxGRIDTABLE_NOTIFY_ROWS_DELETED
,0,m_rowtotal
);
219 grid
->ProcessTableMessage(msg
);
222 //reset our internals...
230 //FIXME: Remove dynamic_cast before sumision to wxwin
231 // if ((provider = dynamic_cast<wxDbGridCellAttrProvider *> (GetAttrProvider())))
232 // Using C casting for now until we can support dynamic_cast with wxWindows
233 provider
= (wxDbGridCellAttrProvider
*)(GetAttrProvider());
236 provider
->AssignDbTable(m_data
);
239 if (count
== wxUSE_QUERY
)
241 m_rowtotal
= m_data
? m_data
->Count() : 0;
249 //Add Information to grid about new data
252 wxGrid
* grid
= GetView();
253 wxGridTableMessage
msg(this, wxGRIDTABLE_NOTIFY_ROWS_APPENDED
, m_rowtotal
);
254 grid
->ProcessTableMessage(msg
);
257 m_dbowner
= takeOwnership
;
258 m_rowmodified
= FALSE
;
262 wxString
wxDbGridTableBase::GetTypeName(int row
, int col
)
267 if (GetNumberCols() > col
)
269 if (m_ColInfo
[col
].wxtypename
== wxGRID_VALUE_DBAUTO
)
271 if (m_data
->GetNumberOfColumns() <= m_ColInfo
[col
].DbCol
)
273 wxFAIL_MSG (_T("You can not use wxGRID_VALUE_DBAUTO for virtual columns"));
275 switch(m_data
->GetColDefs()[(m_ColInfo
[col
].DbCol
)].SqlCtype
)
278 return wxGRID_VALUE_STRING
;
281 return wxGRID_VALUE_NUMBER
;
284 return wxGRID_VALUE_NUMBER
;
287 return wxGRID_VALUE_NUMBER
;
290 return wxGRID_VALUE_NUMBER
;
293 return wxGRID_VALUE_FLOAT
;
296 return wxGRID_VALUE_FLOAT
;
299 return wxGRID_VALUE_DATETIME
;
302 return wxGRID_VALUE_DATETIME
;
304 case SQL_C_TIMESTAMP
:
305 return wxGRID_VALUE_DATETIME
;
308 return wxGRID_VALUE_STRING
;
314 return m_ColInfo
[col
].wxtypename
;
317 wxFAIL_MSG (_T("unknown column"));
321 bool wxDbGridTableBase::CanGetValueAs(int row
, int col
, const wxString
& typeName
)
323 wxLogDebug(wxT("CanGetValueAs() on %i,%i"),row
,col
);
324 //Is this needed? As it will be validated on GetValueAsXXXX
327 if (typeName
== wxGRID_VALUE_STRING
)
329 //FIXME ummm What about blob field etc.
333 if (m_data
->IsColNull(m_ColInfo
[col
].DbCol
))
338 if (m_data
->GetNumberOfColumns() <= m_ColInfo
[col
].DbCol
)
340 //If a virtual column then we can't find it's type. we have to
341 // return FALSE to get using wxVariant.
344 int sqltype
= m_data
->GetColDefs()[(m_ColInfo
[col
].DbCol
)].SqlCtype
;
346 if (typeName
== wxGRID_VALUE_DATETIME
)
348 if ((sqltype
== SQL_C_DATE
) ||
349 (sqltype
== SQL_C_TIME
) ||
350 (sqltype
== SQL_C_TIMESTAMP
))
356 if (typeName
== wxGRID_VALUE_NUMBER
)
358 if ((sqltype
== SQL_C_SSHORT
) ||
359 (sqltype
== SQL_C_USHORT
) ||
360 (sqltype
== SQL_C_SLONG
) ||
361 (sqltype
== SQL_C_ULONG
))
367 if (typeName
== wxGRID_VALUE_FLOAT
)
369 if ((sqltype
== SQL_C_SSHORT
) ||
370 (sqltype
== SQL_C_USHORT
) ||
371 (sqltype
== SQL_C_SLONG
) ||
372 (sqltype
== SQL_C_ULONG
) ||
373 (sqltype
== SQL_C_FLOAT
) ||
374 (sqltype
== SQL_C_DOUBLE
))
383 bool wxDbGridTableBase::CanSetValueAs(int row
, int col
, const wxString
& typeName
)
385 //Is this needed? As will be validated on SetValueAsXXXX
388 if (m_data
->IsColNull(m_ColInfo
[col
].DbCol
))
393 if (typeName
== wxGRID_VALUE_STRING
)
395 //FIXME ummm What about blob field etc.
399 if (!(m_data
->GetColDefs()[(m_ColInfo
[col
].DbCol
)].Updateable
))
404 if (m_data
->GetNumberOfColumns() <= m_ColInfo
[col
].DbCol
)
406 //If a virtual column then we can't find it's type. we have to faulse to
407 //get using wxVairent.
411 int sqltype
= m_data
->GetColDefs()[(m_ColInfo
[col
].DbCol
)].SqlCtype
;
412 if (typeName
== wxGRID_VALUE_DATETIME
)
414 if ((sqltype
== SQL_C_DATE
) ||
415 (sqltype
== SQL_C_TIME
) ||
416 (sqltype
== SQL_C_TIMESTAMP
))
422 if (typeName
== wxGRID_VALUE_NUMBER
)
424 if ((sqltype
== SQL_C_SSHORT
) ||
425 (sqltype
== SQL_C_USHORT
) ||
426 (sqltype
== SQL_C_SLONG
) ||
427 (sqltype
== SQL_C_ULONG
))
433 if (typeName
== wxGRID_VALUE_FLOAT
)
435 if ((sqltype
== SQL_C_SSHORT
) ||
436 (sqltype
== SQL_C_USHORT
) ||
437 (sqltype
== SQL_C_SLONG
) ||
438 (sqltype
== SQL_C_ULONG
) ||
439 (sqltype
== SQL_C_FLOAT
) ||
440 (sqltype
== SQL_C_DOUBLE
))
449 long wxDbGridTableBase::GetValueAsLong(int row
, int col
)
453 if (m_data
->GetNumberOfColumns() <= m_ColInfo
[col
].DbCol
)
455 wxFAIL_MSG (_T("You can not use GetValueAsLong for virtual columns"));
458 int sqltype
= m_data
->GetColDefs()[(m_ColInfo
[col
].DbCol
)].SqlCtype
;
459 if ((sqltype
== SQL_C_SSHORT
) ||
460 (sqltype
== SQL_C_USHORT
) ||
461 (sqltype
== SQL_C_SLONG
) ||
462 (sqltype
== SQL_C_ULONG
))
464 wxVariant val
= m_data
->GetCol(m_ColInfo
[col
].DbCol
);
465 return val
.GetLong();
467 wxFAIL_MSG (_T("unknown column, "));
471 double wxDbGridTableBase::GetValueAsDouble(int row
, int col
)
473 wxLogDebug(wxT("GetValueAsDouble() on %i,%i"),row
,col
);
476 if (m_data
->GetNumberOfColumns() <= m_ColInfo
[col
].DbCol
)
478 wxFAIL_MSG (_T("You can not use GetValueAsDouble for virtual columns"));
481 int sqltype
= m_data
->GetColDefs()[(m_ColInfo
[col
].DbCol
)].SqlCtype
;
482 if ((sqltype
== SQL_C_SSHORT
) ||
483 (sqltype
== SQL_C_USHORT
) ||
484 (sqltype
== SQL_C_SLONG
) ||
485 (sqltype
== SQL_C_ULONG
) ||
486 (sqltype
== SQL_C_FLOAT
) ||
487 (sqltype
== SQL_C_DOUBLE
))
489 wxVariant val
= m_data
->GetCol(m_ColInfo
[col
].DbCol
);
490 return val
.GetDouble();
492 wxFAIL_MSG (_T("unknown column"));
496 bool wxDbGridTableBase::GetValueAsBool(int row
, int col
)
498 wxLogDebug(wxT("GetValueAsBool() on %i,%i"),row
,col
);
501 if (m_data
->GetNumberOfColumns() <= m_ColInfo
[col
].DbCol
)
503 wxFAIL_MSG (_T("You can not use GetValueAsBool for virtual columns"));
506 int sqltype
= m_data
->GetColDefs()[(m_ColInfo
[col
].DbCol
)].SqlCtype
;
507 if ((sqltype
== SQL_C_SSHORT
) ||
508 (sqltype
== SQL_C_USHORT
) ||
509 (sqltype
== SQL_C_SLONG
) ||
510 (sqltype
== SQL_C_ULONG
))
512 wxVariant val
= m_data
->GetCol(m_ColInfo
[col
].DbCol
);
513 return val
.GetBool();
515 wxFAIL_MSG (_T("unknown column, "));
519 void* wxDbGridTableBase::GetValueAsCustom(int row
, int col
, const wxString
& typeName
)
521 wxLogDebug(wxT("GetValueAsCustom() on %i,%i"),row
,col
);
524 if (m_data
->GetNumberOfColumns() <= m_ColInfo
[col
].DbCol
)
526 wxFAIL_MSG (_T("You can not use GetValueAsCustom for virtual columns"));
529 if (m_data
->IsColNull(m_ColInfo
[col
].DbCol
))
532 if (typeName
== wxGRID_VALUE_DATETIME
)
534 wxDbColDef
*pColDefs
= m_data
->GetColDefs();
535 int sqltype
= pColDefs
[(m_ColInfo
[col
].DbCol
)].SqlCtype
;
537 if ((sqltype
== SQL_C_DATE
) ||
538 (sqltype
== SQL_C_TIME
) ||
539 (sqltype
== SQL_C_TIMESTAMP
))
541 wxVariant val
= m_data
->GetCol(m_ColInfo
[col
].DbCol
);
542 return new wxDateTime(val
.GetDateTime());
545 wxFAIL_MSG (_T("unknown column data type "));
550 void wxDbGridTableBase::SetValueAsCustom(int row
, int col
, const wxString
& typeName
, void* value
)
552 wxLogDebug(wxT("SetValueAsCustom() on %i,%i"),row
,col
);
555 if (m_data
->GetNumberOfColumns() <= m_ColInfo
[col
].DbCol
)
557 wxFAIL_MSG (_T("You can not use SetValueAsCustom for virtual columns"));
561 if (typeName
== wxGRID_VALUE_DATETIME
)
563 int sqltype
= m_data
->GetColDefs()[(m_ColInfo
[col
].DbCol
)].SqlCtype
;
564 if ((sqltype
== SQL_C_DATE
) ||
565 (sqltype
== SQL_C_TIME
) ||
566 (sqltype
== SQL_C_TIMESTAMP
))
568 //FIXME: you can't dynamic_cast from (void *)
569 //wxDateTime *date = wxDynamicCast(value, wxDateTime);
570 wxDateTime
*date
= (wxDateTime
*)value
;
573 wxFAIL_MSG (_T("Failed to convert data"));
577 m_rowmodified
= TRUE
;
578 m_data
->SetCol(m_ColInfo
[col
].DbCol
,val
);
581 wxFAIL_MSG (_T("unknown column data type"));
586 wxString
wxDbGridTableBase::GetColLabelValue(int col
)
588 if (GetNumberCols() > col
)
590 return m_ColInfo
[col
].Title
;
592 wxFAIL_MSG (_T("unknown column"));
596 bool wxDbGridTableBase::IsEmptyCell(int row
, int col
)
598 wxLogDebug(wxT("IsEmtpyCell on %i,%i"),row
,col
);
601 return m_data
->IsColNull(m_ColInfo
[col
].DbCol
);
605 wxString
wxDbGridTableBase::GetValue(int row
, int col
)
607 wxLogDebug(wxT("GetValue() on %i,%i"),row
,col
);
610 wxVariant val
= m_data
->GetCol(m_ColInfo
[col
].DbCol
);
611 wxLogDebug(wxT("\tReturning \"%s\"\n"),val
.GetString().c_str());
613 return val
.GetString();
617 void wxDbGridTableBase::SetValue(int row
, int col
,const wxString
& value
)
619 wxLogDebug(wxT("SetValue() on %i,%i"),row
,col
);
622 wxVariant
val(value
);
624 m_rowmodified
= TRUE
;
625 m_data
->SetCol(m_ColInfo
[col
].DbCol
,val
);
629 void wxDbGridTableBase::SetValueAsLong(int row
, int col
, long value
)
631 wxLogDebug(wxT("SetValueAsLong() on %i,%i"),row
,col
);
634 wxVariant
val(value
);
636 m_rowmodified
= TRUE
;
637 m_data
->SetCol(m_ColInfo
[col
].DbCol
,val
);
641 void wxDbGridTableBase::SetValueAsDouble(int row
, int col
, double value
)
643 wxLogDebug(wxT("SetValueAsDouble() on %i,%i"),row
,col
);
646 wxVariant
val(value
);
648 m_rowmodified
= TRUE
;
649 m_data
->SetCol(m_ColInfo
[col
].DbCol
,val
);
654 void wxDbGridTableBase::SetValueAsBool(int row
, int col
, bool value
)
656 wxLogDebug(wxT("SetValueAsBool() on %i,%i"),row
,col
);
659 wxVariant
val(value
);
661 m_rowmodified
= TRUE
;
662 m_data
->SetCol(m_ColInfo
[col
].DbCol
,val
);
666 void wxDbGridTableBase::ValidateRow(int row
)
668 wxLogDebug(wxT("ValidateRow(%i) currently on row (%i). Array count = %i"),row
,m_row
,m_keys
.GetCount());
674 //We add to row as Count is unsigned!
675 if ((unsigned)(row
+1) > m_keys
.GetCount())
677 wxLogDebug(wxT("\trow key unknown"));
678 // Extend Array, iterate through data filling with keys
679 m_data
->SetRowMode(wxDbTable::WX_ROW_MODE_QUERY
);
681 for (trow
= m_keys
.GetCount(); trow
<= row
; trow
++)
683 wxLogDebug(wxT("Fetching row %i.."), trow
);
684 bool ret
= m_data
->GetNext();
686 wxLogDebug(wxT(" ...success=(%i)"),ret
);
687 GenericKey k
= m_data
->GetKey();
694 wxLogDebug(wxT("\trow key known centering data"));
695 GenericKey k
= m_keys
.Item(row
);
696 m_data
->SetRowMode(wxDbTable::WX_ROW_MODE_INDIVIDUAL
);
697 m_data
->ClearMemberVars();
699 if (!m_data
->QueryOnKeyFields())
701 wxDbLogExtendedErrorMsg("ODBC error during Query()\n\n", m_data
->GetDb(),__FILE__
,__LINE__
);
708 m_rowmodified
= FALSE
;
711 bool wxDbGridTableBase::Writeback() const
719 wxLogDebug(wxT("\trow key unknown"));
721 // FIXME: this code requires dbtable support for record status
723 switch (m_data
->get_ModifiedStatus())
725 case wxDbTable::UpdatePending
:
726 result
= m_data
->Update();
728 case wxDbTable::InsertPending
:
729 result
= (m_data
->Insert() == SQL_SUCCESS
);
736 wxLogDebug(wxT("WARNING : Row writeback not implemented "));
741 #include <wx/arrimpl.cpp>
743 WX_DEFINE_OBJARRAY(keyarray
);
745 #endif // #if wxUSE_NEW_GRID
746 #endif // #if wxUSE_ODBC