OCCT

// Copyright (c) 1998-1999 Matra Datavision
// Copyright (c) 1999-2014 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.

#ifndef _Storage_BucketOfPersistent_HeaderFile
#define _Storage_BucketOfPersistent_HeaderFile

#include <Standard_Integer.hxx>
#include <Standard_Persistent.hxx>

class Storage_Schema;
class Storage_BucketOfPersistent;
class Storage_BucketIterator;

class Storage_Bucket
{
  friend class Storage_BucketIterator;
  friend class Storage_Schema;
  friend class Storage_BucketOfPersistent;

  Standard_Persistent** mySpace;
  int                   mySpaceSize;
  int                   myCurrentSpace;

  void Append(Standard_Persistent*);

  Standard_Persistent* Value(const int theIndex) const;

public:
  Storage_Bucket()
      : mySpace(nullptr),
        mySpaceSize(200000),
        myCurrentSpace(-1)
  {
    mySpace = (Standard_Persistent**)Standard::Allocate(sizeof(Standard_Persistent*) * mySpaceSize);
  }

  Storage_Bucket(const int theSpaceSize)
      : mySpace(nullptr),
        mySpaceSize(theSpaceSize),
        myCurrentSpace(-1)
  {
    mySpace = (Standard_Persistent**)Standard::Allocate(sizeof(Standard_Persistent*) * mySpaceSize);
  }

  void Clear();

  ~Storage_Bucket();
};

class Storage_BucketOfPersistent
{
  friend class Storage_BucketIterator;
  Storage_Bucket** myBuckets;
  int              myNumberOfBucket;
  int              myNumberOfBucketAllocated;
  Storage_Bucket*  myCurrentBucket;
  int              myCurrentBucketNumber;
  int              myLength;
  int              myBucketSize;

public:
  Storage_BucketOfPersistent(const int theBucketSize = 300000, const int theBucketNumber = 100);

  int Length() const { return myLength; }

  void Append(const occ::handle<Standard_Persistent>& sp);

  Standard_Persistent* Value(const int theIndex);

  void Clear();

  ~Storage_BucketOfPersistent();
};

class Storage_BucketIterator
{
  Storage_BucketOfPersistent* myBucket;
  Storage_Bucket*             myCurrentBucket;
  int                         myCurrentBucketIndex;
  int                         myCurrentIndex;
  int                         myBucketNumber;
  bool                        myMoreObject;

public:
  Storage_BucketIterator(Storage_BucketOfPersistent*);
  void Init(Storage_BucketOfPersistent*);
  void Reset();

  Standard_Persistent* Value() const
  {
    if (myCurrentBucket)
    {
      return myCurrentBucket->mySpace[myCurrentIndex];
    }
    else
      return nullptr;
  }

  bool More() const { return myMoreObject; }

  void Next();
};

#endif