(blind) fix for GTK2/Unicode crashes when copying text to the clipboard
[wxWidgets.git] / src / common / dobjcmn.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: common/dobjcmn.cpp
3 // Purpose: implementation of data object methods common to all platforms
4 // Author: Vadim Zeitlin, Robert Roebling
5 // Modified by:
6 // Created: 19.10.99
7 // RCS-ID: $Id$
8 // Copyright: (c) wxWidgets Team
9 // Licence: wxWindows licence
10 ///////////////////////////////////////////////////////////////////////////////
11
12 // ============================================================================
13 // declarations
14 // ============================================================================
15
16 // ----------------------------------------------------------------------------
17 // headers
18 // ----------------------------------------------------------------------------
19
20 #if defined(__GNUG__) && !defined(NO_GCC_PRAGMA)
21 #pragma implementation "dataobjbase.h"
22 #endif
23
24 #include "wx/wxprec.h"
25
26 #ifdef __BORLANDC__
27 #pragma hdrstop
28 #endif
29
30 #if wxUSE_DATAOBJ
31
32 #ifndef WX_PRECOMP
33 #include "wx/app.h"
34 #include "wx/debug.h"
35 #endif // WX_PRECOMP
36
37 #include "wx/dataobj.h"
38
39 // ----------------------------------------------------------------------------
40 // lists
41 // ----------------------------------------------------------------------------
42
43 #include "wx/listimpl.cpp"
44
45 WX_DEFINE_LIST(wxSimpleDataObjectList);
46
47 // ----------------------------------------------------------------------------
48 // globals
49 // ----------------------------------------------------------------------------
50
51 static wxDataFormat dataFormatInvalid;
52 WXDLLEXPORT const wxDataFormat& wxFormatInvalid = dataFormatInvalid;
53
54 // ============================================================================
55 // implementation
56 // ============================================================================
57
58 // ----------------------------------------------------------------------------
59 // wxDataObjectBase
60 // ----------------------------------------------------------------------------
61
62 wxDataObjectBase::~wxDataObjectBase()
63 {
64 }
65
66 bool wxDataObjectBase::IsSupported(const wxDataFormat& format,
67 Direction dir) const
68 {
69 size_t nFormatCount = GetFormatCount(dir);
70 if ( nFormatCount == 1 )
71 {
72 return format == GetPreferredFormat(dir);
73 }
74 else
75 {
76 wxDataFormat *formats = new wxDataFormat[nFormatCount];
77 GetAllFormats(formats, dir);
78
79 size_t n;
80 for ( n = 0; n < nFormatCount; n++ )
81 {
82 if ( formats[n] == format )
83 break;
84 }
85
86 delete [] formats;
87
88 // found?
89 return n < nFormatCount;
90 }
91 }
92
93 // ----------------------------------------------------------------------------
94 // wxDataObjectComposite
95 // ----------------------------------------------------------------------------
96
97 wxDataObjectComposite::wxDataObjectComposite()
98 {
99 m_preferred = 0;
100 }
101
102 wxDataObjectComposite::~wxDataObjectComposite()
103 {
104 WX_CLEAR_LIST(wxSimpleDataObjectList, m_dataObjects);
105 }
106
107 wxDataObjectSimple *
108 wxDataObjectComposite::GetObject(const wxDataFormat& format) const
109 {
110 wxSimpleDataObjectList::compatibility_iterator node = m_dataObjects.GetFirst();
111 while ( node )
112 {
113 wxDataObjectSimple *dataObj = node->GetData();
114
115 if ( dataObj->GetFormat() == format )
116 {
117 return dataObj;
118 }
119
120 node = node->GetNext();
121 }
122
123 return (wxDataObjectSimple *)NULL;
124 }
125
126 void wxDataObjectComposite::Add(wxDataObjectSimple *dataObject, bool preferred)
127 {
128 if ( preferred )
129 m_preferred = m_dataObjects.GetCount();
130
131 m_dataObjects.Append( dataObject );
132 }
133
134 wxDataFormat
135 wxDataObjectComposite::GetPreferredFormat(Direction WXUNUSED(dir)) const
136 {
137 wxSimpleDataObjectList::compatibility_iterator node = m_dataObjects.Item( m_preferred );
138
139 wxCHECK_MSG( node, wxFormatInvalid, wxT("no preferred format") );
140
141 wxDataObjectSimple* dataObj = node->GetData();
142
143 return dataObj->GetFormat();
144 }
145
146 #if defined(__WXMSW__)
147
148 size_t wxDataObjectComposite::GetBufferOffset( const wxDataFormat& format )
149 {
150 wxDataObjectSimple *dataObj = GetObject(format);
151
152 wxCHECK_MSG( dataObj, 0,
153 wxT("unsupported format in wxDataObjectComposite"));
154
155 return dataObj->GetBufferOffset( format );
156 }
157
158
159 const void* wxDataObjectComposite::GetSizeFromBuffer( const void* buffer,
160 size_t* size,
161 const wxDataFormat& format )
162 {
163 wxDataObjectSimple *dataObj = GetObject(format);
164
165 wxCHECK_MSG( dataObj, NULL,
166 wxT("unsupported format in wxDataObjectComposite"));
167
168 return dataObj->GetSizeFromBuffer( buffer, size, format );
169 }
170
171
172 void* wxDataObjectComposite::SetSizeInBuffer( void* buffer, size_t size,
173 const wxDataFormat& format )
174 {
175 wxDataObjectSimple *dataObj = GetObject(format);
176
177 wxCHECK_MSG( dataObj, NULL,
178 wxT("unsupported format in wxDataObjectComposite"));
179
180 return dataObj->SetSizeInBuffer( buffer, size, format );
181 }
182
183 #endif
184
185 size_t wxDataObjectComposite::GetFormatCount(Direction WXUNUSED(dir)) const
186 {
187 // TODO what about the Get/Set only formats?
188 return m_dataObjects.GetCount();
189 }
190
191 void wxDataObjectComposite::GetAllFormats(wxDataFormat *formats,
192 Direction WXUNUSED(dir)) const
193 {
194 size_t n = 0;
195 wxSimpleDataObjectList::compatibility_iterator node;
196 for ( node = m_dataObjects.GetFirst(); node; node = node->GetNext() )
197 {
198 // TODO if ( !outputOnlyToo ) && this one counts ...
199 formats[n++] = node->GetData()->GetFormat();
200 }
201 }
202
203 size_t wxDataObjectComposite::GetDataSize(const wxDataFormat& format) const
204 {
205 wxDataObjectSimple *dataObj = GetObject(format);
206
207 wxCHECK_MSG( dataObj, 0,
208 wxT("unsupported format in wxDataObjectComposite"));
209
210 return dataObj->GetDataSize();
211 }
212
213 bool wxDataObjectComposite::GetDataHere(const wxDataFormat& format,
214 void *buf) const
215 {
216 wxDataObjectSimple *dataObj = GetObject(format);
217
218 wxCHECK_MSG( dataObj, false,
219 wxT("unsupported format in wxDataObjectComposite"));
220
221 return dataObj->GetDataHere(buf);
222 }
223
224 bool wxDataObjectComposite::SetData(const wxDataFormat& format,
225 size_t len,
226 const void *buf)
227 {
228 wxDataObjectSimple *dataObj = GetObject(format);
229
230 wxCHECK_MSG( dataObj, false,
231 wxT("unsupported format in wxDataObjectComposite"));
232
233 return dataObj->SetData(len, buf);
234 }
235
236 // ----------------------------------------------------------------------------
237 // wxTextDataObject
238 // ----------------------------------------------------------------------------
239
240 #if defined(__WXGTK20__) && wxUSE_UNICODE
241
242 static inline wxMBConv& GetConv(const wxDataFormat& format)
243 {
244 // use UTF8 for wxDF_UNICODETEXT and UCS4 for wxDF_TEXT
245 return format == wxDF_UNICODETEXT ? wxConvUTF8 : wxConvLibc;
246 }
247
248 size_t wxTextDataObject::GetDataSize(const wxDataFormat& format) const
249 {
250 wxCharBuffer buffer = GetConv(format).cWX2MB( GetText().c_str() );
251 return buffer ? strlen(buffer) + 1 : 0;
252 }
253
254 bool wxTextDataObject::GetDataHere(const wxDataFormat& format, void *buf) const
255 {
256 wxCharBuffer buffer = GetConv(format).cWX2MB( GetText().c_str() );
257 if ( !buffer )
258 return false;
259
260 strcpy( (char*) buf, buffer );
261
262 return true;
263 }
264
265 bool wxTextDataObject::SetData(const wxDataFormat& format,
266 size_t WXUNUSED(len), const void *buf)
267 {
268 wxWCharBuffer buffer = GetConv(format).cMB2WX((const char *)buf);
269 if ( !buffer )
270 return false;
271
272 SetText(buffer);
273
274 return true;
275 }
276
277 #elif wxUSE_UNICODE && defined(__WXMAC__)
278
279 size_t wxTextDataObject::GetDataSize(const wxDataFormat& format) const
280 {
281 if (format == wxDF_UNICODETEXT)
282 {
283 // host native is UTF16
284 wxMBConvUTF16BE converter ;
285 return converter.WC2MB( NULL , GetText().c_str() , 0 ) + 2; // add space for trailing unichar 0
286 }
287 else // == wxDF_TEXT
288 {
289 wxCharBuffer buffer = wxConvLibc.cWX2MB( GetText().c_str() );
290 return strlen( (const char*) buffer ) + 1;
291 }
292 }
293
294 bool wxTextDataObject::GetDataHere(const wxDataFormat& format, void *buf) const
295 {
296 if (format == wxDF_UNICODETEXT)
297 {
298 // host native is UTF16
299 wxMBConvUTF16BE converter ;
300 size_t len = converter.WC2MB( NULL , GetText().c_str() , 0 ) ;
301 wxCharBuffer buffer = converter.cWX2MB( GetText().c_str() );
302 memcpy( (char*) buf, (const char*) buffer , len + 2); // trailing unichar 0
303 }
304 else
305 {
306 wxCharBuffer buffer = wxConvLibc.cWX2MB( GetText().c_str() );
307 strcpy( (char*) buf, (const char*) buffer );
308 }
309
310 return true;
311 }
312
313 bool wxTextDataObject::SetData(const wxDataFormat& format,
314 size_t WXUNUSED(len), const void *buf)
315 {
316 if (format == wxDF_UNICODETEXT)
317 {
318 // host native is UTF16
319 wxMBConvUTF16BE converter ;
320 SetText( converter.cMB2WX( (const char*) buf ) );
321 }
322 else
323 SetText( wxConvLibc.cMB2WX( (const char*) buf ) );
324
325 return true;
326 }
327
328 #else
329
330 size_t wxTextDataObject::GetDataSize() const
331 {
332 return GetTextLength() * sizeof(wxChar);
333 }
334
335 bool wxTextDataObject::GetDataHere(void *buf) const
336 {
337 wxStrcpy((wxChar *)buf, GetText().c_str());
338
339 return true;
340 }
341
342 bool wxTextDataObject::SetData(size_t WXUNUSED(len), const void *buf)
343 {
344 SetText(wxString((const wxChar *)buf));
345
346 return true;
347 }
348
349 #endif
350
351 // ----------------------------------------------------------------------------
352 // wxFileDataObjectBase
353 // ----------------------------------------------------------------------------
354
355 // VZ: I don't need this in MSW finally, so if it is needed in wxGTK, it should
356 // be moved to gtk/dataobj.cpp
357 #if 0
358
359 wxString wxFileDataObjectBase::GetFilenames() const
360 {
361 wxString str;
362 size_t count = m_filenames.GetCount();
363 for ( size_t n = 0; n < count; n++ )
364 {
365 str << m_filenames[n] << wxT('\0');
366 }
367
368 return str;
369 }
370
371 void wxFileDataObjectBase::SetFilenames(const wxChar* filenames)
372 {
373 m_filenames.Empty();
374
375 wxString current;
376 for ( const wxChar *pc = filenames; ; pc++ )
377 {
378 if ( *pc )
379 {
380 current += *pc;
381 }
382 else
383 {
384 if ( !current )
385 {
386 // 2 consecutive NULs - this is the end of the string
387 break;
388 }
389
390 m_filenames.Add(current);
391 current.Empty();
392 }
393 }
394 }
395
396 #endif // 0
397
398 // ----------------------------------------------------------------------------
399 // wxCustomDataObject
400 // ----------------------------------------------------------------------------
401
402 wxCustomDataObject::wxCustomDataObject(const wxDataFormat& format)
403 : wxDataObjectSimple(format)
404 {
405 m_data = (void *)NULL;
406 }
407
408 wxCustomDataObject::~wxCustomDataObject()
409 {
410 Free();
411 }
412
413 void wxCustomDataObject::TakeData(size_t size, void *data)
414 {
415 Free();
416
417 m_size = size;
418 m_data = data;
419 }
420
421 void *wxCustomDataObject::Alloc(size_t size)
422 {
423 return (void *)new char[size];
424 }
425
426 void wxCustomDataObject::Free()
427 {
428 delete [] (char *)m_data;
429 m_size = 0;
430 m_data = (void *)NULL;
431 }
432
433 size_t wxCustomDataObject::GetDataSize() const
434 {
435 return GetSize();
436 }
437
438 bool wxCustomDataObject::GetDataHere(void *buf) const
439 {
440 void *data = GetData();
441 if ( !data )
442 return false;
443
444 memcpy(buf, data, GetSize());
445
446 return true;
447 }
448
449 bool wxCustomDataObject::SetData(size_t size, const void *buf)
450 {
451 Free();
452
453 m_data = Alloc(size);
454 if ( !m_data )
455 return false;
456
457 memcpy(m_data, buf, m_size = size);
458
459 return true;
460 }
461
462 // ============================================================================
463 // some common dnd related code
464 // ============================================================================
465
466 #if wxUSE_DRAG_AND_DROP
467
468 #include "wx/dnd.h"
469
470 // ----------------------------------------------------------------------------
471 // wxTextDropTarget
472 // ----------------------------------------------------------------------------
473
474 // NB: we can't use "new" in ctor initializer lists because this provokes an
475 // internal compiler error with VC++ 5.0 (hey, even gcc compiles this!),
476 // so use SetDataObject() instead
477
478 wxTextDropTarget::wxTextDropTarget()
479 {
480 SetDataObject(new wxTextDataObject);
481 }
482
483 wxDragResult wxTextDropTarget::OnData(wxCoord x, wxCoord y, wxDragResult def)
484 {
485 if ( !GetData() )
486 return wxDragNone;
487
488 wxTextDataObject *dobj = (wxTextDataObject *)m_dataObject;
489 return OnDropText(x, y, dobj->GetText()) ? def : wxDragNone;
490 }
491
492 // ----------------------------------------------------------------------------
493 // wxFileDropTarget
494 // ----------------------------------------------------------------------------
495
496 wxFileDropTarget::wxFileDropTarget()
497 {
498 SetDataObject(new wxFileDataObject);
499 }
500
501 wxDragResult wxFileDropTarget::OnData(wxCoord x, wxCoord y, wxDragResult def)
502 {
503 if ( !GetData() )
504 return wxDragNone;
505
506 wxFileDataObject *dobj = (wxFileDataObject *)m_dataObject;
507 return OnDropFiles(x, y, dobj->GetFilenames()) ? def : wxDragNone;
508 }
509
510 #endif // wxUSE_DRAG_AND_DROP
511
512 #endif // wxUSE_DATAOBJ