cropcraft.py

#!/usr/bin/env python3

# Copyright 2024 INRAE, French National Research Institute for Agriculture, Food and Environment
# 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
# 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.

import click
import os
import subprocess


@click.command(
    help='Generate a 3D model of an agriculutral field described using a YAML config file.'
)
@click.argument('config_file', type=click.Path(exists=True, readable=True))
@click.option(
    '-d',
    '--output-dir',
    default='.',
    type=click.Path(writable=True),
    help='Set the destination path used by the configuration outputs.',
)
@click.option('-f', '--foreground', is_flag=True, help='Open blender in foreground.')
def main(config_file, output_dir, foreground):
    project_path = os.path.dirname(os.path.realpath(__file__))
    entrypoint_path = os.path.join(project_path, 'core', 'blender_entrypoint.py')
    config_path = os.path.realpath(config_file)
    output_path = os.path.realpath(output_dir)

    os.chdir(project_path)

    blender_cmd = ['blender']
    if not foreground:
        blender_cmd.append('--background')
    blender_cmd += [
        '--python-use-system-env',
        '--python',
        entrypoint_path,
        '--',
        config_path,
        output_path,
    ]

    subprocess.run(blender_cmd)


if __name__ == '__main__':
    main()