#pragma once
#include "yolo_runtime.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

static int parse_mask_dilation(int argc, char** argv, int* radius) {
    int seen=0;*radius=0;
    for(int i=1;i<argc;++i) if(strcmp(argv[i],"--mask-dilation")==0) {
        if(seen++ || i+1>=argc) return -1;
        const char* text=argv[++i];size_t length=strlen(text);
        if(!length || length>3 || strspn(text,"0123456789")!=length) return -1;
        *radius=atoi(text);if(*radius>100) return -1;
    }
    return 0;
}

/* Each output belongs to the caller; free(*owned) after RGB/JSON processing.
   Original result-set mask and fit geometry are never changed. */
static int dilate_example_mask(yolo_mask_view_v1_t* mask, int radius,
    const yolo_roi_v1_t* region, unsigned char** owned) {
    yolo_mask_view_v1_t output;
    yolo_error_info_v1_t error;
    memset(&output,0,sizeof(output));output.struct_size=sizeof(output);output.version=YOLO_MASK_VIEW_VERSION;
    memset(&error,0,sizeof(error));error.struct_size=sizeof(error);error.version=YOLO_ERROR_INFO_VERSION;
    *owned=NULL;
    if(yolo_mask_dilate(mask,radius,region,NULL,0,0,&output,&error)!=YOLO_STATUS_OK) {
        fprintf(stderr,"Dilation query failed: %s\n",error.message);return -1;
    }
    const uint64_t bytes=(uint64_t)output.width*(uint64_t)output.height;
    if(bytes>SIZE_MAX) return -1;
    *owned=(unsigned char*)malloc((size_t)bytes);
    if(!*owned) return -1;
    if(yolo_mask_dilate(mask,radius,region,*owned,output.width,bytes,&output,&error)!=YOLO_STATUS_OK) {
        fprintf(stderr,"Dilation failed: %s\n",error.message);free(*owned);*owned=NULL;return -1;
    }
    *mask=output;
    return 0;
}