#include <stdio.h>
#include <iostream>
#include <stdlib.h>
#include <sstream>
#ifndef _WIN32
#include <unistd.h>
#endif /* _WIN32 */
#include <napi.h>

#define PSZ(s) (s).c_str()

using namespace std;

const int nFiles = 10;
const int nEntry = 2;
const int nData = 2;
int array_dims[2] = {512, 512};
const char szFile[] = "leak_test.nxs";
const int iBinarySize = 512*512;
int aiBinaryData[iBinarySize];

#define ON_ERROR(msgstr)\
{\
    std::cerr<<msgstr<<std::endl; \
    return 1; \
}

int main ()
{
    int i, iFile, iEntry, iData, iNXdata;
 
    for(i=0; i < iBinarySize; i++)
    {
        aiBinaryData[i] = rand();
    }	

    for( iFile = 0; iFile < nFiles; iFile++ )
    {
        printf("file %d\n", iFile);

        NXhandle fileid;
        NXlink aLink;
        if( NXopen(szFile, NXACC_CREATE5, &fileid ) != NX_OK) 
            ON_ERROR("NXopen_failed")

        for( iEntry = 0; iEntry < nEntry; iEntry++ )
        {
            ostringstream oss;
            oss << "entry_" << iEntry;

            if (NXmakegroup (fileid, PSZ(oss.str()), "NXentry") != NX_OK) 
                ON_ERROR("NXmakegroup failed!")

            if (NXopengroup (fileid, PSZ(oss.str()), "NXentry") != NX_OK) 
                ON_ERROR("NXopengroup failed!")

            for( iNXdata = 0; iNXdata < nData; iNXdata++ )
            {
                ostringstream oss;
                oss << "data_" << iNXdata;

                if (NXmakegroup (fileid, PSZ(oss.str()), "NXdata") != NX_OK) 
                    ON_ERROR("NXmakegroup failed!")

                if (NXopengroup (fileid, PSZ(oss.str()), "NXdata") != NX_OK) 
                    ON_ERROR("NXopengroup failed!")

                NXgetgroupID(fileid, &aLink);
                for( iData = 0; iData < nData; iData++ )
                {
                    ostringstream oss;
                    oss << "i2_data_" << iData;

                    if (NXcompmakedata (fileid, PSZ(oss.str()), 
                                        NX_INT16, 2, array_dims, NX_COMP_LZW, 
                                        array_dims) != NX_OK)
                        ON_ERROR("NXcompmakedata failed!")

                    if (NXopendata (fileid, PSZ(oss.str())) != NX_OK) 
                        ON_ERROR("NXopendata failed!")

                    if (NXputdata (fileid, aiBinaryData) != NX_OK) 
                        ON_ERROR("NXputdata failed!")

                    if (NXclosedata (fileid) != NX_OK) 
                        ON_ERROR("NXclosedata failed!")
                }

                if (NXclosegroup (fileid) != NX_OK)
                    ON_ERROR("NXclosegroup failed!") 
            }

            if (NXclosegroup (fileid) != NX_OK) 
                ON_ERROR("NXclosegroup failed!") 
        }
    
        if (NXclose (&fileid) != NX_OK)
            ON_ERROR("NXclose failed!") 

        // Delete file
        remove(szFile);
    }
    
    return 0;
}