sandboxed-api/oss-internship-2020/gdal/raster_to_gtiff/raster_to_gtiff.cc

// Copyright 2020 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 <cstdlib>
#include <iostream>
#include <optional>
#include <string>

#include "get_raster_data.h"  // NOLINT(build/include)
#include "gtiff_converter.h"  // NOLINT(build/include)
#include "sandboxed_api/util/fileops.h"
#include "sandboxed_api/util/path.h"
#include "utils.h"  // NOLINT(build/include)

namespace {

absl::Status SaveToGTiff(gdal::sandbox::parser::RasterDataset bands_data,
                         std::string out_file) {
  std::optional<std::string> proj_db_path =
      gdal::sandbox::utils::FindProjDbPath();

  if (proj_db_path == std::nullopt) {
    return absl::FailedPreconditionError("Specified proj.db does not exist");
  }

  gdal::sandbox::RasterToGTiffProcessor processor(
      std::move(out_file), std::move(proj_db_path.value()),
      std::move(bands_data));

  return processor.Run();
}

void Usage() {
  std::cerr << "Example application that converts raster data to GTiff"
               " format inside the sandbox. Usage:\n"
               "raster_to_gtiff input_filename output_filename\n"
               "output_filename must be absolute"
            << std::endl;
}

}  // namespace

int main(int argc, char** argv) {
  if (argc < 3 || !sandbox2::file::IsAbsolutePath(argv[2])) {
    Usage();
    return EXIT_FAILURE;
  }

  std::string input_data_path = std::string(argv[1]);
  std::string output_data_path = std::string(argv[2]);

  if (absl::Status status = gdal::sandbox::SaveToGTiff(
          gdal::sandbox::parser::GetRasterBandsFromFile(
              std::move(input_data_path)),
          std::move(output_data_path));
      !status.ok()) {
    std::cerr << status.ToString() << std::endl;
    return EXIT_FAILURE;
  }

  return EXIT_SUCCESS;
}