%module gdcm
%{
#include "gdcmCommon.h"
#include "gdcmDict.h"
#include "gdcmDictEntry.h"
#include "gdcmDictSet.h"
#include "gdcmDicomDir.h"
#include "gdcmDicomDirElement.h"
#include "gdcmDicomDirImage.h"
#include "gdcmDicomDirMeta.h"
#include "gdcmDicomDirObject.h"
#include "gdcmDicomDirPatient.h"
#include "gdcmDicomDirStudy.h"
#include "gdcmDicomDirSerie.h"
#include "gdcmDocEntrySet.h"
#include "gdcmDocument.h"
#include "gdcmElementSet.h"
#include "gdcmFile.h"
#include "gdcmGlobal.h"
#include "gdcmHeader.h"
#include "gdcmHeaderHelper.h"
#include "gdcmRLEFramesInfo.h"
#include "gdcmJPEGFragmentsInfo.h"
#include "gdcmSQItem.h"
#include "gdcmUtil.h"
#include "gdcmValEntry.h"

////////////////////////////////////////////////////////////////////////////
// Utility functions on strings for removing leading and trailing spaces
void EatLeadingAndTrailingSpaces(std::string & s) {
	while ( s.length() && (s[0] == ' ') )
		s.erase(0,1);
	while ( s.length() && (s[s.length()-1] == ' ') )
		s.erase(s.length()-1, 1);
}

void gdcmPythonVoidFunc(void *arg)
{
  PyObject *arglist, *result;
  PyObject *func = (PyObject *)arg;

  arglist = Py_BuildValue("()");

  result = PyEval_CallObject(func, arglist);
  Py_DECREF(arglist);

  if (result)
    {
    Py_XDECREF(result);
    }
  else
    {
    if (PyErr_ExceptionMatches(PyExc_KeyboardInterrupt))
      {
      std::cerr << "Caught a Ctrl-C within python, exiting program.\n";
      Py_Exit(1);
      }
    PyErr_Print();
    }
}

void gdcmPythonVoidFuncArgDelete(void *arg)
{
  PyObject *func = (PyObject *)arg;
  if (func)
    {
    Py_DECREF(func);
    }
}

using namespace gdcm;
%}
typedef  unsigned short guint16;
typedef  unsigned int guint32;

////////////////////////////////////////////////////////////////////////////
%typemap(out) std::list<std::string> * {
   PyObject* NewItem = (PyObject*)0;
   PyObject* NewList = PyList_New(0); // The result of this typemap

   for (std::list<std::string>::iterator NewString = ($1)->begin();
        NewString != ($1)->end();
        ++NewString)
   {
      NewItem = PyString_FromString(NewString->c_str());
      PyList_Append( NewList, NewItem);
   }
   $result = NewList;
}

////////////////////////////////////////////////////////////////////////////
// Convert a c++ hash table in a python native dictionary
%typemap(out) std::map<std::string, std::list<std::string> > * {
   PyObject* NewDict = PyDict_New(); // The result of this typemap
   PyObject* NewKey = (PyObject*)0;
   PyObject* NewVal = (PyObject*)0;

   for (std::map<std::string,
        std::list<std::string> >::iterator tag = ($1)->begin();
        tag != ($1)->end(); ++tag)
   {
      std::string first = tag->first;
      // Do not publish entries whose keys is made of spaces
      if (first.length() == 0)
         continue;
      NewKey = PyString_FromString(first.c_str());
      PyObject* NewList = PyList_New(0);
      for (std::list<std::string>::iterator Item = tag->second.begin();
           Item != tag->second.end();
           ++Item)
      {
         NewVal = PyString_FromString(Item->c_str());
         PyList_Append( NewList, NewVal);
      }
      PyDict_SetItem( NewDict, NewKey, NewList);
   }
   $result = NewDict;
}

////////////////////////////////////////////////////////////////////////////
// Convert a c++ hash table in a python native dictionary
%typemap(out) gdcm::TagDocEntryHT & {
   PyObject* NewDict = PyDict_New(); // The result of this typemap
   std::string RawName;              // Element name as gotten from gdcm
   PyObject* NewKey = (PyObject*)0;  // Associated name as python object
   std::string RawValue;             // Element value as gotten from gdcm
   PyObject* NewVal = (PyObject*)0;  // Associated value as python object

   for (gdcm::TagDocEntryHT::iterator tag = $1->begin(); tag != $1->end(); ++tag)
   {
      // The element name shall be the key:
      RawName = tag->second->GetName();
      // gdcm unrecognized (including not loaded because their size exceeds
      // the user specified treshold) elements are exported with their
      // TagKey as key.
      if (RawName == "Unknown")
         RawName = tag->second->GetKey();
      NewKey = PyString_FromString(RawName.c_str());

      // Element values are striped from leading/trailing spaces
      if (gdcm::ValEntry* ValEntryPtr =
                dynamic_cast< gdcm::ValEntry* >(tag->second) )
      {
         RawValue = ValEntryPtr->GetValue();
      }
      else
        continue; 
      EatLeadingAndTrailingSpaces(RawValue);
      NewVal = PyString_FromString(RawValue.c_str());
      PyDict_SetItem( NewDict, NewKey, NewVal);
   }
   $result = NewDict;
}

////////////////////////////////////////////////////////////////////////////
%typemap(out) ListDicomDirPatient & {
	PyObject* NewItem = (PyObject*)0;
	$result = PyList_New(0); // The result of this typemap

	for (std::list<gdcm::DicomDirPatient *>::iterator New = ($1)->begin();
	    New != ($1)->end(); ++New) {
		NewItem = SWIG_NewPointerObj(*New,SWIGTYPE_p_DicomDirPatient,1);
		PyList_Append($result, NewItem);
	}
}

%typemap(out) ListDicomDirStudy & {
	PyObject* NewItem = (PyObject*)0;
	$result = PyList_New(0); // The result of this typemap

	for (std::list<gdcm::DicomDirStudy *>::iterator New = ($1)->begin();
	    New != ($1)->end(); ++New) {
		NewItem = SWIG_NewPointerObj(*New,SWIGTYPE_p_DicomDirStudy,1);
		PyList_Append($result, NewItem);
	}
}

%typemap(out) ListDicomDirSerie & {
	PyObject* NewItem = (PyObject*)0;
	$result = PyList_New(0); // The result of this typemap

	for (std::list<gdcm::DicomDirSerie *>::iterator New = ($1)->begin();
	    New != ($1)->end(); ++New) {
		NewItem = SWIG_NewPointerObj(*New,SWIGTYPE_p_DicomDirSerie,1);
		PyList_Append($result, NewItem);
	}
}

%typemap(out) ListDicomDirImage & {
	PyObject* NewItem = (PyObject*)0;
	$result = PyList_New(0); // The result of this typemap

	for (std::list<gdcm::DicomDirImage *>::iterator New = ($1)->begin();
	    New != ($1)->end(); ++New) {
		NewItem = SWIG_NewPointerObj(*New,SWIGTYPE_p_DicomDirImage,1);
		PyList_Append($result, NewItem);
	}
}

////////////////////////////////////////////////////////////////////////////
// Deals with function returning a C++ string.
%typemap(python, in) (gdcm::Method *,void * =NULL,gdcm::Method * =NULL) {
	if($input!=Py_None)
	{
		Py_INCREF($input);
		$1=gdcmPythonVoidFunc;
		$2=$input;
		$3=gdcmPythonVoidFuncArgDelete;
	}
	else
	{
		$1=NULL;
		$2=NULL;
		$3=NULL;
	}
}


////////////////////////////////////////////////////////////////////////////
// Deals with function returning a C++ string.
%typemap(out) string, std::string  {
    $result = PyString_FromString(($1).c_str());
}

%typemap(python, in) const std::string, std::string
{
  $1 = PyString_AsString($input);
}

////////////////////////////////////////////////////////////////////////////
// Warning: Order matters !
%include "gdcmCommon.h"
%include "gdcmRLEFramesInfo.h"
%include "gdcmJPEGFragmentsInfo.h"
%include "gdcmDictEntry.h"
%include "gdcmDict.h"
%include "gdcmDocEntry.h"
%include "gdcmDocEntrySet.h"
%include "gdcmElementSet.h"
%include "gdcmDictSet.h"
%include "gdcmTS.h"
%include "gdcmVR.h"
%include "gdcmSQItem.h"
%include "gdcmDicomDirElement.h"
%include "gdcmDicomDirObject.h"
%include "gdcmDicomDirImage.h"
%include "gdcmDicomDirSerie.h"
%include "gdcmDicomDirStudy.h"
%include "gdcmDicomDirPatient.h"
%include "gdcmDicomDirMeta.h"
%include "gdcmDocument.h"
%include "gdcmHeader.h"
%include "gdcmHeaderHelper.h"
%include "gdcmFile.h"
%include "gdcmUtil.h"
%include "gdcmGlobal.h"
%include "gdcmDicomDir.h"