//
// ********************************************************************
// * 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.          *
// ********************************************************************
//
// G4DataVector class implementation
//
// Author: H.Kurashige, 18 September 2001
// --------------------------------------------------------------------

#include "G4DataVector.hh"
#include <iomanip>

// --------------------------------------------------------------------
G4DataVector::G4DataVector(std::size_t cap)
  : std::vector<G4double>(cap, 0.0)
{}

// --------------------------------------------------------------------
G4DataVector::G4DataVector(std::size_t cap, G4double value)
  : std::vector<G4double>(cap, value)
{}

// --------------------------------------------------------------------
G4bool G4DataVector::Store(std::ofstream& fOut, G4bool ascii)
{
  // Ascii mode
  if(ascii)
  {
    fOut << *this;
    return true;
  }

  // Binary Mode
  auto sizeV = G4int(size());
  fOut.write((char*) (&sizeV), sizeof sizeV);

  auto value = new G4double[sizeV];
  std::size_t i   = 0;
  for(auto itr = cbegin(); itr != cend(); ++itr, ++i)
  {
    value[i] = *itr;
  }
  fOut.write((char*) (value), sizeV * (sizeof(G4double)));
  delete[] value;

  return true;
}

// --------------------------------------------------------------------
G4bool G4DataVector::Retrieve(std::ifstream& fIn, G4bool ascii)
{
  clear();
  G4int sizeV = 0;

  // retrieve in ascii mode
  if(ascii)
  {
    // contents
    fIn >> sizeV;
    if(fIn.fail())
    {
      return false;
    }
    if(sizeV <= 0)
    {
#ifdef G4VERBOSE
      G4cerr << "G4DataVector::Retrieve():";
      G4cerr << " Invalid vector size: " << sizeV << G4endl;
#endif
      return false;
    }

    reserve(sizeV);
    for(G4int i = 0; i < sizeV; ++i)
    {
      G4double vData = 0.0;
      fIn >> vData;
      if(fIn.fail())
      {
        return false;
      }
      push_back(vData);
    }
    return true;
  }

  // retrieve in binary mode
  fIn.read((char*) (&sizeV), sizeof sizeV);

  auto value = new G4double[sizeV];
  fIn.read((char*) (value), sizeV * (sizeof(G4double)));
  if(G4int(fIn.gcount()) != G4int(sizeV * (sizeof(G4double))))
  {
    delete[] value;
    return false;
  }

  reserve(sizeV);
  for(G4int i = 0; i < sizeV; ++i)
  {
    push_back(value[i]);
  }
  delete[] value;
  return true;
}

// --------------------------------------------------------------------
std::ostream& operator<<(std::ostream& out, const G4DataVector& pv)
{
  out << pv.size() << std::setprecision(12) << G4endl;
  for(std::size_t i = 0; i < pv.size(); ++i)
  {
    out << pv[i] << G4endl;
  }
  out << std::setprecision(6);

  return out;
}
