// Copyright 2016 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#pragma once

#include <stddef.h>

#include <zircon/types.h>

namespace bitmap {

// An abstract bitmap.
class Bitmap {
public:
    virtual ~Bitmap() = default;

    // Finds a run of |run_len| |is_set| bits, between |bitoff| and |bitmax|.
    // Sets |out| with the start of the run, or |bitmax| if it is
    // not found in the provided range.
    // If the run is not found, "ZX_ERR_NO_RESOURCES" is returned.
    virtual zx_status_t Find(bool is_set, size_t bitoff, size_t bitmax,
                             size_t run_len, size_t* out) const = 0;

    // Returns true in the bit at bitoff is set.
    virtual bool GetOne(size_t bitoff) const {
        return Get(bitoff, bitoff + 1, nullptr);
    }

    // Returns true if all the bits in [*bitoff*, *bitmax*) are set. Afterwards,
    // *first_unset* will be set to the lesser of bitmax and the index of the
    // first unset bit after *bitoff*.
    virtual bool Get(size_t bitoff, size_t bitmax,
                     size_t* first_unset = nullptr) const = 0;

    // Sets the bit at bitoff.  Only fails on allocation error.
    virtual zx_status_t SetOne(size_t bitoff) {
        return Set(bitoff, bitoff + 1);
    }

    // Sets all bits in the range [*bitoff*, *bitmax*).  Only fails on
    // allocation error or if bitmax < bitoff.
    virtual zx_status_t Set(size_t bitoff, size_t bitmax) = 0;

    // Clears the bit at bitoff.  Only fails on allocation error.
    virtual zx_status_t ClearOne(size_t bitoff) {
        return Clear(bitoff, bitoff + 1);
    }

    // Clears all bits in the range [*bitoff*, *bitmax*).  Only fails on
    // allocation error or if bitmax < bitoff.
    virtual zx_status_t Clear(size_t bitoff, size_t bitmax) = 0;

    // Clear all bits in the bitmap.
    virtual void ClearAll() = 0;
};

} // namespace bitmap