dnaStore: Add functions to store and get floors in a suit block

Author: DarthMDev

panda/src/toontown/dna/dnaStorage.cxx

@@ -386,7 +386,14 @@ int DNAStorage::remove_suit_point(PT(DNASuitPoint) point) {
 void DNAStorage::store_suit_block(const int block_number, const std::string& dept) {
   _suit_blocks_map[block_number] = dept;
 }
-
+////////////////////////////////////////////////////////////////////
+//     Function: store_suit_block_num_floors
+//       Access: Public
+//  Description: Store a block number in the SuitBlockFloorsMap
+////////////////////////////////////////////////////////////////////
+void DNAStorage::store_suit_block_num_floors(const int block_number, const int num_floors) {
+  _suit_block_num_floors_map[block_number] = num_floors;
+}
 ////////////////////////////////////////////////////////////////////
 //     Function: get_suit_blocks
 //       Access: Public
@@ -420,8 +427,22 @@ std::string DNAStorage::get_suit_block_track(int block_number) const {
   }
   return (*i).second;
 }
-
-
+////////////////////////////////////////////////////////////////////
+// Function: get_num_floors
+// Access: Public
+// Description: Get the number of floors in the suit building
+////////////////////////////////////////////////////////////////////
+int DNAStorage::get_num_floors(int block_number) const {
+  // Try to find this code in the map
+  SuitBlockFloorsMap::const_iterator i = _suit_block_num_floors_map.find(block_number);
+  if (i == _suit_block_num_floors_map.end()) {
+    dna_cat.error()
+      << "block number: " << block_number << " not found in SuitBlockFloors map" << std::endl;
+    return 0;
+  }
+  return (*i).second;
+  
+}
 ////////////////////////////////////////////////////////////////////
 //     Function: store_block_number
 //       Access: Public

panda/src/toontown/dna/dnaStorage.h

@@ -48,6 +48,7 @@ typedef pmap< int, int > BlockToZoneMap;
 typedef pmap< int, PosHpr > BlockToPosHprMap;
 typedef pmap< int, LMatrix4f > BlockToTransformMap;
 typedef pmap< int, std::string > SuitBlockMap;
+typedef pmap<int, int > SuitBlockFloorsMap;
 typedef pmap< int, std::string > BlockToTitleMap;
 typedef pmap< int, std::string > BlockToArticleMap;
 typedef pmap< int, std::string > BlockToBuildingTypeMap;
@@ -97,7 +98,8 @@ class EXPCL_TOONTOWN DNAStorage {
   int get_highest_suit_point_index();
   int remove_suit_point(PT(DNASuitPoint));
   void store_suit_block(const int block_number, const std::string& dept);
-  void store_block_number(const std::string& block, const std::string& zone_id);
+  void store_suit_block_num_floors(const int block_number, const int num_floors);
+  void store_block_number(const std::string &block, const std::string &zone_id);
   void store_block_door_pos_hpr(const std::string& block, const LPoint3f& pos, const LPoint3f& hpr);
   void store_block_sign_transform(const std::string& block, const LMatrix4f& mat);
   void store_block_title(const std::string& block, const std::string& title);
@@ -152,6 +154,8 @@ class EXPCL_TOONTOWN DNAStorage {
   bool is_suit_block(int block_number) const;
   std::string get_suit_block_track(int block_number) const;
 
+  int get_num_floors(int block_number) const;
+
   // Block number functions
   int get_zone_from_block_number(int block_number) const;
   int get_block_number_at(unsigned int index) const;
@@ -263,6 +267,8 @@ class EXPCL_TOONTOWN DNAStorage {
   BlockToPosHprMap _block_door_pos_hpr_map;
   BlockToTransformMap _block_sign_transform_map;
   SuitBlockMap _suit_blocks_map;
+  // _suit_block_num_floors_map is a map of block numbers to the number of floors in that block
+  SuitBlockFloorsMap _suit_block_num_floors_map;
   BlockToTitleMap _block_title_map;
   BlockToArticleMap _block_article_map;
   BlockToBuildingTypeMap _block_building_type_map;