AudioDataset.h

Go to the documentation of this file.

#pragma once
 
namespace krotos
{
class AudioDataset : public juce::Thread
{
 
  public:
    AudioDataset();
 
    ~AudioDataset();
 
    std::vector<std::pair<std::string, float>> sample(std::string query, std::size_t k = 3,
                                                      bool descending = true) const;
 
  private:
    std::vector<File> findValidFiles() const;
 
    void run() override;
 
    bool assetsChanged();
 
    bool runAnalysis();
 
    File getFactoryAssetDirectory() const;
 
    File getDatasetFile() const;
 
    std::set<std::string> getPathsSet() const;
 
    std::string fixKeyErrors(std::string key) const;
 
    std::string extractCatID(std::string filename) const;
 
    bool writeCache();
 
    bool readCache();
 
    bool cacheExists();
 
    bool deleteCache();
 
    AudioBuffer<float> resampleAudioBuffer(const AudioBuffer<float>& buffer, double sampleRate, double targetRate);
 
    std::string m_version = "1.0";
    std::map<std::string, std::vector<std::pair<std::string, std::vector<float>>>> m_dataset;
 
    /*
    https://universalcategorysystem.com/
    Version 8.2
    */
    std::array<std::string, 753> CatIDSet = {
            "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"};
 
    JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR(AudioDataset)
};
 
} // namespace krotos