MemoryMapManager.h

/*
 * MemoryMapManager.h
 *
 *  Created on: 28-10-2013
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
 
#ifndef HALMEMORYMAPMANAGER_H_
#define HALMEMORYMAPMANAGER_H_
 
 
#include "Array.h"
#include "CompressionMap.h"
#include "Event.h"
 
namespace Hal {
  class IOManager;
  class CutContainer;
  class MiniEventTask;
  class EventArray : public TObject {
  public:
    Int_t fSize;
    Event** fArray;  //[fSize]
    EventArray(Int_t task_id, Int_t size = 1);
    inline Event* At(const Int_t size) { return fArray[size]; };
    virtual ~EventArray() {
      for (int i = 0; i < fSize; i++)
        delete fArray[i];
      delete[] fArray;
    };
    ClassDef(EventArray, 1)
  };
  class MemoryMapManager : public TObject {
    friend class MiniEventTask;
 
  private:
    Bool_t fUseCompression;
    Bool_t fDirectAcces;
    Int_t fTrackMapSize, fFormatID;
    Int_t fEventCollectionsNo, fTrackCollectionsNo, fMixSize;
    Int_t* fEventToTrackNo;  //[fEventCollectionsNo]
    Int_t* fCounter;         //[fEventCollectionsNo]
    Int_t* fTotalTracks;     //[fEventCollectionsNo]
    Bool_t* fReadyToMix;     //[fEventCollectionsNo]
    EventArray** fEvents;    //[fEventCollectionsNo]
    EventInterface* fInterface;
    //"true" collections x mix_size
    CompressionMap fCompression;
    Int_t fMaxTrackCollectionNo;
    Event* fCurrentEvent;  // pointer to processed event
    Array_4<Int_t>* fTrackMap;
    Array_3<Int_t>* fTrackCounter;
    void ReloadMap(Int_t size);
    void CalculateCompressedMap(Int_t event_collection);
    Int_t GetTrackCollectionsNo(Int_t event_collection) const;
    Int_t GetCounter(Int_t event_collection, Int_t track_collection) const;
    void ResetTrackMaps(Int_t event_collection, Int_t counter);
    void ResetTrackMaps(Int_t event_collection, Int_t track_collection, Int_t counter);
 
  public:
    MemoryMapManager();
    MemoryMapManager(CutContainer* cont);
    Event* GetTemporaryEvent();
    void ManualUpdate(Event* event);
    void SetMixSize(Int_t mix_size);
    Int_t GetTemporaryTotalTracksNo() const;
    void Init(Int_t event_factor, Int_t task_id, Bool_t compress, std::vector<TString> direct);
    void AddTrackToMapTrack(Int_t event_collection, Int_t track_collection, Int_t index);
    void BufferEvent(Int_t collection);
    Event* GetEvent(Int_t collection) const;
    Track* GetTrack(Int_t event_collection, Int_t track_collection, Int_t index) const;
    Track* GetTemporaryTrack(Int_t event_collection, Int_t track_collection, Int_t index) const;
    Track* GetRawTrack(Int_t event_collection, Int_t index) const;
    Event* GetEvent(Int_t collection, Int_t counter) const;
    Track* GetTrack(Int_t event_collection, Int_t track_collection, Int_t counter, Int_t index) const;
    Track* GetRawTrack(Int_t event_collection, Int_t counter, Int_t index) const;
    void RejectLastEvent(Int_t event_collection);
    void ResetEventCollection(Int_t collection);
    void ClearEvent();
    void PrepareMaps(Int_t collection);
    void PrintMap() const;
    Int_t GetCounter(Int_t event_collection) const;
    Int_t GetTracksNo(Int_t event_collection, Int_t track_collection) const;
    Int_t GetTracksNo(Int_t event_collection, Int_t track_collection, Int_t counter) const;
    Int_t GetRawTracksNo(Int_t event_collection) const;
    Bool_t IsReadyToMixing(Int_t collection) const;
    virtual ~MemoryMapManager();
    ClassDef(MemoryMapManager, 1)
  };
}  // namespace Hal
#endif /* HALMEMORYMAPMANAGER_H_ */