You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
3.0 KiB
115 lines
3.0 KiB
/* |
|
* Copyright (C) 2015 The Android Open Source Project |
|
* |
|
* 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. |
|
*/ |
|
|
|
#ifndef __FEC_H__ |
|
#define __FEC_H__ |
|
|
|
#include <utils/Compat.h> |
|
#include <string> |
|
#include <vector> |
|
#include <fec/io.h> |
|
#include <fec/ecc.h> |
|
|
|
#define IMAGE_MIN_THREADS 1 |
|
#define IMAGE_MAX_THREADS 128 |
|
|
|
#define INFO(x...) \ |
|
fprintf(stderr, x); |
|
#define FATAL(x...) { \ |
|
fprintf(stderr, x); \ |
|
exit(1); \ |
|
} |
|
|
|
#define unlikely(x) __builtin_expect(!!(x), 0) |
|
|
|
struct image { |
|
/* if true, decode file in place instead of creating a new output file */ |
|
bool inplace; |
|
/* if true, assume input is a sparse file */ |
|
bool sparse; |
|
/* if true, print more verbose information to stderr */ |
|
bool verbose; |
|
const char *fec_filename; |
|
int fec_fd; |
|
int inp_fd; |
|
/* the number of Reed-Solomon generator polynomial roots, also the number |
|
of parity bytes generated for each N bytes in RS(M, N) */ |
|
int roots; |
|
/* for RS(M, N), N = M - roots */ |
|
int rs_n; |
|
int threads; |
|
uint32_t fec_size; |
|
uint32_t padding; |
|
uint64_t blocks; |
|
uint64_t inp_size; |
|
uint64_t pos; |
|
uint64_t rounds; |
|
uint64_t rv; |
|
uint8_t *fec; |
|
uint8_t *input; |
|
uint8_t *output; |
|
}; |
|
|
|
struct image_proc_ctx; |
|
typedef void (*image_proc_func)(image_proc_ctx *); |
|
|
|
struct image_proc_ctx { |
|
image_proc_func func; |
|
int id; |
|
image *ctx; |
|
uint64_t rv; |
|
uint64_t fec_pos; |
|
uint64_t start; |
|
uint64_t end; |
|
void *rs; |
|
}; |
|
|
|
extern bool image_load(const std::vector<std::string>& filename, image *ctx); |
|
extern bool image_save(const std::string& filename, image *ctx); |
|
|
|
extern bool image_ecc_new(const std::string& filename, image *ctx); |
|
extern bool image_ecc_load(const std::string& filename, image *ctx); |
|
extern bool image_ecc_save(image *ctx); |
|
|
|
extern bool image_process(image_proc_func f, image *ctx); |
|
|
|
extern void image_init(image *ctx); |
|
extern void image_free(image *ctx); |
|
|
|
inline uint8_t image_get_interleaved_byte(uint64_t i, image *ctx) |
|
{ |
|
uint64_t offset = fec_ecc_interleave(i, ctx->rs_n, ctx->rounds); |
|
|
|
if (unlikely(offset >= ctx->inp_size)) { |
|
return 0; |
|
} |
|
|
|
return ctx->input[offset]; |
|
} |
|
|
|
inline void image_set_interleaved_byte(uint64_t i, image *ctx, |
|
uint8_t value) |
|
{ |
|
uint64_t offset = fec_ecc_interleave(i, ctx->rs_n, ctx->rounds); |
|
|
|
if (unlikely(offset >= ctx->inp_size)) { |
|
assert(value == 0); |
|
} else if (ctx->output && ctx->output[offset] != value) { |
|
ctx->output[offset] = value; |
|
} |
|
} |
|
|
|
#endif // __FEC_H__
|
|
|