StMultiKeyMap.h

#ifndef MULTIKEYMAP_H
#define MULTIKEYMAP_H
#include <vector>
class StMultiKeyMapIter;
class StMultiKeyNode;

class StMultiKeyMap
{
friend class StMultiKeyMapIter;
public:
StMultiKeyMap(int nKeys);
~StMultiKeyMap();
void Clear(const char *opt="");
void Add(const void *obj,const float  *keys);
void Add(const void *obj,const double *keys);
const StMultiKeyNode *GetTop() const    {return mTop;}
      StMultiKeyNode *GetTop()          {return mTop;}
                void  SetTop(StMultiKeyNode *top) {mTop = top;}
double Quality();
   int MakeTree();
   int ls(const char *file="") const;
   int Size() const;
   int GetNKey() const { return mNKey;} 
// statics
static void Test();
static void Test2();


protected:
int mNKey;
StMultiKeyNode *mTop;
std::vector<StMultiKeyNode*> mArr; 
};



class StMultiKeyNode
{
friend class StMultiKeyMapIter;
public:
StMultiKeyNode(int nKeys);
StMultiKeyNode(const StMultiKeyNode &source);
virtual ~StMultiKeyNode();
virtual void Set(const void *obj,const float  *keys);
        void Set(const void *obj,const double *keys);
        void Add(const void *obj,const float  *keys);
virtual void Add(StMultiKeyNode *node);
virtual       double Quality();
virtual       int    ls(const char *file="") const;

  const float *GetKeys() const { return mKeys;}
        float  GetKey()  const { return mKeys[int(mIKey)];}
        int    GetIKey() const { return mIKey;} 
        void   SetIKey(int ik) { mIKey=ik;    } 
        int    GetNKey() const { return mNKey;} 
        int    GetNumb(int way) const { return mNumb[way];}
        int    Size() const    { return mNumb[0]+mNumb[1]+1;}
        void  *GetObj () const { return (void*)mObj ;}
        void Clear();
static int GetNInst();
//      Non user functions
protected:
StMultiKeyNode *LLink() const {return mLink[0];}
StMultiKeyNode *RLink() const {return mLink[1];}
private:
void Init();


// statics
public:

protected:
char   mNKey;                   //keys number defined
char   mIKey;                   //key number used on this level
int    mNumb[2];                //Number of left/right objects
StMultiKeyNode *mLink[2];       //Left/Right subtree pointers
const void *mObj;               //Some user object mapped to keys
float *mKeys;                   //keys for this object
int    mId;
};


class StMultiKeyMapIter
{
public:
  StMultiKeyMapIter(const StMultiKeyNode *node,const float *kMin=0,const float *kMax=0);
  void          Set(const StMultiKeyNode *node,const float *kMin=0,const float *kMax=0);
  void       Update(const float *kMin=0,const float *kMax=0);
  void        Reset();

  ~StMultiKeyMapIter();
  StMultiKeyNode *operator*() const     {return (StMultiKeyNode*)mStk[mLev];}
  StMultiKeyMapIter &operator++();
  int Level() const                     {return mLev;}
  float *GetKMin() const                {return mKMin;}
  float *GetKMax() const                {return mKMax;}
  const int *Touched() const            {return mTouched;}
  private:
  const StMultiKeyNode *LLink(const StMultiKeyNode *node) const;
  const StMultiKeyNode *RLink(const StMultiKeyNode *node) const;
  const StMultiKeyNode *Left(const StMultiKeyNode *node);
  int FullCheck();
  protected:
  mutable int mTouched[3];
  std::vector<float> mMinMax;
  float *mKMin;
  float *mKMax;
  int mNK;
  int mLev;
  const StMultiKeyNode *mTop;
  std::vector<const StMultiKeyNode*> mStk;
};
#endif //MULTYKEYBINTREE_H