# dxfdedup removes duplicate entities from DXF files.
#
# Copyright (C) 2017 ravas (lastauth.com/profile/ravas)
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
#
# https://www.gnu.org/licenses/gpl-3.0.en.html

# Always back-up your files before using this software!!
# Do some tests!!


import importlib
import pathlib
import time

import ezdxf

class Deduplicator(object):

    def __init__(self, entities=None, iders='dxfdedup.identifiers'):
        """
        :entities: list of entity types to deduplicate; None ? dedup all types
        """
        print('Back-up your files before using this object.')
        self.entities = entities
        self.iders = iders
        self.identifiers = []
        self.set_iders(entities, iders)

    def set_iders(self, entities=None, iders=None):
        """
        Import the indetifiers subpackage and its submodules,
        and append an instance of each submodule's
        Indetifier(Object) to self.entities.
        :entities: list of entity types to deduplicate
        :iders: name of the identifiers subpackage
        """

        entities = entities or self.entities
        iders = iders or self.iders
        
        ider_pkg = importlib.import_module(iders)

        names = []
        
        if entities is None:
            names = ider_pkg.__all__
        else:
            names = [n for n in ider_pkg.__all__ if n in entities]

        print(f'The following types will be deduplicated: {names}')

        for name in names:
            i = importlib.import_module(f'{iders}.{name}')
            self.identifiers.append(i.Identifier())

    def dedup_d(self, path):
        """
        Remove duplicates from all dxf files in path.
        :path: a directory path (string)
        """
        
        po = pathlib.Path(path).expanduser()
        
        if po.is_dir():
            start = time.time()
            for p in po.glob('**/*.dxf'):
                print(p)
                self.dedup(p)
            stop = time.time()
            print(f'Finished! Time Elapsed: {stop-start}')
        else:
            print('Invalid Directory!')

    def dedup(self, path):
        """
        Remove duplicates from path.
        :path: a dxf file path (string)
        """
        
        dxf = ezdxf.readfile(path)
        modelspace = dxf.modelspace()
        deleted = 0
        
        for i in self.identifiers:
            entities = modelspace.query(i.e_type)
            seen = set()
            for entity in entities:
                uid = i.unique_id(entity)
                if uid not in seen:
                    seen.add(uid)
                else:
                    deleted += 1
                    modelspace.delete_entity(entity)

        if deleted > 0:
            dxf.saveas(path)          
        print(f'Duplicates deleted: {deleted}')