/**
 * 
 */
package at.gv.egiz.pdfas.impl.output;

import java.io.ByteArrayOutputStream;
import java.io.OutputStream;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import at.gv.egiz.pdfas.framework.output.DataSink;
import at.gv.egiz.pdfas.performance.PerformanceCounters;

/**
 * @author wprinz
 *
 */
public class ByteArrayDataSink implements DataSink
{
  /**
   * The log.
   */
  private static final Log log = LogFactory.getLog(ByteArrayDataSink.class);

  protected String mimeType = null;
  protected String characterEncoding = null;
  
  protected ByteArrayOutputStream baos = null;

  
  public ByteArrayDataSink()
  {
    PerformanceCounters.byteArrays.increment();
  }

  /**
   * @see at.gv.egiz.pdfas.framework.output.DataSink#createOutputStream(java.lang.String)
   */
  public OutputStream createOutputStream(String mimeType)
  {
    return createOutputStream(mimeType, null);
  }

  /**
   * @see at.gv.egiz.pdfas.framework.output.DataSink#createOutputStream(java.lang.String, java.lang.String)
   */
  public OutputStream createOutputStream(String mimeType, String characterEncoding)
  {
    this.mimeType = mimeType;
    this.characterEncoding = characterEncoding;
    
    if (this.baos != null)
    {
      log.warn("An output stream is created twice. The old one will be rendered useless.");
    }
    this.baos = new ByteArrayOutputStream(4096);
    return this.baos;
  }
  
  /**
   * Returns the byte array.
   * @return Returns the byte array.
   */
  public byte [] getByteArray ()
  {
    return this.baos.toByteArray();
  }

  /**
   * @return the mimeType
   */
  public String getMimeType()
  {
    return this.mimeType;
  }

  /**
   * @return the characterEncoding
   */
  public String getCharacterEncoding()
  {
    return this.characterEncoding;
  }

}