// Copyright (c) the JPEG XL Project Authors. All rights reserved.

//

// Use of this source code is governed by a BSD-style

// license that can be found in the LICENSE file.

#ifndef LIB_JXL_BASE_SPAN_H_

#define LIB_JXL_BASE_SPAN_H_

// Span (array view) is a non-owning container that provides cheap "cut"

// operations and could be used as "ArrayLike" data source for PaddedBytes.

#include <cstddef>

#include <cstdint>

#include <type_traits>

#include <vector>

#include "lib/jxl/base/status.h"

namespace jxl {

template <typename T>

class Span {

 public:

  constexpr Span() noexcept : Span(nullptr, 0) {}

  constexpr Span(T* array, size_t length) noexcept

      : ptr_(array), len_(length) {}

jxl::Span<unsigned char const>::Span(unsigned char const*, unsigned long)

Line

Count

Source

27

647k

      : ptr_(array), len_(length) {}

jxl::Span<float>::Span(float*, unsigned long)

Line

Count

Source

27

14.4k

      : ptr_(array), len_(length) {}

  template <size_t N>

  explicit constexpr Span(T (&a)[N]) noexcept : Span(a, N) {}

  template <typename U>

  constexpr Span(U* array, size_t length) noexcept

      : ptr_(reinterpret_cast<T*>(array)), len_(length) {

    static_assert(sizeof(U) == sizeof(T), "Incompatible type of source.");

  }

  template <typename ArrayLike>

  explicit constexpr Span(const ArrayLike& other) noexcept

      : Span(reinterpret_cast<T*>(other.data()), other.size()) {

    static_assert(sizeof(*other.data()) == sizeof(T),

                  "Incompatible type of source.");

  }

jxl::Span<unsigned char const>::Span<std::__1::vector<unsigned char, std::__1::allocator<unsigned char> > >(std::__1::vector<unsigned char, std::__1::allocator<unsigned char> > const&)

Line

Count

Source

40

374k

      : Span(reinterpret_cast<T*>(other.data()), other.size()) {

41

374k

    static_assert(sizeof(*other.data()) == sizeof(T),

42

374k

                  "Incompatible type of source.");

43

374k

  }

jxl::Span<unsigned char const>::Span<jxl::PaddedBytes>(jxl::PaddedBytes const&)

Line

Count

Source

40

9

      : Span(reinterpret_cast<T*>(other.data()), other.size()) {

41

9

    static_assert(sizeof(*other.data()) == sizeof(T),

42

9

                  "Incompatible type of source.");

43

9

  }

  using NCT = typename std::remove_const<T>::type;

  constexpr T* data() const noexcept { return ptr_; }

jxl::Span<unsigned char const>::data() const

Line

Count

Source

47

1.27M

  constexpr T* data() const noexcept { return ptr_; }

jxl::Span<float>::data() const

Line

Count

Source

47

57.7k

  constexpr T* data() const noexcept { return ptr_; }

  constexpr size_t size() const noexcept { return len_; }

jxl::Span<unsigned char const>::size() const

Line

Count

Source

49

580k

  constexpr size_t size() const noexcept { return len_; }

jxl::Span<float>::size() const

Line

Count

Source

49

57.7k

  constexpr size_t size() const noexcept { return len_; }

  constexpr bool empty() const noexcept { return len_ == 0; }

  constexpr T* begin() const noexcept { return data(); }

jxl::Span<unsigned char const>::begin() const

Line

Count

Source

53

374k

  constexpr T* begin() const noexcept { return data(); }

jxl::Span<float>::begin() const

Line

Count

Source

53

57.7k

  constexpr T* begin() const noexcept { return data(); }

  constexpr T* end() const noexcept { return data() + size(); }

  constexpr T& operator[](size_t i) const noexcept {

    // MSVC 2015 accepts this as constexpr, but not ptr_[i]

    return *(data() + i);

  }

  Status remove_prefix(size_t n) noexcept {

    JXL_ENSURE(size() >= n);

    ptr_ += n;

    len_ -= n;

    return true;

  }

jxl::Span<float>::remove_prefix(unsigned long)

Line

Count

Source

62

43.3k

  Status remove_prefix(size_t n) noexcept {

63

43.3k

    JXL_ENSURE(size() >= n);

64

43.3k

    ptr_ += n;

65

43.3k

    len_ -= n;

66

43.3k

    return true;

67

43.3k

  }

Unexecuted instantiation: jxl::Span<unsigned char const>::remove_prefix(unsigned long)

  void AppendTo(std::vector<NCT>& dst) const {

    dst.insert(dst.end(), begin(), end());

  }

  std::vector<NCT> Copy() const { return std::vector<NCT>(begin(), end()); }

 private:

  T* ptr_;

  size_t len_;

};

using Bytes = Span<const uint8_t>;

}  // namespace jxl

#endif  // LIB_JXL_BASE_SPAN_H_