// 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_MODULAR_ENCODING_DEC_MA_H_

#define LIB_JXL_MODULAR_ENCODING_DEC_MA_H_

#include <jxl/memory_manager.h>

#include <cstddef>

#include <cstdint>

#include <vector>

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

#include "lib/jxl/dec_bit_reader.h"

#include "lib/jxl/modular/options.h"

namespace jxl {

// inner nodes

struct PropertyDecisionNode {

  PropertyVal splitval;

  int16_t property;  // -1: leaf node, lchild points to leaf node

  uint32_t lchild;

  uint32_t rchild;

  Predictor predictor;

  int64_t predictor_offset;

  uint32_t multiplier;

  PropertyDecisionNode(int p, int split_val, int lchild, int rchild,

                       Predictor predictor, int64_t predictor_offset,

                       uint32_t multiplier)

      : splitval(split_val),

        property(p),

        lchild(lchild),

        rchild(rchild),

        predictor(predictor),

        predictor_offset(predictor_offset),

        multiplier(multiplier) {}

  PropertyDecisionNode()

      : splitval(0),

        property(-1),

        lchild(0),

        rchild(0),

        predictor(Predictor::Zero),

        predictor_offset(0),

        multiplier(1) {}

  static PropertyDecisionNode Leaf(Predictor predictor, int64_t offset = 0,

                                   uint32_t multiplier = 1) {

    return PropertyDecisionNode(-1, 0, 0, 0, predictor, offset, multiplier);

  }

  static PropertyDecisionNode Split(int p, int split_val, int lchild,

                                    int rchild = -1) {

    if (rchild == -1) rchild = lchild + 1;

    return PropertyDecisionNode(p, split_val, lchild, rchild, Predictor::Zero,

                                0, 1);

  }

};

using Tree = std::vector<PropertyDecisionNode>;

Status DecodeTree(JxlMemoryManager *memory_manager, BitReader *br, Tree *tree,

                  size_t tree_size_limit);

}  // namespace jxl

#endif  // LIB_JXL_MODULAR_ENCODING_DEC_MA_H_