io/source/TextOutputStream/TextOutputStream.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  *
   9  * This file incorporates work covered by the following license notice:
  10  *
  11  *   Licensed to the Apache Software Foundation (ASF) under one or more
  12  *   contributor license agreements. See the NOTICE file distributed
  13  *   with this work for additional information regarding copyright
  14  *   ownership. The ASF licenses this file to you under the Apache
  15  *   License, Version 2.0 (the "License"); you may not use this file
  16  *   except in compliance with the License. You may obtain a copy of
  17  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
  18  */
  19
  20
  21
  22 #include <cppuhelper/implbase.hxx>
  23 #include <cppuhelper/supportsservice.hxx>
  24
  25 #include <rtl/textenc.h>
  26 #include <rtl/tencinfo.h>
  27
  28 #include <com/sun/star/io/IOException.hpp>
  29 #include <com/sun/star/io/XTextOutputStream2.hpp>
  30 #include <com/sun/star/lang/XServiceInfo.hpp>
  31
  32 namespace com::sun::star::uno { class XComponentContext; }
  33
  34 using namespace ::cppu;
  35 using namespace ::com::sun::star::uno;
  36 using namespace ::com::sun::star::lang;
  37 using namespace ::com::sun::star::io;
  38
  39 // Implementation XTextOutputStream
  40
  41 namespace {
  42
  43 class OTextOutputStream : public WeakImplHelper< XTextOutputStream2, XServiceInfo >
  44 {
  45     Reference< XOutputStream > mxStream;
  46
  47     // Encoding
  48     bool mbEncodingInitialized;
  49     rtl_UnicodeToTextConverter  mConvUnicode2Text;
  50     rtl_UnicodeToTextContext    mContextUnicode2Text;
  51
  52     Sequence<sal_Int8> implConvert( const OUString& rSource );
  53     /// @throws IOException
  54     void checkOutputStream() const;
  55
  56 public:
  57     OTextOutputStream();
  58     virtual ~OTextOutputStream() override;
  59
  60     // Methods XTextOutputStream
  61     virtual void SAL_CALL writeString( const OUString& aString ) override;
  62     virtual void SAL_CALL setEncoding( const OUString& Encoding ) override;
  63
  64     // Methods XOutputStream
  65     virtual void SAL_CALL writeBytes( const Sequence< sal_Int8 >& aData ) override;
  66     virtual void SAL_CALL flush(  ) override;
  67     virtual void SAL_CALL closeOutput(  ) override;
  68
  69     // Methods XActiveDataSource
  70     virtual void SAL_CALL setOutputStream( const Reference< XOutputStream >& aStream ) override;
  71     virtual Reference< XOutputStream > SAL_CALL getOutputStream(  ) override;
  72
  73     // Methods XServiceInfo
  74         virtual OUString              SAL_CALL getImplementationName() override;
  75         virtual Sequence< OUString >  SAL_CALL getSupportedServiceNames() override;
  76         virtual sal_Bool              SAL_CALL supportsService(const OUString& ServiceName) override;
  77 };
  78
  79 }
  80
  81 OTextOutputStream::OTextOutputStream()
  82     : mbEncodingInitialized(false)
  83     , mConvUnicode2Text(nullptr)
  84     , mContextUnicode2Text(nullptr)
  85 {
  86 }
  87
  88 OTextOutputStream::~OTextOutputStream()
  89 {
  90     if( mbEncodingInitialized )
  91     {
  92         rtl_destroyUnicodeToTextContext( mConvUnicode2Text, mContextUnicode2Text );
  93         rtl_destroyUnicodeToTextConverter( mConvUnicode2Text );
  94     }
  95 }
  96
  97 Sequence<sal_Int8> OTextOutputStream::implConvert( const OUString& rSource )
  98 {
  99     const sal_Unicode *puSource = rSource.getStr();
 100     sal_Int32 nSourceSize = rSource.getLength();
 101
 102     sal_Size nTargetCount = 0;
 103     sal_Size nSourceCount = 0;
 104
 105     sal_uInt32 uiInfo;
 106     sal_Size nSrcCvtChars;
 107
 108     // take nSourceSize * 3 as preference
 109     // this is an upper boundary for converting to utf8,
 110     // which most often used as the target.
 111     sal_Int32 nSeqSize =  nSourceSize * 3;
 112
 113     Sequence<sal_Int8> seqText( nSeqSize );
 114     char *pTarget = reinterpret_cast<char *>(seqText.getArray());
 115     while( true )
 116     {
 117         nTargetCount += rtl_convertUnicodeToText(
 118                                     mConvUnicode2Text,
 119                                     mContextUnicode2Text,
 120                                     &( puSource[nSourceCount] ),
 121                                     nSourceSize - nSourceCount ,
 122                                     &( pTarget[nTargetCount] ),
 123                                     nSeqSize - nTargetCount,
 124                                     RTL_UNICODETOTEXT_FLAGS_UNDEFINED_DEFAULT |
 125                                     RTL_UNICODETOTEXT_FLAGS_INVALID_DEFAULT ,
 126                                     &uiInfo,
 127                                     &nSrcCvtChars);
 128         nSourceCount += nSrcCvtChars;
 129
 130         if( uiInfo & RTL_UNICODETOTEXT_INFO_DESTBUFFERTOSMALL )
 131         {
 132             nSeqSize *= 2;
 133             seqText.realloc( nSeqSize );  // double array size
 134             pTarget = reinterpret_cast<char*>(seqText.getArray());
 135             continue;
 136         }
 137         break;
 138     }
 139
 140     // reduce the size of the buffer (fast, no copy necessary)
 141     seqText.realloc( nTargetCount );
 142     return seqText;
 143 }
 144
 145
 146 // XTextOutputStream
 147
 148 void OTextOutputStream::writeString( const OUString& aString )
 149 {
 150     checkOutputStream();
 151     if( !mbEncodingInitialized )
 152     {
 153         setEncoding( u"utf8"_ustr );
 154     }
 155     if( !mbEncodingInitialized )
 156         return;
 157
 158     Sequence<sal_Int8> aByteSeq = implConvert( aString );
 159     mxStream->writeBytes( aByteSeq );
 160 }
 161
 162 void OTextOutputStream::setEncoding( const OUString& Encoding )
 163 {
 164     OString aOEncodingStr = OUStringToOString( Encoding, RTL_TEXTENCODING_ASCII_US );
 165     rtl_TextEncoding encoding = rtl_getTextEncodingFromMimeCharset( aOEncodingStr.getStr() );
 166     if( RTL_TEXTENCODING_DONTKNOW == encoding )
 167         return;
 168
 169     mbEncodingInitialized = true;
 170     mConvUnicode2Text   = rtl_createUnicodeToTextConverter( encoding );
 171     mContextUnicode2Text = rtl_createUnicodeToTextContext( mConvUnicode2Text );
 172 }
 173
 174
 175 // XOutputStream
 176 void OTextOutputStream::writeBytes( const Sequence< sal_Int8 >& aData )
 177 {
 178     checkOutputStream();
 179     mxStream->writeBytes( aData );
 180 }
 181
 182 void OTextOutputStream::flush(  )
 183 {
 184     checkOutputStream();
 185     mxStream->flush();
 186 }
 187
 188 void OTextOutputStream::closeOutput(  )
 189 {
 190     checkOutputStream();
 191     mxStream->closeOutput();
 192 }
 193
 194
 195 void OTextOutputStream::checkOutputStream() const
 196 {
 197     if (! mxStream.is() )
 198         throw IOException(u"output stream is not initialized, you have to use setOutputStream first"_ustr);
 199 }
 200
 201
 202 // XActiveDataSource
 203
 204 void OTextOutputStream::setOutputStream( const Reference< XOutputStream >& aStream )
 205 {
 206     mxStream = aStream;
 207 }
 208
 209 Reference< XOutputStream > OTextOutputStream::getOutputStream()
 210 {
 211     return mxStream;
 212 }
 213
 214 OUString OTextOutputStream::getImplementationName()
 215 {
 216     return u"com.sun.star.comp.io.TextOutputStream"_ustr;
 217 }
 218
 219 sal_Bool OTextOutputStream::supportsService(const OUString& ServiceName)
 220 {
 221     return cppu::supportsService(this, ServiceName);
 222 }
 223
 224 Sequence< OUString > OTextOutputStream::getSupportedServiceNames()
 225 {
 226     return { u"com.sun.star.io.TextOutputStream"_ustr };
 227 }
 228
 229
 230
 231 extern "C" SAL_DLLPUBLIC_EXPORT css::uno::XInterface*
 232 io_OTextOutputStream_get_implementation(
 233     css::uno::XComponentContext* , css::uno::Sequence<css::uno::Any> const&)
 234 {
 235     return cppu::acquire(new OTextOutputStream());
 236 }
 237
 238 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */