source: Sophya/trunk/SophyaLib/BaseTools/segdatablock.h@ 3213

// This may look like C code, but it is really -*- C++ -*-
// Gestion de block de donnees avec partage de references
//                         R. Ansari  Mars 2005
// LAL (Orsay) / IN2P3-CNRS  DAPNIA/SPP (Saclay) / CEA
#ifndef SEGDATABLOCK_H
#define SEGDATABLOCK_H

#include "machdefs.h"
#include "anydataobj.h"
#include <string.h> 
#include <vector>
#include <iostream>
#include <iomanip>
#include <typeinfo>

#include "thsafeop.h"  //  for ThreadSafe operations (Ref.Count/Share) 

/*!
   \class SOPHYA::SegDBInterface
   \ingroup BaseTools
   Interface definition for segmented data container (template class).
*/
/*!
   \class SOPHYA::SegDataBlock
   \ingroup BaseTools
   Template class impementing segmented data container in memory with 
   management of reference sharing. 
*/

namespace SOPHYA {

////////////////////////////////////////////////////////////////
//// ---------------- Class SegDBInterface ------------------ //
//// ----------------- Class SegDataBlock ------------------- //
////////////////////////////////////////////////////////////////

template <class T>
class SegDBInterface : public AnyDataObj {
public:
  virtual ~SegDBInterface() {}
  //! Changes the data segment size and reallocates the memory segments
  // segsz : Segment size ;  nbseg : Number of data segments
  virtual void SetSize(size_t segsz, size_t nbseg=0) = 0;
  //! Alias for SetSize()
  inline void  ReSize(size_t segsz, size_t nbseg=0) { SetSize(segsz, nbseg); }
  //! Adds one segment to the data structure - returns the new number of segments.
  virtual size_t Extend() = 0;
  //! Return the segment size data structure
  virtual  size_t SegmentSize() const = 0; 
  //! Return the number of data segments
  virtual size_t NbSegments() const = 0;
  //! Return the current size of the segmented data structure
  inline size_t Size() const    { return  SegmentSize()*NbSegments(); }
  //! Return the pointer to data segment \b k
  virtual T* GetSegment(size_t k) = 0;   
  //! Return the const (read-only) pointer to data segment \b k      
  virtual T const * GetCstSegment(size_t k) const = 0; 
};


// classe de container avec partage de reference
template <class T>
class SegDataBlock : public SegDBInterface<T> {

public:

  //! Default constructor - optional specification of segment size and number of segments
  explicit SegDataBlock(size_t segsz=32, size_t nbseg=0)
  {
    mSRef = NULL;
    if (gThsop == NULL) gThsop = new ThSafeOp;
    SetSize(segsz, nbseg);
  }
  //! copy constructor - shares the data
  SegDataBlock(const SegDataBlock<T>& a)
  {
    gThsop->lock();   //  (ThreadSafe) - Start of atomic operation 
    mSRef = a.mSRef;
    mSRef->nref++;
    gThsop->unlock();   //  (ThreadSafe) - End of atomic operation
  }
  //! copy constructor with specification of flags for data sharing and element value copy
  SegDataBlock(const SegDataBlock<T>& a, bool share, bool cpval=true)
  {
    if (share) {
      gThsop->lock();   //  (ThreadSafe) - Start of atomic operation 
      mSRef = a.mSRef;
      mSRef->nref++;
      gThsop->unlock();   //  (ThreadSafe) - End of atomic operation
    }
    else {
      mSRef = NULL;
      Clone(a, cpval);
    }
  }
  //! copy constructor - shares the data if \b is a SegDataBlock, clones otherwise
  SegDataBlock(const SegDBInterface<T>& a)
  {
    if (gThsop == NULL) gThsop = new ThSafeOp;
    SegDataBlock<T> * sdb = dynamic_cast< SegDataBlock<T> *>(&a);
    if (sdb != NULL) { 
      gThsop->lock();   //  (ThreadSafe) - Start of atomic operation 
      mSRef = sdb->mSRef;
      mSRef->nref++;
      gThsop->unlock();   //  (ThreadSafe) - End of atomic operation
    }
    else Clone(a, true);
  }
  //! Destructor. The memory is freed when the last object referencing the data segment is destroyed
  virtual ~SegDataBlock()
  {
    //DEL    cout << " DEBUG-~SegDataBlock() " << hex << mSRef << dec << " NRef()= " << NRef() << endl;
    gThsop->lock();   //  (ThreadSafe) - Start of atomic operation 
    Delete();
    gThsop->unlock();   //  (ThreadSafe) - End of atomic operation
  }


  //! Adds one segment to the data structure - returns the number of allocated segments.
  virtual size_t Extend()
  {
    T * p = new T[mSRef->segsize];
    gThsop->lock();   //  (ThreadSafe) - Start of atomic operation 
    mSRef->dseg.push_back(p);
    gThsop->unlock();   //  (ThreadSafe) - End of atomic operation
    return( mSRef->dseg.size());
  }

  //! Changes the data segment size and reallocates the memory segments
  // segsz : Segment size ;  nbseg : Number of data segments
  virtual void SetSize(size_t segsz, size_t nbseg=0)
  {
    gThsop->lock();   //  (ThreadSafe) - Start of atomic operation 
    Delete();
    mSRef = new SDREF;
    mSRef->nref = 1;
    mSRef->segsize = segsz;
    mSRef->dsid = AnyDataObj::getUniqueId();
    for(size_t k=0; k<nbseg; k++) Extend_P();
    gThsop->unlock();   //  (ThreadSafe) - End of atomic operation
  }
//! Shares the data between two SegDataBlock objects
  void Share(const SegDataBlock<T>& a)
  {
    gThsop->lock();   //  (ThreadSafe) - Start of atomic operation 
    Delete();
    mSRef = a.mSRef;
    mSRef->nref++;
    gThsop->unlock();   //  (ThreadSafe) - End of atomic operation
  }

//! Makes a clone of the data structure and optionaly copie the data 
  void Clone(const SegDBInterface<T> & a, bool cpval=true)
  {
    gThsop->lock();   //  (ThreadSafe) - Start of atomic operation 
    Delete();
    mSRef = new SDREF;
    mSRef->nref = 1;
    mSRef->segsize = a.SegmentSize();
    gThsop->unlock();   //  (ThreadSafe) - End of atomic operation
    for(size_t k=0; k<a.NbSegments(); k++) {
      Extend();
      if (cpval) {
        T * dst = GetSegment(k);
        const T * src = a.GetCstSegment(k);
        memcpy(dst, src, mSRef->segsize*sizeof(T));
      }
    }   
  }


  //! Return the segment size of the data structure
  virtual size_t SegmentSize() const  { return mSRef->segsize; }
  //! Return the number of data segments
  virtual size_t NbSegments() const  { return mSRef->dseg.size(); }
  //! Return the current size of the segmented data structure
  inline size_t Size() const    { return mSRef->segsize*mSRef->dseg.size(); }

  //! Return the pointer to data segment \b k      
  virtual T* GetSegment(size_t k)              { return mSRef->dseg[k]; } 
  //! Return the const (read-only) pointer to data segment \b k      
  virtual T const * GetCstSegment(size_t k) const { return mSRef->dseg[k]; } 

  //! Return the segment index for  element \b i
  inline size_t SegIndex(size_t i) {  return i/mSRef->segsize; }
  //! Return the offset (in data segment)  for  element \b i
  inline size_t EltOffset(size_t i) {  return i%mSRef->segsize; }

  //! Return the \b i th element of the segmented data structure   
  inline T& operator()(size_t i)       { return *(mSRef->dseg[SegIndex(i)]+EltOffset(i));}
  //! Return the \b i th element of  the data structure
  inline T& operator()(size_t i) const { return *(mSRef->dseg[SegIndex(i)]+EltOffset(i));}
  //! Return the \b i th element of the segmented data structure   
  inline T& operator[](size_t i)       { return *(mSRef->dseg[SegIndex(i)]+EltOffset(i));}
  //! Return the \b i th element of  the data structure
  inline T& operator[](size_t i) const { return *(mSRef->dseg[SegIndex(i)]+EltOffset(i));}


  //! Return the number of references to the data structure
  inline size_t NRef() const { return mSRef->nref; }

  //! Equal operator. Set all element values to \b v 
  SegDataBlock<T>& operator = (T const & v)
  {
    for(size_t k=0; k<NbSegments(); k++) {
      T * p = mSRef->dseg[k];
      for(size_t j=0; j<SegmentSize(); j++) p[j] = v;
    }
    return (*this);
  }
  //! Equal operator. Clones and copie values from \b a
  inline SegDataBlock<T>& operator = (const SegDBInterface<T> & a)
  {
    Clone(a, true);
    return (*this);
  }
  //! Equal operator. Clones and copie values from \b a
  inline SegDataBlock<T>& operator = (const SegDataBlock<T> & a)
  {
    Clone(a, true);
    return (*this);
  }

  //! ASCII formatted output (print)
  void Print(ostream& os, int lev=0, const char * sep=NULL) const
  {
    os << "SegDataBlock< " << typeid(T).name() << "> mSRef= " << hex << mSRef
       << " NRef=" << dec << NRef() << " DSId= " <<  DRefId() << endl;
    os << " ... SegSize= " << SegmentSize() << " NbSeg= " << NbSegments() 
       << "  Size= " << Size() << endl;
    if (sep == NULL) sep = "  ";
    if (lev > 0) {  
      for(size_t k=0; k<NbSegments(); k++) {
        T * p = mSRef->dseg[k];
        os << " ..... DataSeg[ " << k << " ] : " << hex << p << dec << endl;
        if (lev > 1) 
          for(size_t j=0; j<SegmentSize(); j++)  os << p[j] << sep;
        os << endl;     
      }
    }
  }
  //! ASCII formatted output (print) on cout
  inline void Print(int lev=0, const char * sep=NULL) const
  {
    Print(cout, lev, sep);
  }

  //! Returns the unique object identifier 
  inline uint_8 DRefId() const { return mSRef->dsid; }

protected:
  //! NON-thread safe: Decrement the number of reference counts, and free the memory if NRef=0
  void Delete() 
  {
    if (mSRef == NULL) return;
    mSRef->nref--;
    if (mSRef->nref > 0)  { mSRef = NULL; return; }
    //DEL cout << " DEBUG-SegDataBlock::Delete() NbSegments() = " << NbSegments() << endl;
    for(size_t k=0; k<NbSegments(); k++) {
      delete[] mSRef->dseg[k];
      mSRef->dseg[k] = NULL;
    }
    delete mSRef;
    mSRef = NULL;
  }
  //! NON-thread safe, version of Extend() : Adds one segment to the data structure 
  size_t Extend_P()
  {
    T * p = new T[mSRef->segsize];
    mSRef->dseg.push_back(p);
    return( mSRef->dseg.size());
  }

  /*! \cond 
    SDREF structure for reference management - for internal use by SegDataBlock
  */
  typedef struct {
    size_t nref;      // Number of references to the data structure
    uint_8 dsid;      // Data structure Id - Used by FIO_SegDataBlock
    std::vector<T *> dseg;
    size_t segsize;
  } SDREF;
  /*! \endcond */
  SDREF *  mSRef;    //!< SDREF structure for reference sharing

  static ThSafeOp* gThsop;  //!< Mutex for thread safe operation 
};

template <class T> ThSafeOp* SegDataBlock<T>::gThsop  = NULL;  // static member initialized to NULL

//! Definition of operator \<\< for ascii formatted output of SegDataBlock
template<class T> 
inline ostream& operator << (ostream& os, const SegDataBlock<T>& a)
  {  a.Print(os); return(os); }

} // Fin du namespace

#endif