src/rawCompression.cpp

// Copyright 2019 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include <algorithm>
#include <memory>
#include <string>
#include <utility>
#include <vector>

#include <boost/gil/image.hpp>
#include "src/rawCompression.h"

struct PixelInserter {
  std::vector<uint8_t> *storage;
  explicit PixelInserter(std::vector<uint8_t> *s) : storage(s) {}
  void operator()(boost::gil::rgb8_pixel_t p) const {
    using boost::gil::at_c;
    storage->push_back(at_c<0>(p));
    storage->push_back(at_c<1>(p));
    storage->push_back(at_c<2>(p));
  }
};

std::string RawCompression::toString() const {
  return std::string("RAW");
}

DCM_Compression RawCompression::method() const {
  return RAW;
}

std::unique_ptr<uint8_t[]> RawCompression::compress(
    const boost::gil::rgb8_view_t &view, size_t *size) {
  return getRawData(view, size);
}

std::unique_ptr<uint8_t[]> RawCompression::getRawData(
    const boost::gil::rgb8_view_t &view, size_t *size) {
  std::vector<uint8_t> storage;
  storage.reserve(view.width() * view.height() *
                  boost::gil::num_channels<boost::gil::rgb8_image_t>());
  for_each_pixel(view, PixelInserter(&storage));
  std::unique_ptr<uint8_t[]> output =
      std::make_unique<uint8_t[]>(storage.size());
  std::move(storage.begin(), storage.end(), output.get());
  *size = storage.size();
  return output;
}