[gdcm.git] / src / gdcmlib.h

// gdcmlib Intro:  
// * gdcmlib is a library dedicated to reading and writing dicom files.
// * LGPL for the license
// * lightweigth as opposed to CTN or DCMTK wich come bundled which try
//   to implement the full DICOM standard (networking...). gdcmlib concentrates
//   on reading and 
// * Formats: this lib should be able to read ACR-NEMA v1 and v2, Dicom v3 (as
//   stated in part10). [cf dcmtk/dcmdata/docs/datadict.txt]
// * Targeted plateforms: Un*xes and Win32/VC++6.0

#include <string>
#include <stddef.h>    // For size_t
#include <glib.h>
#include <stdio.h>

// The requirement for the hash table (or map) that we shall use:
// 1/ First, next, last (iterators)
// 2/ should be sortable (i.e. sorted by TagKey). This condition
//    shall be droped since the Win32/VC++ implementation doesn't look
//    a sorted one. Pffff....
// 3/ Make sure we can setup some default size value, which should be
//    around 4500 entries which is the average dictionary size (said JPR)
#include <map>


// Tag based hash tables.
// We shall use as keys the strings (as the C++ type) obtained by
// concatenating the group value and the element value (both of type
// unsigned 16 bit integers in Dicom) expressed in hexadecimal.
// Example: consider the tag given as (group, element) = (0x0010, 0x0010).
// Then the corresponding TagKey shall be the string 0010|0010 (where
// the | (pipe symbol) acts as a separator). Refer to 
// gdcmDictEntry::TranslateToKey for this conversion function.
typedef string TagKey;

class gdcmDictEntry {
private:
        guint16 group;    // e.g. 0x0010
        guint16 element;  // e.g. 0x0010
        string  vr;       // Value Representation i.e. some clue about the nature
                          // of the data represented e.g. "FD" short for
                          // "Floating Point Double"
        // CLEAN ME: find the official dicom name for this field !
        string  fourth;   // Fourth field containing some semantics.
        string  name;     // e.g. "Patient_Name"
        TagKey  key;      // This is redundant zith (group, element) but we add
                          // on efficiency purposes.
        // DCMTK has many fields for handling a DictEntry (see below). What are the
        // relevant ones for gdcmlib ?
        //      struct DBI_SimpleEntry {
        //         Uint16 group;
        //         Uint16 element;
        //         Uint16 upperGroup;
        //         Uint16 upperElement;
        //         DcmEVR evr;
        //         const char* tagName;
        //         int vmMin;
        //         int vmMax;
        //         const char* standardVersion;
        //         DcmDictRangeRestriction groupRestriction;
        //         DcmDictRangeRestriction elementRestriction;
        //       };
public:
        //CLEANME gdcmDictEntry();
        gdcmDictEntry(guint16 group, guint16 element,
                      string vr, string fourth, string vr);
        static TagKey TranslateToKey(guint16 group, guint16 element);
        guint16 GetGroup(void)  { return group;};
        guint16 GetElement(void){return element;};
        string  GetVR(void)     {return vr; };
        void    SetVR(string in){vr = in; };
        string  GetFourth(void) {return fourth;};
        string  GetName(void)   {return name;};
        string  GetKey(void)    {return key;};
};
  
typedef map<TagKey, gdcmDictEntry*> TagHT;

// A single DICOM dictionary i.e. a container for a collection of dictionary
// entries. There should be a single public dictionary (THE dictionary of
// the actual DICOM v3) but as many shadow dictionaries as imagers 
// combined with all software versions...
class gdcmDict {
        string name;
        string filename;
        TagHT entries;
public:
        gdcmDict();
        gdcmDict(char *FileName);   // Read Dict from disk
        int AppendEntry(gdcmDictEntry* NewEntry);
        gdcmDictEntry * GetTag(guint32 group, guint32 element);
        void Print(ostream&);
};

// Container for managing a set of loaded dictionaries. Sharing dictionaries
// should avoid :
// * reloading an allready loaded dictionary.
// * having many in memory representations of the same dictionary.
#define PUBDICTNAME     "DicomV3Dict"
#define PUBDICTFILENAME "../Dicts/dicomV3.dic"
typedef string DictKey;
typedef map<DictKey, gdcmDict*> DictSetHT;
class gdcmDictSet {
private:
        DictSetHT dicts;
        int AppendDict(gdcmDict* NewDict);
public:
        gdcmDictSet(void);    // loads THE DICOM v3 dictionary
        int LoadDictFromFile(string filename);
///// QUESTION: the following function might not be thread safe !? Maybe
/////           we need some mutex here, to avoid concurent creation of
/////           the same dictionary !?!?!
        int LoadDictFromName(string filename);
        int LoadDictFromFile(char*, DictKey);
        int LoadAllDictFromDirectory(string DirectoryName);
        int LoadDicomV3Dict(void);
        string* GetAllDictNames();
        void Print(ostream&);
        gdcmDict* GetDict(DictKey DictName);
        gdcmDict* GetDefaultPublicDict(void);
};

// The dicom header of a Dicom file contains a set of such ELement VALUES
// (when successfuly parsed against a given Dicom dictionary)
class ElValue {
private:
        gdcmDictEntry *entry;
        guint32 LgrElem;
        // Might prove of some interest (see _ID_DCM_ELEM)
        // int Swap;
public:
        string  value;     // used to be char * valeurElem
        size_t Offset;     // Offset from the begining of file for direct user access
        ElValue(gdcmDictEntry*);
        void   SetVR(string);
        string GetVR(void);
        void SetLength(guint32 l){LgrElem = l; };
        void SetValue(string val){ value = val; };
        void SetOffset(size_t of){ Offset = of; };
        string  GetValue(void)   { return value; };
        guint32 GetLength(void) { return LgrElem; };
        size_t  GetOffset(void)  { return Offset; };
        guint16 GetGroup(void)   { return entry->GetGroup(); };
        guint16 GetElement(void) { return entry->GetElement(); };
        string  GetKey(void)     { return entry->GetKey(); };
        string  GetName(void)    { return entry->GetName();};
};

typedef map<TagKey, ElValue*> TagElValueHT;
typedef map<string, ElValue*> TagElValueNameHT;
// Container for a set of succefully parsed ElValues.
class ElValSet {
        // We need both accesses with a TagKey and the Dicentry.Name
        TagElValueHT tagHt;
        TagElValueNameHT NameHt;
public:
        void Add(ElValue*);
        void Print(ostream &);
        void PrintByName(ostream &);
        string GetElValue(guint32 group, guint32 element);
        string GetElValue(string);
};

// The various entries of the explicit value representation (VR) shall
// be managed within a dictionary. 
typedef string VRKey;
typedef string VRAtr;
typedef map<TagKey, VRAtr> VRHT;    // Value Representation Hash Table

// The typical usage of objects of this class is to classify a set of
// dicom files according to header information e.g. to create a file hierachy
// reflecting the Patient/Study/Serie informations, or extracting a given
// SerieId. Accesing the content (image[s] or volume[s]) is beyond the
// functionality of this class (see dmcFile below).
// Notes:
// * the gdcmHeader::Set*Tag* family members cannot be defined as protected
//   (Swig limitations for as Has_a dependency between gdcmFile and gdcmHeader)
class gdcmHeader {      
//FIXME sw should be qn EndianType !!!
        //enum EndianType {
                //LittleEndian, 
                //BadLittleEndian,
                //BigEndian, 
                //BadBigEndian};
private:
        // All instances share the same valur representation dictionary
        static VRHT *dicom_vr;
        static gdcmDictSet* Dicts;  // Global dictionary container
        gdcmDict* RefPubDict;       // Public Dictionary
        gdcmDict* RefShaDict;       // Shadow Dictionary (optional)
        ElValSet PubElVals;     // Element Values parsed with Public Dictionary
        ElValSet ShaElVals;     // Element Values parsed with Shadow Dictionary
        // In order to inspect/navigate through the file
        string filename;
        FILE * fp;
        // The tag Image Location ((0028,0200) containing the adress of
        // the pixels) is not allways present. When we store this information
        // FIXME
        // outside of the elements:
        guint16 grPixel;
        guint16 numPixel;
        bool PixelsTrouves;
        bool grPixelTrouve;
        size_t PixelPosition;
        int sw;

        void Initialise(void);
        void CheckSwap(void);
        void FindLength(ElValue *);
        void FindVR(ElValue *);
        guint32 SwapLong(guint32);
        short int SwapShort(short int);
        void InitVRDict(void);
        ElValue * ReadNextElement(void);
        gdcmDictEntry * IsInDicts(guint32, guint32);
        void SetAsidePixelData(ElValue*);
protected:
        enum FileType {
                Unknown = 0,
                TrueDicom,
                ExplicitVR,
                ImplicitVR,
                ACR,
                ACR_LIBIDO};
        FileType filetype;
///// QUESTION: Maybe Print is a better name than write !?
        int write(ostream&);   
///// QUESTION: Maybe anonymize should be a friend function !?!?
/////           See below for an example of how anonymize might be implemented.
        int anonymize(ostream&);
public:
        bool IsAnInteger(guint16, guint16, string, guint32);
        virtual void BuildHeader(void);
        gdcmHeader(char* filename);
        virtual ~gdcmHeader();

        int SetPubDict(string filename);
        // When some proprietary shadow groups are disclosed, whe can set
        // up an additional specific dictionary to access extra information.
        int SetShaDict(string filename);

        // Retrieve all potentially available tag [tag = (group, element)] names
        // from the standard (or public) dictionary (hence static). Typical usage:
        // enable the user of a GUI based interface to select his favorite fields
        // for sorting or selection.
        string* GetPubTagNames();
        // Get the element values themselves:
        string GetPubElValByName(string TagName);
        string GetPubElValByNumber(guint16 group, guint16 element);
        // Get the element value representation: (VR) might be needed by caller
        // to convert the string typed content to caller's native type (think
        // of C/C++ vs Python).
        string GetPubElValRepByName(string TagName);
        string GetPubElValRepByNumber(guint16 group, guint16 element);
        void   PrintPubElVal(ostream &);
        void   PrintPubDict(ostream &);
          
        // Same thing with the shadow :
        string* GetShaTagNames(); 
        string GetShaElValByName(string TagName);
        string GetShaElValByNumber(guint16 group, guint16 element);
        string GetShaElValRepByName(string TagName);
        string GetShaElValRepByNumber(guint16 group, guint16 element);

        // Wrappers of the above (both public and shadow) to avoid bugging the
        // caller with knowing if ElVal is from the public or shadow dictionary.
        string GetElValByName(string TagName);
        string GetElValByNumber(guint16 group, guint16 element);
        string GetElValRepByName(string TagName);
        string GetElValRepByNumber(guint16 group, guint16 element);

        int SetPubElValByName(string content, string TagName);
        int SetPubElValByNumber(string content, guint16 group, guint16 element);
        int SetShaElValByName(string content, string ShadowTagName);
        int SetShaElValByNumber(string content, guint16 group, guint16 element);

        int GetSwapCode();
};

// In addition to Dicom header exploration, this class is designed
// for accessing the image/volume content. One can also use it to
// write Dicom files.
////// QUESTION: this looks still like an open question wether the
//////           relationship between a gdcmFile and gdcmHeader is of
//////           type IS_A or HAS_A !
class gdcmFile: gdcmHeader
{
private:
        void* Data;
        int Parsed;                             // weather allready parsed
        string OrigFileName;    // To avoid file overwrite
public:
        // Constructor dedicated to writing a new DICOMV3 part10 compliant
        // file (see SetFileName, SetDcmTag and Write)
        gdcmFile();
        // Opens (in read only and when possible) an existing file and checks
        // for DICOM compliance. Returns NULL on failure.
        // Note: the in-memory representation of all available tags found in
        //    the DICOM header is post-poned to first header information access.
        //    This avoid a double parsing of public part of the header when
        //    one sets an a posteriori shadow dictionary (efficiency can be
        //    seen a a side effect).
        gdcmFile(string & filename);
        // For promotion (performs a deepcopy of pointed header object)
        gdcmFile(gdcmHeader* header);
        ~gdcmFile();

        // On writing purposes. When instance was created through
        // gdcmFile(string filename) then the filename argument MUST be different
        // from the constructor's one (no overwriting aloud).
        int SetFileName(string filename);

        // Allocates necessary memory, copies the data (image[s]/volume[s]) to
        // newly allocated zone and return a pointer to it:
        void * GetImageData();
        // Returns size (in bytes) of required memory to contain data
        // represented in this file.
        size_t GetImageDataSize();
        // Copies (at most MaxSize bytes) of data to caller's memory space.
        // Returns an error code on failure (if MaxSize is not big enough)
        int PutImageDataHere(void* destination, size_t MaxSize );
        // Allocates ExpectedSize bytes of memory at this->Data and copies the
        // pointed data to it.
        int SetImageData(void * Data, size_t ExpectedSize);
        // Push to disk.
        int Write();
};

//class gdcmSerie : gdcmFile;
//class gdcmMultiFrame : gdcmFile;

//
//Examples:
// * gdcmFile WriteDicom;
//   WriteDicom.SetFileName("MyDicomFile.dcm");
//      string * AllTags = gdcmHeader.GetDcmTagNames();
//   WriteDicom.SetDcmTag(AllTags[5], "253");
//   WriteDicom.SetDcmTag("Patient Name", "bozo");
//   WriteDicom.SetDcmTag("Patient Name", "bozo");
//      WriteDicom.SetImageData(Image);
//   WriteDicom.Write();
//
//
//   Anonymize(ostream& output) {
//   a = gdcmFile("toto1");
//   a.SetPubValueByName("Patient Name", "");
//   a.SetPubValueByName("Date", "");
//   a.SetPubValueByName("Study Date", "");
//   a.write(output);
//   }