import ecto

from opendm import log
from opendm import io
from opendm import system
from opendm import context

class ODMTexturingCell(ecto.Cell):  

    def declare_params(self, params):
        params.declare("resize", 'resizes images by the largest side', 2400)
        params.declare("resolution", 'The resolution of the output textures. Must be '
                            'greater than textureWithSize.', 4096)
        params.declare("size", 'The resolution to rescale the images performing '
                            'the texturing.', 3600)

    def declare_io(self, params, inputs, outputs):
        inputs.declare("tree", "Struct with paths", [])
        inputs.declare("args", "The application arguments.", {})
        inputs.declare("reconstruction", "Clusters output. list of ODMReconstructions", [])
        outputs.declare("reconstruction", "Clusters output. list of ODMReconstructions", [])

    def process(self, inputs, outputs):
        
        log.ODM_INFO('Running OMD Texturing Cell')

        # get inputs
        args = self.inputs.args
        tree = self.inputs.tree

        # define paths and create working directories
        system.mkdir_p(tree.odm_texturing)

        # check if we rerun cell or not
        rerun_cell = args['rerun'] is not None \
            and args['rerun'] == 'odm_texturing'

        if not io.file_exists(tree.odm_textured_model_obj) or rerun_cell:
            log.ODM_DEBUG('Writting ODM Textured file in: %s' \
                % tree.odm_textured_model_obj)

            # odm_texturing definitions
            kwargs = {
                'bin': context.odm_modules_path,
                'out_dir': tree.odm_texturing,
                'bundle': tree.opensfm_bundle,
                'imgs_path': tree.dataset_resize,
                'imgs_list': tree.opensfm_bundle_list,
                'model': tree.odm_mesh,
                'log': tree.odm_texuring_log,
                'resize': self.params.resize,
                'resolution': self.params.resolution,
                'size': self.params.size
            }

            # run texturing binary
            system.run('{bin}/odm_texturing -bundleFile {bundle} '           \
                '-imagesPath {imgs_path} -imagesListPath {imgs_list} '       \
                '-inputModelPath {model} -outputFolder {out_dir}/ '          \
                '-textureResolution {resolution} -bundleResizedTo {resize} ' \
                '-textureWithSize {size} -logFile {log}'.format(**kwargs))
        else:
            log.ODM_WARNING('Found a valid ODM Texture file in: %s' \
                % tree.odm_textured_model_obj)
                
        log.ODM_INFO('Running OMD Texturing Cell - Finished')
        return ecto.OK if args['end_with'] != 'odm_texturing' else ecto.QUIT