test-prefire.py

from pathlib import Path


import logging
from harmony.message import Format, Message, Source, ScaleExtent

from harmony_browse_image_generator.adapter import BrowseImageGeneratorAdapter
from harmony_browse_image_generator.browse import create_browse_imagery

# Set up the logger
loglevel = logging.INFO
logger = logging.getLogger()
logger.setLevel(loglevel)

# Set up the console handler with a level of DEBUG
ch = logging.StreamHandler()
ch.setLevel(loglevel)

# Add the handler to the logger
logger.addHandler(ch)


def main():
    ##  CHANGE THIS TO WHERE YOU UNTARRED THE DATA
    data_dir = Path('/Users/savoie/projects/data-services/hybig/prefire/')

    # prefire
    testtif = data_dir / 'PREFIRE_SAT2_2B-FLX_S07_R00_20210721044449_03042.nc.G00.tif'
    scale_extent = ScaleExtent({})

    tiffs = [testtif]
    for mime_type in ['image/png']:
        message = Message({})
        source = Source({})
        message.format = Format({'mime': mime_type})
        message.format.scaleExtent = scale_extent
        adapter = BrowseImageGeneratorAdapter(message)
        for tiff in tiffs:
            file_list = create_browse_imagery(
                adapter.message, tiff, source, None, logger
            )
            print(f'completed: \n{file_list}')


if __name__ == '__main__':
    main()