creaImageIOTreeComparators.h

Go to the documentation of this file.

#ifndef __creaImageIOTreeNodeComparators_h_INCLUDED__
#define __creaImageIOTreeNodeComparators_h_INCLUDED__

#include <vector>
#include <iostream>

namespace creaImageIO
{

  namespace tree
  {

    
    class Node;
    
    
  //=====================================================================
  struct Comparator
  {
    virtual ~Comparator() {}
    virtual bool operator() (Node* const & x, Node* const & y) = 0;
  };
  //=====================================================================



  //=====================================================================
  struct ComparatorWithOrder : public Comparator
  {
    ComparatorWithOrder(bool reverse_order = false) 
      : mReverseOrder(reverse_order) {}
    virtual ~ComparatorWithOrder() {}


    virtual bool compare(Node* const &, Node* const &) = 0;

    virtual bool operator() (Node* const & x, Node* const & y)
    {
      if (mReverseOrder) return this->compare(y,x);
      return this->compare(x,y);
    };

    bool mReverseOrder;
  };
  //=====================================================================



  //=====================================================================
  class LexicographicalComparator : public Comparator
  {
  public:
    LexicographicalComparator(const std::string& name) 
      : mName(name) {}
    ~LexicographicalComparator() {}

    const std::string& GetName() const { return mName; }
    void SetName(const std::string& s) { mName = s; }
    void Clear() { mComparator.clear(); }
    void DeleteComparators() 
    { 
       std::vector<Comparator*>::iterator i;
       for (i =mComparator.begin();
            i!=mComparator.end();
            ++i)
         {
           delete *i;
         }
       mComparator.clear(); 
    }
    void Add(Comparator* c) { mComparator.push_back(c); }
    
    bool operator() (Node* const & x, Node * const & y);

  private:
    std::string mName;
    std::vector<Comparator*> mComparator;
  };
  //=====================================================================




  //=================================================================== 
  struct IntComparator : 
    public ComparatorWithOrder
  {    
    IntComparator(const std::string& key,
                                     bool reverse_order)
      :
      ComparatorWithOrder(reverse_order),
      mKey(key)
    {}
    virtual bool compare(Node* const & x, Node* const & y);

  private:
    std::string mKey;
  };
  //===================================================================

  //===================================================================
  struct FloatComparator : 
    public ComparatorWithOrder
  {    
    FloatComparator(const std::string& key,
                                       bool reverse_order )
      : 
      ComparatorWithOrder(reverse_order),
      mKey(key)
    {}

    virtual bool compare(Node* const & x, Node* const & y);

  private:
    std::string mKey;
  };
  //===================================================================

  //===================================================================
  struct StringComparator : 
    public ComparatorWithOrder
  {    
    StringComparator(const std::string& key,
                                   bool reverse_order )
      : 
      ComparatorWithOrder(reverse_order),
      mKey(key)
    {}
    
    virtual bool compare(Node* const & x, Node* const & y);
    
  private:
    std::string mKey;
  };
  //===================================================================

  //===================================================================
#define INT_FIELD_COMP(NAME,FIELD)      \
  struct Node##NAME##Comparator :       \
    public IntComparator        \
  {                                             \
    Node##NAME##Comparator(bool o = false) :    \
      IntComparator(FIELD,o)    \
    {}                                          \
  }
  //==================================================================
  
  //===================================================================
#define FLOAT_FIELD_COMP(NAME,FIELD)            \
  struct Node##NAME##Comparator :       \
    public FloatComparator      \
  {                                             \
    Node##NAME##Comparator(bool o = false) :    \
      FloatComparator(FIELD,o)  \
    {}                                          \
  }
  //===================================================================

  //===================================================================
#define STRING_FIELD_COMP(NAME,FIELD)           \
  struct Node##NAME##Comparator :       \
    public StringComparator     \
  {                                             \
    Node##NAME##Comparator(bool o = false) :    \
      StringComparator(FIELD,o) \
    {}                                          \
  }
  //===================================================================



  //===================================================================
  // Patient comparators
  STRING_FIELD_COMP(PatientName,"A0010_0010");
  STRING_FIELD_COMP(PatientSex, "A0010_0040");
  STRING_FIELD_COMP(PatientBirthday, "A0010_0030");
  //===================================================================

  //===================================================================
  // Study comparators
  STRING_FIELD_COMP(StudyDate,"A0008_0020");
  STRING_FIELD_COMP(StudyDescription,"A0008_1030");
  //===================================================================

  //===================================================================
  // Series comparators
  STRING_FIELD_COMP(Modality,"A0008_0060");
  STRING_FIELD_COMP(SeriesDescription,"A0008_103E");
  STRING_FIELD_COMP(SeriesDate,"A0008_0021");
  //===================================================================

  //===================================================================
  // Image comparators
  INT_FIELD_COMP(ImageNumber,"A0020_0013");
  FLOAT_FIELD_COMP(SliceLocation,"A0020_1041");
  STRING_FIELD_COMP(FullFileName,"FullFileName");
  //===================================================================
  } // namespace tree

} // namespace creaImageIO



#endif // #ifndef __creaImageIOComparators_h_INCLUDED__