WriteNeXusFile.hh

/*
$Id: WriteNeXusFile.hh 2247 2011-04-27 05:20:41Z jisuzuki $
 */

#ifndef WRITENEXUSFILE
#define WRITENEXUSFILE

#include <map>
#include "Header.hh"
#include "HeaderBase.hh"
#include "ElementContainer.hh"
#include "NeutronVector.hh"
#include "UInt4Container.hh"
#include "napi.h"
#include "ElementContainerArray.hh"
#include "ElementContainerMatrix.hh"
#include "UInt4ContainerArray.hh"
#include "UInt4ContainerMatrix.hh"
#include "TwoDimElementContainer.hh"


class WriteNeXusFile
{
private:
  UInt4 Comp;   
  UInt4 r_handleFlag;
protected:
  void CreateNeXusFile( const string &FileName, const string &UserName );
  //void CloseNeXusFile();
  NXhandle handle;
  NXlink link;

public:
  WriteNeXusFile( const string &FileName, const string &UserName="UserName", UInt4 CompMode=1 );
  WriteNeXusFile( NXhandle r_handle, UInt4 CompMode=1 );
  

 ~WriteNeXusFile();
  NXhandle PutHandle(){ return handle; }
  NXlink   PutLink(){ return link; }

  void CloseNeXusFile();

  void AppendAttribute( const string &Name, string value );
  void AppendAttribute( const string &Name, UInt4  value );
  void AppendAttribute( const string &Name, Int4   value );
  void AppendAttribute( const string &Name, Double value );
  void AppendAttribute( const string &Name, vector<Int4>   value );
  void AppendAttribute( const string &Name, vector<Double> value );
  void AppendAttribute( const string &Name, vector<string> value );
  void AppendAttribute( HeaderBase *header );

  void MakeOpenGroup( const string &Name, string Class );
  void CloseGroup();
  void CloseData();


  template <class T>
  void WriteData( const string &Name, const T &t );

  template <class T>
  void WriteData( const string &Name, const std::vector<T *> &t );

  template <class T>
  void WriteData( const string &Name, const std::map<string, T> &t );

  //void WriteData( const string &Name );
  void WriteData( const string &Name, UInt4  value );
  void WriteData( const string &Name, Int4   value );
  void WriteData( const string &Name, Double value );
  void WriteData( const string &Name, string value );
  void WriteData( const string &Name, const UInt4  *value, UInt4 size );
  void WriteData( const string &Name, const Int4   *value, UInt4 size );
  void WriteData( const string &Name, const Double *value, UInt4 size );
  void WriteData( const string &Name, const vector< string > &value );
  void WriteData( const string &Name, const vector< UInt4  > &value );
  void WriteData( const string &Name, const vector< Int4   > &value );
  void WriteData( const string &Name, const vector< Double > &value );
  void WriteData1( const string &Name, ElementContainer EC );
  void WriteData1( const string &Name, HeaderBase Header );
  void WriteData1( const string &Name, ElementContainerArray ECA );
  void WriteData1( const string &Name, ElementContainerMatrix ECM );
  void WriteData1( const string &Name, UInt4ContainerMatrix UCM );
  void WriteData1( const string &Name, UInt4ContainerArray  UCA );
  void WriteData1( const string &Name, UInt4Container UC );

  void WriteData1( const string &Name, TwoDimElementContainer TDEC );

  void MakeLink( NXlink Link );
};

template <class T>
void WriteNeXusFile::
WriteData( const string &Name, const T &t )
{
  if( Name == "" ){
    MakeOpenGroup( "ManyoLibData", "NXdata" );
  }
  else{
    MakeOpenGroup( Name, "NXdata" ); 
  }

  // this new WriteData() scheme writes data with an attribute "version"=2
  AppendAttribute("version", 2);        // writes 'version="NX_INT32:2"'
  //NXputattr( handle, "version", (void*)( &v ), 1, NX_UINT32 );
  
  t.NXwrite(*this);


  NXgetgroupID( handle, &link );
  CloseGroup();
}

template <class T>
void WriteNeXusFile::
WriteData( const string &Name, const std::vector<T *> &t )
{
  if( Name == "" ){
    MakeOpenGroup( "std::vector_pointer_data", "NXdata" );
  }
  else{
    MakeOpenGroup( Name, "NXdata" ); 
  }

  UInt4 size = t.size();
  WriteData( "size", size );
  static char index[128];
  for( UInt4 i=0; i<size; i++ ){
        sprintf(index, "%s%d", Name.c_str(), i);
        WriteData(string(index), *t[i]);
  }

  NXgetgroupID( handle, &link );
  CloseGroup();
}
template <class T>
void WriteNeXusFile::
WriteData( const string &Name, const std::map<string, T> &t )
{
  if( Name == "" ){
    MakeOpenGroup( "std::map_data", "NXdata" );
  }
  else{
    MakeOpenGroup( Name, "NXdata" ); 
  }

  for (typename std::map<string, T>::const_iterator p=t.begin(),
                 end=t.end();p!=end;++p) {
        WriteData(p->first, p->second);
  }

  NXgetgroupID( handle, &link );
  CloseGroup();
}
#endif



/* Test code on Python environment.
e = Neutron.ElementContainer()

w = Neutron.WriteNeXusFile( "nx.nx", "suzuki" )
w.MakeOpenGroup("Entry1","NXentry")
w.MakeOpenGroup("Data1","NXdata")

w.WriteData( "ElementContainerData", e )

w.CloseGroup() # Close Data1
w.CloseGroup() # Close Entry1
del w
 */



/* test code on Python interface
import Neutron
w = Neutron.WriteNeXusFile("nx.nx", "suzuki")
w.AppendAttribute("x", "xxx")
w.AppendAttribute("x", 12)
w.MakeOpenGroup("Entry1","NXentry")
w.MakeOpenGroup("Data1","NXdata")

w.WriteData("user_name", "suzuki")

vui = Neutron.MakeUInt4Vector( 20 )
vi  = Neutron.MakeInt4Vector( 30 )
vd  = Neutron.MakeDoubleVector( 10 )

for x in range( 10 ):
        vd[x] = float( x )

w.WriteData( "vui", vui )
w.AppendAttribute("x", 12)

w.WriteData( "vi",  vi )
w.AppendAttribute("x", 13)

w.WriteData( "vd",  vd )
w.AppendAttribute("x", 14)

s = Neutron.MakeStringVector()
s.append( "ooho" )
s.append( "kasuga" )
s.append( "tsukuba" )
s.append( "Ibarakiken Tsukubashi kasuga" )
w.WriteData( "s", s  )

w.CloseGroup() # Close Data1
w.CloseGroup() # Close Entry1
del w
 */