io/source/TextOutputStream/TextOutputStream.cxx

   1 /*************************************************************************
   2  *
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * Copyright 2008 by Sun Microsystems, Inc.
   6  *
   7  * OpenOffice.org - a multi-platform office productivity suite
   8  *
   9  * $RCSfile: TextOutputStream.cxx,v $
  10  * $Revision: 1.11 $
  11  *
  12  * This file is part of OpenOffice.org.
  13  *
  14  * OpenOffice.org is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU Lesser General Public License version 3
  16  * only, as published by the Free Software Foundation.
  17  *
  18  * OpenOffice.org is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU Lesser General Public License version 3 for more details
  22  * (a copy is included in the LICENSE file that accompanied this code).
  23  *
  24  * You should have received a copy of the GNU Lesser General Public License
  25  * version 3 along with OpenOffice.org.  If not, see
  26  * <http://www.openoffice.org/license.html>
  27  * for a copy of the LGPLv3 License.
  28  *
  29  ************************************************************************/
  30
  31 // MARKER(update_precomp.py): autogen include statement, do not remove
  32 #include "precompiled_io.hxx"
  33
  34
  35 #include <osl/mutex.hxx>
  36 #include <osl/diagnose.h>
  37
  38 #include <uno/mapping.hxx>
  39
  40 #include <cppuhelper/factory.hxx>
  41 #include <cppuhelper/implbase3.hxx>
  42 #include <cppuhelper/implementationentry.hxx>
  43
  44 #include <rtl/textenc.h>
  45 #include <rtl/tencinfo.h>
  46 #include <rtl/unload.h>
  47
  48 #include <com/sun/star/io/XTextOutputStream.hpp>
  49 #include <com/sun/star/io/XActiveDataSource.hpp>
  50 #include <com/sun/star/lang/XServiceInfo.hpp>
  51
  52
  53 #define IMPLEMENTATION_NAME "com.sun.star.comp.io.TextOutputStream"
  54 #define SERVICE_NAME "com.sun.star.io.TextOutputStream"
  55
  56 using namespace ::osl;
  57 using namespace ::rtl;
  58 using namespace ::cppu;
  59 using namespace ::com::sun::star::uno;
  60 using namespace ::com::sun::star::lang;
  61 using namespace ::com::sun::star::io;
  62 using namespace ::com::sun::star::registry;
  63
  64 namespace io_TextOutputStream
  65 {
  66     rtl_StandardModuleCount g_moduleCount = MODULE_COUNT_INIT;
  67 //===========================================================================
  68 // Implementation XTextOutputStream
  69
  70 typedef WeakImplHelper3< XTextOutputStream, XActiveDataSource, XServiceInfo > TextOutputStreamHelper;
  71 class OCommandEnvironment;
  72
  73 class OTextOutputStream : public TextOutputStreamHelper
  74 {
  75     Reference< XOutputStream > mxStream;
  76
  77     // Encoding
  78     OUString mEncoding;
  79     sal_Bool mbEncodingInitialized;
  80     rtl_UnicodeToTextConverter  mConvUnicode2Text;
  81     rtl_UnicodeToTextContext    mContextUnicode2Text;
  82
  83     Sequence<sal_Int8> implConvert( const OUString& rSource );
  84
  85 public:
  86     OTextOutputStream();
  87     ~OTextOutputStream();
  88
  89     // Methods XTextOutputStream
  90     virtual void SAL_CALL writeString( const OUString& aString )
  91         throw(IOException, RuntimeException);
  92     virtual void SAL_CALL setEncoding( const OUString& Encoding )
  93         throw(RuntimeException);
  94
  95     // Methods XOutputStream
  96     virtual void SAL_CALL writeBytes( const Sequence< sal_Int8 >& aData )
  97         throw(NotConnectedException, BufferSizeExceededException, IOException, RuntimeException);
  98     virtual void SAL_CALL flush(  )
  99         throw(NotConnectedException, BufferSizeExceededException, IOException, RuntimeException);
 100     virtual void SAL_CALL closeOutput(  )
 101         throw(NotConnectedException, BufferSizeExceededException, IOException, RuntimeException);
 102
 103     // Methods XActiveDataSource
 104     virtual void SAL_CALL setOutputStream( const Reference< XOutputStream >& aStream )
 105         throw(RuntimeException);
 106     virtual Reference< XOutputStream > SAL_CALL getOutputStream(  )
 107         throw(RuntimeException);
 108
 109     // Methods XServiceInfo
 110         virtual OUString              SAL_CALL getImplementationName() throw();
 111         virtual Sequence< OUString >  SAL_CALL getSupportedServiceNames(void) throw();
 112         virtual sal_Bool              SAL_CALL supportsService(const OUString& ServiceName) throw();
 113 };
 114
 115 OTextOutputStream::OTextOutputStream()
 116 {
 117     mbEncodingInitialized = false;
 118 }
 119
 120 OTextOutputStream::~OTextOutputStream()
 121 {
 122     if( mbEncodingInitialized )
 123     {
 124         rtl_destroyUnicodeToTextContext( mConvUnicode2Text, mContextUnicode2Text );
 125         rtl_destroyUnicodeToTextConverter( mConvUnicode2Text );
 126     }
 127 }
 128
 129 Sequence<sal_Int8> OTextOutputStream::implConvert( const OUString& rSource )
 130 {
 131     const sal_Unicode *puSource = rSource.getStr();
 132     sal_Int32 nSourceSize = rSource.getLength();
 133
 134     sal_Size nTargetCount = 0;
 135     sal_Size nSourceCount = 0;
 136
 137     sal_uInt32 uiInfo;
 138     sal_Size nSrcCvtChars;
 139
 140     // take nSourceSize * 3 as preference
 141     // this is an upper boundary for converting to utf8,
 142     // which most often used as the target.
 143     sal_Int32 nSeqSize =  nSourceSize * 3;
 144
 145     Sequence<sal_Int8> seqText( nSeqSize );
 146     sal_Char *pTarget = (sal_Char *) seqText.getArray();
 147     while( sal_True )
 148     {
 149         nTargetCount += rtl_convertUnicodeToText(
 150                                     mConvUnicode2Text,
 151                                     mContextUnicode2Text,
 152                                     &( puSource[nSourceCount] ),
 153                                     nSourceSize - nSourceCount ,
 154                                     &( pTarget[nTargetCount] ),
 155                                     nSeqSize - nTargetCount,
 156                                     RTL_UNICODETOTEXT_FLAGS_UNDEFINED_DEFAULT |
 157                                     RTL_UNICODETOTEXT_FLAGS_INVALID_DEFAULT ,
 158                                     &uiInfo,
 159                                     &nSrcCvtChars);
 160         nSourceCount += nSrcCvtChars;
 161
 162         if( uiInfo & RTL_UNICODETOTEXT_INFO_DESTBUFFERTOSMALL )
 163         {
 164             nSeqSize *= 2;
 165             seqText.realloc( nSeqSize );  // double array size
 166             pTarget = (sal_Char*) seqText.getArray();
 167             continue;
 168         }
 169         break;
 170     }
 171
 172     // reduce the size of the buffer (fast, no copy necessary)
 173     seqText.realloc( nTargetCount );
 174     return seqText;
 175 }
 176
 177
 178 //===========================================================================
 179 // XTextOutputStream
 180
 181 void OTextOutputStream::writeString( const OUString& aString )
 182     throw(IOException, RuntimeException)
 183 {
 184     if( !mbEncodingInitialized )
 185     {
 186         OUString aUtf8Str( RTL_CONSTASCII_USTRINGPARAM("utf8") );
 187         setEncoding( aUtf8Str );
 188     }
 189     if( !mbEncodingInitialized )
 190         return;
 191
 192     Sequence<sal_Int8> aByteSeq = implConvert( aString );
 193     mxStream->writeBytes( aByteSeq );
 194 }
 195
 196 void OTextOutputStream::setEncoding( const OUString& Encoding )
 197     throw(RuntimeException)
 198 {
 199     OString aOEncodingStr = OUStringToOString( Encoding, RTL_TEXTENCODING_ASCII_US );
 200     rtl_TextEncoding encoding = rtl_getTextEncodingFromMimeCharset( aOEncodingStr.getStr() );
 201     if( RTL_TEXTENCODING_DONTKNOW == encoding )
 202         return;
 203
 204     mbEncodingInitialized = true;
 205     mConvUnicode2Text   = rtl_createUnicodeToTextConverter( encoding );
 206     mContextUnicode2Text = rtl_createUnicodeToTextContext( mConvUnicode2Text );
 207     mEncoding = Encoding;
 208 }
 209
 210 //===========================================================================
 211 // XOutputStream
 212 void OTextOutputStream::writeBytes( const Sequence< sal_Int8 >& aData )
 213     throw(NotConnectedException, BufferSizeExceededException, IOException, RuntimeException)
 214 {
 215     mxStream->writeBytes( aData );
 216 }
 217
 218 void OTextOutputStream::flush(  )
 219     throw(NotConnectedException, BufferSizeExceededException, IOException, RuntimeException)
 220 {
 221     mxStream->flush();
 222 }
 223
 224 void OTextOutputStream::closeOutput(  )
 225     throw(NotConnectedException, BufferSizeExceededException, IOException, RuntimeException)
 226 {
 227     mxStream->closeOutput();
 228 }
 229
 230
 231 //===========================================================================
 232 // XActiveDataSource
 233
 234 void OTextOutputStream::setOutputStream( const Reference< XOutputStream >& aStream )
 235     throw(RuntimeException)
 236 {
 237     mxStream = aStream;
 238 }
 239
 240 Reference< XOutputStream > OTextOutputStream::getOutputStream()
 241     throw(RuntimeException)
 242 {
 243     return mxStream;
 244 }
 245
 246
 247 Reference< XInterface > SAL_CALL TextOutputStream_CreateInstance( const Reference< XComponentContext > &)
 248 {
 249     return Reference < XInterface >( ( OWeakObject * ) new OTextOutputStream() );
 250 }
 251
 252 OUString TextOutputStream_getImplementationName() SAL_THROW(  () )
 253 {
 254     return OUString( RTL_CONSTASCII_USTRINGPARAM( IMPLEMENTATION_NAME ) );
 255 }
 256
 257
 258 Sequence< OUString > TextOutputStream_getSupportedServiceNames()
 259 {
 260     static Sequence < OUString > *pNames = 0;
 261     if( ! pNames )
 262     {
 263         MutexGuard guard( Mutex::getGlobalMutex() );
 264         if( !pNames )
 265         {
 266             static Sequence< OUString > seqNames(1);
 267             seqNames.getArray()[0] = OUString( RTL_CONSTASCII_USTRINGPARAM( SERVICE_NAME ) );
 268             pNames = &seqNames;
 269         }
 270     }
 271     return *pNames;
 272 }
 273
 274 OUString OTextOutputStream::getImplementationName() throw()
 275 {
 276     return TextOutputStream_getImplementationName();
 277 }
 278
 279 sal_Bool OTextOutputStream::supportsService(const OUString& ServiceName) throw()
 280 {
 281     Sequence< OUString > aSNL = getSupportedServiceNames();
 282     const OUString * pArray = aSNL.getConstArray();
 283
 284     for( sal_Int32 i = 0; i < aSNL.getLength(); i++ )
 285         if( pArray[i] == ServiceName )
 286             return sal_True;
 287
 288     return sal_False;
 289 }
 290
 291 Sequence< OUString > OTextOutputStream::getSupportedServiceNames(void) throw()
 292 {
 293     return TextOutputStream_getSupportedServiceNames();
 294 }
 295
 296
 297 }
 298
 299 using namespace io_TextOutputStream;
 300
 301 static struct ImplementationEntry g_entries[] =
 302 {
 303     {
 304         TextOutputStream_CreateInstance, TextOutputStream_getImplementationName ,
 305         TextOutputStream_getSupportedServiceNames, createSingleComponentFactory ,
 306         &g_moduleCount.modCnt , 0
 307     },
 308     { 0, 0, 0, 0, 0, 0 }
 309 };
 310
 311 extern "C"
 312 {
 313 sal_Bool SAL_CALL component_canUnload( TimeValue *pTime )
 314 {
 315     return g_moduleCount.canUnload( &g_moduleCount , pTime );
 316 }
 317
 318 //==================================================================================================
 319 void SAL_CALL component_getImplementationEnvironment(
 320     const sal_Char ** ppEnvTypeName, uno_Environment ** )
 321 {
 322     *ppEnvTypeName = CPPU_CURRENT_LANGUAGE_BINDING_NAME;
 323 }
 324 //==================================================================================================
 325 sal_Bool SAL_CALL component_writeInfo(
 326     void * pServiceManager, void * pRegistryKey )
 327 {
 328     return component_writeInfoHelper( pServiceManager, pRegistryKey, g_entries );
 329 }
 330 //==================================================================================================
 331 void * SAL_CALL component_getFactory(
 332     const sal_Char * pImplName, void * pServiceManager, void * pRegistryKey )
 333 {
 334     return component_getFactoryHelper( pImplName, pServiceManager, pRegistryKey , g_entries );
 335 }
 336 }
 337
 338