/*
This test does a read write cycle of a known good hypertree grid file,
comparing the shape of read and written trees, to exercise the reader
and writer and ensure that they work.
*/

#include "vtkDataArray.h"
#include "vtkHyperTreeGrid.h"
#include "vtkNew.h"
#include "vtkTestUtilities.h"
#include "vtkXMLHyperTreeGridReader.h"
#include "vtkXMLHyperTreeGridWriter.h"

#include <string>
#include <vtksys/SystemTools.hxx>

#define VTK_SUCCESS 0
#define VTK_FAILURE 1
int TestXMLHyperTreeGridIO(int argc, char* argv[])
{
  char* temp_dir =
    vtkTestUtilities::GetArgOrEnvOrDefault("-T", argc, argv, "VTK_TEMP_DIR", "Testing/Temporary");
  if (!temp_dir)
  {
    cerr << "Could not determine temporary directory." << endl;
    return VTK_FAILURE;
  }

  char* data_dir = vtkTestUtilities::GetDataRoot(argc, argv);
  if (!data_dir)
  {
    delete[] temp_dir;
    cerr << "Could not determine data directory." << endl;
    return VTK_FAILURE;
  }

  std::string fname = "HyperTreeGrid.1.htg";

  std::string ifname = std::string(data_dir) + std::string("/Data/") + fname;

  cout << "- READ INPUT --------------------------------" << endl;
  vtkNew<vtkXMLHyperTreeGridReader> reader;
  reader->SetFileName(ifname.c_str());
  reader->Update();

  vtkHyperTreeGrid* read_in = reader->GetOutput();
  int size[3];
  vtkDataArray* coords;
  int ntp;

  std::string read1;

  read_in->GetDimensions(size);
  read1 += "SIZE " + std::to_string(size[0]) + "," + std::to_string(size[1]) + "," +
    std::to_string(size[2]) + "\n";
  read1 += "DIMS " + std::to_string(read_in->GetDimension()) + "\n";
  read1 += "#TREES " + std::to_string(read_in->GetMaxNumberOfTrees()) + "\n";
  read1 += "ORIENTATION " + std::to_string(read_in->GetOrientation()) + "\n";
  read1 += "BRANCHFACTOR " + std::to_string(read_in->GetBranchFactor()) + "\n";
  coords = read_in->GetXCoordinates();
  ntp = coords->GetNumberOfTuples();
  read1 += "XCOORDS " + std::to_string(ntp) + ":" + std::to_string(coords->GetTuple1(0)) + "..." +
    std::to_string(coords->GetTuple1(ntp - 1)) + "\n";
  coords = read_in->GetYCoordinates();
  ntp = coords->GetNumberOfTuples();
  read1 += "YCOORDS " + std::to_string(ntp) + ":" + std::to_string(coords->GetTuple1(0)) + "..." +
    std::to_string(coords->GetTuple1(ntp - 1)) + "\n";
  coords = read_in->GetZCoordinates();
  ntp = coords->GetNumberOfTuples();
  read1 += "ZCOORDS " + std::to_string(ntp) + ":" + std::to_string(coords->GetTuple1(0)) + "..." +
    std::to_string(coords->GetTuple1(ntp - 1)) + "\n";
  read1 += "TRANSPOSED " + std::to_string(read_in->GetTransposedRootIndexing()) + "\n";
  read1 += "#CHILDREN " + std::to_string(read_in->GetNumberOfChildren()) + "\n";
  read1 += "#LEVELS " + std::to_string(read_in->GetNumberOfLevels()) + "\n";
  read1 += "#VERTS " + std::to_string(read_in->GetNumberOfCells()) + "\n";
  read1 += "#LEAVES " + std::to_string(read_in->GetNumberOfLeaves()) + "\n";
  cout << read1 << endl;

  std::string output_dir = temp_dir;
  output_dir += "/HTG";

  vtksys::SystemTools::MakeDirectory(output_dir);

  std::string ofname = output_dir + "/" + fname;

  cout << "- WRITE BINARY --------------------------------" << endl;

  vtkNew<vtkXMLHyperTreeGridWriter> writer;
  writer->SetFileName(ofname.c_str());
  writer->SetInputData(read_in);
  writer->SetDataModeToBinary();
  writer->Write();

  vtkNew<vtkXMLHyperTreeGridReader> reader2;
  reader2->SetFileName(ofname.c_str());
  reader2->Update();
  vtkHyperTreeGrid* wrote_out = reader2->GetOutput();

  std::string read2;
  wrote_out->GetDimensions(size);
  read2 += "SIZE " + std::to_string(size[0]) + "," + std::to_string(size[1]) + "," +
    std::to_string(size[2]) + "\n";
  read2 += "DIMS " + std::to_string(wrote_out->GetDimension()) + "\n";
  read2 += "#TREES " + std::to_string(wrote_out->GetMaxNumberOfTrees()) + "\n";
  read2 += "ORIENTATION " + std::to_string(wrote_out->GetOrientation()) + "\n";
  read2 += "BRANCHFACTOR " + std::to_string(wrote_out->GetBranchFactor()) + "\n";
  coords = wrote_out->GetXCoordinates();
  ntp = coords->GetNumberOfTuples();
  read2 += "XCOORDS " + std::to_string(ntp) + ":" + std::to_string(coords->GetTuple1(0)) + "..." +
    std::to_string(coords->GetTuple1(ntp - 1)) + "\n";
  coords = wrote_out->GetYCoordinates();
  ntp = coords->GetNumberOfTuples();
  read2 += "YCOORDS " + std::to_string(ntp) + ":" + std::to_string(coords->GetTuple1(0)) + "..." +
    std::to_string(coords->GetTuple1(ntp - 1)) + "\n";
  coords = wrote_out->GetZCoordinates();
  ntp = coords->GetNumberOfTuples();
  read2 += "ZCOORDS " + std::to_string(ntp) + ":" + std::to_string(coords->GetTuple1(0)) + "..." +
    std::to_string(coords->GetTuple1(ntp - 1)) + "\n";
  read2 += "TRANSPOSED " + std::to_string(wrote_out->GetTransposedRootIndexing()) + "\n";
  read2 += "#CHILDREN " + std::to_string(wrote_out->GetNumberOfChildren()) + "\n";
  read2 += "#LEVELS " + std::to_string(wrote_out->GetNumberOfLevels()) + "\n";
  read2 += "#VERTS " + std::to_string(wrote_out->GetNumberOfCells()) + "\n";
  read2 += "#LEAVES " + std::to_string(wrote_out->GetNumberOfLeaves()) + "\n";
  cout << read2 << endl;

  cout << "- WRITE APPENDED --------------------------------" << endl;

  writer->SetDataModeToAppended();
  writer->Write();

  reader2->Update();
  wrote_out = reader2->GetOutput();

  std::string read3;
  wrote_out->GetDimensions(size);
  read3 += "SIZE " + std::to_string(size[0]) + "," + std::to_string(size[1]) + "," +
    std::to_string(size[2]) + "\n";
  read3 += "DIMS " + std::to_string(wrote_out->GetDimension()) + "\n";
  read3 += "#TREES " + std::to_string(wrote_out->GetMaxNumberOfTrees()) + "\n";
  read3 += "ORIENTATION " + std::to_string(wrote_out->GetOrientation()) + "\n";
  read3 += "BRANCHFACTOR " + std::to_string(wrote_out->GetBranchFactor()) + "\n";
  coords = wrote_out->GetXCoordinates();
  ntp = coords->GetNumberOfTuples();
  read3 += "XCOORDS " + std::to_string(ntp) + ":" + std::to_string(coords->GetTuple1(0)) + "..." +
    std::to_string(coords->GetTuple1(ntp - 1)) + "\n";
  coords = wrote_out->GetYCoordinates();
  ntp = coords->GetNumberOfTuples();
  read3 += "YCOORDS " + std::to_string(ntp) + ":" + std::to_string(coords->GetTuple1(0)) + "..." +
    std::to_string(coords->GetTuple1(ntp - 1)) + "\n";
  coords = wrote_out->GetZCoordinates();
  ntp = coords->GetNumberOfTuples();
  read3 += "ZCOORDS " + std::to_string(ntp) + ":" + std::to_string(coords->GetTuple1(0)) + "..." +
    std::to_string(coords->GetTuple1(ntp - 1)) + "\n";
  read3 += "TRANSPOSED " + std::to_string(wrote_out->GetTransposedRootIndexing()) + "\n";
  read3 += "#CHILDREN " + std::to_string(wrote_out->GetNumberOfChildren()) + "\n";
  read3 += "#LEVELS " + std::to_string(wrote_out->GetNumberOfLevels()) + "\n";
  read3 += "#VERTS " + std::to_string(wrote_out->GetNumberOfCells()) + "\n";
  read3 += "#LEAVES " + std::to_string(wrote_out->GetNumberOfLeaves()) + "\n";
  cout << read3 << endl;

  bool ret = VTK_FAILURE;
  if (read1 == read2 && read1 == read3)
  {
    ret = VTK_SUCCESS;
    vtksys::SystemTools::RemoveADirectory(output_dir);
  }
  else
  {
    cerr << "Problem: Written file does not match read in file." << endl;
  }

  delete[] temp_dir;
  delete[] data_dir;

  return ret;
}