Don't leak a gchar* in wxDataFormat::GetId()
[wxWidgets.git] / src / gtk1 / dataobj.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: dataobj.cpp
3 // Purpose: wxDataObject class
4 // Author: Robert Roebling
5 // Id: $Id$
6 // Copyright: (c) 1998 Robert Roebling
7 // Licence: wxWindows licence
8 ///////////////////////////////////////////////////////////////////////////////
9
10 // For compilers that support precompilation, includes "wx.h".
11 #include "wx/wxprec.h"
12
13 #include "wx/dataobj.h"
14
15 #if wxUSE_DATAOBJ
16
17 #include "wx/app.h"
18 #include "wx/debug.h"
19 #include "wx/mstream.h"
20 #include "wx/image.h"
21 #include "wx/log.h"
22 #include "wx/uri.h"
23
24 #include <gdk/gdk.h>
25
26 //-------------------------------------------------------------------------
27 // global data
28 //-------------------------------------------------------------------------
29
30 GdkAtom g_textAtom = 0;
31 GdkAtom g_altTextAtom = 0;
32 GdkAtom g_pngAtom = 0;
33 GdkAtom g_fileAtom = 0;
34
35 //-------------------------------------------------------------------------
36 // wxDataFormat
37 //-------------------------------------------------------------------------
38
39 wxDataFormat::wxDataFormat()
40 {
41 // do *not* call PrepareFormats() from here for 2 reasons:
42 //
43 // 1. we will have time to do it later because some other Set function
44 // must be called before we really need them
45 //
46 // 2. doing so prevents us from declaring global wxDataFormats because
47 // calling PrepareFormats (and thus gdk_atom_intern) before GDK is
48 // initialised will result in a crash
49 m_type = wxDF_INVALID;
50 m_format = (GdkAtom) 0;
51 }
52
53 wxDataFormat::wxDataFormat( wxDataFormatId type )
54 {
55 PrepareFormats();
56 SetType( type );
57 }
58
59 wxDataFormat::wxDataFormat( const wxChar *id )
60 {
61 PrepareFormats();
62 SetId( id );
63 }
64
65 wxDataFormat::wxDataFormat( const wxString &id )
66 {
67 PrepareFormats();
68 SetId( id );
69 }
70
71 wxDataFormat::wxDataFormat( NativeFormat format )
72 {
73 PrepareFormats();
74 SetId( format );
75 }
76
77 void wxDataFormat::SetType( wxDataFormatId type )
78 {
79 PrepareFormats();
80
81 m_type = type;
82
83 #if wxUSE_UNICODE
84 if (m_type == wxDF_UNICODETEXT)
85 m_format = g_textAtom;
86 else if (m_type == wxDF_TEXT)
87 m_format = g_altTextAtom;
88 #else
89 if (m_type == wxDF_TEXT || m_type == wxDF_UNICODETEXT)
90 m_format = g_textAtom;
91 #endif
92 else
93 if (m_type == wxDF_BITMAP)
94 m_format = g_pngAtom;
95 else
96 if (m_type == wxDF_FILENAME)
97 m_format = g_fileAtom;
98 else
99 {
100 wxFAIL_MSG( wxT("invalid dataformat") );
101 }
102 }
103
104 wxDataFormatId wxDataFormat::GetType() const
105 {
106 return m_type;
107 }
108
109 wxString wxDataFormat::GetId() const
110 {
111 gchar* atom_name = gdk_atom_name( m_format );
112 wxString ret = wxString::FromAscii( atom_name );
113 g_free(atom_name);
114 return ret;
115 }
116
117 void wxDataFormat::SetId( NativeFormat format )
118 {
119 PrepareFormats();
120 m_format = format;
121
122 if (m_format == g_textAtom)
123 #if wxUSE_UNICODE
124 m_type = wxDF_UNICODETEXT;
125 #else
126 m_type = wxDF_TEXT;
127 #endif
128 else
129 if (m_format == g_altTextAtom)
130 m_type = wxDF_TEXT;
131 else
132 if (m_format == g_pngAtom)
133 m_type = wxDF_BITMAP;
134 else
135 if (m_format == g_fileAtom)
136 m_type = wxDF_FILENAME;
137 else
138 m_type = wxDF_PRIVATE;
139 }
140
141 void wxDataFormat::SetId( const wxChar *id )
142 {
143 PrepareFormats();
144 m_type = wxDF_PRIVATE;
145 wxString tmp( id );
146 m_format = gdk_atom_intern( (const char*) tmp.ToAscii(), FALSE );
147 }
148
149 void wxDataFormat::PrepareFormats()
150 {
151 // VZ: GNOME included in RedHat 6.1 uses the MIME types below and not the
152 // atoms STRING and file:ALL as the old code was, but normal X apps
153 // use STRING for text selection when transfering the data via
154 // clipboard, for example, so do use STRING for now (GNOME apps will
155 // probably support STRING as well for compatibility anyhow), but use
156 // text/uri-list for file dnd because compatibility is not important
157 // here (with whom?)
158 if (!g_textAtom)
159 #if wxUSE_UNICODE
160 g_textAtom = gdk_atom_intern( "UTF8_STRING", FALSE );
161 g_altTextAtom = gdk_atom_intern( "STRING", FALSE );
162 #else
163 g_textAtom = gdk_atom_intern( "STRING" /* "text/plain" */, FALSE );
164 #endif
165 if (!g_pngAtom)
166 g_pngAtom = gdk_atom_intern( "image/png", FALSE );
167 if (!g_fileAtom)
168 g_fileAtom = gdk_atom_intern( "text/uri-list", FALSE );
169 }
170
171 //-------------------------------------------------------------------------
172 // wxDataObject
173 //-------------------------------------------------------------------------
174
175 wxDataObject::wxDataObject()
176 {
177 }
178
179 wxDataObject::~wxDataObject()
180 {
181 // dtor is empty but needed for Darwin and AIX -- otherwise it doesn't link
182 }
183
184 bool wxDataObject::IsSupportedFormat(const wxDataFormat& format, Direction dir) const
185 {
186 size_t nFormatCount = GetFormatCount(dir);
187 if ( nFormatCount == 1 )
188 {
189 return format == GetPreferredFormat();
190 }
191 else
192 {
193 wxDataFormat *formats = new wxDataFormat[nFormatCount];
194 GetAllFormats(formats,dir);
195
196 size_t n;
197 for ( n = 0; n < nFormatCount; n++ )
198 {
199 if ( formats[n] == format )
200 break;
201 }
202
203 delete [] formats;
204
205 // found?
206 return n < nFormatCount;
207 }
208 }
209
210 // ----------------------------------------------------------------------------
211 // wxFileDataObject
212 // ----------------------------------------------------------------------------
213
214 bool wxFileDataObject::GetDataHere(void *buf) const
215 {
216 wxString filenames;
217
218 for (size_t i = 0; i < m_filenames.GetCount(); i++)
219 {
220 filenames += wxT("file:");
221 filenames += m_filenames[i];
222 filenames += wxT("\r\n");
223 }
224
225 memcpy( buf, filenames.mbc_str(), filenames.Len() + 1 );
226
227 return TRUE;
228 }
229
230 size_t wxFileDataObject::GetDataSize() const
231 {
232 size_t res = 0;
233
234 for (size_t i = 0; i < m_filenames.GetCount(); i++)
235 {
236 // This is junk in UTF-8
237 res += m_filenames[i].Len();
238 res += 5 + 2; // "file:" (5) + "\r\n" (2)
239 }
240
241 return res + 1;
242 }
243
244 bool wxFileDataObject::SetData(size_t WXUNUSED(size), const void *buf)
245 {
246 m_filenames.Empty();
247
248 // we get data in the text/uri-list format, i.e. as a sequence of URIs
249 // (filenames prefixed by "file:") delimited by "\r\n"
250 wxString filename;
251 for ( const char *p = (const char *)buf; ; p++ )
252 {
253 // some broken programs (testdnd GTK+ sample!) omit the trailing
254 // "\r\n", so check for '\0' explicitly here instead of doing it in
255 // the loop statement to account for it
256 if ( (*p == '\r' && *(p+1) == '\n') || !*p )
257 {
258 size_t lenPrefix = 5; // strlen("file:")
259 if ( filename.Left(lenPrefix).MakeLower() == _T("file:") )
260 {
261 // sometimes the syntax is "file:filename", sometimes it's
262 // URL-like: "file://filename" - deal with both
263 if ( filename[lenPrefix] == _T('/') &&
264 filename[lenPrefix + 1] == _T('/') )
265 {
266 // skip the slashes
267 lenPrefix += 2;
268 }
269
270 AddFile(wxURI::Unescape(filename.c_str() + lenPrefix));
271 filename.Empty();
272 }
273 else
274 {
275 wxLogDebug(_T("Unsupported URI '%s' in wxFileDataObject"),
276 filename.c_str());
277 }
278
279 if ( !*p )
280 break;
281
282 // skip '\r'
283 p++;
284 }
285 else
286 {
287 filename += *p;
288 }
289 }
290
291 return TRUE;
292 }
293
294 void wxFileDataObject::AddFile( const wxString &filename )
295 {
296 m_filenames.Add( filename );
297 }
298
299 // ----------------------------------------------------------------------------
300 // wxBitmapDataObject
301 // ----------------------------------------------------------------------------
302
303 wxBitmapDataObject::wxBitmapDataObject()
304 {
305 Init();
306 }
307
308 wxBitmapDataObject::wxBitmapDataObject( const wxBitmap& bitmap )
309 : wxBitmapDataObjectBase(bitmap)
310 {
311 Init();
312
313 DoConvertToPng();
314 }
315
316 wxBitmapDataObject::~wxBitmapDataObject()
317 {
318 Clear();
319 }
320
321 void wxBitmapDataObject::SetBitmap( const wxBitmap &bitmap )
322 {
323 ClearAll();
324
325 wxBitmapDataObjectBase::SetBitmap(bitmap);
326
327 DoConvertToPng();
328 }
329
330 bool wxBitmapDataObject::GetDataHere(void *buf) const
331 {
332 if ( !m_pngSize )
333 {
334 wxFAIL_MSG( wxT("attempt to copy empty bitmap failed") );
335
336 return FALSE;
337 }
338
339 memcpy(buf, m_pngData, m_pngSize);
340
341 return TRUE;
342 }
343
344 bool wxBitmapDataObject::SetData(size_t size, const void *buf)
345 {
346 Clear();
347
348 wxCHECK_MSG( wxImage::FindHandler(wxBITMAP_TYPE_PNG) != NULL,
349 FALSE, wxT("You must call wxImage::AddHandler(new wxPNGHandler); to be able to use clipboard with bitmaps!") );
350
351 m_pngSize = size;
352 m_pngData = malloc(m_pngSize);
353
354 memcpy(m_pngData, buf, m_pngSize);
355
356 wxMemoryInputStream mstream((char*) m_pngData, m_pngSize);
357 wxImage image;
358 if ( !image.LoadFile( mstream, wxBITMAP_TYPE_PNG ) )
359 {
360 return FALSE;
361 }
362
363 m_bitmap = wxBitmap(image);
364
365 return m_bitmap.Ok();
366 }
367
368 void wxBitmapDataObject::DoConvertToPng()
369 {
370 if ( !m_bitmap.Ok() )
371 return;
372
373 wxCHECK_RET( wxImage::FindHandler(wxBITMAP_TYPE_PNG) != NULL,
374 wxT("You must call wxImage::AddHandler(new wxPNGHandler); to be able to use clipboard with bitmaps!") );
375
376 wxImage image = m_bitmap.ConvertToImage();
377
378 wxCountingOutputStream count;
379 image.SaveFile(count, wxBITMAP_TYPE_PNG);
380
381 m_pngSize = count.GetSize() + 100; // sometimes the size seems to vary ???
382 m_pngData = malloc(m_pngSize);
383
384 wxMemoryOutputStream mstream((char*) m_pngData, m_pngSize);
385 image.SaveFile(mstream, wxBITMAP_TYPE_PNG);
386 }
387
388 #endif // wxUSE_DATAOBJ