/*
 * Copyright (c) 2019, SAP SE. All rights reserved.
 * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 *
 */

#ifndef SHARE_MEMORY_METASPACE_LEFTOVERBINS_HPP
#define SHARE_MEMORY_METASPACE_LEFTOVERBINS_HPP

#include "memory/allocation.hpp"
#include "memory/metaspace/counter.hpp"
#include "utilities/bitMap.hpp"
#include "utilities/debug.hpp"
#include "utilities/globalDefinitions.hpp"


class outputStream;

namespace metaspace {

// The LeftOverManager is responsible for managing small leftover-
// and deallocated blocks.
// They come from two sources:
// a) the leftover space left in a chunk when a chunk gets retired
//    because it cannot serve a requested allocation. These blocks
//    can be largeish (100s - 1000s of words).
// b) when a metaspace allocation is deallocated prematurely - e.g.
//    due to interrupted class loading. These blocks are small or
//    very small.

class BinMap {

  typedef uint32_t mask_type;
  mask_type _mask;

  static mask_type mask_for_pos(int pos) { return 1 << pos; }

public:

  BinMap() : _mask(0) {}

  bool all_zero() const          { return _mask == 0; }

  bool get_bit(int pos) const    { return (_mask & mask_for_pos(pos)) != 0 ? true : false; }
  void set_bit(int pos)          { _mask |= mask_for_pos(pos); }
  void clr_bit(int pos)          { _mask &= ~mask_for_pos(pos); }

  // Starting at (including) pos, find the position of the next 1 bit.
  // Return -1 if not found.
  inline int find_next_set_bit(int pos) const;

  static int size() { return sizeof(mask_type) * 8; }

};

struct block_t {
  block_t* next;
  size_t size;
};

struct block_stats_t {
  size_t word_size;
  int num_blocks;
};

template <
  size_t min_word_size,
  size_t spread,
  int num_bins
>
class Bins {

  STATIC_ASSERT(sizeof(block_t) <= (min_word_size * BytesPerWord));

  block_t* _bins[num_bins];

  BinMap _mask;

  // e.g. spread = 4
  //
  // sz    bno (put)  bno (get)
  //         (guarant)
  // 0     00         00
  // 1     00         01
  // 2     00         01
  // 3     00         01
  // 4     01         01
  // 5     01         02
  // 6     01         02
  // 7     01         02
  // 8     02         02
  // 9     02         03
  // 10    02         03
  // 11    02         03
  //
  // put -> no = wordsize / spread
  //
  // get -> no = (req_wordsize + spread - 1) / spread

  // The bin number for a given word size.
  static int bin_for_size(size_t word_size) {
    assert(word_size >= min_word_size && word_size < maximal_word_size(),
           "Word size oob (" SIZE_FORMAT ")", word_size);
    return (word_size - min_word_size) / spread;
  }

  // [minimal, maximal) size of blocks which are held in a bin.
  // Note that when taking a block out of the bin, only the minimum block size
  // is guaranteed.
  static size_t minimal_word_size_in_bin(int bno) {
    return min_word_size + (bno * spread);
  }
  static size_t maximal_word_size_in_bin(int bno) {
    return minimal_word_size_in_bin(bno) + spread;
  }

public:

  Bins() : _mask() {
    assert(BinMap::size() >= num_bins, "mask too small");
    ::memset(_bins, 0, sizeof(_bins));
  }

  // [min, max) word size
  static size_t minimal_word_size() { return min_word_size; }
  static size_t maximal_word_size() { return min_word_size + (spread * num_bins); }

  inline void put(MetaWord* p, size_t word_size);

  inline block_t* get(size_t word_size);

#ifdef ASSERT
  void verify() const;
#endif

  void statistics(block_stats_t* stats) const;

  void print(outputStream* st) const;

};


class LeftOverManager : public CHeapObj<mtInternal> {

  typedef Bins<2, 2, 16> VerySmallBinsType;
  VerySmallBinsType _very_small_bins;

  block_t* _large_block_reserve;

  // The current large block we gnaw on
  MetaWord* _current;
  size_t _current_size;

  SizeCounter _total_word_size;

  // Take the topmost block from the large block reserve list
  // and make it current.
  inline void prime_current();

  // Allocate from current block. Returns NULL if current block
  // is too small.
  inline MetaWord* alloc_from_current(size_t word_size);

  void large_block_statistics(block_stats_t* stats) const;

public:

  static size_t minimal_word_size() {
    return VerySmallBinsType::minimal_word_size();
  }

  LeftOverManager() :
    _very_small_bins(),
    _large_block_reserve(NULL),
    _current(NULL),
    _current_size(0)
  {}

  inline void add_block(MetaWord* p, size_t word_size);

  inline MetaWord* get_block(size_t requested_word_size);

#ifdef ASSERT
  void verify() const;
#endif

  void statistics(block_stats_t* stats) const;

  void print(outputStream* st, bool detailed = false) const;

  size_t total_word_size() const { return _total_word_size.get(); }

};




} // namespace metaspace

#endif // SHARE_MEMORY_METASPACE_CHUNKMANAGER_HPP