movie.cpp

   1 #include "lsnes.hpp"
   2 #include "movie.hpp"
   3 #include "rom.hpp"
   4 #include "misc.hpp"
   5 #include <stdexcept>
   6 #include <cassert>
   7 #include <cstring>
   8 #include <fstream>
   9
  10 #define FLAG_SYNC CONTROL_FRAME_SYNC
  11
  12 //std::ofstream debuglog("movie-debugging-log", std::ios::out | std::ios::app);
  13
  14 namespace
  15 {
  16         void hash_string(uint8_t* res, const std::string& s) throw(std::bad_alloc)
  17         {
  18                 std::vector<char> t;
  19                 t.resize(s.length());
  20                 std::copy(s.begin(), s.end(), t.begin());
  21                 sha256::hash(res, t);
  22         }
  23
  24         uint8_t* enlarge(std::vector<uint8_t>& v, size_t amount) throw(std::bad_alloc)
  25         {
  26                 size_t i = v.size();
  27                 v.resize(i + amount);
  28                 return &v[i];
  29         }
  30
  31         inline void write64(uint8_t* buffer, uint64_t value) throw()
  32         {
  33                 buffer[0] = value >> 56;
  34                 buffer[1] = value >> 48;
  35                 buffer[2] = value >> 40;
  36                 buffer[3] = value >> 32;
  37                 buffer[4] = value >> 24;
  38                 buffer[5] = value >> 16;
  39                 buffer[6] = value >> 8;
  40                 buffer[7] = value;
  41         }
  42
  43         inline void write32(uint8_t* buffer, uint32_t value) throw()
  44         {
  45                 buffer[0] = value >> 24;
  46                 buffer[1] = value >> 16;
  47                 buffer[2] = value >> 8;
  48                 buffer[3] = value;
  49         }
  50
  51         inline uint32_t read32(const uint8_t* buffer) throw()
  52         {
  53                 return (static_cast<uint32_t>(buffer[0]) << 24) |
  54                         (static_cast<uint32_t>(buffer[1]) << 16) |
  55                         (static_cast<uint32_t>(buffer[2]) << 8) |
  56                         (static_cast<uint32_t>(buffer[3]));
  57         }
  58
  59         inline uint64_t read64(const uint8_t* buffer) throw()
  60         {
  61                 return (static_cast<uint64_t>(buffer[0]) << 56) |
  62                         (static_cast<uint64_t>(buffer[1]) << 48) |
  63                         (static_cast<uint64_t>(buffer[2]) << 40) |
  64                         (static_cast<uint64_t>(buffer[3]) << 32) |
  65                         (static_cast<uint64_t>(buffer[4]) << 24) |
  66                         (static_cast<uint64_t>(buffer[5]) << 16) |
  67                         (static_cast<uint64_t>(buffer[6]) << 8) |
  68                         (static_cast<uint64_t>(buffer[7]));
  69         }
  70
  71         inline void write16s(uint8_t* buffer, int16_t x) throw()
  72         {
  73                 uint16_t y = static_cast<uint16_t>(x);
  74                 buffer[0] = y >> 8;
  75                 buffer[1] = y;
  76         }
  77
  78         void hash_subframe(sha256& ctx, const controls_t& ctrl) throw()
  79         {
  80                 uint8_t buf[2 * TOTAL_CONTROLS];
  81                 for(unsigned i = 0; i < TOTAL_CONTROLS; i++)
  82                         write16s(buf + 2 * i, ctrl(i));
  83                 ctx.write(buf, 2 * TOTAL_CONTROLS);
  84         }
  85
  86         //Hashes frame and returns starting subframe of next frame.
  87         uint64_t hash_frame(sha256& ctx, std::vector<controls_t>& input, uint64_t first_subframe,
  88                 uint32_t bound) throw()
  89         {
  90                 if(!bound) {
  91                         //Ignore this frame completely.
  92                         if(first_subframe >= input.size())
  93                                 return first_subframe;
  94                         first_subframe++;
  95                         while(first_subframe < input.size() && !input[first_subframe](CONTROL_FRAME_SYNC))
  96                                 first_subframe++;
  97                         return first_subframe;
  98                 }
  99                 if(first_subframe >= input.size()) {
 100                         //Hash an empty frame.
 101                         hash_subframe(ctx, controls_t(true));
 102                         return first_subframe;
 103                 }
 104
 105                 uint64_t subframes_to_hash = 1;
 106                 uint64_t last_differing = 1;
 107                 uint64_t next;
 108                 controls_t prev = input[first_subframe];
 109                 prev(CONTROL_FRAME_SYNC) = 0;
 110
 111                 while(first_subframe + subframes_to_hash < input.size() && !input[first_subframe + subframes_to_hash]
 112                         (CONTROL_FRAME_SYNC)) {
 113                         if(!(input[first_subframe + subframes_to_hash] == prev))
 114                                 last_differing = subframes_to_hash + 1;
 115                         prev = input[first_subframe + subframes_to_hash];
 116                         subframes_to_hash++;
 117                 }
 118                 next = first_subframe + subframes_to_hash;
 119                 subframes_to_hash = last_differing;
 120                 for(uint64_t i = 0; i < subframes_to_hash && i < bound; i++)
 121                         hash_subframe(ctx, input[first_subframe + i]);
 122                 return next;
 123         }
 124
 125         void hash_movie(uint8_t* res, uint64_t current_frame, uint32_t* pollcounters,
 126                 std::vector<controls_t>& input) throw(std::bad_alloc)
 127         {
 128                 sha256 ctx;
 129                 //If current_frame == 0, hash is empty.
 130                 if(!current_frame) {
 131                         ctx.read(res);
 132                         return;
 133                 }
 134                 //Hash past frames.
 135                 uint64_t current_subframe = 0;
 136                 for(uint64_t i = 1; i < current_frame; i++)
 137                         current_subframe = hash_frame(ctx, input, current_subframe, 0x7FFFFFFF);
 138                 //Current frame is special.
 139                 for(size_t i = 0; i < TOTAL_CONTROLS; i++) {
 140                         uint32_t polls = pollcounters[i] & 0x7FFFFFFF;
 141                         uint32_t last_seen = 0;
 142                         for(size_t j = 0; j < polls; j++) {
 143                                 if(current_subframe + j < input.size() && !input[current_subframe + j]
 144                                         (CONTROL_FRAME_SYNC))
 145                                         last_seen = input[current_subframe + j](i);
 146                                 uint8_t buf[2];
 147                                 write16s(buf, last_seen);
 148                                 ctx.write(buf, 2);
 149                         }
 150                 }
 151                 ctx.read(res);
 152         }
 153 }
 154
 155 void movie::set_all_DRDY() throw()
 156 {
 157         for(size_t i = 0; i < TOTAL_CONTROLS; i++)
 158                 pollcounters[i] |= 0x80000000UL;
 159 }
 160
 161 std::string movie::rerecord_count() throw(std::bad_alloc)
 162 {
 163         return rerecords;
 164 }
 165
 166 void movie::rerecord_count(const std::string& count) throw(std::bad_alloc)
 167 {
 168         rerecords = count;
 169 }
 170
 171 std::string movie::project_id() throw(std::bad_alloc)
 172 {
 173         return _project_id;
 174 }
 175
 176 void movie::project_id(const std::string& id) throw(std::bad_alloc)
 177 {
 178         _project_id = id;
 179 }
 180
 181 bool movie::readonly_mode() throw()
 182 {
 183         return readonly;
 184 }
 185
 186 short movie::next_input(unsigned port, unsigned controller, unsigned index) throw(std::bad_alloc, std::logic_error)
 187 {
 188         return next_input(ccindex2(port, controller, index));
 189 }
 190
 191 void movie::set_controls(controls_t controls) throw()
 192 {
 193         current_controls = controls;
 194 }
 195
 196 uint32_t movie::count_changes(uint64_t first_subframe) throw()
 197 {
 198         if(first_subframe >= movie_data.size())
 199                 return 0;
 200         uint32_t ret = 1;
 201         while(first_subframe + ret < movie_data.size() && !movie_data[first_subframe + ret](CONTROL_FRAME_SYNC))
 202                 ret++;
 203         return ret;
 204 }
 205
 206 controls_t movie::get_controls() throw()
 207 {
 208         if(!readonly)
 209                 return current_controls;
 210         controls_t c;
 211         //Before the beginning? Somebody screwed up (but return released / neutral anyway)...
 212         if(current_frame == 0)
 213                 return c;
 214         //Otherwise find the last valid frame of input.
 215         uint32_t changes = count_changes(current_frame_first_subframe);
 216         if(!changes)
 217                 return c;       //End of movie.
 218         for(size_t i = 0; i < TOTAL_CONTROLS; i++) {
 219                 uint32_t polls = pollcounters[i] & 0x7FFFFFFF;
 220                 uint32_t index = (changes > polls) ? polls : changes - 1;
 221                 c(i) = movie_data[current_frame_first_subframe + index](i);
 222         }
 223         return c;
 224 }
 225
 226 uint64_t movie::get_current_frame() throw()
 227 {
 228         return current_frame;
 229 }
 230
 231 uint64_t movie::get_lag_frames() throw()
 232 {
 233         return lag_frames;
 234 }
 235
 236 uint64_t movie::get_frame_count() throw()
 237 {
 238         return frames_in_movie;
 239 }
 240
 241 void movie::next_frame() throw(std::bad_alloc)
 242 {
 243         //If all poll counters are zero for all real controls, this frame is lag.
 244         bool this_frame_lag = true;
 245         for(size_t i = MAX_SYSTEM_CONTROLS; i < TOTAL_CONTROLS; i++)
 246                 if(pollcounters[i] & 0x7FFFFFFF)
 247                         this_frame_lag = false;
 248         //Hack: Reset frames must not be considered lagged, so we abuse pollcounter bit for reset to mark those.
 249         if(pollcounters[CONTROL_SYSTEM_RESET] & 0x7FFFFFFF)
 250                 this_frame_lag = false;
 251         //Oh, frame 0 must not be considered lag.
 252         if(current_frame && this_frame_lag) {
 253                 lag_frames++;
 254                 //debuglog << "Frame " << current_frame << " is lag" << std::endl << std::flush;
 255                 if(!readonly) {
 256                         //If in read-write mode, write a dummy record for the frame. Force sync flag.
 257                         //As index should be movie_data.size(), it is correct afterwards.
 258                         controls_t c = current_controls;
 259                         c(CONTROL_FRAME_SYNC) = 1;
 260                         movie_data.push_back(c);
 261                         frames_in_movie++;
 262                 }
 263         }
 264
 265         //Reset the poll counters and DRDY flags.
 266         for(unsigned i = 0; i < TOTAL_CONTROLS; i++)
 267                 pollcounters[i] = 0;
 268
 269         //Increment the current frame counter and subframe counter. Note that first subframe is undefined for
 270         //frame 0 and 0 for frame 1.
 271         if(current_frame)
 272                 current_frame_first_subframe = current_frame_first_subframe +
 273                         count_changes(current_frame_first_subframe);
 274         else
 275                 current_frame_first_subframe = 0;
 276         current_frame++;
 277 }
 278
 279 bool movie::get_DRDY(unsigned controlindex) throw(std::logic_error)
 280 {
 281         if(controlindex >= TOTAL_CONTROLS)
 282                 throw std::logic_error("movie::get_DRDY: Bad index");
 283         return ((pollcounters[controlindex] & 0x80000000UL) != 0);
 284 }
 285
 286 bool movie::get_DRDY(unsigned port, unsigned controller, unsigned index) throw(std::logic_error)
 287 {
 288         return get_DRDY(ccindex2(port, controller, index));
 289 }
 290
 291 short movie::next_input(unsigned controlindex) throw(std::bad_alloc, std::logic_error)
 292 {
 293         //Check validity of index.
 294         if(controlindex == FLAG_SYNC)
 295                 return 0;
 296         if(controlindex >= TOTAL_CONTROLS)
 297                 throw std::logic_error("movie::next_input: Invalid control index");
 298
 299         //Clear the DRDY flag.
 300         pollcounters[controlindex] &= 0x7FFFFFFF;
 301
 302         if(readonly) {
 303                 //In readonly mode...
 304                 //If at the end of the movie, return released / neutral (but also record the poll)...
 305                 if(current_frame_first_subframe >= movie_data.size()) {
 306                         pollcounters[controlindex]++;
 307                         return 0;
 308                 }
 309                 //Before the beginning? Somebody screwed up (but return released / neutral anyway)...
 310                 if(current_frame == 0)
 311                         return 0;
 312                 //Otherwise find the last valid frame of input.
 313                 uint32_t changes = count_changes(current_frame_first_subframe);
 314                 uint32_t polls = (pollcounters[controlindex]++) & 0x7FFFFFFF;
 315                 uint32_t index = (changes > polls) ? polls : changes - 1;
 316                 //debuglog << "Frame=" << current_frame << " Subframe=" << polls << " control=" << controlindex << " value=" << movie_data[current_frame_first_subframe + index](controlindex) << " fetchrow=" << current_frame_first_subframe + index << std::endl << std::flush;
 317                 return movie_data[current_frame_first_subframe + index](controlindex);
 318         } else {
 319                 //Readwrite mode.
 320                 //Before the beginning? Somebody screwed up (but return released / neutral anyway)...
 321                 //Also, frame 0 must not be added to movie file.
 322                 if(current_frame == 0)
 323                         return 0;
 324                 //If at movie end, insert complete input with frame sync set (this is the first subframe).
 325                 if(current_frame_first_subframe >= movie_data.size()) {
 326                         controls_t c = current_controls;
 327                         c(CONTROL_FRAME_SYNC) = 1;
 328                         movie_data.push_back(c);
 329                         //current_frame_first_subframe should be movie_data.size(), so it is right.
 330                         pollcounters[controlindex]++;
 331                         frames_in_movie++;
 332                         assert(pollcounters[controlindex] == 1);
 333                         //debuglog << "Frame=" << current_frame << " Subframe=" << (pollcounters[controlindex] - 1) << " control=" << controlindex << " value=" << movie_data[current_frame_first_subframe](controlindex) << " fetchrow=" << current_frame_first_subframe << std::endl << std::flush;
 334                         return movie_data[current_frame_first_subframe](controlindex);
 335                 }
 336                 short new_value = current_controls(controlindex);
 337                 //Fortunately, we know this frame is the last one in movie_data.
 338                 uint32_t pollcounter = pollcounters[controlindex] & 0x7FFFFFFF;
 339                 uint64_t fetchrow = movie_data.size() - 1;
 340                 if(current_frame_first_subframe + pollcounter < movie_data.size()) {
 341                         //The index is within existing size. Change the value and propagate to all subsequent
 342                         //subframes.
 343                         for(uint64_t i = current_frame_first_subframe + pollcounter; i < movie_data.size(); i++)
 344                                 movie_data[i](controlindex) = new_value;
 345                         fetchrow = current_frame_first_subframe + pollcounter;
 346                 } else if(new_value != movie_data[movie_data.size() - 1](controlindex)) {
 347                         //The index is not within existing size and value does not match. We need to create a new
 348                         //subframes(s), copying the last subframe.
 349                         while(current_frame_first_subframe + pollcounter >= movie_data.size()) {
 350                                 controls_t c = movie_data[movie_data.size() - 1];
 351                                 c(CONTROL_FRAME_SYNC) = 0;
 352                                 movie_data.push_back(c);
 353                         }
 354                         fetchrow = current_frame_first_subframe + pollcounter;
 355                         movie_data[current_frame_first_subframe + pollcounter](controlindex) = new_value;
 356                 }
 357                 pollcounters[controlindex]++;
 358                 //debuglog << "Frame=" << current_frame << " Subframe=" << (pollcounters[controlindex] - 1) << " control=" << controlindex << " value=" << new_value << " fetchrow=" << fetchrow << std::endl << std::flush;
 359                 return new_value;
 360         }
 361 }
 362
 363 movie::movie() throw(std::bad_alloc)
 364 {
 365         readonly = false;
 366         rerecords = "0";
 367         _project_id = "";
 368         current_frame = 0;
 369         frames_in_movie = 0;
 370         current_frame_first_subframe = 0;
 371         for(unsigned i = 0; i < TOTAL_CONTROLS; i++) {
 372                 pollcounters[i] = 0;
 373                 current_controls(i) = 0;
 374         }
 375         lag_frames = 0;
 376         clear_caches();
 377 }
 378
 379 void movie::load(const std::string& rerecs, const std::string& project_id, const std::vector<controls_t>& input)
 380         throw(std::bad_alloc, std::runtime_error)
 381 {
 382         if(input.size() > 0 && !input[0](CONTROL_FRAME_SYNC))
 383                 throw std::runtime_error("First subframe MUST have frame sync flag set");
 384         clear_caches();
 385         frames_in_movie = 0;
 386         for(auto i = input.begin(); i != input.end(); i++)
 387                 if((*i)(CONTROL_FRAME_SYNC))
 388                         frames_in_movie++;
 389         readonly = true;
 390         rerecords = rerecs;
 391         _project_id = project_id;
 392         current_frame = 0;
 393         current_frame_first_subframe = 0;
 394         for(unsigned i = 0; i < TOTAL_CONTROLS; i++) {
 395                 pollcounters[i] = 0;
 396         }
 397         lag_frames = 0;
 398         movie_data = input;
 399 }
 400
 401 std::vector<controls_t> movie::save() throw(std::bad_alloc)
 402 {
 403         return movie_data;
 404 }
 405
 406 void movie::commit_reset(long delay) throw(std::bad_alloc)
 407 {
 408         if(readonly || delay < 0)
 409                 return;
 410         //If this frame is lagged, we need to write entry for it.
 411         bool this_frame_lag = true;
 412         for(size_t i = MAX_SYSTEM_CONTROLS; i < TOTAL_CONTROLS; i++)
 413                 if(pollcounters[i] & 0x7FFFFFFF)
 414                         this_frame_lag = false;
 415         //Hack: Reset frames must not be considered lagged, so we abuse pollcounter bit for reset to mark those.
 416         if(pollcounters[CONTROL_SYSTEM_RESET] & 0x7FFFFFFF)
 417                 this_frame_lag = false;
 418         if(this_frame_lag) {
 419                 controls_t c = current_controls;
 420                 c(CONTROL_FRAME_SYNC) = 1;
 421                 movie_data.push_back(c);
 422                 frames_in_movie++;
 423                 //Current_frame_first_subframe is correct.
 424         }
 425         //Also set poll counters on reset cycles to avoid special cases elsewhere.
 426         pollcounters[CONTROL_SYSTEM_RESET] = 1;
 427         pollcounters[CONTROL_SYSTEM_RESET_CYCLES_HI] = 1;
 428         pollcounters[CONTROL_SYSTEM_RESET_CYCLES_LO] = 1;
 429         //Current frame is always last in rw mode.
 430         movie_data[current_frame_first_subframe](CONTROL_SYSTEM_RESET) = 1;
 431         movie_data[current_frame_first_subframe](CONTROL_SYSTEM_RESET_CYCLES_HI) = delay / 10000;
 432         movie_data[current_frame_first_subframe](CONTROL_SYSTEM_RESET_CYCLES_LO) = delay % 10000;
 433 }
 434
 435 unsigned movie::next_poll_number()
 436 {
 437         unsigned max = 0;
 438         for(unsigned i = 0; i < TOTAL_CONTROLS; i++)
 439                 if(max < (pollcounters[i] & 0x7FFFFFFF))
 440                         max = (pollcounters[i] & 0x7FFFFFFF);
 441         return max + 1;
 442 }
 443
 444 void movie::readonly_mode(bool enable) throw(std::bad_alloc)
 445 {
 446         bool was_in_readonly = readonly;
 447         readonly = enable;
 448         if(was_in_readonly && !readonly) {
 449                 clear_caches();
 450                 //Transitioning to readwrite mode, we have to adjust the length of the movie data.
 451                 if(current_frame == 0) {
 452                         //WTF... At before first frame. Blank the entiere movie.
 453                         movie_data.clear();
 454                         return;
 455                 }
 456                 //Fun special case: Current frame is not in movie (current_frame_first_subframe >= movie_data.size()).
 457                 //In this case, we have to extend the movie data.
 458                 if(current_frame_first_subframe >= movie_data.size()) {
 459                         //Yes, this will insert one extra frame... But we will lose it later if it is not needed.
 460                         while(frames_in_movie < current_frame) {
 461                                 controls_t c(true);
 462                                 movie_data.push_back(c);
 463                                 frames_in_movie++;
 464                         }
 465                         current_frame_first_subframe = movie_data.size() - 1;
 466                 }
 467
 468                 //We have to take the part up to furthest currently readable subframe. Also, we need to propagate
 469                 //forward values with smaller poll counters.
 470                 uint64_t next_frame_first_subframe = current_frame_first_subframe +
 471                         count_changes(current_frame_first_subframe);
 472                 uint64_t max_readable_subframes = current_frame_first_subframe;
 473                 for(size_t i = 0; i < TOTAL_CONTROLS; i++) {
 474                         uint32_t polls = pollcounters[i] & 0x7FFFFFFF;
 475                         if(current_frame_first_subframe + polls >= next_frame_first_subframe)
 476                                 max_readable_subframes = next_frame_first_subframe;
 477                         else if(current_frame_first_subframe + polls > max_readable_subframes)
 478                                 max_readable_subframes = current_frame_first_subframe + polls;
 479                 }
 480                 movie_data.resize(max_readable_subframes);
 481                 next_frame_first_subframe = max_readable_subframes;
 482                 for(size_t i = 1; i < TOTAL_CONTROLS; i++) {
 483                         uint32_t polls = pollcounters[i] & 0x7FFFFFFF;
 484                         if(!polls)
 485                                 polls = 1;
 486                         for(uint64_t j = current_frame_first_subframe + polls; j < next_frame_first_subframe; j++)
 487                                 movie_data[j](i) = movie_data[current_frame_first_subframe + polls - 1](i);
 488                 }
 489                 frames_in_movie = current_frame - ((current_frame_first_subframe >= movie_data.size()) ? 1 : 0);
 490         }
 491 }
 492
 493 //Save state of movie code.
 494 std::vector<uint8_t> movie::save_state() throw(std::bad_alloc)
 495 {
 496         //debuglog << "--------------------------------------------" << std::endl;
 497         //debuglog << "SAVING STATE:" << std::endl;
 498         std::vector<uint8_t> ret;
 499         hash_string(enlarge(ret, 32), _project_id);
 500         write64(enlarge(ret, 8), current_frame);
 501         //debuglog << "Current frame is " << current_frame << std::endl;
 502         //debuglog << "Poll counters: ";
 503         for(unsigned i = 0; i < TOTAL_CONTROLS; i++) {
 504                 uint32_t v = pollcounters[i];
 505                 //debuglog << v;
 506                 if(v & 0x80000000UL) {
 507                         //debuglog << "R ";
 508                 } else
 509                         ;//debuglog << " ";
 510                 write32(enlarge(ret, 4), v);
 511         }
 512         //debuglog << std::endl;
 513         {
 514                 uint64_t v = lag_frames;
 515                 //debuglog << "Lag frame count: " << lag_frames << std::endl;
 516                 write64(enlarge(ret, 8), v);
 517         }
 518         hash_movie(enlarge(ret, 32), current_frame, pollcounters, movie_data);
 519         uint8_t hash[32];
 520         sha256::hash(hash, ret);
 521         memcpy(enlarge(ret, 32), hash, 32);
 522         //debuglog << "--------------------------------------------" << std::endl;
 523         //debuglog.flush();
 524         return ret;
 525 }
 526
 527 //Restore state of movie code. Throws if state is invalid. Flag gives new state of readonly flag.
 528 size_t movie::restore_state(const std::vector<uint8_t>& state, bool ro) throw(std::bad_alloc, std::runtime_error)
 529 {
 530         //Check the whole-data checksum.
 531         size_t ptr = 0;
 532         uint8_t tmp[32];
 533         if(state.size() != 112+4*TOTAL_CONTROLS)
 534                 throw std::runtime_error("Movie save data corrupt: Wrong length");
 535         sha256::hash(tmp, &state[0], state.size() - 32);
 536         if(memcmp(tmp, &state[state.size() - 32], 32))
 537                 throw std::runtime_error("Movie save data corrupt: Checksum does not match");
 538         //debuglog << "--------------------------------------------" << std::endl;
 539         //debuglog << "RESTORING STATE:" << std::endl;
 540         //Check project id.
 541         hash_string(tmp, _project_id);
 542         if(memcmp(tmp, &state[ptr], 32))
 543                 throw std::runtime_error("Save is not from this movie");
 544         ptr += 32;
 545         //Read current frame.
 546         uint64_t tmp_curframe = read64(&state[ptr]);
 547         uint64_t tmp_firstsubframe = 0;
 548         for(uint64_t i = 1; i < tmp_curframe; i++)
 549                 tmp_firstsubframe = tmp_firstsubframe + count_changes(tmp_firstsubframe);
 550         ptr += 8;
 551         //Read poll counters and drdy flags.
 552         uint32_t tmp_pollcount[TOTAL_CONTROLS];
 553         for(unsigned i = 0; i < TOTAL_CONTROLS; i++) {
 554                 uint32_t v = read32(&state[ptr]);
 555                 ptr += 4;
 556                 tmp_pollcount[i] = v;
 557         }
 558         uint64_t tmp_lagframes = read64(&state[ptr]);
 559         tmp_lagframes &= 0x7FFFFFFFFFFFFFFFULL;
 560         ptr += 8;
 561         hash_movie(tmp, tmp_curframe, tmp_pollcount, movie_data);
 562         if(memcmp(tmp, &state[ptr], 32))
 563                 throw std::runtime_error("Save is not from this movie");
 564
 565         //Ok, all checks pass. Copy the state. Do this in readonly mode so we can use normal routine to switch
 566         //to readwrite mode.
 567         readonly = true;
 568         current_frame = tmp_curframe;
 569         current_frame_first_subframe = tmp_firstsubframe;
 570         memcpy(pollcounters, tmp_pollcount, sizeof(tmp_pollcount));
 571         lag_frames = tmp_lagframes;
 572
 573         //debuglog << "Current frame is " << current_frame << std::endl;
 574         //debuglog << "Poll counters: ";
 575         for(unsigned i = 0; i < TOTAL_CONTROLS; i++) {
 576                 uint32_t v = pollcounters[i];
 577                 //debuglog << v;
 578                 if(v & 0x80000000UL) {
 579                         //debuglog << "R ";
 580                 } else
 581                         ;//debuglog << " ";
 582         }
 583         //debuglog << std::endl;
 584         {
 585                 //debuglog << "Lag frame count: " << lag_frames << std::endl;
 586         }
 587
 588         //debuglog << "--------------------------------------------" << std::endl;
 589         //debuglog.flush();
 590
 591         //Move to readwrite mode if needed.
 592         readonly_mode(ro);
 593         return 0;
 594 }
 595
 596 long movie::get_reset_status() throw()
 597 {
 598         if(current_frame == 0 || current_frame_first_subframe >= movie_data.size())
 599                 return -1;      //No resets out of range.
 600         if(!movie_data[current_frame_first_subframe](CONTROL_SYSTEM_RESET))
 601                 return -1;      //Not a reset.
 602         //Also set poll counters on reset cycles to avoid special cases elsewhere.
 603         pollcounters[CONTROL_SYSTEM_RESET] = 1;
 604         pollcounters[CONTROL_SYSTEM_RESET_CYCLES_HI] = 1;
 605         pollcounters[CONTROL_SYSTEM_RESET_CYCLES_LO] = 1;
 606         long hi = movie_data[current_frame_first_subframe](CONTROL_SYSTEM_RESET_CYCLES_HI);
 607         long lo = movie_data[current_frame_first_subframe](CONTROL_SYSTEM_RESET_CYCLES_LO);
 608         return hi * 10000 + lo;
 609 }
 610
 611 uint64_t movie::frame_subframes(uint64_t frame) throw()
 612 {
 613         if(frame < cached_frame)
 614                 clear_caches();
 615         uint64_t p = cached_subframe;
 616         for(uint64_t i = cached_frame; i < frame; i++)
 617                 p = p + count_changes(p);
 618         cached_frame = frame;
 619         cached_subframe = p;
 620         return count_changes(p);
 621 }
 622
 623 void movie::clear_caches() throw()
 624 {
 625         cached_frame = 1;
 626         cached_subframe = 0;
 627 }
 628
 629 controls_t movie::read_subframe(uint64_t frame, uint64_t subframe) throw()
 630 {
 631         if(frame < cached_frame)
 632                 clear_caches();
 633         uint64_t p = cached_subframe;
 634         for(uint64_t i = cached_frame; i < frame; i++)
 635                 p = p + count_changes(p);
 636         cached_frame = frame;
 637         cached_subframe = p;
 638         uint64_t max = count_changes(p);
 639         if(!max)
 640                 return controls_t(true);
 641         if(max <= subframe)
 642                 subframe = max - 1;
 643         return movie_data[p + subframe];
 644 }
 645
 646
 647 movie_logic::movie_logic() throw()
 648 {
 649 }
 650
 651 movie& movie_logic::get_movie() throw()
 652 {
 653         return mov;
 654 }
 655
 656 long movie_logic::new_frame_starting(bool dont_poll) throw(std::bad_alloc, std::runtime_error)
 657 {
 658         mov.next_frame();
 659         controls_t c = update_controls(false);
 660         if(!mov.readonly_mode()) {
 661                 mov.set_controls(c);
 662                 if(dont_poll)
 663                         mov.set_all_DRDY();
 664                 if(c(CONTROL_SYSTEM_RESET)) {
 665                         long hi = c(CONTROL_SYSTEM_RESET_CYCLES_HI);
 666                         long lo = c(CONTROL_SYSTEM_RESET_CYCLES_LO);
 667                         mov.commit_reset(hi * 10000 + lo);
 668                 }
 669         }
 670         return mov.get_reset_status();
 671 }
 672
 673 short movie_logic::input_poll(bool port, unsigned dev, unsigned id) throw(std::bad_alloc, std::runtime_error)
 674 {
 675         if(dev >= MAX_CONTROLLERS_PER_PORT || id >= CONTROLLER_CONTROLS)
 676                 return 0;
 677         if(!mov.get_DRDY(port ? 1 : 0, dev, id)) {
 678                 mov.set_controls(update_controls(true));
 679                 mov.set_all_DRDY();
 680         }
 681         int16_t in = mov.next_input(port ? 1 : 0, dev, id);
 682         //debuglog << "BSNES asking for (" << port << "," << dev << "," << id << ") (frame " << mov.get_current_frame()
 683         //      << ") giving " << in << std::endl;
 684         //debuglog.flush();
 685         return in;
 686 }