RGBLuminanceSource

This commit is contained in:
Henry
2022-08-21 19:39:44 -05:00
parent 48b08e0a9f
commit 95e1b8b460
2 changed files with 168 additions and 136 deletions

View File

@@ -1842,3 +1842,171 @@ impl LuminanceSource for PlanarYUVLuminanceSource{
}
}
/*
* Copyright 2009 ZXing authors
*
* 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.
*/
//package com.google.zxing;
/**
* This class is used to help decode images from files which arrive as RGB data from
* an ARGB pixel array. It does not support rotation.
*
* @author dswitkin@google.com (Daniel Switkin)
* @author Betaminos
*/
pub struct RGBLuminanceSource {
luminances:Vec<u8>,
dataWidth:usize,
dataHeight:usize,
left:usize,
top:usize,
width:usize,
height:usize,
}
impl LuminanceSource for RGBLuminanceSource {
fn new( width:usize, height:usize) -> Self {
let new_ils : Self;
new_ils.width = width;
new_ils.height = height;
new_ils
}
fn getRow(&self, y:usize, row:&Vec<u8>) -> Vec<u8> {
if y < 0 || y >= self.getHeight() {
panic!("Requested row is outside the image: {}" , y);
}
let width = self.getWidth();
let offset = (y + self.top) *self. dataWidth + self.left;
row[0..width].clone_from_slice(&self.luminances[offset..width]);
//System.arraycopy(self.luminances, offset, row, 0, width);
return row;
}
fn getMatrix(&self) -> Vec<u8> {
let width =self. getWidth();
let height = self.getHeight();
// If the caller asks for the entire underlying image, save the copy and give them the
// original data. The docs specifically warn that result.length must be ignored.
if (width == self.dataWidth && height == self.dataHeight) {
return self.luminances;
}
let area = width * height;
let mut matrix = Vec::with_capacity(area);
let mut inputOffset = self.top * self.dataWidth + self.left;
// If the width matches the full width of the underlying data, perform a single copy.
if (width == self.dataWidth) {
matrix[0..area].clone_from_slice(&self.luminances[inputOffset..area]);
//System.arraycopy(self.luminances, inputOffset, matrix, 0, area);
return matrix;
}
// Otherwise copy one cropped row at a time.
for y in 0..height{
//for (int y = 0; y < height; y++) {
let outputOffset = y * width;
matrix[outputOffset..width].clone_from_slice(&self.luminances[inputOffset..width]);
//System.arraycopy(luminances, inputOffset, matrix, outputOffset, width);
inputOffset += self.dataWidth;
}
return matrix;
}
fn getWidth(&self) -> usize {
self.width
}
fn getHeight(&self) -> usize {
self.height
}
fn isCropSupported(&self) -> bool {
return true;
}
fn crop(&self, left:usize, top:usize, width:usize, height:usize) -> Result<dyn LuminanceSource,UnsupportedOperationException> {
match RGBLuminanceSource::new_complex(&self.luminances,
self.dataWidth,
self.dataHeight,
self.left + left,
self.top + top,
width,
height) {
Ok(crop) => Ok(crop),
Err(error) => Err(UnsupportedOperationException::new()),
}
}
}
impl RGBLuminanceSource {
pub fn new_with_width_height_pixels( width:usize, height:usize, pixels:&Vec<usize>) -> Self{
//super(width, height);
let dataWidth = width;
let dataHeight = height;
let left = 0;
let top = 0;
// In order to measure pure decoding speed, we convert the entire image to a greyscale array
// up front, which is the same as the Y channel of the YUVLuminanceSource in the real app.
//
// Total number of pixels suffices, can ignore shape
let size = width * height;
let mut luminances:Vec<u8> = Vec::with_capacity(size);
for offset in 0..size{
//for (int offset = 0; offset < size; offset++) {
let pixel = pixels[offset];
let r = (pixel >> 16) & 0xff; // red
let g2 = (pixel >> 7) & 0x1fe; // 2 * green
let b = pixel & 0xff; // blue
// Calculate green-favouring average cheaply
luminances[offset] = ((r + g2 + b) / 4);
}
Self { luminances, dataWidth, dataHeight, left: left, top: top, width, height }
}
fn new_complex( pixels:&Vec<u8>,
dataWidth:usize,
dataHeight:usize,
left:usize,
top:usize,
width:usize,
height:usize) -> Result<Self,IllegalArgumentException>{
if (left + width > dataWidth || top + height > dataHeight) {
return Err(IllegalArgumentException::new("Crop rectangle does not fit within image data."));
}
Ok(Self{
luminances: todo!(),
dataWidth,
dataHeight,
left,
top,
width,
height,
})
}
}