Function get_encoder_params

Test/astc_test_image.py:289–333

The the encoder and image set parameters for a test run. Args: encoder_name: The encoder name. reference_name: The reference encoder name. test_set_name: The test image set. Return: The test parameters for the requested encoder and test set. An instance

(encoder_name: str, reference_name: str,
                       test_set_name: str)

def get_encoder_params(encoder_name: str, reference_name: str,
                       test_set_name: str) \
                       -> tuple[te.EncoderBase, str, str, str]:
    '''
    The the encoder and image set parameters for a test run.

    Args:
        encoder_name: The encoder name.
        reference_name: The reference encoder name.
        test_set_name: The test image set.

    Return:
        The test parameters for the requested encoder and test set. An instance
        of the encoder wrapper class, the output data name, the output result
        directory, and the reference to use.
    '''
    used_reference_name: Optional[str] = None
    encoder: te.EncoderBase

    if encoder_name.startswith('ref'):
        _, version, simd = encoder_name.split('-')

        # Version-specific variants
        compatible_prefixes = ['5.']
        if any(True for x in compatible_prefixes if version.startswith(x)):
            encoder = te.Encoder2xRel(version, simd)
            name = f'reference-{version}-{simd}'
            out_dir = f'Test/Images/{test_set_name}'

            return (encoder, name, out_dir, reference_name)

        # Latest main
        if version == 'main':
            encoder = te.Encoder2x(simd)
            name = f'reference-{version}-{simd}'
            out_dir = f'Test/Images/{test_set_name}'
            return (encoder, name, out_dir, reference_name)

        assert False, f'Encoder {encoder_name} not recognized'

    encoder = te.Encoder2x(encoder_name)
    name = f'develop-{encoder_name}'
    out_dir = f'TestOutput/{test_set_name}'
    used_reference_name = reference_name.replace('ref', 'reference')
    return (encoder, name, out_dir, used_reference_name)


def parse_command_line():