mct.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 <vector>
 
namespace grk
{
struct ShiftInfo
{
  ShiftInfo(int32_t mn, int32_t mx, int32_t shift) : _min(mn), _max(mx), _shift(shift) {}
  ShiftInfo() : ShiftInfo(0, 0, 0) {}
  int32_t _min;
  int32_t _max;
  int32_t _shift;
};
 
struct ScheduleInfo
{
  ScheduleInfo(Tile* t, FlowComponent* flow, uint32_t linesPerTask)
      : tile(t), compno(0), flow_(flow), linesPerTask_(linesPerTask), yBegin(0), yEnd(0)
  {}
  Tile* tile;
  uint16_t compno;
  std::vector<ShiftInfo> shiftInfo;
  FlowComponent* flow_;
  uint32_t linesPerTask_;
  uint32_t yBegin;
  uint32_t yEnd;
};
 
class Mct
{
public:
  Mct(Tile* tile, GrkImage* image, TileCodingParams* tcp);

  void compress_rev(FlowComponent* flow, bool applyDcShift = false);
  void schedule_decompress_rev(FlowComponent* flow, bool applyDcShift = false);

  void compress_irrev(FlowComponent* flow, bool applyDcShift = false);
  void schedule_decompress_irrev(FlowComponent* flow, bool applyDcShift = false);

  void schedule_decompress_dc_shift_rev(FlowComponent* flow, uint16_t compno);

  void schedule_decompress_dc_shift_irrev(FlowComponent* flow, uint16_t compno);

  static const double* get_norms_rev(void);

  static const double* get_norms_irrev(void);

  static bool compress_custom(uint8_t* p_coding_data, uint64_t n, uint8_t** p_data,
                              uint16_t numComps, uint32_t is_signed);
  static bool schedule_decompress_custom(uint8_t* pDecodingData, uint64_t n, uint8_t** pData,
                                         uint16_t pNbComp, uint32_t isSigned);
  static void calculate_norms(double* pNorms, uint16_t nb_comps, float* pMatrix);
 
private:
  void genShift(uint16_t compno, int32_t sign, std::vector<ShiftInfo>& shiftInfo);
  void genShift(int32_t sign, std::vector<ShiftInfo>& shiftInfo);
 
  Tile* tile_;
  GrkImage* image_;
  TileCodingParams* tcp_;
};
 
/* ----------------------------------------------------------------------- */

 
} // namespace grk