# Purpose: open example files with big polyface models
# Created: 23.04.2014
# Copyright (c) 2014, Manfred Moitzi
# License: MIT License
from __future__ import unicode_literals
import ezdxf
import time


def count_meshes(entities):
    polyface_count = 0
    polymesh_count = 0
    for entity in entities:
        if entity.get_mode() == 'AcDbPolyFaceMesh':
            polyface_count += 1
        elif entity.get_mode() == 'AcDbPolygonMesh':
            polymesh_count += 1
    return polyface_count, polymesh_count


def print_stats(filename):
    print('opening DXF file: {}'.format(filename))
    start_time = time.time()
    dwg = ezdxf.readfile(filename)
    end_time = time.time()
    print('time for reading: {:.1f} seconds'.format(end_time - start_time))
    print("DXF version: {}".format(dwg.dxfversion))
    print("Database contains {} entities.".format(len(dwg.entitydb)))
    polylines = dwg.entities.query('POLYLINE')
    polyface_count, polymesh_count = count_meshes(polylines)
    print("PolyFaceMeshes: {}".format(polyface_count))
    print("PolygonMeshes {}".format(polymesh_count))


if __name__ == '__main__':
    print_stats(r'D:\Source\ezdxf-dev\integration_tests\polyface_AC1015.dxf')
    print_stats(r'D:\Source\ezdxf-dev\integration_tests\polymesh_AC1015.dxf')
    print_stats(r'D:\Source\dxftest\fanuc-430-arm.dxf')
    print_stats(r'D:\Source\dxftest\cnc machine.dxf')