CompressScheduler.h

Go to the documentation of this file.

/*
 *    Copyright (C) 2016-2026 Grok Image Compression Inc.
 *
 *    This source code is free software: you can redistribute it and/or  modify
 *    it under the terms of the GNU Affero General Public License, version 3,
 *    as published by the Free Software Foundation.
 *
 *    This source code is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU Affero General Public License for more details.
 *
 *    You should have received a copy of the GNU Affero General Public License
 *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
 
#pragma once
 
#include "SchedulerStandard.h"
#include "RateControlStats.h"
 
namespace grk
{
class CompressScheduler : public SchedulerStandard
{
public:
  CompressScheduler(Tile* tile, bool needsRateControl, TileCodingParams* tcp,
                    const double* mct_norms, uint16_t mct_numcomps);
  ~CompressScheduler() override = default;
 
  bool scheduleT1(ITileProcessor* proc) override;
  bool populateT1Flow(FlowComponent* flow);

  const RateControlStats& getRateControlStats() const
  {
    return rateControlStats_;
  }
 
private:
  bool compress(size_t workerId, uint64_t maxBlocks);

  void compress(t1::ICoder* coder, t1::CompressBlockExec* block);

  Tile* tile_;

  mutable std::mutex distortion_mutex_;
  bool needsRateControl_;
  std::vector<t1::CompressBlockExec*> encodeBlocks_;

  std::atomic<int64_t> blockCount_;

  TileCodingParams* tcp_;

  const double* mct_norms_;

  uint16_t mct_numcomps_;

  RateControlStats rateControlStats_;
};
 
} // namespace grk