using Flowframes.IO;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace Flowframes.Data
{
    public struct AI
    {
        public enum Backend { Pytorch, Ncnn, Tensorflow, Other }
        public Backend backend;
        public string aiName;
        public string aiNameShort;
        public string friendlyName;
        public string description;
        public string pkgDir;
        public int[] supportedFactors;
        public bool multiPass;  // Are multiple passes needed to get to the desired interp factor?

        public AI(Backend backend, string aiName, string friendlyName, string desc, string pkgDir, int[] factors, bool multiPass = false)
        {
            this.backend = backend;
            this.aiName = aiName;
            this.aiNameShort = aiName.Split(' ')[0].Split('_')[0];
            this.friendlyName = friendlyName;
            this.description = desc;
            this.pkgDir = pkgDir;
            this.supportedFactors = factors;
            this.multiPass = multiPass;
        }
    }
}