409 lines
9.9 KiB
C++
409 lines
9.9 KiB
C++
// basisu_backend.h
|
|
// Copyright (C) 2019-2021 Binomial LLC. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
#pragma once
|
|
|
|
#include "../transcoder/basisu.h"
|
|
#include "basisu_enc.h"
|
|
#include "../transcoder/basisu_transcoder_internal.h"
|
|
#include "basisu_frontend.h"
|
|
|
|
namespace basisu
|
|
{
|
|
struct etc1_selector_palette_entry
|
|
{
|
|
etc1_selector_palette_entry()
|
|
{
|
|
clear();
|
|
}
|
|
|
|
void clear()
|
|
{
|
|
basisu::clear_obj(*this);
|
|
}
|
|
|
|
uint8_t operator[] (uint32_t i) const { assert(i < 16); return m_selectors[i]; }
|
|
uint8_t& operator[] (uint32_t i) { assert(i < 16); return m_selectors[i]; }
|
|
|
|
void set_uint32(uint32_t v)
|
|
{
|
|
for (uint32_t byte_index = 0; byte_index < 4; byte_index++)
|
|
{
|
|
uint32_t b = (v >> (byte_index * 8)) & 0xFF;
|
|
|
|
m_selectors[byte_index * 4 + 0] = b & 3;
|
|
m_selectors[byte_index * 4 + 1] = (b >> 2) & 3;
|
|
m_selectors[byte_index * 4 + 2] = (b >> 4) & 3;
|
|
m_selectors[byte_index * 4 + 3] = (b >> 6) & 3;
|
|
}
|
|
}
|
|
|
|
uint32_t get_uint32() const
|
|
{
|
|
return get_byte(0) | (get_byte(1) << 8) | (get_byte(2) << 16) | (get_byte(3) << 24);
|
|
}
|
|
|
|
uint32_t get_byte(uint32_t byte_index) const
|
|
{
|
|
assert(byte_index < 4);
|
|
|
|
return m_selectors[byte_index * 4 + 0] |
|
|
(m_selectors[byte_index * 4 + 1] << 2) |
|
|
(m_selectors[byte_index * 4 + 2] << 4) |
|
|
(m_selectors[byte_index * 4 + 3] << 6);
|
|
}
|
|
|
|
uint8_t operator()(uint32_t x, uint32_t y) const { assert((x < 4) && (y < 4)); return m_selectors[x + y * 4]; }
|
|
uint8_t& operator()(uint32_t x, uint32_t y) { assert((x < 4) && (y < 4)); return m_selectors[x + y * 4]; }
|
|
|
|
bool operator< (const etc1_selector_palette_entry& other) const
|
|
{
|
|
for (uint32_t i = 0; i < 16; i++)
|
|
{
|
|
if (m_selectors[i] < other.m_selectors[i])
|
|
return true;
|
|
else if (m_selectors[i] != other.m_selectors[i])
|
|
return false;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool operator== (const etc1_selector_palette_entry& other) const
|
|
{
|
|
for (uint32_t i = 0; i < 16; i++)
|
|
{
|
|
if (m_selectors[i] != other.m_selectors[i])
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
private:
|
|
uint8_t m_selectors[16];
|
|
};
|
|
|
|
typedef basisu::vector<etc1_selector_palette_entry> etc1_selector_palette_entry_vec;
|
|
|
|
struct encoder_block
|
|
{
|
|
encoder_block()
|
|
{
|
|
clear();
|
|
}
|
|
|
|
uint32_t m_endpoint_predictor;
|
|
|
|
int m_endpoint_index;
|
|
int m_selector_index;
|
|
|
|
int m_selector_history_buf_index;
|
|
|
|
bool m_is_cr_target;
|
|
void clear()
|
|
{
|
|
m_endpoint_predictor = 0;
|
|
|
|
m_endpoint_index = 0;
|
|
m_selector_index = 0;
|
|
|
|
m_selector_history_buf_index = 0;
|
|
m_is_cr_target = false;
|
|
}
|
|
};
|
|
|
|
typedef basisu::vector<encoder_block> encoder_block_vec;
|
|
typedef vector2D<encoder_block> encoder_block_vec2D;
|
|
|
|
struct etc1_endpoint_palette_entry
|
|
{
|
|
etc1_endpoint_palette_entry()
|
|
{
|
|
clear();
|
|
}
|
|
|
|
color_rgba m_color5;
|
|
uint32_t m_inten5;
|
|
bool m_color5_valid;
|
|
|
|
void clear()
|
|
{
|
|
clear_obj(*this);
|
|
}
|
|
};
|
|
|
|
typedef basisu::vector<etc1_endpoint_palette_entry> etc1_endpoint_palette_entry_vec;
|
|
|
|
struct basisu_backend_params
|
|
{
|
|
bool m_etc1s;
|
|
bool m_debug, m_debug_images;
|
|
float m_endpoint_rdo_quality_thresh;
|
|
float m_selector_rdo_quality_thresh;
|
|
uint32_t m_compression_level;
|
|
|
|
bool m_used_global_codebooks;
|
|
|
|
bool m_validate;
|
|
|
|
basisu_backend_params()
|
|
{
|
|
clear();
|
|
}
|
|
|
|
void clear()
|
|
{
|
|
m_etc1s = false;
|
|
m_debug = false;
|
|
m_debug_images = false;
|
|
m_endpoint_rdo_quality_thresh = 0.0f;
|
|
m_selector_rdo_quality_thresh = 0.0f;
|
|
m_compression_level = 0;
|
|
m_used_global_codebooks = false;
|
|
m_validate = true;
|
|
}
|
|
};
|
|
|
|
struct basisu_backend_slice_desc
|
|
{
|
|
basisu_backend_slice_desc()
|
|
{
|
|
clear();
|
|
}
|
|
|
|
void clear()
|
|
{
|
|
clear_obj(*this);
|
|
}
|
|
|
|
uint32_t m_first_block_index;
|
|
|
|
uint32_t m_orig_width;
|
|
uint32_t m_orig_height;
|
|
|
|
uint32_t m_width;
|
|
uint32_t m_height;
|
|
|
|
uint32_t m_num_blocks_x;
|
|
uint32_t m_num_blocks_y;
|
|
|
|
uint32_t m_num_macroblocks_x;
|
|
uint32_t m_num_macroblocks_y;
|
|
|
|
uint32_t m_source_file_index; // also the basis image index
|
|
uint32_t m_mip_index;
|
|
bool m_alpha;
|
|
bool m_iframe;
|
|
};
|
|
|
|
typedef basisu::vector<basisu_backend_slice_desc> basisu_backend_slice_desc_vec;
|
|
|
|
struct basisu_backend_output
|
|
{
|
|
basist::basis_tex_format m_tex_format;
|
|
|
|
bool m_etc1s;
|
|
bool m_uses_global_codebooks;
|
|
bool m_srgb;
|
|
|
|
uint32_t m_num_endpoints;
|
|
uint32_t m_num_selectors;
|
|
|
|
uint8_vec m_endpoint_palette;
|
|
uint8_vec m_selector_palette;
|
|
|
|
basisu_backend_slice_desc_vec m_slice_desc;
|
|
|
|
uint8_vec m_slice_image_tables;
|
|
basisu::vector<uint8_vec> m_slice_image_data;
|
|
uint16_vec m_slice_image_crcs;
|
|
|
|
basisu_backend_output()
|
|
{
|
|
clear();
|
|
}
|
|
|
|
void clear()
|
|
{
|
|
m_tex_format = basist::basis_tex_format::cETC1S;
|
|
m_etc1s = false;
|
|
m_uses_global_codebooks = false;
|
|
m_srgb = true;
|
|
|
|
m_num_endpoints = 0;
|
|
m_num_selectors = 0;
|
|
|
|
m_endpoint_palette.clear();
|
|
m_selector_palette.clear();
|
|
m_slice_desc.clear();
|
|
m_slice_image_tables.clear();
|
|
m_slice_image_data.clear();
|
|
m_slice_image_crcs.clear();
|
|
}
|
|
|
|
uint32_t get_output_size_estimate() const
|
|
{
|
|
uint32_t total_compressed_bytes = (uint32_t)(m_slice_image_tables.size() + m_endpoint_palette.size() + m_selector_palette.size());
|
|
for (uint32_t i = 0; i < m_slice_image_data.size(); i++)
|
|
total_compressed_bytes += (uint32_t)m_slice_image_data[i].size();
|
|
|
|
return total_compressed_bytes;
|
|
}
|
|
};
|
|
|
|
class basisu_backend
|
|
{
|
|
BASISU_NO_EQUALS_OR_COPY_CONSTRUCT(basisu_backend);
|
|
|
|
public:
|
|
|
|
basisu_backend();
|
|
|
|
void clear();
|
|
|
|
void init(basisu_frontend *pFront_end, basisu_backend_params ¶ms, const basisu_backend_slice_desc_vec &slice_desc);
|
|
|
|
uint32_t encode();
|
|
|
|
const basisu_backend_output &get_output() const { return m_output; }
|
|
const basisu_backend_params& get_params() const { return m_params; }
|
|
|
|
private:
|
|
basisu_frontend *m_pFront_end;
|
|
basisu_backend_params m_params;
|
|
basisu_backend_slice_desc_vec m_slices;
|
|
basisu_backend_output m_output;
|
|
|
|
etc1_endpoint_palette_entry_vec m_endpoint_palette;
|
|
etc1_selector_palette_entry_vec m_selector_palette;
|
|
|
|
struct etc1_global_selector_cb_entry_desc
|
|
{
|
|
uint32_t m_pal_index;
|
|
uint32_t m_mod_index;
|
|
bool m_was_used;
|
|
};
|
|
|
|
typedef basisu::vector<etc1_global_selector_cb_entry_desc> etc1_global_selector_cb_entry_desc_vec;
|
|
|
|
etc1_global_selector_cb_entry_desc_vec m_global_selector_palette_desc;
|
|
|
|
basisu::vector<encoder_block_vec2D> m_slice_encoder_blocks;
|
|
|
|
// Maps OLD to NEW endpoint/selector indices
|
|
uint_vec m_endpoint_remap_table_old_to_new;
|
|
uint_vec m_endpoint_remap_table_new_to_old;
|
|
bool_vec m_old_endpoint_was_used;
|
|
bool_vec m_new_endpoint_was_used;
|
|
|
|
uint_vec m_selector_remap_table_old_to_new;
|
|
|
|
// Maps NEW to OLD endpoint/selector indices
|
|
uint_vec m_selector_remap_table_new_to_old;
|
|
|
|
uint32_t get_total_slices() const
|
|
{
|
|
return (uint32_t)m_slices.size();
|
|
}
|
|
|
|
uint32_t get_total_slice_blocks() const
|
|
{
|
|
return m_pFront_end->get_total_output_blocks();
|
|
}
|
|
|
|
uint32_t get_block_index(uint32_t slice_index, uint32_t block_x, uint32_t block_y) const
|
|
{
|
|
const basisu_backend_slice_desc &slice = m_slices[slice_index];
|
|
|
|
assert((block_x < slice.m_num_blocks_x) && (block_y < slice.m_num_blocks_y));
|
|
|
|
return slice.m_first_block_index + block_y * slice.m_num_blocks_x + block_x;
|
|
}
|
|
|
|
uint32_t get_total_blocks(uint32_t slice_index) const
|
|
{
|
|
return m_slices[slice_index].m_num_blocks_x * m_slices[slice_index].m_num_blocks_y;
|
|
}
|
|
|
|
uint32_t get_total_blocks() const
|
|
{
|
|
uint32_t total_blocks = 0;
|
|
for (uint32_t i = 0; i < m_slices.size(); i++)
|
|
total_blocks += get_total_blocks(i);
|
|
return total_blocks;
|
|
}
|
|
|
|
// Returns the total number of input texels, not counting padding up to blocks/macroblocks.
|
|
uint32_t get_total_input_texels(uint32_t slice_index) const
|
|
{
|
|
return m_slices[slice_index].m_orig_width * m_slices[slice_index].m_orig_height;
|
|
}
|
|
|
|
uint32_t get_total_input_texels() const
|
|
{
|
|
uint32_t total_texels = 0;
|
|
for (uint32_t i = 0; i < m_slices.size(); i++)
|
|
total_texels += get_total_input_texels(i);
|
|
return total_texels;
|
|
}
|
|
|
|
int find_slice(uint32_t block_index, uint32_t *pBlock_x, uint32_t *pBlock_y) const
|
|
{
|
|
for (uint32_t i = 0; i < m_slices.size(); i++)
|
|
{
|
|
if ((block_index >= m_slices[i].m_first_block_index) && (block_index < (m_slices[i].m_first_block_index + m_slices[i].m_num_blocks_x * m_slices[i].m_num_blocks_y)))
|
|
{
|
|
const uint32_t ofs = block_index - m_slices[i].m_first_block_index;
|
|
const uint32_t x = ofs % m_slices[i].m_num_blocks_x;
|
|
const uint32_t y = ofs / m_slices[i].m_num_blocks_x;
|
|
|
|
if (pBlock_x) *pBlock_x = x;
|
|
if (pBlock_y) *pBlock_y = y;
|
|
|
|
return i;
|
|
}
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
void create_endpoint_palette();
|
|
|
|
void create_selector_palette();
|
|
|
|
// endpoint palette
|
|
// 5:5:5 and predicted 4:4:4 colors, 1 or 2 3-bit intensity table indices
|
|
// selector palette
|
|
// 4x4 2-bit selectors
|
|
|
|
// per-macroblock:
|
|
// 4 diff bits
|
|
// 4 flip bits
|
|
// Endpoint template index, 1-8 endpoint indices
|
|
// Alternately, if no template applies, we can send 4 ETC1S bits followed by 4-8 endpoint indices
|
|
// 4 selector indices
|
|
|
|
void reoptimize_and_sort_endpoints_codebook(uint32_t total_block_endpoints_remapped, uint_vec &all_endpoint_indices);
|
|
void sort_selector_codebook();
|
|
void create_encoder_blocks();
|
|
void compute_slice_crcs();
|
|
bool encode_image();
|
|
bool encode_endpoint_palette();
|
|
bool encode_selector_palette();
|
|
int find_video_frame(int slice_index, int delta);
|
|
void check_for_valid_cr_blocks();
|
|
};
|
|
|
|
} // namespace basisu
|
|
|