StreamExporter.cpp

#include "StreamExporter.h"

#include "Buffer.h"
#include "ExportDescription.h"
#include "ProcessExport.h"

using namespace Isis;
using namespace std;


namespace Isis {
  StreamExporter::StreamExporter() : ImageExporter() {
  }


  StreamExporter::~StreamExporter() {
  }


  void StreamExporter::setGrayscale(ExportDescription &desc) {
    initialize(desc);
  }


  void StreamExporter::setRgb(ExportDescription &desc) {
    initialize(desc);
  }


  void StreamExporter::setRgba(ExportDescription &desc) {
    initialize(desc);
  }


  void StreamExporter::initialize(ExportDescription &desc) {
    ImageExporter::initialize(desc);
    createBuffer();
  }


  void StreamExporter::writeGrayscale(vector<Buffer *> &in) const {
    Buffer &grayLine = *in[0];

    int lineIndex = grayLine.Line() - 1;
    for (int sampleIndex = 0; sampleIndex < grayLine.SampleDimension(); sampleIndex++) {
      int pixelValue = outputPixelValue(grayLine[sampleIndex]);
      setBuffer(sampleIndex, 0, pixelValue);
    }

    writeLine(lineIndex);
  }


  void StreamExporter::writeRgb(vector<Buffer *> &in) const {
    Buffer &redLine = *in[0];
    Buffer &greenLine = *in[1];
    Buffer &blueLine = *in[2];

    for (int i = 0; i < redLine.SampleDimension(); i++) {
      int red = outputPixelValue(redLine[i]);
      int green = outputPixelValue(greenLine[i]);
      int blue = outputPixelValue(blueLine[i]);

      setBuffer(i, 0, red);
      setBuffer(i, 1, green);
      setBuffer(i, 2, blue);
    }

    writeLine(redLine.Line() - 1);
  }


  void StreamExporter::writeRgba(vector<Buffer *> &in) const {
    Buffer &redLine = *in[0];
    Buffer &greenLine = *in[1];
    Buffer &blueLine = *in[2];
    Buffer &alphaLine = *in[3];

    for (int i = 0; i < redLine.SampleDimension(); i++) {
      int red = outputPixelValue(redLine[i]);
      int green = outputPixelValue(greenLine[i]);
      int blue = outputPixelValue(blueLine[i]);
      int alpha = outputPixelValue(alphaLine[i]);

      setBuffer(i, 0, red);
      setBuffer(i, 1, green);
      setBuffer(i, 2, blue);
      setBuffer(i, 3, alpha);
    }

    writeLine(redLine.Line() - 1);
  }
};