/*
 * Copyright 2009, Google Inc.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *
 *     * Redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above
 * copyright notice, this list of conditions and the following disclaimer
 * in the documentation and/or other materials provided with the
 * distribution.
 *     * Neither the name of Google Inc. nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */


// This file contains the main routine for the converter that writes
// out a scene graph as a JSON file.

#include <string>
#include <iostream>
#include <vector>

#include "base/at_exit.h"
#include "base/command_line.h"
#include "base/file_path.h"
#include "base/file_util.h"
#include "base/string_util.h"
#include "converter/cross/converter.h"
#include "utils/cross/file_path_utils.h"

#if defined(OS_WIN)
int wmain(int argc, wchar_t **argv) {
  // On Windows, CommandLine::Init ignores its arguments and uses
  // GetCommandLineW.
  CommandLine::Init(0, NULL);
#endif
#if defined(OS_LINUX)
int main(int argc, char **argv) {
  CommandLine::Init(argc, argv);
#endif
#if defined(OS_MACOSX)
// The "real" main on Mac is in mac/converter_main.mm, so we can get
// memory pool initialization for Cocoa.
int CrossMain(int argc, char**argv) {
  CommandLine::Init(argc, argv);
#endif
  // Create an at_exit_manager so that base singletons will get
  // deleted properly.
  base::AtExitManager at_exit_manager;
  const CommandLine* command_line = CommandLine::ForCurrentProcess();

  FilePath in_filename, out_filename;
  // Use the absolute path to the converter tool for the case that we
  // are in the current working directory and "." is not on the PATH.
  FilePath converter_dir = FilePath(argv[0]).DirName();
  file_util::AbsolutePath(&converter_dir);
  const FilePath converter_tool = converter_dir.Append(
      o3d::UTF8ToFilePath("convert.py"));


  std::vector<std::wstring> values = command_line->GetLooseValues();
  if (values.size() == 1) {
    // If we're only given one argument, then construct the output
    // filename by substituting the extension on the filename (if any)
    // with .o3dtgz.
    in_filename = o3d::WideToFilePath(values[0]);
    out_filename = in_filename.ReplaceExtension(FILE_PATH_LITERAL(".o3dtgz"));
  } else if (values.size()== 2) {
    in_filename = o3d::WideToFilePath(values[0]);
    out_filename = o3d::WideToFilePath(values[1]);
  } else {
    std::cerr << "Usage: " << argv[0]
              << " [ options ] <infile.dae> [ <outfile> ]\n";
    std::cerr
        << "--no-condition\n"
        << "    Stops the converter from conditioning shaders.\n"
        << "--base-path=<path>\n"
        << "    Sets the path to remove from URIs of external files\n"
        << "--asset-paths=<comma separted list of paths>\n"
        << "    Sets the paths for finding textures and other external\n"
        << "    files.\n"
        << "--up-axis=x,y,z\n"
        << "    Converts the file to have this up axis.\n"
        << "--pretty-print\n"
        << "    Makes the exported JSON easier to read.\n"
        << "--keep-filters\n"
        << "    Stops the converter from forcing all texture samplers to use\n"
        << "    tri-linear filtering.\n"
        << "--keep-materials\n"
        << "    Stops the converter from changing materials to <constant> if\n"
        << "    they are used by a mesh that has no normals.\n"
        << "--no-binary\n"
        << "    Use JSON for buffers, skins, curves instead of binary\n"
        << "--no-archive\n"
        << "    Don't make a gzipped tar file, just flat files. Still takes\n"
        << "    the name of an archive file; for archive.o3dtgz, creates\n"
        << "    directory named archive/ and writes files inside.\n"
        << "--convert-dds-to-png\n"
        << "    Convert all DDS textures to PNGs. For cube map textures,\n"
        << "    writes six separate PNGs with suffixes _posx, _negx, etc.\n"
        << "--convert-cg-to-glsl\n"
        << "    Convert shaders using an external tool.\n"
        << "    Requires python on PATH.\n"
        << "--converter-tool=<filename> [default: "
        << converter_tool.value() << "]\n"
        << "    Specifies the shader converter tool.\n";
    return EXIT_FAILURE;
  }

  o3d::converter::Options options;
  options.condition = !command_line->HasSwitch("no-condition");
  options.pretty_print = command_line->HasSwitch("pretty-print");
  options.binary = !command_line->HasSwitch("no-binary");
  options.archive = !command_line->HasSwitch("no-archive");
  options.convert_dds_to_png = command_line->HasSwitch("convert-dds-to-png");
  options.convert_cg_to_glsl = command_line->HasSwitch("convert-cg-to-glsl");
  options.converter_tool = command_line->HasSwitch("converter-tool") ?
      command_line->GetSwitchValuePath("converter-tool") : converter_tool;
  if (command_line->HasSwitch("base-path")) {
    options.base_path = command_line->GetSwitchValuePath("base-path");
  }
  if (command_line->HasSwitch("asset-paths")) {
    std::vector<CommandLine::StringType> paths;
    SplitString(command_line->GetSwitchValueNative("asset-paths"), ',', &paths);
    for (size_t ii = 0; ii < paths.size(); ++ii) {
      options.file_paths.push_back(FilePath(paths[ii]));
    }
  }
  if (command_line->HasSwitch("up-axis")) {
    std::string up_axis_string = command_line->GetSwitchValueASCII("up-axis");
    int x, y, z;
    if (sscanf(up_axis_string.c_str(), "%d,%d,%d", &x, &y, &z) != 3) {
      std::cerr << "Invalid --up-axis value. Should be --up-axis=x,y,z\n";
      return EXIT_FAILURE;
    }
    options.up_axis = o3d::Vector3(static_cast<float>(x),
                                   static_cast<float>(y),
                                   static_cast<float>(z));
  }

  o3d::String error_messages;
  bool result = o3d::converter::Convert(in_filename,
                                        out_filename,
                                        options,
                                        &error_messages);
  if (result) {
    std::cerr << "Converted '" << o3d::FilePathToUTF8(in_filename).c_str()
              << "' to '" << o3d::FilePathToUTF8(out_filename).c_str()
              << "'." << std::endl;
    return EXIT_SUCCESS;
  } else {
    std::cerr << error_messages.c_str() << std::endl;
    std::cerr << "FAILED to convert '"
              << o3d::FilePathToUTF8(in_filename).c_str()
              << "' to '" << o3d::FilePathToUTF8(out_filename).c_str()
              << "'." << std::endl;
    return EXIT_FAILURE;
  }
}