123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572 |
- // SPDX-License-Identifier: LGPL-2.1 OR BSD-3-Clause
- /*
- * Copyright (c) 2019, Chips&Media
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions are met:
- *
- * 1. Redistributions of source code must retain the above copyright notice, this
- * list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright notice,
- * this list of conditions and the following disclaimer in the documentation
- * and/or other materials provided with the distribution.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
- * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
- * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
- * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
- * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
- * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
- #include <stdio.h>
- #include <stdint.h>
- #include <stddef.h>
- #include "main_helper.h"
- #include "pbu.h"
- enum
- {
- SPP_IN,
- SPP_OUT_RBSP,
- SPP_OUT_NAL,
- SPP_UNALIGN
- };
- # define NAL_BUF_SIZE 512
- typedef struct
- {
- // stream buffer status
- unsigned char* rd_ptr;
- unsigned char* wr_ptr;
- unsigned char* bb_start;
- unsigned char* bb_end;
- int explicit_end_flag;
- int stream_end_flag;
- // instance handler ... for interrupt handling
- void* handle;
- } gbu_if_t;
- typedef struct
- {
- // for interfacing
- gbu_if_t *param;
- // internal read pointer
- Uint8* rd_ptr;
- // nal buffer things
- Uint8 nal_buf[2][NAL_BUF_SIZE];
- int nal_buf_idx;
- int nal_buf_cnt;
- int nal_ptr;
- Uint32 nal_cnt;
- int last_nal_byte;
- // rbsp buffer things
- Uint32 wbuf[2];
- Int8 wbuf_emul_info;
- Int8 bptr;
- Int8 rbsp_bit_cnt;
- int rbsp_init;
- // rbsp consumed bit count
- int tcnt;
- int tc;
- // trailing zero counter
- int zero_cnt;
- // eos
- int eos;
- Uint32 est_nal_byte_cnt;
- int epbRequired;
- } gbu_t;
- static void enc_flush_rbsp(spp_enc_context context);
- static void enc_flush_nal_buf(spp_enc_context context);
- static void put_bits(spp_enc_context context, Uint32 var, int n);
- static void put_ue(spp_enc_context context, Uint32 var);
- spp_enc_context spp_enc_init(Uint8 *buffer, int buffer_size, int enableEPB)
- {
- gbu_t* gbu;
-
- gbu = (gbu_t*)osal_malloc(sizeof(gbu_t));
- if (!gbu)
- return NULL;
- gbu->param = (gbu_if_t *)osal_malloc(sizeof(gbu_if_t));
- if (!gbu->param) {
- osal_free(gbu);
- return NULL;
- }
- // connect interface structure
- gbu->param->bb_start = buffer;
- gbu->param->bb_end = buffer + buffer_size;
- gbu->param->rd_ptr = buffer;
- gbu->param->wr_ptr = buffer;
-
- // reset write pointer
- gbu->rd_ptr = gbu->param->wr_ptr;
- // reset nal memory index/cnt
- gbu->nal_buf_idx = 0;
- gbu->nal_buf_cnt = 0;
- gbu->nal_ptr = 0;
- gbu->nal_cnt = 0;
- gbu->est_nal_byte_cnt = 0;
- // reset trailing zero count
- gbu->zero_cnt = 0;
- // reset rbsp consumed count
- gbu->bptr = 32;
- gbu->rbsp_bit_cnt = 0;
- gbu->tcnt = 0;
- // reset wbuf
- gbu->wbuf[0] = gbu->wbuf[1] = 0x00;
- gbu->epbRequired = enableEPB;
- return (spp_enc_context)gbu;
-
- }
- void spp_enc_deinit(spp_enc_context context)
- {
- gbu_t* gbu = (gbu_t *)context;
- if (gbu) {
- if (gbu->param) {
- osal_free(gbu->param);
- }
- osal_free(gbu);
- }
- }
- void spp_enc_init_rbsp(spp_enc_context context)
- {
- gbu_t* gbu = (gbu_t *)context;
- gbu->tcnt = 0;
- // reset trailing zero count
- gbu->zero_cnt = 0;
- // reset rbsp consumed count
- gbu->bptr = 32;
- // reset wbuf
- gbu->wbuf[0] = gbu->wbuf[1] = 0x00;
- }
- void spp_enc_put_nal_byte(spp_enc_context context, Uint32 var, int n)
- {
- int i;
- gbu_t* gbu = (gbu_t *)context;
- for (i = n-1 ; i >= 0 ; i--)
- {
- gbu->nal_buf[gbu->nal_buf_idx][gbu->nal_ptr] = ((var >> (i<<3)) & 0xFF);
- gbu->nal_ptr++;
- gbu->nal_cnt++;
- gbu->est_nal_byte_cnt++;
- if (gbu->nal_ptr == NAL_BUF_SIZE)
- {
- enc_flush_nal_buf(context);
- gbu->nal_buf_idx = (gbu->nal_buf_idx+1)&1;
- gbu->nal_ptr = 0;
- }
- }
- }
- void spp_enc_put_bits(spp_enc_context context, Uint32 var, int n)
- {
- Uint32 mask;
- Uint32 data;
- gbu_t* gbu = (gbu_t *)context;
- if (n == 0)
- return;
- // update total rbsp count
- gbu->tcnt += n;
- if (n > gbu->bptr)
- {
- // write data
- mask = (1 << gbu->bptr) - 1;
- data = (var >> (n - gbu->bptr)) & mask;
- gbu->wbuf[0] |= data;
- gbu->rbsp_bit_cnt += gbu->bptr;
- // update input data
- n -= gbu->bptr;
- // update gbu bptr
- gbu->bptr = 0;
- // write rbsp data to nbuf
- enc_flush_rbsp(context);
- }
- mask = ((Uint64)1 << n) - 1;
- data = var & mask;
- // write data
- gbu->wbuf[0] |= data << (gbu->bptr - n);
- gbu->rbsp_bit_cnt += n;
- // update bptr
- gbu->bptr -= n;
- if (gbu->bptr == 0)
- enc_flush_rbsp(context);
- }
- void spp_enc_flush(spp_enc_context context)
- {
- enc_flush_rbsp(context);
-
- enc_flush_nal_buf(context);
- }
- void enc_flush_rbsp(spp_enc_context context)
- {
- gbu_t* gbu = (gbu_t *)context;
-
- int cnt = (32 - gbu->bptr + 7) >> 3;
- int pos = 4 - cnt;
- Uint8 data;
- gbu->wbuf[1] = gbu->wbuf[0];
- for (cnt = cnt-1 ; cnt >= 0 ; cnt--)
- {
- data = (gbu->wbuf[0] >> ((cnt+pos)<<3)) & 0xFF;
-
- if (gbu->epbRequired == 1)
- {
- // insert EPB if needed
- if ((gbu->zero_cnt == 2) && (data <= 0x03))
- {
- spp_enc_put_nal_byte(context, 0x03, 1);
- gbu->zero_cnt = 0;
- }
- }
- spp_enc_put_nal_byte(context, data, 1);
- // update number of trailing zeroes
- if (data == 0x00)
- gbu->zero_cnt++;
- else
- gbu->zero_cnt = 0;
- }
- // reset bit ptr
- gbu->bptr = 32;
- // reset wbuf
- gbu->wbuf[0] = 0;
- }
- static void enc_flush_nal_buf(spp_enc_context context)
- {
- int i, cnt;
- int left, room;
- Uint8* ptr;
- Uint8* align_wr_ptr;
- gbu_t* gbu = (gbu_t *)context;
- gbu_if_t* io = gbu->param;
- ptr = (Uint8 *)io->wr_ptr;
- align_wr_ptr = (Uint8*)(ptr);
- for (cnt = 0 ; cnt < gbu->nal_ptr ; )
- {
- room = left = io->rd_ptr - io->wr_ptr;
- // wraparound case
- if (left <= 0)
- {
- left = io->bb_end - io->wr_ptr;
- room += io->bb_end - io->bb_start;
- }
- // write stream to CPB
- for (i = 0 ; i < left && cnt < gbu->nal_ptr ; i++, cnt++)
- *(align_wr_ptr + i) = gbu->nal_buf[gbu->nal_buf_idx][cnt];
- gbu->param->wr_ptr = align_wr_ptr + i;
- // handle wraparound case
- if (io->wr_ptr == io->bb_end)
- io->wr_ptr = io->bb_start;
- // if room is small enough, then flush data to external memory
- if ((room - i) < 256)
- {
- //host_isr(gbu->param, INT_BIT_BIT_BUF_FULL, NULL);
- continue;
- }
- }
- // clear nal_ptr
- gbu->nal_ptr = 0;
- }
- // put unsigned exp-golomb code
- void spp_enc_put_ue(spp_enc_context context, Uint32 var)
- {
- Int32 num;
- Uint32 data;
-
- num = -1;
- data = var + 1;
- while (data)
- {
- data = data >> 1;
- num++;
- }
- // leading zero bits
- put_bits(context, 0, num);
- put_bits(context, var+1, num+1);
- }
- Uint32 spp_enc_get_ue_bit_size(Uint32 var)
- {
- Uint32 size;
- Int32 num;
- Uint32 data;
- num = -1;
- data = var + 1;
- while (data)
- {
- data = data >> 1;
- num++;
- }
- // leading zero bits
- size = num;
- size += (num+1);
- return size;
- }
- // put signed exp-golomb code
- void spp_enc_put_se(spp_enc_context context, Int32 var)
- {
- Uint32 data;
-
- // (-1)^(code_num + 1)
- if (var > 0)
- data = (( var)<<1) - 1;
- else
- data = ((-var)<<1);
- put_ue(context, data);
- }
- // put stop one bit & padding zeroes
- void spp_enc_put_byte_align(spp_enc_context context, int has_stop_bit)
- {
- gbu_t* gbu = (gbu_t *)context;
- // stop one bit
- if (has_stop_bit)
- spp_enc_put_bits(context, 1, 1);
- // padding zeroes
- if (gbu->bptr & 7)
- spp_enc_put_bits(context, 0, gbu->bptr & 7);
- }
- // number of remain bit in wbuf
- Uint32 spp_enc_get_wbuf_remain(spp_enc_context context)
- {
- gbu_t* gbu = (gbu_t *)context;
- return (Uint32)gbu->bptr;
- }
- // number of total bit after gbu initialization
- Uint32 spp_enc_get_rbsp_bit(spp_enc_context context)
- {
- gbu_t* gbu = (gbu_t *)context;
- return (Uint32)gbu->tcnt;
- }
- // number of nal byte after gbu initialization
- Uint32 spp_enc_get_nal_cnt(spp_enc_context context)
- {
- gbu_t* gbu = (gbu_t *)context;
- return (Uint32)gbu->nal_cnt;
- }
- // return wr_ptr of stream buffer
- Uint8* spp_enc_get_wr_ptr(spp_enc_context context)
- {
- gbu_t* gbu = (gbu_t *)context;
- enc_flush_rbsp(context);
- enc_flush_nal_buf(context);
- return gbu->param->wr_ptr;
- }
- Uint8* spp_enc_get_wr_ptr_only(spp_enc_context context)
- {
- gbu_t* gbu = (gbu_t *)context;
- enc_flush_rbsp(context);
- return gbu->param->wr_ptr + gbu->nal_ptr;
- }
- void spp_enc_set_wr_ptr(spp_enc_context context, Uint32 wr_ptr)
- {
- gbu_t* gbu = (gbu_t *)context;
- gbu_if_t* io = gbu->param;
- intptr_t wr_int_ptr = wr_ptr;
- intptr_t bb_end_int_ptr = (intptr_t)io->bb_end;
- intptr_t gbu_param_bb_end_int_ptr = (intptr_t)gbu->param->bb_end;
- intptr_t gbu_param_bb_start_int_ptr = (intptr_t)gbu->param->bb_start;
-
- while(wr_ptr > bb_end_int_ptr)
- wr_int_ptr = wr_int_ptr + gbu_param_bb_end_int_ptr - gbu_param_bb_start_int_ptr;
- io->wr_ptr = (Uint8 *)wr_int_ptr;
- }
- // get a estimated NAL count in byte
- Uint32 spp_enc_get_est_nal_cnt(spp_enc_context context)
- {
- gbu_t* gbu = (gbu_t *)context;
- Uint32 est_nal_byte = 0;
- Uint8 data;
- Int32 cnt = (32 - gbu->bptr ) >> 3;
- Uint32 pos = 4 - cnt;
- Uint32 zero_cnt = gbu->zero_cnt;
- for (cnt = cnt-1 ; cnt >= 0 ; cnt--)
- {
- data = (gbu->wbuf[0] >> ((cnt+pos)<<3)) & 0xFF;
- // insert EPB if needed
- if ((zero_cnt == 2) && (data <= 0x03))
- {
- est_nal_byte++;
- zero_cnt = 0;
- }
- est_nal_byte++;
- // update number of trailing zeroes
- if (data == 0x00)
- zero_cnt++;
- else
- zero_cnt = 0;
- }
- return gbu->est_nal_byte_cnt + est_nal_byte;
- }
- static void put_bits(spp_enc_context context, Uint32 var, int n)
- {
- Uint32 mask;
- Uint32 data;
- gbu_t* gbu = (gbu_t *)context;
- if (n == 0)
- return;
- // update total rbsp count
- gbu->tcnt += n;
- if (n > gbu->bptr)
- {
- // write data
- mask = (1 << gbu->bptr) - 1;
- data = (var >> (n - gbu->bptr)) & mask;
- gbu->wbuf[0] |= data;
- gbu->rbsp_bit_cnt += gbu->bptr;
- // update input data
- n -= gbu->bptr;
- // update gbu bptr
- gbu->bptr = 0;
- // write rbsp data to nbuf
- enc_flush_rbsp(context);
- }
- mask = (1 << n) - 1;
- data = var & mask;
- // write data
- gbu->wbuf[0] |= data << (gbu->bptr - n);
- gbu->rbsp_bit_cnt += n;
- // update bptr
- gbu->bptr -= n;
- if (gbu->bptr == 0)
- enc_flush_rbsp(context);
- }
- static void put_ue(spp_enc_context context, Uint32 var)
- {
- Int32 num;
- Uint32 data;
- num = -1;
- data = var + 1;
- while (data)
- {
- data = data >> 1;
- num++;
- }
- // leading zero bits
- put_bits(context, 0, num);
- put_bits(context, var+1, num+1);
- }
-
|