An Atoms Cache is composed by a cache header file, a folder containing all the agent types used by the agents and for each frame an header, frame, meta, pose fileAtoms has its own internal format for agent caches, please check the reference to know more about what files is made of.
Cache header
The cache header is a MapMetadata cointaining the following key-value pairs:
- agentIds: It's an IntArrayMetadata containing all the agents ids
- cacheType: It's a StringMetadata. It's ,for internal use only and it must store the value "dynamic"
- endFrame: It's an IntMetadata containing the end frame
- startFrame: It's an IntMetadata containing the start frame
This file must have the following name extension: atoms (ex: tasti.e. test_cache.atoms)
Agent types
To make the caches more portable, all the agent type files should be are saved in a "agentTypes" subfolder next to the cache files.
The agent type can be are serialized in a binary format:
Code Block | ||
---|---|---|
| ||
Atoms::AgentTypePtr agentType = agent->agentType(); AtomsCore::Archive agentTypeArchive(agentType->memSize()); agentType->serialize(agentTypeArchive); agentTypeArchive->writeToFile("myAgentType.agentType") |
To make an agent type visible inside the Atoms UI, an event simulation python wrapper script must be is exported in the same folder.
Code Block | ||
---|---|---|
| ||
std::string cachePath = "/atomsDemo/cache/"; std::string agentType = agent->agentType().name(); Atoms::AgentTypePtr agentTypePtr = agent->agentType(); AtomsCore::Archive agentTypeArchive(agentTypePtr->memSize()); agentTypePtr->serialize(agentTypeArchive); agentTypeArchive->writeToFile(cachePath + "/agentTypes/" + agentType + ".agentType") //store the python event wrapper const std::string pythonEvent = "import AtomsCore\n" "import Atoms\n" "import imath\n" "\n" "class "+ agentType +"Cached(Atoms.SimulationEvent):\n" "\tagentTypeFile = r'" + agentTypePath + "'\n" "\teventName = '" + agentType + "Cached'\n" "\tagentTypeName = '+ agentType + '\n" "\tgeoPath = ''\n" "\tskinPath = ''\n" "\tstateMachine = ''\n" "\n" "\tdef __init__(self):\n" "\t\tAtoms.SimulationEvent.__init__(self)\n" "\t\tself.setName(self.eventName)\n" "\n" "\tdef load(self):\n" "\t\taType = Atoms.AgentType()\n" "\t\taTypeArchive = AtomsCore.Archive()\n" "\t\tif aTypeArchive.readFromFile(self.agentTypeFile):\n" "\t\t\taType.deserialise(aTypeArchive)\n" "\t\telse:\n" "\t\t\treturn\n" "\n" "\t\tif self.geoPath:\n" "\t\t\tmeshMap = AtomsCore.MapMetadata()\n" "\t\t\ttypeArchive = AtomsCore.Archive()\n" "\t\t\tif typeArchive.readFromFile(self.geoPath):\n" "\t\t\t\tmeshMap.deserialise(typeArchive)\n" "\t\t\t\taType.metadata()["lowGeo"] = meshMap\n" "\n" "\t\tif self.skinPath:\n" "\t\t\tskinMap = AtomsCore.MapMetadata()\n" "\t\t\tskinArchive = AtomsCore.Archive()\n" "\t\t\tif skinArchive.readFromFile(self.skinPath):\n" "\t\t\t\tskinMap.deserialise(skinArchive)\n" "\t\t\t\taType.metadata()["skinGeo"] = skinMap\n" "\n" "\t\taType.metadata()["stateMachine"] = AtomsCore.StringMetadata(self.stateMachine)\n" "\n" "\t\tAtoms.AgentTypes.instance().addAgentType(self.agentTypeName, aType)\n"; std::string agentTypeWrapperPath = cachePath + "/agenTypes/" + agentType + ".py"; std::ofstream out(agentTypeWrapperPath); out << pythonEvent; out.close(); |
Code Block | ||
---|---|---|
| ||
CACHED_AGENT_TYPE_SCRIPT = ''' import AtomsCore import Atoms class AgentTypeEvent$(Atoms.SimulationEvent): \tagentTypeFile = r'agentTypeFile$' \teventName = 'agentType$' \tagentTypeName = 'agentTypeName$' \tgeoPath = '' \tskinPath = '' \tstateMachine = '' \tdef __init__(self): \t\tAtoms.SimulationEvent.__init__(self) \t\tself.setName(self.eventName) \tdef load(self): \t\taType = Atoms.AgentType() \t\taTypeArchive = AtomsCore.Archive() \t\tif aTypeArchive.readFromFile(self.agentTypeFile): \t\t\taType.deserialise(aTypeArchive) \t\telse: \t\t\treturn \t\tif self.geoPath: \t\t\tmeshMap = AtomsCore.MapMetadata() \t\t\ttypeArchive = AtomsCore.Archive() \t\t\tif typeArchive.readFromFile(self.geoPath): \t\t\t\tmeshMap.deserialise(typeArchive) \t\t\t\taType.metadata()["lowGeo"] = meshMap \t\tif self.skinPath: \t\t\tskinMap = AtomsCore.MapMetadata() \t\t\tskinArchive = AtomsCore.Archive() \t\t\tif skinArchive.readFromFile(self.skinPath): \t\t\t\tskinMap.deserialise(skinArchive) \t\t\t\taType.metadata()["skinGeo"] = skinMap \t\taType.metadata()["stateMachine"] = AtomsCore.StringMetadata(self.stateMachine) \t\tAtoms.AgentTypes.instance().addAgentType(self.agentTypeName, aType) ''' agentType = "atomsRobot" # This serialize the agent type in python aType = Atoms.AgentTypes.instance().agentType(agentType) if aType: atArchive = AtomsCore.Archive(aType.memSize()) aType.serialise(atArchive) atArchive.writeToFile(os.path.join(agentTypesPath, "%s.agentType" % (agentType))) agentTypesPath = "/atomsDemo/cache/test_cache/agentTypes/" agentTypeScript = CACHED_AGENT_TYPE_SCRIPT af = os.path.normpath(os.path.join(agentTypesPath, "%s.agentType" % (agentType))) agentTypeScript = agentTypeScript.replace("agentTypeFile$", af) agentTypeScript = agentTypeScript.replace("agentType$","%sCached" % agentType) agentTypeScript = agentTypeScript.replace("agentTypeName$", agentType) agentTypeScript = agentTypeScript.replace("AgentTypeEvent$", "%sCached" % (agentType)) with open(os.path.join(agentTypesPath, "%s.py" % (agentType)), "w") as pyFile: pyFile.write(agentTypeScript) pyFile.close() |
Frame files
For each frame must have The cache has these four files per frame:
- header: basic info such as number of agents at the current frame, number of created agents and number of deleted agents
- frame: agent data such as agent type, position, velocity, bbox and variation
- meta: agents metadata
- pose: agent skeleton data
Header file
The cache frame header is a MapMetadata cointaining the following key-value pairs:
- agents: It's an IntArrayMetadata containing all the agents ids
- agentsCreated: It's an IntArrayMetadata containing the id ids of the created agents created at this frame.
- agentsDeleted: It's an IntArrayMetadata containing the id ids of the deleted agents deleted at this frame.
- box: It's an BoxMetadata containing the bounding box at the current frame
This file must have the has the following extension: .$FRAME.header.atoms (ex: tasti.e. test_cache.0010.header.atoms)
Frame file
The cache frame is a MapMetadata. Each entry in this map use uses the agent id as key while the value is another MapMetadata containing the following key-value pairs:
- agentType: StringMetadata containing the agent type name
- box: BoxMetadata containing the agent bounding box
- position: Vector3Metadata containing the agent position
- variation: StringMetadata containing the agent variation
- velocity: Vector3Metadata containing the agent velocity
This is an example of frame data:
- "0":
- "agentType": StringMetadata("atomsRobot")
- "box": BoxMetadata(AtomsCore::Box(AtomsCore::Vector3(-10,-10,-10),AtomsCore::Vector3( 10, 10, 10)))
- "position": Vector3Metadata(AtomsCore::Vector3(0,0,0))
- "velocity": Vector3Metadata(AtomsCore::Vector3(1.4354,0.234,0.54433)) - "1":
- "agentType": StringMetadata("atomsRobot")
- "box": BoxMetadata(AtomsCore::Box(AtomsCore::Vector3(-50,-10,-50),AtomsCore::Vector3(-40, 10, -40)))
- "position": Vector3Metadata(AtomsCore::Vector3(-45,0,0))
- "velocity": Vector3Metadata(AtomsCore::Vector3(1.5464,0.124,0.76433)) - "5":
- "agentType": StringMetadata("atomsRobot")
- "box": BoxMetadata(...)
- "position": Vector3Metadata(...)
- "velocity": Vector3Metadata(...) - ...
This file must have the has the following extension: .$FRAME.frame.atoms (ex: tasti.e. test_cache.0010.frame.atoms)
Metafile
The cache meta file is a MapMetadata. Each entry in this map use uses the agent id as key while the value is another MapMetadata containing all the metadatas of the agent
This is an example:
- "0":
- "position": Vector3Metadata(...)
- "direction": Vector3Metadata(...)
- "state": IntMetadata(...)
- ... - "1":
- "position": Vector3Metadata(...)
- "direction": Vector3Metadata(...)
- "state": IntMetadata(...)
- ... "1":
- "position": Vector3Metadata(...)
- "direction": Vector3Metadata(...)
- "state": IntMetadata(...)
- ...- ...
This file must have the following extension: .$FRAME.meta.atoms (ex: tasti.e. test_cache.0010.meta.atoms)
Pose file
The cache pose file is a MapMetadata. Each entry in this map use uses the agent id as key while the value is a PoseMetadata that store the agent pose.
This file must have has the following extension: .$FRAME.pose.atoms (ex: tasti.e. test_cache.0010.pose.atoms)
Writing an Atoms cache
An atoms cache can be exported using the AtomsCache::exportCacheFrame() function. This function export exports only the frame files, the header and the agents type must be exported separately.
Code Block | ||
---|---|---|
| ||
#include <Atoms/AtomsCache std::string cachePath = "/atomsDemo/cache/"; std::string cacheName = "cache_test"; int startFrame = 1; int endFrame = 2; // list of agents groups to export std::vector<AtomsPtr<Atoms::AgentGroup>> agentGroups; // if you are in maya or houdini you can get the agent groups from the agents simulation auto sims = Atoms::AgentsSimulations::instance() auto sim = sims.begin().second; for(unsigned int i=0; i < sim.numAgentGroups(); i++) { agentGroups.push_back(sim.agentGroup(i)); } // This maps the agent global id with the cache id std::unordered_map<size_t, size_t> cacheIdMap; std::set<std::string> agentTypes; // Add compression and use multithread size_t tags = AtomsCore::Archive::kRandomAccessCompress | AtomsCore::Archive::kMultithread; for (unsigned int frame = startFrame; frame < endFrame; frame++) { AtomsCache::exportCacheFrame(cachePath, cacheName, frame, agentGroups, cacheIdMap, agentTypes, tags); } // save the agent types Atoms::AgentTypes& agentTypes = Atoms::AgentTypes::instance(); for(const std::string agentType: agentTypes) { auto agentTypePtr = agentTypes->agentType(agentType); if (!agentTypePtr) continue; AtomsCore::Archive agentTypeArchive(agentTypePtr->memSize()); agentType->serialize(agentTypeArchive); std::string agentTypePath = cachePath + "/agenTypes/" + agentType + ".agentType"; agentTypeArchive->writeToFile(agentTypePath); //store the python event wrapper const std::string pythonEvent = "import AtomsCore\n" "import Atoms\n" "import imath\n" "\n" "class "+ agentType +"Cached(Atoms.SimulationEvent):\n" "\tagentTypeFile = r'" + agentTypePath + "'\n" "\teventName = '" + agentType + "Cached'\n" "\tagentTypeName = '+ agentType + '\n" "\tgeoPath = ''\n" "\tskinPath = ''\n" "\tstateMachine = ''\n" "\n" "\tdef __init__(self):\n" "\t\tAtoms.SimulationEvent.__init__(self)\n" "\t\tself.setName(self.eventName)\n" "\n" "\tdef load(self):\n" "\t\taType = Atoms.AgentType()\n" "\t\taTypeArchive = AtomsCore.Archive()\n" "\t\tif aTypeArchive.readFromFile(self.agentTypeFile):\n" "\t\t\taType.deserialise(aTypeArchive)\n" "\t\telse:\n" "\t\t\treturn\n" "\n" "\t\tif self.geoPath:\n" "\t\t\tmeshMap = AtomsCore.MapMetadata()\n" "\t\t\ttypeArchive = AtomsCore.Archive()\n" "\t\t\tif typeArchive.readFromFile(self.geoPath):\n" "\t\t\t\tmeshMap.deserialise(typeArchive)\n" "\t\t\t\taType.metadata()["lowGeo"] = meshMap\n" "\n" "\t\tif self.skinPath:\n" "\t\t\tskinMap = AtomsCore.MapMetadata()\n" "\t\t\tskinArchive = AtomsCore.Archive()\n" "\t\t\tif skinArchive.readFromFile(self.skinPath):\n" "\t\t\t\tskinMap.deserialise(skinArchive)\n" "\t\t\t\taType.metadata()["skinGeo"] = skinMap\n" "\n" "\t\taType.metadata()["stateMachine"] = AtomsCore.StringMetadata(self.stateMachine)\n" "\n" "\t\tAtoms.AgentTypes.instance().addAgentType(self.agentTypeName, aType)\n"; std::string agentTypeWrapperPath = cachePath + "/agenTypes/" + agentType + ".py"; std::ofstream out(agentTypeWrapperPath); out << pythonEvent; out.close(); } // write the cache header AtomsCore::IntArrayMetadata agentIdsMeta; std::vector<int>& agentIdsVec = agentIdsMeta.get(); agentIdsVec.reserve(cacheIdMap.size()); for (auto it=cacheIdMap.begin(); it != cacheIdMap.end(); it++) { agentIdsVec.pushBack(it->second); } std::sort(agentIdsVec.begin(), agentIdsVec.end()); AtomsCore::MapMetadata cacheHeader; cacheHeader.addEntry("startFrame", &AtomsCore::IntMetadata(startFrame)); cacheHeader.addEntry("endFrame", &AtomsCore::IntMetadata(endFrame)); cacheHeader.addEntry("cacheType", &AtomsCore::StringMetadata("dynamic")); cacheHeader.addEntry("agentIds", &agentIdsMeta); |
Reading an Atoms cache
Since the Atoms cache is composed by made of different serialized MapMetadata serialized you can read them using the atoms serialization system.
If you are going to query multiple times some different cache frames during a simulation, you can use the cache manager to increase the performance.