// Display next frame bool next() { static bool first = true; if(_decoder.getNextFrame(&_frame)) { ++_currentFrame; if(first) { _fh = _frame.height(); _fw = _frame.width(); createMask(); getBestFrames((folder+"/BgCandidates/_GraphcutInfo.txt").toStdString()); first = false; _width = 2000; _height = 1500; screen = new unsigned char[_width*_height*4]; } bool important = false; for(int i = 0; i < _bestFrames.size(); ++i) { if(_bestFrames[i].id == _currentFrame) important = true; } if(!important) return true; _framePos1 = _homographies[_currentFrame] * vec3(-_fw/2, _fh/2, 1); _framePos2 = _homographies[_currentFrame] * vec3(-_fw/2, -_fh/2, 1); _framePos3 = _homographies[_currentFrame] * vec3(_fw/2, -_fh/2, 1); _framePos4 = _homographies[_currentFrame] * vec3(_fw/2, _fh/2, 1); memset(screen, 0, _width*_height*3); flipV(screen, _width, _height, 3); SoftwareRenderer::render(RX::vec3(255, 0, 255), _framePos1, _framePos2, _framePos3, _framePos4, screen, _width, _height, 3); SoftwareRenderer::renderNoColorAB(_frame, RX::vec3(255, 0, 255), RX::vec3(0, 255, 0), _framePos1, _framePos2, _framePos3, _framePos4, screen, _width, _height, 4, 3); for(int i = 0; i < _width*_height; ++i) { char swap = screen[i*3]; screen[i*3] = screen[i*3+2]; screen[i*3+2] = swap; } char buf[50]; sprintf(buf, "/BgCandidates/mask%d.png", _currentFrame); QImage maskImg(screen, _width, _height, QImage::Format_RGB888); maskImg.save(folder+buf); return true; } else return false; }