//
// ********************************************************************
// * License and Disclaimer                                           *
// *                                                                  *
// * The  Geant4 software  is  copyright of the Copyright Holders  of *
// * the Geant4 Collaboration.  It is provided  under  the terms  and *
// * conditions of the Geant4 Software License,  included in the file *
// * LICENSE and available at  http://cern.ch/geant4/license .  These *
// * include a list of copyright holders.                             *
// *                                                                  *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work  make  any representation or  warranty, express or implied, *
// * regarding  this  software system or assume any liability for its *
// * use.  Please see the license in the file  LICENSE  and URL above *
// * for the full disclaimer and the limitation of liability.         *
// *                                                                  *
// * This  code  implementation is the result of  the  scientific and *
// * technical work of the GEANT4 collaboration.                      *
// * By using,  copying,  modifying or  distributing the software (or *
// * any work based  on the software)  you  agree  to acknowledge its *
// * use  in  resulting  scientific  publications,  and indicate your *
// * acceptance of all terms of the Geant4 Software license.          *
// ********************************************************************
//
//
// Author: Ivana Hrivnacova, 15/09/2020 (ivana@ipno.in2p3.fr)

#include "G4AnalysisUtilities.hh"

#include "toolx/hdf5/header"
#include "toolx/hdf5/h2file"
#include "tools/histo/h1d"
#include "tools/histo/h2d"
#include "tools/histo/h3d"
#include "tools/histo/p1d"
#include "tools/histo/p2d"

//_____________________________________________________________________________
template <>
inline
G4bool G4Hdf5HnFileManager<tools::histo::p1d>::WriteImpl(
  hid_t hdirectory, tools::histo::p1d* ht, const G4String& htName)
{
  return toolx::hdf5::write_profile(G4cout, hdirectory, htName, *ht);
}

//_____________________________________________________________________________
template <>
inline
G4bool G4Hdf5HnFileManager<tools::histo::p2d>::WriteImpl(
  hid_t hdirectory, tools::histo::p2d* ht, const G4String& htName)
{
  return toolx::hdf5::write_profile(G4cout, hdirectory, htName, *ht);
}

//_____________________________________________________________________________
template <typename HT>
inline
G4bool G4Hdf5HnFileManager<HT>::WriteImpl(
  hid_t hdirectory, HT* ht, const G4String& htName)
{
  return toolx::hdf5::write_histo(G4cout, hdirectory, htName, *ht);
}

//_____________________________________________________________________________
template <typename HT>
inline
G4bool G4Hdf5HnFileManager<HT>::WriteExtra(
  HT* ht, const G4String& htName, const G4String& fileName)
{
  // create a new file
  // create new file
  hid_t hfile = ::H5Fcreate(fileName, H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT);

  // Do nothing if there is no file
  // (te error should be handled by caller)
  if ( hfile < 0 ) {
    G4cerr << "::H5Fcreate failed " << G4endl;
    return false;
  }

  // create a header with general infos :
  if(!toolx::hdf5::write_header(hfile)) {
    G4cerr << "toolx::hdf5::write_header() failed." << std::endl;
    return false;
  }

  // Create directories
  hid_t hdirectory = toolx_H5Gcreate(hfile, "histograms", 0);
  if ( hdirectory < 0 ) {
    G4cerr << "toolx_H5Gcreate failed " << G4endl;
    return false;
  }

  auto result = toolx::hdf5::write_atb(hdirectory, "type", "directory");
  if ( ! result) {
    G4cerr << "toolx::hdf5::write_atb failed " << G4endl;
    return false;
  }

  result = WriteImpl(hdirectory, ht, htName);
  if ( ! result) {
    G4cerr << "Writing HT failed" << G4endl;
    return false;
  }

  // close file
  ::H5Gclose(hdirectory);
  ::H5Gclose(hfile);
  return true;
}

//_____________________________________________________________________________
template <typename HT>
inline
G4bool G4Hdf5HnFileManager<HT>::Write(
  HT* ht, const G4String& htName, G4String& fileName)
{
  if (fileName.empty()) {
    // should not happen
    G4cerr << "!!! Hdf5 file name not defined." << G4endl;
    G4cerr << "!!! Write " << htName << " failed." << G4endl;
    return false;
  }

  auto hdirectory = std::get<1>(*fFileManager->GetTFile(fileName));
    if ( hdirectory < 0 ) {
    G4Analysis::Warn(
      "Failed to get Hdf5 file " + fileName + " histo directory.",
      fkClass, "Write");
    return false;
  }

  auto result = WriteImpl(hdirectory, ht, htName);
  fFileManager->LockDirectoryNames();
  return result;
}