]> Creatis software - gdcm.git/blobdiff - src/gdcmElValSet.cxx
fonction UpdateGroupLengthNew (utilisant une HTable)
[gdcm.git] / src / gdcmElValSet.cxx
index 70f21fc15ffc47f8ae25538ca06f97651e3aee99..fe5df93739c928d7815f837bc2d43db4d2a47c08 100644 (file)
 // gdcmElValSet.cxx
 
-#include "gdcm.h"
+#include <sstream>
 #include "gdcmUtil.h"
+#include "gdcmElValSet.h"
 
-TagElValueHT & ElValSet::GetTagHt(void) {
+#define DEBUG 0
+
+
+TagElValueHT & gdcmElValSet::GetTagHt(void) {
        return tagHt;
 }
 
-void ElValSet::Add(ElValue * newElValue) {
-       tagHt[newElValue->GetKey()]   = newElValue;
+void gdcmElValSet::Add(gdcmElValue * newElValue) {
+       tagHt [newElValue->GetKey()]  = newElValue;
        NameHt[newElValue->GetName()] = newElValue;
 }
 
-void ElValSet::Print(ostream & os) {
+void gdcmElValSet::Print(ostream & os) {
        for (TagElValueHT::iterator tag = tagHt.begin();
                  tag != tagHt.end();
                  ++tag){
                os << tag->first << ": ";
                os << "[" << tag->second->GetValue() << "]";
                os << "[" << tag->second->GetName()  << "]";
-               os << "[" << tag->second->GetVR()    << "]" << endl;
+               os << "[" << tag->second->GetVR()    << "]"; 
+               os << " lgr : " << tag->second->GetLength();
+               os << endl;
        }
-}
-
-
-int ElValSet::Write(FILE * _fp) {
-
-       guint16 gr, el;
-       guint32 lgr;
-       const char * val;
-       // restent à tester les echecs en écriture
-       for (TagElValueHT::iterator tag = tagHt.begin();
-                 tag != tagHt.end();
-                 ++tag){
+} 
 
-               // Question :
-               // peut-on se passer des affectations?
-               // - passer l'adresse du resultat d'une fonction (???)
-               // - acceder au champ sans passer par un accesseur ?
-               gr = tag->second->GetGroup();
-               el = tag->second->GetElement();
-               lgr = tag->second->GetLength();
-               val = tag->second->GetValue().c_str();
-               
-               fwrite ( &gr,(size_t)2 ,(size_t)1 ,_fp);        //group
-               fwrite ( &el,(size_t)2 ,(size_t)1 ,_fp);        //element
-               //fwrite ( tag->second->GetVR(),(size_t)2 ,(size_t)1 ,_fp);     //VR
-               
-               // voir pb lgr  + VR
-               fwrite ( &lgr,(size_t)4 ,(size_t)1 ,_fp);                       //lgr 
-               // voir pb des int16 et int32 : les identifier, les convertir, modifier la longueur
-               fwrite ( val,(size_t)lgr ,(size_t)1 ,_fp); //valeur Elem
-       }
-       return(1);
-}
-void ElValSet::PrintByName(ostream & os) {
+void gdcmElValSet::PrintByName(ostream & os) {
        for (TagElValueNameHT::iterator tag = NameHt.begin();
                  tag != NameHt.end();
                  ++tag){
                os << tag->first << ": ";
                os << "[" << tag->second->GetValue() << "]";
-               os << "[" << tag->second->GetKey()  << "]";
+               os << "[" << tag->second->GetKey()   << "]";
                os << "[" << tag->second->GetVR()    << "]" << endl;
        }
 }
 
-ElValue* ElValSet::GetElementByNumber(guint32 group, guint32 element) {
+gdcmElValue* gdcmElValSet::GetElementByNumber(guint16 group, guint16 element) {
        TagKey key = gdcmDictEntry::TranslateToKey(group, element);
        if ( ! tagHt.count(key))
-               return (ElValue*)0;
-       if (tagHt.count(key) > 1)
-               dbg.Verbose(0, "ElValSet::GetElementByNumber",
-                           "multiple entries for this key (FIXME) !");
+               return (gdcmElValue*)0;
        return tagHt.find(key)->second;
 }
 
-ElValue* ElValSet::GetElementByName(string TagName) {
+gdcmElValue* gdcmElValSet::GetElementByName(string TagName) {
    if ( ! NameHt.count(TagName))
-      return (ElValue*)0;
-   if (NameHt.count(TagName) > 1)
-      dbg.Verbose(0, "ElValSet::GetElement",
-                  "multipe entries for this key (FIXME) !");
+      return (gdcmElValue*)0;
    return NameHt.find(TagName)->second;
 }
 
-string ElValSet::GetElValueByNumber(guint32 group, guint32 element) {
+string gdcmElValSet::GetElValueByNumber(guint16 group, guint16 element) {
        TagKey key = gdcmDictEntry::TranslateToKey(group, element);
        if ( ! tagHt.count(key))
                return "gdcm::Unfound";
-       if (tagHt.count(key) > 1)
-               dbg.Verbose(0, "ElValSet::GetElValueByNumber",
-                           "multiple entries for this key (FIXME) !");
        return tagHt.find(key)->second->GetValue();
 }
 
-int ElValSet::SetElValueByNumber(string content, guint32 group, guint32 element) {
+string gdcmElValSet::GetElValueByName(string TagName) {
+       if ( ! NameHt.count(TagName))
+               return "gdcm::Unfound";
+       return NameHt.find(TagName)->second->GetValue();
+}
+
+
+int gdcmElValSet::SetElValueByNumber(string content,
+                                     guint16 group, guint16 element) {
        TagKey key = gdcmDictEntry::TranslateToKey(group, element);
        if ( ! tagHt.count(key))
                return 0;
-       if (tagHt.count(key) > 1) {
-               dbg.Verbose(0, "ElValSet::SetElValueByNumber",
-                           "multiple entries for this key (FIXME) !");
-               return (0); 
-       }
-                                      
        tagHt[key]->SetValue(content);
-       return(1);              
+       tagHt[key]->SetLength(content.length());         
+       return 1;
 }
 
-string ElValSet::GetElValueByName(string TagName) {
+int gdcmElValSet::SetElValueByName(string content, string TagName) {
        if ( ! NameHt.count(TagName))
-               return "gdcm::Unfound";
-       if (NameHt.count(TagName) > 1)
-               dbg.Verbose(0, "ElValSet::GetElValue",
-                           "multipe entries for this key (FIXME) !");
-       return NameHt.find(TagName)->second->GetValue();
+               return 0;
+       NameHt[TagName]->SetValue(content);
+       NameHt[TagName]->SetLength(content.length());
+       return 1;               
 }
 
-int ElValSet::SetElValueByName(string content, string TagName) {
-       if ( ! NameHt.count(TagName))
+/**
+ * \ingroup gdcmElValSet
+ * \brief   Generate a free TagKey i.e. a TagKey that is not present
+ *          in the TagHt dictionary.
+ * @param   group The generated tag must belong to this group.  
+ * @return  The element of tag with given group which is fee.
+ */
+guint32 gdcmElValSet::GenerateFreeTagKeyInGroup(guint16 group) {
+   for (guint32 elem = 0; elem < UINT32_MAX; elem++) {
+      TagKey key = gdcmDictEntry::TranslateToKey(group, elem);
+      if (tagHt.count(key) == 0)
+         return elem;
+   }
+   return UINT32_MAX;
+}
+
+int gdcmElValSet::SetElValueLengthByNumber(guint32 length,
+                                           guint16 group, guint16 element) {
+       TagKey key = gdcmDictEntry::TranslateToKey(group, element);
+       if ( ! tagHt.count(key))
                return 0;
-       if (NameHt.count(TagName) > 1) {
-               dbg.Verbose(0, "ElValSet::SetElValue",
-                           "multipe entries for this key (FIXME) !");
+       tagHt[key]->SetLength(length);   
+       return 1 ;              
+}
+
+
+int gdcmElValSet::SetElValueLengthByName(guint32 length, string TagName) {
+       if ( ! NameHt.count(TagName))
                return 0;
+       NameHt.find(TagName)->second->SetLength(length);         
+       return 1 ;              
+}
+
+void gdcmElValSet::UpdateGroupLength(bool SkipSequence) {
+
+   // TODO : reecrire entierement : une HTable dans la quelle on stocke les groupes, puis leur lgr
+   // Voir juste apres
+
+       // On parcourt la table pour recalculer la longueur des 'elements 0x0000'
+       // au cas ou un tag ai été ajouté par rapport à ce qui a été lu
+       // dans l'image native
+       //
+       // cf : code IdDcmWriteFile dans libido/src/dcmwrite.c
+                               
+       // On fait de l'implicit VR little Endian 
+       // (pour moins se fairche sur processeur INTEL)
+       // On force le TRANSFERT SYNTAX UID
+   guint16 gr, el;
+   string vr;
+   guint32 lgrCalcGroupe=0;
+   gdcmElValue *elem, *elemZ, *elemZPrec;
+   guint16 grCourant = 0;
+                               
+       TagElValueHT::iterator tag = tagHt.begin();
+       
+       elem = tag->second;
+       gr   = elem->GetGroup();
+       el   = elem->GetElement();
+                       
+       if (el != 0x0000) {
+               if(DEBUG)printf("ajout elem OOOO premiere fois\n");
+               gdcmDictEntry * tagZ = new gdcmDictEntry(gr, 0x0000, "UL");
+               elemZPrec = new gdcmElValue(tagZ);      // on le cree
+               elemZPrec->SetLength(4);
+               Add(elemZPrec);                         // On l'accroche à sa place
+       } else {
+               elemZPrec = elem;
+               if(DEBUG)printf("Pas d'ajout elem OOOO premiere fois\n");
+       }
+       lgrCalcGroupe = 0;
+       if(DEBUG)printf("init-1 lgr (%d) pour gr %04x\n",lgrCalcGroupe, gr);
+       grCourant = gr;
+       
+       for (tag = ++tagHt.begin();
+                 tag != tagHt.end();
+                 ++tag){
+                 
+               elem = tag->second;
+               gr = elem->GetGroup();
+               el = elem->GetElement();
+      vr = elem->GetVR();
+
+      if (SkipSequence && vr == "SQ")  continue;
+         // pas SEQUENCE en ACR-NEMA
+         // WARNING : pb CERTAIN
+         //           si on est descendu 'a l'interieur' des SQ 
+        
+       if ( (gr != grCourant) /*&&     // On arrive sur un nv Groupe     
+               (el != 0xfffe) */       ) {
+                           
+                       if (el != 0x0000) {
+                               gdcmDictEntry * tagZ = new gdcmDictEntry(gr, 0x0000, "UL");
+                               elemZ = new gdcmElValue(tagZ); // on le cree
+                               elemZ->SetLength(4);
+                               Add(elemZ);                     // On l'accroche à sa place 
+                               if(DEBUG)printf("ajout elem OOOO pour gr %04x\n",gr);
+                       } else { 
+                               elemZ=elem;
+                               if(DEBUG)printf("maj elemZ\n");
+                       }
+                       
+                       ostringstream f;
+                       f << lgrCalcGroupe; 
+                       //sprintf(str_lgrCalcGroupe,"%d",lgrCalcGroupe);
+                       elemZPrec->SetValue(f.str());
+                       if(DEBUG)printf("ecriture lgr (%d, %s) pour gr %04x\n",lgrCalcGroupe, f.str().c_str(), grCourant);
+                       if(DEBUG)printf ("%04x %04x [%s]\n",elemZPrec->GetGroup(), elemZPrec->GetElement(),
+                                                       elemZPrec->GetValue().c_str());
+                       if(DEBUG)cout << "Addresse elemZPrec " << elemZPrec<< endl;
+                       elemZPrec=elemZ;
+                       lgrCalcGroupe = 0;
+                       grCourant     = gr;
+       // BUG :
+       // calcule lgr erronnée si ExplicitVR et VR =OB, SQ, etc
+       //      
+                       if(DEBUG)printf("init-2 lgr (%d) pour gr %04x\n",lgrCalcGroupe, gr);                    
+               } else {                        // On n'EST PAS sur un nv Groupe
+                       lgrCalcGroupe += 2 + 2 + 4 + elem->GetLength();  // Gr + Num + Lgr + LgrElem
+                       if(DEBUG)printf("increment (%d) el %04x-->lgr (%d) pour gr %04x\n",
+                                                               elem->GetLength(), el, lgrCalcGroupe, gr);
+               }               
        }
-       NameHt.find(TagName)->second->SetValue(content);
+       
+       // BUG :
+       // Ecriture incorrecte si (7FE0 0000) absent et aucun element apres groupe 7FE0 :-(
+       // A JETTER !
+       //
+}
+
+//
+// Remplacera UpdateGroupLength
+// Commite par precaution.
+// Ne pas utiliser
+//
+
+void gdcmElValSet::UpdateGroupLengthNew(bool SkipSequence, FileType type) {
+   guint16 gr, el;
+   string vr;
+   
+   gdcmElValue *elem;
+   char trash[10];
+   GroupKey key;
+   GroupHT groupHt;
+   
+   for (TagElValueHT::iterator tag2 = tagHt.begin();
+        tag2 != tagHt.end();
+        ++tag2){
+
+      elem  = tag2->second;
+      gr = elem->GetGroup();
+      el = elem->GetElement();
+      vr = elem->GetVR(); 
+           
+      sprintf(trash, "%04x", gr);
+      key = trash;
+      
+      if (SkipSequence && vr == "SQ") continue;
+         // pas SEQUENCE en ACR-NEMA
+         // WARNING : pb CERTAIN
+         //           si on est descendu 'a l'interieur' des SQ 
+         //
+         // --> la descente a l'interieur' des SQ 
+         // devra etre faite avec une liste chainee, pas avec une HTable...
+             
+      if ( groupHt.count(key) == 0) { 
+         groupHt[key] = 2 + 2 + 4 + 4; // creation automatique, par affectation ???
+      } else {       
+         if (type = ExplicitVR) {
+            if ( (vr == "OB") || (vr == "OW") || (vr == "SQ") ) {
+               groupHt[key] +=  4;
+            }
+         }
+         groupHt[key] += 2 + 2 + 4 + elem->GetLength(); 
+      }   
+   }
+   
+   // Liberer groupHt !
+}
+
+
+
+void gdcmElValSet::WriteElements(FileType type, FILE * _fp) {
+   guint16 gr, el;
+   guint32 lgr;
+   const char * val;
+   string vr;
+   guint32 val_uint32;
+   guint16 val_uint16;
+   
+   vector<string> tokens;
+
+   void *ptr;
+
+   // Tout ceci ne marche QUE parce qu'on est sur un proc Little Endian 
+   // restent à tester les echecs en écriture (apres chaque fwrite)
+
+   for (TagElValueHT::iterator tag2=tagHt.begin();
+        tag2 != tagHt.end();
+        ++tag2){
+
+      gr =  tag2->second->GetGroup();
+      el =  tag2->second->GetElement();
+      lgr = tag2->second->GetLength();
+      val = tag2->second->GetValue().c_str();
+      vr =  tag2->second->GetVR();
+      if(DEBUG)printf ("%04x %04x [%s] : [%s]\n",gr, el, vr.c_str(), val);
+
+      if ( type == ACR ) { 
+         if (gr < 0x0008) continue;
+         if (gr %2)   continue;
+         if (vr == "SQ" ) continue;
+      } 
+
+      fwrite ( &gr,(size_t)2 ,(size_t)1 ,_fp);  //group
+      fwrite ( &el,(size_t)2 ,(size_t)1 ,_fp);  //element
+
+      if ( (type == ExplicitVR) && (gr <= 0x0002) ) {
+         // On est en EXPLICIT VR
+         guint16 z=0, shortLgr;
+         fwrite (vr.c_str(),(size_t)2 ,(size_t)1 ,_fp);
+
+         if ( (vr == "OB") || (vr == "OW") || (vr == "SQ") ) {
+            fwrite ( &z,  (size_t)2 ,(size_t)1 ,_fp);
+            fwrite ( &lgr,(size_t)4 ,(size_t)1 ,_fp);
+
+         } else {
+            shortLgr=lgr;
+            fwrite ( &shortLgr,(size_t)2 ,(size_t)1 ,_fp);
+         }
+      } else {
+         fwrite ( &lgr,(size_t)4 ,(size_t)1 ,_fp);
+      }
+
+      tokens.erase(tokens.begin(),tokens.end());
+      Tokenize (tag2->second->GetValue(), tokens, "\\");
+
+      if (vr == "US" || vr == "SS") {
+         for (unsigned int i=0; i<tokens.size();i++) {
+            val_uint16 = atoi(tokens[i].c_str());
+            ptr = &val_uint16;
+            fwrite ( ptr,(size_t)2 ,(size_t)1 ,_fp);
+         }
+         continue;
+      }
+      if (vr == "UL" || vr == "SL") {
+         for (unsigned int i=0; i<tokens.size();i++) {
+            val_uint32 = atoi(tokens[i].c_str());
+            ptr = &val_uint32;
+            fwrite ( ptr,(size_t)4 ,(size_t)1 ,_fp);
+         }
+         continue;
+      }
+
+      // Les pixels ne sont pas chargés dans l'element !
+      if ((gr == 0x7fe0) && (el == 0x0010) ) break;
+
+      fwrite ( val,(size_t)lgr ,(size_t)1 ,_fp); //valeur Elem
+   }
 }
 
+int gdcmElValSet::Write(FILE * _fp, FileType type) {
 
+   if (type == ImplicitVR) {
+      string implicitVRTransfertSyntax = "1.2.840.10008.1.2";
+      SetElValueByNumber(implicitVRTransfertSyntax, 0x0002, 0x0010);
+      
+      //FIXME Refer to standards on page 21, chapter 6.2 "Value representation":
+      //      values with a VR of UI shall be padded with a single trailing null
+      //      Dans le cas suivant on doit pader manuellement avec un 0.
+      SetElValueLengthByNumber(18, 0x0002, 0x0010);
+   }
+   
+       // Question :
+       // Comment pourrait-on si le DcmHeader vient d'un fichoer DicomV3 ou non ,
+       // (FileType est un champ de gdcmHeader ...)
+       // WARNING : Si on veut ecrire du DICOM V3 a partir d'un DcmHeader ACR-NEMA
+       // no way
+       
+       
+   if (type == ExplicitVR) {
+      string explicitVRTransfertSyntax = "1.2.840.10008.1.2.1";
+      SetElValueByNumber(explicitVRTransfertSyntax, 0x0002, 0x0010);
+      // See above comment 
+      SetElValueLengthByNumber(20, 0x0002, 0x0010);
+   }
+
+   if ( (type == ImplicitVR) || (type == ExplicitVR) )
+      UpdateGroupLength();
+   if ( type == ACR)
+      UpdateGroupLength(true);
+
+   WriteElements(type, _fp);
+
+   return(1);
+}