Method packTetris

source/lib/PccLibEncoder/source/PCCEncoder.cpp:3258–3397

}

void PCCEncoder::packTetris( PCCFrameContext& frame, size_t presetWidth, size_t presetHeight, int safeguard ) {
  auto& width   = frame.getWidth();
  auto& height  = frame.getHeight();
  auto& patches = frame.getPatches();
  // set no matched patches, since this function does not take into account the previous frame
  frame.setNumMatchedPatches( 0 );
  if ( patches.empty() ) { return; }
  // sorting by patch largest dimension
  std::sort( patches.begin(), patches.end(), []( PCCPatch& a, PCCPatch& b ) { return a.gt( b ); } );
  if ( g_printDetailedInfo ) {
    std::cout << "Patch order:" << std::endl;
    for ( auto& patch : patches ) {
      std::cout << "Patch[" << patch.getIndex() << "]=(" << patch.getSizeU0() << "," << patch.getSizeV0() << ")"
                << std::endl;
    }
  }
  size_t occupancySizeU = presetWidth / params_.occupancyResolution_;
  size_t occupancySizeV = (std::max)( patches[0].getSizeV0(), patches[0].getSizeU0() );
  for ( auto& patch : patches ) { occupancySizeU = (std::max)( occupancySizeU, patch.getSizeU0() + 1 ); }
  width                             = occupancySizeU * params_.occupancyResolution_;
  height                            = occupancySizeV * params_.occupancyResolution_;
  size_t            maxOccupancyRow = 0;
  std::vector<bool> occupancyMap;
  occupancyMap.resize( occupancySizeU * occupancySizeV, false );
  std::vector<int> horizon;
  horizon.resize( occupancySizeU, 0 );
  if ( g_printDetailedInfo ) {
    std::cout << "Horizon :[";
    for ( int i = 0; i < occupancySizeU; i++ ) { std::cout << horizon[i] << ","; }
    std::cout << "]" << std::endl;
  }
  for ( auto& patch : patches ) {
    assert( patch.getSizeU0() <= occupancySizeU );
    assert( patch.getSizeV0() <= occupancySizeV );
    auto& occupancy = patch.getOccupancy();
    // getting the horizons using the rotation 0 position
    if ( g_printDetailedInfo ) { patch.print(); }
    std::vector<int> topHorizon;
    std::vector<int> bottomHorizon;
    std::vector<int> rightHorizon;
    std::vector<int> leftHorizon;
    patch.getPatchHorizons( topHorizon, bottomHorizon, rightHorizon, leftHorizon );
    bool locationFound = false;
    // try to place the patch tetris-style
    int numOrientations = params_.useEightOrientations_ ? 8 : 2;
    while ( !locationFound ) {
      int    best_wasted_space = (std::numeric_limits<int>::max)();
      size_t bestU;
      size_t bestV;
      int    bestOrientation;
      for ( size_t u = 0; u < occupancySizeU; ++u ) {
        for ( size_t v = 0; v < occupancySizeV; ++v ) {
          patch.setU0( u );
          patch.setV0( v );
          for ( size_t orientationIdx = 0; orientationIdx < numOrientations; orientationIdx++ ) {
            patch.setPatchOrientation( g_orientationVertical[orientationIdx] );
            if ( !patch.isPatchLocationAboveHorizon( horizon, topHorizon, bottomHorizon, rightHorizon, leftHorizon ) ) {
              if ( g_printDetailedInfo ) {