src/gtk/dataobj.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/gtk/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 #if wxUSE_DATAOBJ
  14
  15 #include "wx/dataobj.h"
  16
  17 #ifndef WX_PRECOMP
  18     #include "wx/log.h"
  19     #include "wx/app.h"
  20     #include "wx/image.h"
  21 #endif
  22
  23 #include "wx/mstream.h"
  24 #include "wx/uri.h"
  25
  26 #include "wx/gtk/private.h"
  27
  28 //-------------------------------------------------------------------------
  29 // global data
  30 //-------------------------------------------------------------------------
  31
  32 GdkAtom  g_textAtom        = 0;
  33 GdkAtom  g_altTextAtom     = 0;
  34 GdkAtom  g_pngAtom         = 0;
  35 GdkAtom  g_fileAtom        = 0;
  36
  37 //-------------------------------------------------------------------------
  38 // wxDataFormat
  39 //-------------------------------------------------------------------------
  40
  41 wxDataFormat::wxDataFormat()
  42 {
  43     // do *not* call PrepareFormats() from here for 2 reasons:
  44     //
  45     // 1. we will have time to do it later because some other Set function
  46     //    must be called before we really need them
  47     //
  48     // 2. doing so prevents us from declaring global wxDataFormats because
  49     //    calling PrepareFormats (and thus gdk_atom_intern) before GDK is
  50     //    initialised will result in a crash
  51     m_type = wxDF_INVALID;
  52     m_format = (GdkAtom) 0;
  53 }
  54
  55 wxDataFormat::wxDataFormat( wxDataFormatId type )
  56 {
  57     PrepareFormats();
  58     SetType( type );
  59 }
  60
  61 void wxDataFormat::InitFromString( const wxString &id )
  62 {
  63     PrepareFormats();
  64     SetId( id );
  65 }
  66
  67 wxDataFormat::wxDataFormat( NativeFormat format )
  68 {
  69     PrepareFormats();
  70     SetId( format );
  71 }
  72
  73 void wxDataFormat::SetType( wxDataFormatId type )
  74 {
  75     PrepareFormats();
  76
  77     m_type = type;
  78
  79 #if wxUSE_UNICODE
  80     if (m_type == wxDF_UNICODETEXT)
  81         m_format = g_textAtom;
  82     else if (m_type == wxDF_TEXT)
  83         m_format = g_altTextAtom;
  84 #else // !wxUSE_UNICODE
  85     // notice that we don't map wxDF_UNICODETEXT to g_textAtom here, this
  86     // would lead the code elsewhere to treat data objects with this format as
  87     // containing UTF-8 data which is not true
  88     if (m_type == wxDF_TEXT)
  89         m_format = g_textAtom;
  90 #endif // wxUSE_UNICODE/!wxUSE_UNICODE
  91     else
  92     if (m_type == wxDF_BITMAP)
  93         m_format = g_pngAtom;
  94     else
  95     if (m_type == wxDF_FILENAME)
  96         m_format = g_fileAtom;
  97     else
  98     {
  99        wxFAIL_MSG( wxT("invalid dataformat") );
 100     }
 101 }
 102
 103 wxDataFormatId wxDataFormat::GetType() const
 104 {
 105     return m_type;
 106 }
 107
 108 wxString wxDataFormat::GetId() const
 109 {
 110     wxGtkString atom_name(gdk_atom_name(m_format));
 111     return wxString::FromAscii(atom_name);
 112 }
 113
 114 void wxDataFormat::SetId( NativeFormat format )
 115 {
 116     PrepareFormats();
 117     m_format = format;
 118
 119     if (m_format == g_textAtom)
 120 #if wxUSE_UNICODE
 121         m_type = wxDF_UNICODETEXT;
 122 #else
 123         m_type = wxDF_TEXT;
 124 #endif
 125     else
 126     if (m_format == g_altTextAtom)
 127         m_type = wxDF_TEXT;
 128     else
 129     if (m_format == g_pngAtom)
 130         m_type = wxDF_BITMAP;
 131     else
 132     if (m_format == g_fileAtom)
 133         m_type = wxDF_FILENAME;
 134     else
 135         m_type = wxDF_PRIVATE;
 136 }
 137
 138 void wxDataFormat::SetId( const wxString& id )
 139 {
 140     PrepareFormats();
 141     m_type = wxDF_PRIVATE;
 142     m_format = gdk_atom_intern( id.ToAscii(), FALSE );
 143 }
 144
 145 void wxDataFormat::PrepareFormats()
 146 {
 147     // VZ: GNOME included in RedHat 6.1 uses the MIME types below and not the
 148     //     atoms STRING and file:ALL as the old code was, but normal X apps
 149     //     use STRING for text selection when transferring the data via
 150     //     clipboard, for example, so do use STRING for now (GNOME apps will
 151     //     probably support STRING as well for compatibility anyhow), but use
 152     //     text/uri-list for file dnd because compatibility is not important
 153     //     here (with whom?)
 154     if (!g_textAtom)
 155     {
 156 #if wxUSE_UNICODE
 157         g_textAtom = gdk_atom_intern( "UTF8_STRING", FALSE );
 158         g_altTextAtom = gdk_atom_intern( "STRING", FALSE );
 159 #else
 160         g_textAtom = gdk_atom_intern( "STRING" /* "text/plain" */, FALSE );
 161 #endif
 162     }
 163     if (!g_pngAtom)
 164         g_pngAtom = gdk_atom_intern( "image/png", FALSE );
 165     if (!g_fileAtom)
 166         g_fileAtom = gdk_atom_intern( "text/uri-list", FALSE );
 167 }
 168
 169 //-------------------------------------------------------------------------
 170 // wxDataObject
 171 //-------------------------------------------------------------------------
 172
 173 wxDataObject::wxDataObject()
 174 {
 175 }
 176
 177 wxDataObject::~wxDataObject()
 178 {
 179     // dtor is empty but needed for Darwin and AIX -- otherwise it doesn't link
 180 }
 181
 182 bool wxDataObject::IsSupportedFormat(const wxDataFormat& format, Direction dir) const
 183 {
 184     size_t nFormatCount = GetFormatCount(dir);
 185     if ( nFormatCount == 1 )
 186     {
 187         return format == GetPreferredFormat();
 188     }
 189     else
 190     {
 191         wxDataFormat *formats = new wxDataFormat[nFormatCount];
 192         GetAllFormats(formats,dir);
 193
 194         size_t n;
 195         for ( n = 0; n < nFormatCount; n++ )
 196         {
 197             if ( formats[n] == format )
 198                 break;
 199         }
 200
 201         delete [] formats;
 202
 203         // found?
 204         return n < nFormatCount;
 205     }
 206 }
 207
 208 // ----------------------------------------------------------------------------
 209 // wxTextDataObject
 210 // ----------------------------------------------------------------------------
 211
 212 #if wxUSE_UNICODE
 213
 214 void
 215 wxTextDataObject::GetAllFormats(wxDataFormat *formats,
 216                                 wxDataObjectBase::Direction WXUNUSED(dir)) const
 217 {
 218     *formats++ = GetPreferredFormat();
 219     *formats = g_altTextAtom;
 220 }
 221
 222 #endif // wxUSE_UNICODE
 223
 224 // ----------------------------------------------------------------------------
 225 // wxFileDataObject
 226 // ----------------------------------------------------------------------------
 227
 228 bool wxFileDataObject::GetDataHere(void *buf) const
 229 {
 230     wxString filenames;
 231
 232     for (size_t i = 0; i < m_filenames.GetCount(); i++)
 233     {
 234         filenames += wxT("file:");
 235         filenames += m_filenames[i];
 236         filenames += wxT("\r\n");
 237     }
 238
 239     memcpy( buf, filenames.mbc_str(), filenames.length() + 1 );
 240
 241     return true;
 242 }
 243
 244 size_t wxFileDataObject::GetDataSize() const
 245 {
 246     size_t res = 0;
 247
 248     for (size_t i = 0; i < m_filenames.GetCount(); i++)
 249     {
 250         // This is junk in UTF-8
 251         res += m_filenames[i].length();
 252         res += 5 + 2; // "file:" (5) + "\r\n" (2)
 253     }
 254
 255     return res + 1;
 256 }
 257
 258 bool wxFileDataObject::SetData(size_t WXUNUSED(size), const void *buf)
 259 {
 260     // we get data in the text/uri-list format, i.e. as a sequence of URIs
 261     // (filenames prefixed by "file:") delimited by "\r\n". size includes
 262     // the trailing zero (in theory, not for Nautilus in early GNOME
 263     // versions).
 264
 265     m_filenames.Empty();
 266
 267     const gchar *nexttemp = (const gchar*) buf;
 268     for ( ; ; )
 269     {
 270         int len = 0;
 271         const gchar *temp = nexttemp;
 272         for (;;)
 273         {
 274             if (temp[len] == 0)
 275             {
 276                 if (len > 0)
 277                 {
 278                     // if an app omits '\r''\n'
 279                     nexttemp = temp+len;
 280                     break;
 281                 }
 282
 283                 return true;
 284             }
 285             if (temp[len] == '\r')
 286             {
 287                 if (temp[len+1] == '\n')
 288                     nexttemp = temp+len+2;
 289                 else
 290                     nexttemp = temp+len+1;
 291                 break;
 292             }
 293             len++;
 294         }
 295
 296         if (len == 0)
 297             break;
 298
 299         // required to give it a trailing zero
 300         gchar *uri = g_strndup( temp, len );
 301
 302         gchar *fn = g_filename_from_uri( uri, NULL, NULL );
 303
 304         g_free( uri );
 305
 306         if (fn)
 307         {
 308             AddFile( wxConvFileName->cMB2WX( fn ) );
 309             g_free( fn );
 310         }
 311     }
 312
 313     return true;
 314 }
 315
 316 void wxFileDataObject::AddFile( const wxString &filename )
 317 {
 318    m_filenames.Add( filename );
 319 }
 320
 321 // ----------------------------------------------------------------------------
 322 // wxBitmapDataObject
 323 // ----------------------------------------------------------------------------
 324
 325 wxBitmapDataObject::wxBitmapDataObject()
 326 {
 327     Init();
 328 }
 329
 330 wxBitmapDataObject::wxBitmapDataObject( const wxBitmap& bitmap )
 331                   : wxBitmapDataObjectBase(bitmap)
 332 {
 333     Init();
 334
 335     DoConvertToPng();
 336 }
 337
 338 wxBitmapDataObject::~wxBitmapDataObject()
 339 {
 340     Clear();
 341 }
 342
 343 void wxBitmapDataObject::SetBitmap( const wxBitmap &bitmap )
 344 {
 345     ClearAll();
 346
 347     wxBitmapDataObjectBase::SetBitmap(bitmap);
 348
 349     DoConvertToPng();
 350 }
 351
 352 bool wxBitmapDataObject::GetDataHere(void *buf) const
 353 {
 354     if ( !m_pngSize )
 355     {
 356         wxFAIL_MSG( wxT("attempt to copy empty bitmap failed") );
 357
 358         return false;
 359     }
 360
 361     memcpy(buf, m_pngData, m_pngSize);
 362
 363     return true;
 364 }
 365
 366 bool wxBitmapDataObject::SetData(size_t size, const void *buf)
 367 {
 368     Clear();
 369
 370     wxCHECK_MSG( wxImage::FindHandler(wxBITMAP_TYPE_PNG) != NULL,
 371                  false, wxT("You must call wxImage::AddHandler(new wxPNGHandler); to be able to use clipboard with bitmaps!") );
 372
 373     m_pngSize = size;
 374     m_pngData = malloc(m_pngSize);
 375
 376     memcpy(m_pngData, buf, m_pngSize);
 377
 378     wxMemoryInputStream mstream((char*) m_pngData, m_pngSize);
 379     wxImage image;
 380     if ( !image.LoadFile( mstream, wxBITMAP_TYPE_PNG ) )
 381     {
 382         return false;
 383     }
 384
 385     m_bitmap = wxBitmap(image);
 386
 387     return m_bitmap.IsOk();
 388 }
 389
 390 void wxBitmapDataObject::DoConvertToPng()
 391 {
 392     if ( !m_bitmap.IsOk() )
 393         return;
 394
 395     wxCHECK_RET( wxImage::FindHandler(wxBITMAP_TYPE_PNG) != NULL,
 396                  wxT("You must call wxImage::AddHandler(new wxPNGHandler); to be able to use clipboard with bitmaps!") );
 397
 398     wxImage image = m_bitmap.ConvertToImage();
 399
 400     wxCountingOutputStream count;
 401     image.SaveFile(count, wxBITMAP_TYPE_PNG);
 402
 403     m_pngSize = count.GetSize() + 100; // sometimes the size seems to vary ???
 404     m_pngData = malloc(m_pngSize);
 405
 406     wxMemoryOutputStream mstream((char*) m_pngData, m_pngSize);
 407     image.SaveFile(mstream, wxBITMAP_TYPE_PNG);
 408 }
 409
 410 // ----------------------------------------------------------------------------
 411 // wxURLDataObject
 412 // ----------------------------------------------------------------------------
 413
 414 wxURLDataObject::wxURLDataObject(const wxString& url) :
 415    wxDataObjectSimple( wxDataFormat( gdk_atom_intern("text/x-moz-url",FALSE) ) )
 416 {
 417    m_url = url;
 418 }
 419
 420 size_t wxURLDataObject::GetDataSize() const
 421 {
 422     if (m_url.empty())
 423         return 0;
 424
 425     return 2*m_url.Len()+2;
 426 }
 427
 428 bool wxURLDataObject::GetDataHere(void *buf) const
 429 {
 430     if (m_url.empty())
 431         return false;
 432
 433     wxCSConv conv( "UCS2" );
 434     conv.FromWChar( (char*) buf, 2*m_url.Len()+2, m_url.wc_str() );
 435
 436     return true;
 437 }
 438
 439     // copy data from buffer to our data
 440 bool wxURLDataObject::SetData(size_t len, const void *buf)
 441 {
 442     if (len == 0)
 443     {
 444         m_url = wxEmptyString;
 445         return false;
 446     }
 447
 448     wxCSConv conv( "UCS2" );
 449     wxWCharBuffer res = conv.cMB2WC( (const char*) buf );
 450     m_url = res;
 451     int pos = m_url.Find( '\n' );
 452     if (pos != wxNOT_FOUND)
 453         m_url.Remove( pos, m_url.Len() - pos );
 454
 455     return true;
 456 }
 457
 458
 459 #endif // wxUSE_DATAOBJ