IceRevisitedRadix.h

Go to the documentation of this file.

//----------------------------------------------------------------------
//----------------------------------------------------------------------

//----------------------------------------------------------------------
// Include Guard
#ifndef RecoTracker_MkFitCore_src_Ice_IceRevisitedRadix_h
#define RecoTracker_MkFitCore_src_Ice_IceRevisitedRadix_h

#include "IcePreprocessor.h"
#include "IceTypes.h"

#define RADIX_LOCAL_RAM

enum RadixHint {
  RADIX_SIGNED,    
  RADIX_UNSIGNED,  

  RADIX_FORCE_DWORD = 0x7fffffff
};

class RadixSort {
public:
  // Constructor/Destructor
  RadixSort();
  ~RadixSort();
  // Sorting methods
  RadixSort& Sort(const udword* input, udword nb, RadixHint hint = RADIX_SIGNED);
  RadixSort& Sort(const float* input, udword nb);

  //i.e. in the order you may further process your data
  const udword* GetRanks() const { return mRanks; }

  udword* RelinquishRanks();

  //otherwise you can recycle it the way you want.
  udword* GetRecyclable() const { return mRanks2; }

  // Stats
  udword GetUsedRam() const;
  udword GetNbTotalCalls() const { return mTotalCalls; }
  udword GetNbHits() const { return mNbHits; }

private:
#ifndef RADIX_LOCAL_RAM
  udword* mHistogram;  
  udword* mLink;       
#endif
  udword mCurrentSize;  
  udword* mRanks;       
  udword* mRanks2;
  // Stats
  udword mTotalCalls;  
  udword mNbHits;      

  // Internal methods
  void CheckResize(udword nb);
  bool Resize(udword nb);
};

#endif  // __ICERADIXSORT_H__