saveload_system.rs 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. use specs::prelude::*;
  2. use specs::saveload::{SimpleMarker, SimpleMarkerAllocator, SerializeComponents, DeserializeComponents, MarkedBuilder};
  3. use specs::error::NoError;
  4. use super::components::*;
  5. use std::fs::File;
  6. use std::path::Path;
  7. use std::fs;
  8. macro_rules! serialize_individually {
  9. ($ecs:expr, $ser:expr, $data:expr, $( $type:ty),*) => {
  10. $(
  11. SerializeComponents::<NoError, SimpleMarker<SerializeMe>>::serialize(
  12. &( $ecs.read_storage::<$type>(), ),
  13. &$data.0,
  14. &$data.1,
  15. &mut $ser,
  16. )
  17. .unwrap();
  18. )*
  19. };
  20. }
  21. pub fn save_game(ecs : &mut World) {
  22. // Create helper
  23. let mapcopy = ecs.get_mut::<super::map::Map>().unwrap().clone();
  24. let savehelper = ecs
  25. .create_entity()
  26. .with(SerializationHelper{ map : mapcopy })
  27. .marked::<SimpleMarker<SerializeMe>>()
  28. .build();
  29. // Actually serialize
  30. {
  31. let data = ( ecs.entities(), ecs.read_storage::<SimpleMarker<SerializeMe>>() );
  32. let writer = File::create("./savegame.json").unwrap();
  33. let mut serializer = serde_json::Serializer::new(writer);
  34. serialize_individually!(ecs, serializer, data, Position, Renderable, Player, Viewshed, Monster,
  35. Name, BlocksTile, CombatStats, SufferDamage, WantsToMelee, Item, Consumable, Ranged, InflictsDamage,
  36. AreaOfEffect, Confusion, ProvidesHealing, InBackpack, WantsToPickupItem, WantsToUseItem,
  37. WantsToDropItem, SerializationHelper, Equippable, Equipped, MeleePowerBonus, DefenseBonus,
  38. WantsToRemoveItem, ParticleLifetime, HungerClock, ProvidesFood, MagicMapper
  39. );
  40. }
  41. // Clean up
  42. ecs.delete_entity(savehelper).expect("Crash on cleanup");
  43. }
  44. pub fn does_save_exist() -> bool {
  45. Path::new("./savegame.json").exists()
  46. }
  47. macro_rules! deserialize_individually {
  48. ($ecs:expr, $de:expr, $data:expr, $( $type:ty),*) => {
  49. $(
  50. DeserializeComponents::<NoError, _>::deserialize(
  51. &mut ( &mut $ecs.write_storage::<$type>(), ),
  52. &mut $data.0, // entities
  53. &mut $data.1, // marker
  54. &mut $data.2, // allocater
  55. &mut $de,
  56. )
  57. .unwrap();
  58. )*
  59. };
  60. }
  61. pub fn load_game(ecs: &mut World) {
  62. {
  63. // Delete everything
  64. let mut to_delete = Vec::new();
  65. for e in ecs.entities().join() {
  66. to_delete.push(e);
  67. }
  68. for del in to_delete.iter() {
  69. ecs.delete_entity(*del).expect("Deletion failed");
  70. }
  71. }
  72. let data = fs::read_to_string("./savegame.json").unwrap();
  73. let mut de = serde_json::Deserializer::from_str(&data);
  74. {
  75. let mut d = (&mut ecs.entities(), &mut ecs.write_storage::<SimpleMarker<SerializeMe>>(), &mut SimpleMarkerAllocator::<SerializeMe>::new());
  76. deserialize_individually!(ecs, de, d, Position, Renderable, Player, Viewshed, Monster,
  77. Name, BlocksTile, CombatStats, SufferDamage, WantsToMelee, Item, Consumable, Ranged, InflictsDamage,
  78. AreaOfEffect, Confusion, ProvidesHealing, InBackpack, WantsToPickupItem, WantsToUseItem,
  79. WantsToDropItem, SerializationHelper, Equippable, Equipped, MeleePowerBonus, DefenseBonus,
  80. WantsToRemoveItem, ParticleLifetime, HungerClock, ProvidesFood, MagicMapper
  81. );
  82. }
  83. let mut deleteme : Option<Entity> = None;
  84. {
  85. let entities = ecs.entities();
  86. let helper = ecs.read_storage::<SerializationHelper>();
  87. let player = ecs.read_storage::<Player>();
  88. let position = ecs.read_storage::<Position>();
  89. for (e,h) in (&entities, &helper).join() {
  90. let mut worldmap = ecs.write_resource::<super::map::Map>();
  91. *worldmap = h.map.clone();
  92. worldmap.tile_content = vec![Vec::new(); super::map::MAPCOUNT];
  93. deleteme = Some(e);
  94. }
  95. for (e,_p,pos) in (&entities, &player, &position).join() {
  96. let mut ppos = ecs.write_resource::<rltk::Point>();
  97. *ppos = rltk::Point::new(pos.x, pos.y);
  98. let mut player_resource = ecs.write_resource::<Entity>();
  99. *player_resource = e;
  100. }
  101. }
  102. ecs.delete_entity(deleteme.unwrap()).expect("Unable to delete helper");
  103. }
  104. pub fn delete_save() {
  105. if Path::new("./savegame.json").exists() { std::fs::remove_file("./savegame.json").expect("Unable to delete file"); }
  106. }