#ifndef INC_LIB_SEG_MASK_DATA_H
#define INC_LIB_SEG_MASK_DATA_H

#include <lib/algo/avl_tree.h>
#include <lib/monad.h>

struct MaskData {
  uint16_t label;
  size_t area;
  size_t perimeter;
};

// Allocate Mask Data for Label
struct MaskData* create_mask_data(uint16_t label);

// Compare mask data labels
bool_t compare_labels(struct MaskData* left, struct MaskData* right);

// Create AVL Mask node
struct AVLNode* create_avl_mask_node(struct MaskData* data);

// Insert MaskData into the AVL Tree
struct Result insert_mask(struct AVLNode* node, struct MaskData* data);

// Allocate a label's Mask data in a tree
//  If it already exists, skip the allocation
struct AVLNode* insert_mask_alloc(struct AVLNode* node, uint16_t label);

// Print AVL Node Mask Data Label
void print_label(struct AVLNode* root);

// Increase the label's area
bool_t increase_label_area(struct AVLNode* root, uint16_t label);

// Increase the label's perimeter
bool_t increase_label_perimeter(struct AVLNode* root, uint16_t label);

// Increase the label's area
//  Create an AVL node if it doesn't exist
struct AVLNode* increase_label_area_alloc(struct AVLNode* root, uint16_t label);

// Increase the label's perimeter
//  Create an AVL node if it doesn't exist
struct AVLNode* increase_label_perimeter_alloc(struct AVLNode* root, uint16_t label);

#endif