/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 * Copyright by The HDF Group.                                               *
 * Copyright by the Board of Trustees of the University of Illinois.         *
 * All rights reserved.                                                      *
 *                                                                           *
 * This file is part of HDF5.  The full HDF5 copyright notice, including     *
 * terms governing use, modification, and redistribution, is contained in    *
 * the files COPYING and Copyright.html.  COPYING can be found at the root   *
 * of the source code distribution tree; Copyright.html can be found at the  *
 * root level of an installed copy of the electronic HDF5 document set and   *
 * is linked from the top-level documents page.  It can also be found at     *
 * http://hdfgroup.org/HDF5/doc/Copyright.html.  If you do not have          *
 * access to either file, you may request a copy from help@hdfgroup.org.     *
 * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */

#include <string>

#include "H5Include.h"
#include "H5Exception.h"
#include "H5IdComponent.h"
#include "H5PropList.h"
#include "H5Object.h"
#include "H5OcreatProp.h"
#include "H5DcreatProp.h"
#include "H5CommonFG.h"
#include "H5DataType.h"
#include "H5ArrayType.h"

#ifndef H5_NO_NAMESPACE
namespace H5 {
#endif

//--------------------------------------------------------------------------
// Function:	ArrayType default constructor
///\brief	Default constructor: Creates a stub ArrayType
// Programmer	Binh-Minh Ribler - May 2004
//--------------------------------------------------------------------------
ArrayType::ArrayType() : DataType(), rank(-1), dimensions(NULL) {}

//--------------------------------------------------------------------------
// Function:	ArrayType overloaded constructor
///\brief	Creates an ArrayType object using an existing id.
///\param	existing_id - IN: Id of an existing datatype
///\exception	H5::DataTypeIException
// Programmer	Binh-Minh Ribler - May 2004
//--------------------------------------------------------------------------
ArrayType::ArrayType( const hid_t existing_id ) : DataType( existing_id )
{
    setArrayInfo();
}

//--------------------------------------------------------------------------
// Function:	ArrayType copy constructor
///\brief	Copy constructor: makes a copy of the original ArrayType object.
// Programmer	Binh-Minh Ribler - May 2004
//--------------------------------------------------------------------------
ArrayType::ArrayType( const ArrayType& original ) : DataType( original ), rank(original.rank)
{
    // Allocate space then copy the dimensions from the original array
    dimensions = new hsize_t[rank];
    for (int i = 0; i < rank; i++)
	dimensions[i] = original.dimensions[i];
}

//--------------------------------------------------------------------------
// Function:	ArrayType overloaded constructor
///\brief	Creates a new array data type based on the specified
///		\a base_type.
///\param	base_type - IN: Existing datatype
///\param	ndims     - IN: Rank of the array, [0..H5S_MAX_RANK]
///\param	dims      - IN: Size of each array dimension
///\exception	H5::DataTypeIException
// Programmer	Binh-Minh Ribler - May 2004
//--------------------------------------------------------------------------
ArrayType::ArrayType(const DataType& base_type, int ndims, const hsize_t* dims) : DataType()
{
    // Call C API to create an array data type
    hid_t new_type_id = H5Tarray_create2(base_type.getId(), ndims, dims);
    if (new_type_id < 0)
	throw DataTypeIException("ArrayType constructor", "H5Tarray_create2 failed");

    // Set the id and rank for this object
    id = new_type_id;
    rank = ndims;

    // Allocate space then set the dimensions as provided by caller
    dimensions = new hsize_t[rank];
    for (int i = 0; i < rank; i++)
	dimensions[i] = dims[i];
}

//--------------------------------------------------------------------------
// Function:	ArrayType::operator=
///\brief	Assignment operator
///\param	rhs - IN: Reference to the existing array datatype
///\return	Reference to ArrayType instance
///\exception	H5::DataTypeIException
///		std::bad_alloc
// Description
// 		Closes the id on the lhs object first with setId, then copies
//		each data member from the rhs object.
// Programmer	Binh-Minh Ribler - Mar 2016
// Modification
//--------------------------------------------------------------------------
ArrayType& ArrayType::operator=(const ArrayType& rhs)
{
    if (this != &rhs)
    {
        // handling references to this id
        try {
            setId(rhs.id);
            // Note: a = b, so there are two objects with the same hdf5 id
            // that's why incRefCount is needed, and it is called by setId
        }
        catch (Exception close_error) {
            throw DataTypeIException(inMemFunc("operator="), close_error.getDetailMsg());
        }

	// Copy the rank of the rhs array
	rank = rhs.rank;

	// Allocate space then copy the dimensions from the rhs array
	dimensions = new hsize_t[rank];
	for (int i = 0; i < rank; i++)
	    dimensions[i] = rhs.dimensions[i];
    }
    return(*this);
}

//--------------------------------------------------------------------------
// Function:	ArrayType::setArrayInfo
///\brief	Retrieves the rank and dimensions from the array datatype
///		and store the info in this ArrayType object.
///\exception	H5::DataTypeIException
// Programmer	Binh-Minh Ribler - January 2016
//--------------------------------------------------------------------------
void ArrayType::setArrayInfo()
{
    // Get the rank of the array type specified by id from the C API
    int ndims = H5Tget_array_ndims(id);
    if (ndims < 0)
    {
	throw DataTypeIException("ArrayType::setArrayInfo", "H5Tget_array_ndims failed");
    }

    // Get the dimensions from the C API
    hsize_t* dims;
    dims = new hsize_t[ndims];
    if (dims != NULL)
    {
	// Get the dimensions
	ndims = H5Tget_array_dims2(id, dims);
	if (ndims < 0)
	    throw DataTypeIException("ArrayType::setArrayInfo", "H5Tget_array_dims2 failed");

	// Store the array's info in memory
	rank = ndims;
	dimensions = new hsize_t[rank];
	for (int i = 0; i < rank; i++)
	    dimensions[i] = dims[i];
	delete []dims;
    }
} // setArrayInfo

//--------------------------------------------------------------------------
// Function:	ArrayType::getArrayNDims
///\brief	Returns the number of dimensions for an array datatype.
///\return	Number of dimensions
///\exception	H5::DataTypeIException
// Programmer	Binh-Minh Ribler - May 2004
// Modification
//		Modified to use setArrayInfo().
//		If rank is positive, return rank
//		If rank is invalid but object has a valid identifier, obtain the
//		  rank and dimensions, store them in the object, and return rank
//		Otherwise, i.e., rank is invalid and object doesn't have a
//		  valid identifier, throw an exception
//--------------------------------------------------------------------------
int ArrayType::getArrayNDims()
{
    // Validate the id first, this object could be a default object
    if (!p_valid_id(id))
	throw DataTypeIException("ArrayType::getArrayNDims", "ArrayType object is not a valid array type.");

    // If the array's info has not been stored, i.e. "rank" still has its
    // initial value, -1, and "dimensions" is still NULL, retrieve rank and
    // dimensions via the C API and store them in this ArrayType object.
    if (rank < 0 && dimensions == NULL)
	setArrayInfo();

    return(rank);
}

//--------------------------------------------------------------------------
// Function:	ArrayType::getArrayDims
///\brief	Retrieves the size of all dimensions of an array datatype.
///\param	dims - OUT: Sizes of dimensions
///\return	Number of dimensions
///\exception	H5::DataTypeIException
// Programmer	Binh-Minh Ribler - May 2004
// Modification
//	Jan, 2016
//		Modified to use setArrayInfo().
//		If the array information has not been stored, retrieve rank and
//		dimensions of the array type identified by "id" via the C API.
//		Copy "dimensions" to the user's buffer
//--------------------------------------------------------------------------
int ArrayType::getArrayDims(hsize_t* dims)
{
    // Validate the id first, this object could be a default object
    if (!p_valid_id(id))
	throw DataTypeIException("ArrayType::getArrayDims", "ArrayType object is not a valid array type.");

    // If the array's info has not been stored, i.e. "rank" still has its
    // initial value, -1, and "dimensions" is still NULL, retrieve rank and
    // dimensions via the C API and store them in this ArrayType object.
    if (rank < 0 && dimensions == NULL)
	setArrayInfo();

    // Copy what's in "dimensions" to user's buffer "dims"
    for (int i = 0; i < rank; i++)
	dims[i] = dimensions[i];

    return(rank);
}

//--------------------------------------------------------------------------
// Function:	ArrayType destructor
///\brief	Properly terminates access to this array datatype.
// Programmer	Binh-Minh Ribler - May 2004
//--------------------------------------------------------------------------
ArrayType::~ArrayType()
{
   // Free allocated memory
   if (dimensions != NULL)
      delete []dimensions;
}

#ifndef H5_NO_NAMESPACE
} // end namespace
#endif