CatIDSearch.cpp

Go to the documentation of this file.

namespace krotos
{
namespace UCS
{
std::array<std::string, 753> CatIDs = {
        "AIRBlow",    "AIRBrst",    "AIRHiss",     "AIRMisc",   "AIRSuck",   "AERODoor",  "AEROHeli",  "AEROInt",
        "AEROJet",    "AEROMech",   "AEROMil",     "AEROMisc",  "AEROProp",  "AERORadio", "AERORckt",  "ALRMBell",
        "ALRMBuzr",   "ALRMClok",   "ALRMElec",    "ALRMMisc",  "ALRMSirn",  "AMBAir",    "AMBAlpn",   "AMBAmus",
        "AMBBird",    "AMBCele",    "AMBCnst",     "AMBDsrt",   "AMBDsgn",   "AMBEmrg",   "AMBFant",   "AMBFarm",
        "AMBForst",   "AMBGras",    "AMBHist",     "AMBTech",   "AMBHosp",   "AMBInd",    "AMBInsc",   "AMBLake",
        "AMBMrkt",    "AMBMisc",    "AMBNaut",     "AMBOffc",   "AMBPark",   "AMBPrisn",  "AMBPrtst",  "AMBPubl",
        "AMBRlgn",    "AMBHome",    "AMBRest",     "AMBRoom",   "AMBRurl",   "AMBSchl",   "AMBSci",    "AMBSea",
        "AMBSprt",    "AMBSubn",    "AMBSwmp",     "AMBTown",   "AMBTraf",   "AMBTran",   "AMBTrop",   "AMBTndra",
        "AMBUndr",    "AMBUndwtr",  "AMBUrbn",     "AMBWar",    "ANMLAmph",  "ANMLAqua",  "ANMLBat",   "ANMLCat",
        "ANMLWcat",   "ANMLDog",    "ANMLWdog",    "ANMLFarm",  "ANMLHors",  "ANMLInsc",  "ANMLMisc",  "ANMLPrim",
        "ANMLRept",   "ANMLRdnt",   "ANMLWild",    "ADR",       "ASSET",     "BNCE",      "IR",        "LPGRP",
        "MIX",        "PFX",        "PROD",        "RAW",       "REF",       "SCNE",      "TEST",      "TMARK",
        "WIP",        "WTF",        "BEEPAppl",    "BEEP",      "BEEPLofi",  "BEEPMed",   "BEEPTimer", "BEEPVeh",
        "BELLAnml",   "BELLDoor",   "BELLGong",    "BELLHand",  "BELLLrg",   "BELLMisc",  "BIRDPrey",  "BIRDCrow",
        "BIRDFowl",   "BIRDMisc",   "BIRDSea",     "BIRDSong",  "BIRDTrop",  "BIRDWade",  "BOATAir",   "BOATWash",
        "BOATDoor",   "BOATElec",   "BOATFish",    "BOATHorn",  "BOATInt",   "BOATMech",  "BOATMil",   "BOATMisc",
        "BOATMotr",   "BOATRace",   "BOATRow",     "BOATSail",  "BOATShip",  "BOATStm",   "BOATSub",   "BOATUndwtr",
        "BLLTBy",     "BLLTImpt",   "BLLTMisc",    "BLLTRico",  "BLLTShel",  "TOONAnml",  "TOONBoing", "TOONClang",
        "TOONCreak",  "TOONHorn",   "TOONImpt",    "TOONMach",  "TOONMisc",  "TOONMx",    "TOONPluk",  "TOONPop",
        "TOONShake",  "TOONSplt",   "TOONSqk",     "TOONStrch", "TOONSwsh",  "TOONTwang", "TOONVeh",   "TOONVox",
        "TOONWarb",   "TOONWhis",   "TOONZip",     "CERMBrk",   "CERMCrsh",  "CERMFric",  "CERMHndl",  "CERMImpt",
        "CERMMisc",   "CERMMvmt",   "CERMTonl",    "CHAINBrk",  "CHAINHndl", "CHAINImpt", "CHAINMisc", "CHAINMvmt",
        "CHEMAcid",   "CHEMMisc",   "CHEMReac",    "CLOCKChim", "CLOCKMech", "CLOCKMisc", "CLOCKTick", "CLOTHFlp",
        "CLOTHHndl",  "CLOTHImpt",  "CLOTHMisc",   "CLOTHMvmt", "CLOTHRip",  "COMAv",     "COMCam",    "COMCell",
        "COMMic",     "COMMisc",    "COMPhono",    "COMRadio",  "COMStatic", "COMTelm",   "COMTelph",  "COMTv",
        "COMTran",    "COMType",    "CMPTDriv",    "CMPTKey",   "CMPTMisc",  "CREAAqua",  "CREAAvian", "CREABeast",
        "CREABlob",   "CREADino",   "CREADrgn",    "CREAElem",  "CREAEthr",  "CREAHmn",   "CREAInsc",  "CREAMisc",
        "CREAMnstr",  "CREARept",   "CREASmall",   "CREASrce",  "CRWDAngr",  "CRWDApls",  "CRWDBatl",  "CRWDCele",
        "CRWDCheer",  "CRWDChld",   "CRWDConv",    "CRWDLaff",  "CRWDLoop",  "CRWDMisc",  "CRWDPanic", "CRWDQuiet",
        "CRWDReac",   "CRWDSing",   "CRWDSprt",    "CRWDWalla", "DSGNBass",  "DSGNBoom",  "DSGNBram",  "DSGNDist",
        "DSGNDron",   "DSGNErie",   "DSGNEthr",    "DSGNGran",  "DSGNImpt",  "DSGNMisc",  "DSGNMorph", "DSGNRise",
        "DSGNRmbl",   "DSGNRythm",  "DSGNSrce",    "DSGNStngr", "DSGNSynth", "DSGNTonl",  "DSGNVocl",  "DSGNWhsh",
        "DESTRClpse", "DESTRCrsh",  "DESTRMisc",   "DIRTCrsh",  "DIRTDust",  "DIRTHndl",  "DIRTImpt",  "DIRTMisc",
        "DIRTMvmt",   "DIRTTonl",   "DOORAntq",    "DOORAppl",  "DOORCab",   "DOORComp",  "DOORCreak", "DOORDungn",
        "DOORElec",   "DOORGate",   "DOORGlas",    "DOORHdwr",  "DOORTech",  "DOORHydr",  "DOORKnck",  "DOORMetl",
        "DOORMisc",   "DOORPlas",   "DOORPrisn",   "DOORRevl",  "DOORSlid",  "DOORSton",  "DOORSwng",  "DOORWood",
        "DRWRMetl",   "DRWRMisc",   "DRWRPlas",    "DRWRWood",  "ELECArc",   "ELECBuzz",  "ELECEmf",   "ELECMisc",
        "ELECSprk",   "ELECZap",    "EQUIPBridle", "EQUIPTech", "EQUIPMisc", "EQUIPRec",  "EQUIPSprt", "EQUIPTact",
        "EXPLDsgn",   "EXPLMisc",   "EXPLReal",    "FARTDsgn",  "FARTMisc",  "FARTReal",  "FGHTBf",    "FGHTClth",
        "FGHTGrab",   "FGHTImpt",   "FGHTMisc",    "FIREBurn",  "FIREBrst",  "FIRECrkl",  "FIREGas",   "FIREIgn",
        "FIREMisc",   "FIRESizz",   "FIRETrch",    "FIRETurb",  "FIREWhsh",  "FRWKComr",  "FRWKMisc",  "FRWKRec",
        "FOLYClth",   "FOLYFeet",   "FOLYHand",    "FOLYMisc",  "FOLYProp",  "FOODCook",  "FOODDrnk",  "FOODEat",
        "FOODGware",  "FOODIngr",   "FOODKware",   "FOODMisc",  "FOODPour",  "FOODTware", "FEETAnml",  "FEETCrea",
        "FEETHors",   "FEETHmn",    "FEETMisc",    "GAMEArcd",  "GAMEBoard", "GAMECas",   "GAMEMisc",  "GAMEVideo",
        "GEOFuma",    "GEOGeyser",  "GEOLava",     "GEOMisc",   "GEOMudpot", "GLASBrk",   "GLASCrsh",  "GLASFric",
        "GLASHndl",   "GLASImpt",   "GLASMisc",    "GLASMvmt",  "GLASTonl",  "GOREBlood", "GOREBone",  "GOREBurn",
        "GOREFlsh",   "GOREMisc",   "GOREOoze",    "GORESrce",  "GORESplt",  "GORESqsh",  "GOREStab",  "GUNAntq",
        "GUNArtl",    "GUNAuto",    "GUNCano",     "GUNHndl",   "GUNTech",   "GUNMech",   "GUNMisc",   "GUNPis",
        "GUNRif",     "GUNShotg",   "GUNSupr",     "HORNAir",   "HORNCele",  "HORNMisc",  "HORNTrad",  "HMNBlow",
        "HMNBrth",    "HMNBurp",    "HMNCough",    "HMNHart",   "HMNKiss",   "HMNMisc",   "HMNPee",    "HMNSkin",
        "HMNSneez",   "HMNSniff",   "HMNSnor",     "HMNSpit",   "HMNVomit",  "ICEBrk",    "ICECrsh",   "ICEFric",
        "ICEHndl",    "ICEImpt",    "ICEMisc",     "ICEMvmt",   "ICETonl",   "LASRBeam",  "LASRGun",   "LASRImpt",
        "LASRMisc",   "LETHRCreak", "LETHRHndl",   "LETHRImpt", "LETHRMisc", "LETHRMvmt", "LIQBubl",   "LIQImpt",
        "LIQMisc",    "LIQMvmt",    "LIQSuct",     "MACHAmus",  "MACHAntq",  "MACHAppl",  "MACHCnst",  "MACHElev",
        "MACHEscl",   "MACHFan",    "MACHGrdn",    "MACHGym",   "MACHTech",  "MACHHvac",  "MACHInd",   "MACHMech",
        "MACHMed",    "MACHMisc",   "MACHOffc",    "MACHPump",  "MAGAngl",   "MAGElem",   "MAGEvil",   "MAGMisc",
        "MAGPoof",    "MAGShim",    "MAGSpel",     "MECHClik",  "MECHGear",  "MECHHydr",  "MECHLtch",  "MECHLvr",
        "MECHLock",   "MECHMisc",   "MECHPuly",    "MECHRtch",  "MECHRelay", "MECHRolr",  "MECHSwtch", "METLBrk",
        "METLCrsh",   "METLFric",   "METLHndl",    "METLImpt",  "METLMisc",  "METLMvmt",  "METLTonl",  "MOTRAntq",
        "MOTRComb",   "MOTRElec",   "MOTRMisc",    "MOTRSrvo",  "MOTRTurb",  "MOVEActv",  "MOVEAnml",  "MOVECrea",
        "MOVECrwd",   "MOVEHmn",    "MOVEInsc",    "MOVEMisc",  "MOVEPres",  "MUSCBell",  "MUSCBrass", "MUSCChim",
        "MUSCChor",   "MUSCExpr",   "MUSCInst",    "MUSCKeyd",  "MUSCLoop",  "MUSCMisc",  "MUSCPerc",  "MUSCTnprc",
        "MUSCPrfm",   "MUSCPluck",  "MUSCSmpl",    "MUSCShake", "MUSCSong",  "MUSCStngr", "MUSCStr",   "MUSCSynth",
        "MUSCToy",    "MUSCWind",   "NATDAval",    "NATDQuak",  "NATDMisc",  "NATDTorn",  "NATDTsun",  "NATDThyp",
        "NATDVolc",   "OBJBag",     "OBJBook",     "OBJCoin",   "OBJCont",   "OBJFash",   "OBJFurn",   "OBJGrdn",
        "OBJGym",     "OBJHsehld",  "OBJJewl",     "OBJKey",    "OBJLug",    "OBJMed",    "OBJMisc",   "OBJOffc",
        "OBJPack",    "OBJTape",    "OBJUmbr",     "OBJWhled",  "OBJWrite",  "OBJZipr",   "PAPRFltr",  "PAPRFric",
        "PAPRHndl",   "PAPRImpt",   "PAPRMisc",    "PAPRRip",   "PAPRTonl",  "PLASBrk",   "PLASCrsh",  "PLASFric",
        "PLASHndl",   "PLASImpt",   "PLASMisc",    "PLASMvmt",  "PLASTonl",  "RAINClth",  "RAINConc",  "RAIN",
        "RAINGlas",   "RAINInt",    "RAINMetl",    "RAINPlas",  "RAINVege",  "RAINWatr",  "RAINWood",  "ROBTMisc",
        "ROBTMvmt",   "ROBTVox",    "ROCKBrk",     "ROCKCrsh",  "ROCKFric",  "ROCKHndl",  "ROCKImpt",  "ROCKMisc",
        "ROCKMvmt",   "ROCKTonl",   "ROPECreak",   "ROPEHndl",  "ROPEImpt",  "ROPEMisc",  "ROPEMvmt",  "RUBRCrsh",
        "RUBRFric",   "RUBRHndl",   "RUBRImpt",    "RUBRMisc",  "RUBRMvmt",  "RUBRTonl",  "SCIAlrm",   "SCICmpt",
        "SCIDoor",    "SCIEnrg",    "SCIImpt",     "SCIMach",   "SCIMech",   "SCIMisc",   "SCIRetro",  "SCIShip",
        "SCIVeh",     "SCIWeap",    "SNOWCrsh",    "SNOWFric",  "SNOWHndl",  "SNOWImpt",  "SNOWMisc",  "SNOWMvmt",
        "SPRTCourt",  "SPRTField",  "SPRTGym",     "SPRTIndor", "SPRTMisc",  "SPRTSkate", "SPRTTrck",  "SPRTWatr",
        "SPRTWntr",   "SWSH",       "WHSH",        "TOOLGrdn",  "TOOLHand",  "TOOLMisc",  "TOOLPneu",  "TOOLPowr",
        "TOYElec",    "TOYMech",    "TOYMisc",     "TRNBrake",  "TRNClak",   "TRNDoor",   "TRNDiesl",  "TRNElec",
        "TRNHspd",    "TRNHorn",    "TRNInt",      "TRNMech",   "TRNMisc",   "TRNSteam",  "TRNSbwy",   "TRNTram",
        "UIAlert",    "UIBeep",     "UIClick",     "UIData",    "UIGlitch",  "UIMisc",    "UIMvmt",    "VEGEGras",
        "VEGELeaf",   "VEGEMisc",   "VEGETree",    "VEHAlrm",   "VEHAntq",   "VEHAtv",    "VEHBike",   "VEHBrake",
        "VEHBus",     "VEHCar",     "VEHCnst",     "VEHDoor",   "VEHElec",   "VEHEmrg",   "VEHFarm",   "VEHFrght",
        "VEHBy",      "VEHHorn",    "VEHInt",      "VEHJalop",  "VEHMech",   "VEHMil",    "VEHMisc",   "VEHMoto",
        "VEHRace",    "VEHSirn",    "VEHSkid",     "VEHSusp",   "VEHTire",   "VEHTruck",  "VEHUtil",   "VEHWagn",
        "VEHWndw",    "VOXAlien",   "VOXBaby",     "VOXCheer",  "VOXChld",   "VOXCry",    "VOXEfrt",   "VOXFem",
        "VOXFutz",    "VOXHist",    "VOXLaff",     "VOXMale",   "VOXMisc",   "VOXReac",   "VOXScrm",   "VOXSing",
        "VOXWhsp",    "WATRBubl",   "WATRDran",    "WATRDrip",  "WATRFizz",  "WATRFlow",  "WATRFoun",  "WATRImpt",
        "WATRLap",    "WATRMisc",   "WATRMvmt",    "WATRPlmb",  "WATRPour",  "WATRSplsh", "WATRSpray", "WATRStm",
        "WATRSurf",   "WATRTurb",   "WATRUndwtr",  "WATRFall",  "WATRWave",  "WEAPArmr",  "WEAPArro",  "WEAPAxe",
        "WEAPBlnt",   "WEAPBow",    "WEAPKnif",    "WEAPMisc",  "WEAPPole",  "WEAPSiege", "WEAPSwrd",  "WEAPWhip",
        "HAIL",       "WTHR",       "STORM",       "THUN",      "WHSTHmn",   "WHSTMech",  "WHSTMisc",  "WINDDsgn",
        "WIND",       "WINDGust",   "WINDInt",     "WINDTonl",  "WINDTurb",  "WINDVege",  "WNDWCover", "WNDWHdwr",
        "WNDWKnck",   "WNDWMetl",   "WNDWMisc",    "WNDWPlas",  "WNDWWood",  "WINGBird",  "WINGCrea",  "WINGInsc",
        "WINGMisc",   "WOODBrk",    "WOODCrsh",    "WOODFric",  "WOODHandl", "WOODImpt",  "WOODMisc",  "WOODMvmt",
        "WOODTonl"};
}
 
CatIDSearch::CatIDSearch(std::string root)
{
    generator.seed(std::random_device{}());
 
    std::sort(UCS::CatIDs.begin(), UCS::CatIDs.end());
 
    // To-Do: searching for all factory assets - should potentially be done in a separate thread
 
    std::size_t count = 0;
    for (const auto& path : File(root).findChildFiles(2, true, "*.wav"))
    {
        auto basename = path.getFileName().toStdString();
        // try to avoid duplicate files coming from different folders
        if (filenames.count(basename) == 0)
        {
            auto success = addDatasetItem(path.getFullPathName().toStdString());
            if (success)
                ++count;
        }
        filenames.insert(basename);
    }
    filenames.clear();
    DBG("UCS file count: " << count);
}
 
CatIDSearch::~CatIDSearch()
{
    //
}
 
bool CatIDSearch::addDatasetItem(std::string path)
{
    auto key = extractCatID(path);
    auto success = std::binary_search(UCS::CatIDs.begin(), UCS::CatIDs.end(), key);
    if (success)
    {
        dataset[key].push_back(path);
    }
    return success;
}
 
std::string CatIDSearch::extractCatID(std::string path)
{
    auto filename = File(path).getFileName().toStdString();
    std::string delimiter = "_";
    std::string CatID = filename.substr(0, filename.find(delimiter)); // UCS category or partial/full filename
    return CatID;
}
 
std::string CatIDSearch::findSimilar(std::string path)
{
    auto key = extractCatID(path);
    auto success = std::binary_search(UCS::CatIDs.begin(), UCS::CatIDs.end(), key);
    if (success)
    {
        // key was a valid CatID
        auto filepaths = dataset[key];
        DBG("UCS CatID: " << key);
        DBG("number of results: " << filepaths.size());
        if (!filepaths.empty())
        {
            std::uniform_int_distribution<std::size_t> distribution(0, filepaths.size() - 1);
            std::size_t idx = distribution(generator);
            auto result = filepaths[idx];
            // Note: might want to check file attributes like duration before returning
            // e.g. if the query file is several minutes long we might not want to return a file that's 1-second long
            return result;
        }
    }
    return path;
}
 
std::vector<std::string> CatIDSearch::findAllSimilar(std::string path)
{
    std::vector<std::string> results = {path};
 
    auto key = extractCatID(path);
    auto success = std::binary_search(UCS::CatIDs.begin(), UCS::CatIDs.end(), key);
    if (success)
    {
        // key was a valid CatID
        auto filepaths = dataset[key];
        std::sort(filepaths.begin(), filepaths.end());
        return filepaths;
    }
    return results;
}
 
} // namespace krotos