// Licensed to the Apache Software Foundation (ASF) under one

// or more contributor license agreements.  See the NOTICE file

// distributed with this work for additional information

// regarding copyright ownership.  The ASF licenses this file

// to you 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.

use crate::data::{contains_nulls, ArrayData};

use arrow_buffer::ArrowNativeType;

use super::equal_range;

pub(super) fn dictionary_equal<T: ArrowNativeType>(

    lhs: &ArrayData,

    rhs: &ArrayData,

    lhs_start: usize,

    rhs_start: usize,

    len: usize,

) -> bool {

    let lhs_keys = lhs.buffer::<T>(0);

    let rhs_keys = rhs.buffer::<T>(0);

    let lhs_values = &lhs.child_data()[0];

    let rhs_values = &rhs.child_data()[0];

    // Only checking one null mask here because by the time the control flow reaches

    // this point, the equality of the two masks would have already been verified.

    if !contains_nulls(lhs.nulls(), lhs_start, len) {

        
(0..len)4
.
all4
(|i| {

            let lhs_pos = lhs_start + i;

            let rhs_pos = rhs_start + i;

            equal_range(

                lhs_values,

                rhs_values,

                lhs_keys[lhs_pos].to_usize().unwrap(),

                rhs_keys[rhs_pos].to_usize().unwrap(),

                1,

            )

        })

    } else {

        // get a ref of the null buffer bytes, to use in testing for nullness

        let lhs_nulls = lhs.nulls().unwrap();

        let rhs_nulls = rhs.nulls().unwrap();

        
(0..len)2
.
all2
(|i| {

            let lhs_pos = lhs_start + i;

            let rhs_pos = rhs_start + i;

            let lhs_is_null = lhs_nulls.is_null(lhs_pos);

            let rhs_is_null = rhs_nulls.is_null(rhs_pos);

            lhs_is_null

                || (lhs_is_null == rhs_is_null)

                    && equal_range(

                        lhs_values,

                        rhs_values,

                        lhs_keys[lhs_pos].to_usize().unwrap(),

                        rhs_keys[rhs_pos].to_usize().unwrap(),

                        1,

                    )

        })

    }

}