X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/e6ef9ea4ed13bee3ea5d830270986eab571876e6..7fc641aff97d5f100b94cabb0288d3c4261e5328:/src/common/cmndata.cpp

diff --git a/src/common/cmndata.cpp b/src/common/cmndata.cpp
index 40e6073529..ba6c494e40 100644
--- a/src/common/cmndata.cpp
+++ b/src/common/cmndata.cpp
@@ -78,7 +78,7 @@ wxColourData::wxColourData()
 }
 
 wxColourData::wxColourData(const wxColourData& data)
-    : wxObject()
+            : wxObject()
 {
     (*this) = data;
 }
@@ -89,26 +89,28 @@ wxColourData::~wxColourData()
 
 void wxColourData::SetCustomColour(int i, const wxColour& colour)
 {
-    wxCHECK_RET( (i >= 0 && i < WXSIZEOF(m_custColours)), _T("custom colour index out of range") );
+    wxCHECK_RET( i >= 0 && i < NUM_CUSTOM, _T("custom colour index out of range") );
 
     m_custColours[i] = colour;
 }
 
 wxColour wxColourData::GetCustomColour(int i)
 {
-    wxCHECK_MSG( (i >= 0 && i < WXSIZEOF(m_custColours)), wxColour(0,0,0),
+    wxCHECK_MSG( i >= 0 && i < NUM_CUSTOM, wxColour(0,0,0),
                  _T("custom colour index out of range") );
 
     return m_custColours[i];
 }
 
-void wxColourData::operator=(const wxColourData& data)
+wxColourData& wxColourData::operator=(const wxColourData& data)
 {
-    for (int i = 0; i < WXSIZEOF(m_custColours); i++)
+    for ( int i = 0; i < NUM_CUSTOM; i++)
         m_custColours[i] = data.m_custColours[i];
 
     m_dataColour = data.m_dataColour;
     m_chooseFull = data.m_chooseFull;
+
+    return *this;
 }
 
 // ----------------------------------------------------------------------------
@@ -122,7 +124,7 @@ wxString wxColourData::ToString() const
 {
     wxString str(m_chooseFull ? '1' : '0');
 
-    for ( int i = 0; i < WXSIZEOF(m_custColours); i++ )
+    for ( int i = 0; i < NUM_CUSTOM; i++ )
     {
         str += wxCOL_DATA_SEP;