// @(#)root/io:$Id$
// Author: Fons Rademakers   08/07/97

/*************************************************************************
 * Copyright (C) 1995-2000, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/

#ifndef ROOT_TMapFile
#define ROOT_TMapFile

#ifdef WIN32
#include "Windows4Root.h"
#endif

#include "TObject.h"
#include "TVirtualMapFile.h"

class TDirectory;
class TList;
class TMapRec;

class TMapFile : public TVirtualMapFile {

friend class TMapRec;

private:
   Longptr_t   fFd;             ///< Descriptor of mapped file
   Int_t       fVersion;        ///< ROOT version (or -1 for shadow map file)
   char       *fName;           ///< Name of mapped file
   char       *fTitle;          ///< Title of mapped file
   char       *fOption;         ///< Directory creation options
   void       *fMmallocDesc;    ///< Pointer to mmalloc descriptor
   ULongptr_t  fBaseAddr;       ///< Base address of mapped memory region
   Int_t       fSize;           ///< Original start size of memory mapped region
   TMapRec    *fFirst;          ///< List of streamed objects is shared memory
   TMapRec    *fLast;           ///< Last object in list of shared objects
   Longptr_t   fOffset;         ///< Offset in bytes for region mapped by reader
   TDirectory *fDirectory;      ///< Pointer to directory associated to this mapfile
   TList      *fBrowseList;     ///< List of KeyMapFile objects
   Bool_t      fWritable;       ///< TRUE if mapped file opened in RDWR mode
   Longptr_t   fSemaphore;      ///< Modification semaphore (or getpid() for WIN32)
   ULongptr_t  fhSemaphore;     ///< HANDLE of WIN32 Mutex object to implement semaphore
   TObject    *fGetting;        ///< Don't deadlock in update mode, when from Get() Add() is called
   Int_t       fWritten;        ///< Number of objects written so far
   Double_t    fSumBuffer;      ///< Sum of buffer sizes of objects written so far
   Double_t    fSum2Buffer;     ///< Sum of squares of buffer sizes of objects written so far

   static Longptr_t fgMapAddress;  ///< Map to this address, set address via SetMapAddress()
   static void  *fgMmallocDesc; ///< Used in Close() and operator delete()

protected:
   TMapFile();
   TMapFile(const char *name, const char *title, Option_t *option, Int_t size, TMapFile *&newMapFile);
   TMapFile(const TMapFile &f, Longptr_t offset = 0);

   TMapFile &operator=(const TMapFile &rhs) = delete;

   TMapFile  *FindShadowMapFile();
   void       InitDirectory();
   TObject   *Remove(TObject *obj, Bool_t lock);
   TObject   *Remove(const char *name, Bool_t lock);
   void       SumBuffer(Int_t bufsize);
   Int_t      GetBestBuffer();

   void   CreateSemaphore(Int_t pid = 0);
   Int_t  AcquireSemaphore();
   Int_t  ReleaseSemaphore();
   void   DeleteSemaphore();

   static void *MapToAddress();

   ~TMapFile() override;
   void *operator new(size_t sz) { return TObject::operator new(sz); }
   void *operator new[](size_t sz) { return TObject::operator new[](sz); }
   void *operator new(size_t sz, void *vp) { return TObject::operator new(sz, vp); }
   void *operator new[](size_t sz, void *vp) { return TObject::operator new[](sz, vp); }
   void     operator delete(void *vp);

public:
   enum { kDefaultMapSize = 0x80000 }; // default size of mapped heap is 500 KB

   void          Browse(TBrowser *b) override;
   void          Close(Option_t *option = "") override;
   void         *GetBaseAddr() const { return (void *)fBaseAddr; }
   void         *GetBreakval() const;
   TDirectory   *GetDirectory() const {return fDirectory;}
   Int_t         GetFd() const { return fFd; }
   void         *GetMmallocDesc() const { return fMmallocDesc; }
   const char   *GetName() const override { return fName; }
   Int_t         GetSize() const { return fSize; }
   const char   *GetOption() const override { return fOption; }
   const char   *GetTitle() const override { return fTitle; }
   TMapRec      *GetFirst() const { return (TMapRec*)((Longptr_t) fFirst + fOffset); }
   TMapRec      *GetLast() const { return (TMapRec*)((Longptr_t) fLast + fOffset); }
   Bool_t        IsFolder() const override;
   Bool_t        IsWritable() const { return fWritable; }
   void         *OrgAddress(void *addr) const { return (void *)((Longptr_t)addr - fOffset); }
   void          Print(Option_t *option="") const override;
   void          ls(Option_t *option="") const override;
   Bool_t        cd(const char *path = nullptr);

   void          Add(const TObject *obj, const char *name = "");
   void          Update(TObject *obj = nullptr);
   TObject      *Remove(TObject *obj) { return Remove(obj, kTRUE); }
   TObject      *Remove(const char *name) { return Remove(name, kTRUE); }
   void          RemoveAll();
   TObject      *Get(const char *name, TObject *retObj = nullptr);

   static TMapFile *Create(const char *name, Option_t *option="READ", Int_t size=kDefaultMapSize, const char *title="");
   static TMapFile *WhichMapFile(void *addr);
   static void      SetMapAddress(Longptr_t addr);

   ClassDefOverride(TMapFile,0)  // Memory mapped directory structure
};



/**
\class TMapRec
\ingroup IO

Keep track of an object in the mapped file.

A TMapFile contains a list of TMapRec objects which keep track of
the actual objects stored in the mapped file.
*/

class TMapRec {

friend class TMapFile;

private:
   char            *fName;       ///< Object name
   char            *fClassName;  ///< Class name
   TObject         *fObject;     ///< Pointer to original object
   void            *fBuffer;     ///< Buffer containing object of class name
   Int_t            fBufSize;    ///< Buffer size
   TMapRec         *fNext;       ///< Next MapRec in list

   TMapRec(const TMapRec&) = delete;
   TMapRec &operator=(const TMapRec&) = delete;

public:
   TMapRec(const char *name, const TObject *obj, Int_t size, void *buf);
   ~TMapRec();
   const char   *GetName(Longptr_t offset = 0) const { return (char *)((Longptr_t) fName + offset); }
   const char   *GetClassName(Longptr_t offset = 0) const { return (char *)((Longptr_t) fClassName + offset); }
   void         *GetBuffer(Longptr_t offset = 0) const { return (void *)((Longptr_t) fBuffer + offset); }
   Int_t         GetBufSize() const { return fBufSize; }
   TObject      *GetObject() const;
   TMapRec      *GetNext(Longptr_t offset = 0) const { return (TMapRec *)((Longptr_t) fNext + offset); }
};

#endif
