MeVis/Foundation/Sources/ML/include/host/mlTileRequest.h

Go to the documentation of this file.

// **InsertLicense** code
//-------------------------------------------------------------------------
//-------------------------------------------------------------------------
#ifndef __mlTileRequest_H
#define __mlTileRequest_H

// ML-includes
#ifndef __mlInitSystemML_H
#include "mlInitSystemML.h"
#endif

#include "mlSubImage.h"
#include "mlPagedImage.h"
#include "mlPageIDIterator.h"

ML_START_NAMESPACE

class TileRequest;
class PageRequest;
class PageRequestQueueInterface;

typedef void TileRequestFinishedCB(void* data, TileRequest* request);

//-------------------------------------------------------------------------------------------
//-------------------------------------------------------------------------------------------
class ML_UNIX_ONLY_EXPORT(MLEXPORT) TileRequest
{
public:
  TileRequest(PagedImage* inputImage, const SubImageBox& box, MLDataType dataType,
              const ScaleShiftData& scaleShift, bool readOnlyInputTile = false, bool tryToBecomeMemoryImage = false);

  ~TileRequest();

  enum AllocationPolicy {
    UseMemoryManager = 0,  //<! This will allocate the memory using the MemoryManager
    UseMLAlloc       = 1,  //<! This will allocate the memory using MLAlloc
    NoAllocation     = 2,  //<! This can be used if no data should be allocated at all (e.g., for processAllPages)
    ExternalBuffer   = 3   //<! This can be used to provide the data by the user (and to avoid freeing the data when something goes wrong)
  };

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

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

  bool prepareForCursorVisit();

  void  createInputPageIds();

  void  setNeededBy(PageRequest* request) {
    _neededByPageRequest = request;
  }

  bool  isBlockedByMemoryImg();

  void pageRequestFinished(SubImage& pageImage, PageRequestQueueInterface& queue);


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

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

  void setAllocationPolicy(AllocationPolicy policy) { _allocationPolicy = policy; }

  AllocationPolicy getAllocationPolicy() const { return _allocationPolicy; }

  MLErrorCode allocateData();

  void  forgetData() { _subImage.setData(NULL); }

  void  freeData() { if (_allocationPolicy!=ExternalBuffer) { _subImage.free(); } }

  const SubImage& getSubImage() const { return _subImage; }

  void setExternalDataBuffer(void* data) {
    setAllocationPolicy(TileRequest::ExternalBuffer);
    _subImage.setData(data);
  }

  void setUseTileReadOnly(bool flag) { _readOnlyInputTile = flag; }


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

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

  bool hasFinished() const;

  void setTileRequestFinishedCB(TileRequestFinishedCB* cb, void* data) {
    _finishedCB = cb;
    _finishedCBData = data;
  }

  double getProgress() const;


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

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

  bool hasError() const { return _error!=ML_RESULT_OK; }

  void setError(MLErrorCode error);

  MLErrorCode getError() const { return _error; }

  void propagateErrorUpwards(MLErrorCode error);

  bool propagateCancellationDownIfRequestNotNeededAnymore();

  void collectPageRequestsWithErrors(std::set<PageRequest*>& result);

#ifdef ML_TILEREQUEST_DEBUG_PRINTING
  // TODO: remove or replace with stream ops
  void print(int depth);

  // TODO: remove or replace with stream ops
  void printCachedPage(int depth, int pageId);
#endif


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

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

  MLErrorCode updateImageProperties();


protected:
  friend class PageRequestCursor;

  PageRequest* createPageRequest(MLint pageId, PageRequestQueueInterface& queue);

  void         copyDataFromSubImg(SubImage& pageImage);

  void         sendTileRequestFinished(PageRequestQueueInterface& queue);

  SubImage _subImage;

  AllocationPolicy _allocationPolicy;

  ScaleShiftData _scaleShiftData;

  bool           _readOnlyInputTile;
  bool           _tileWillBecomeMemoryImage;

  PagedImage*  _image;

  MLint      _inputPagesNeeded;

  PageIDIterator _inputPageIds;

  MLint          _traversalCursorPosition;

  PageRequest*   _neededByPageRequest;

  MLErrorCode    _error;

  TileRequestFinishedCB* _finishedCB;
  void*                  _finishedCBData;
};

ML_END_NAMESPACE

#endif //of __mlTileRequest_H