155 lines
4.9 KiB
Python
155 lines
4.9 KiB
Python
# Copyright (c) 2019-2022, Manfred Moitzi
|
|
# License: MIT-License
|
|
from __future__ import annotations
|
|
from typing import TYPE_CHECKING, Optional
|
|
from ezdxf.lldxf import validator
|
|
from ezdxf.lldxf.const import SUBCLASS_MARKER, DXF2007
|
|
from ezdxf.lldxf.attributes import (
|
|
DXFAttributes,
|
|
DefSubclass,
|
|
DXFAttr,
|
|
RETURN_DEFAULT,
|
|
group_code_mapping,
|
|
)
|
|
from .dxfentity import base_class, SubclassProcessor
|
|
from .dxfobj import DXFObject
|
|
from .factory import register_entity
|
|
|
|
if TYPE_CHECKING:
|
|
from ezdxf.entities import DXFNamespace
|
|
from ezdxf.lldxf.tagwriter import AbstractTagWriter
|
|
|
|
__all__ = ["Sun"]
|
|
|
|
acdb_sun = DefSubclass(
|
|
"AcDbSun",
|
|
{
|
|
"version": DXFAttr(90, default=1),
|
|
"status": DXFAttr(
|
|
290,
|
|
default=1,
|
|
validator=validator.is_integer_bool,
|
|
fixer=RETURN_DEFAULT,
|
|
),
|
|
"color": DXFAttr(
|
|
63,
|
|
default=7,
|
|
validator=validator.is_valid_aci_color,
|
|
fixer=RETURN_DEFAULT,
|
|
),
|
|
"true_color": DXFAttr(421, default=16777215),
|
|
"intensity": DXFAttr(40, default=1),
|
|
"shadows": DXFAttr(
|
|
291,
|
|
default=1,
|
|
validator=validator.is_integer_bool,
|
|
fixer=RETURN_DEFAULT,
|
|
),
|
|
"julian_day": DXFAttr(91, default=2456922),
|
|
# Time in seconds past midnight:
|
|
"time": DXFAttr(92, default=43200),
|
|
"daylight_savings_time": DXFAttr(
|
|
292,
|
|
default=0,
|
|
validator=validator.is_integer_bool,
|
|
fixer=RETURN_DEFAULT,
|
|
),
|
|
# Shadow type:
|
|
# 0 = Ray traced shadows
|
|
# 1 = Shadow maps
|
|
"shadow_type": DXFAttr(
|
|
70,
|
|
default=0,
|
|
validator=validator.is_integer_bool,
|
|
fixer=RETURN_DEFAULT,
|
|
),
|
|
"shadow_map_size": DXFAttr(71, default=256),
|
|
"shadow_softness": DXFAttr(280, default=1),
|
|
},
|
|
)
|
|
acdb_sun_group_codes = group_code_mapping(acdb_sun)
|
|
|
|
|
|
@register_entity
|
|
class Sun(DXFObject):
|
|
"""DXF SUN entity"""
|
|
|
|
DXFTYPE = "SUN"
|
|
DXFATTRIBS = DXFAttributes(base_class, acdb_sun)
|
|
MIN_DXF_VERSION_FOR_EXPORT = DXF2007
|
|
|
|
def load_dxf_attribs(
|
|
self, processor: Optional[SubclassProcessor] = None
|
|
) -> DXFNamespace:
|
|
dxf = super().load_dxf_attribs(processor)
|
|
if processor:
|
|
processor.fast_load_dxfattribs(dxf, acdb_sun_group_codes, 1)
|
|
return dxf
|
|
|
|
def export_entity(self, tagwriter: AbstractTagWriter) -> None:
|
|
"""Export entity specific data as DXF tags."""
|
|
super().export_entity(tagwriter)
|
|
tagwriter.write_tag2(SUBCLASS_MARKER, acdb_sun.name)
|
|
self.dxf.export_dxf_attribs(
|
|
tagwriter,
|
|
[
|
|
"version",
|
|
"status",
|
|
"color",
|
|
"true_color",
|
|
"intensity",
|
|
"shadows",
|
|
"julian_day",
|
|
"time",
|
|
"daylight_savings_time",
|
|
"shadow_type",
|
|
"shadow_map_size",
|
|
"shadow_softness",
|
|
],
|
|
)
|
|
|
|
|
|
# todo: implement SUNSTUDY?
|
|
acdb_sunstudy = DefSubclass(
|
|
"AcDbSun",
|
|
{
|
|
"version": DXFAttr(90),
|
|
"name": DXFAttr(1),
|
|
"description": DXFAttr(2),
|
|
"output_type": DXFAttr(70),
|
|
"sheet_set_name": DXFAttr(
|
|
3
|
|
), # Included only if Output type is Sheet Set.
|
|
"use_subset": DXFAttr(
|
|
290
|
|
), # Included only if Output type is Sheet Set.
|
|
"sheet_subset_name": DXFAttr(4),
|
|
# Included only if Output type is Sheet Set.
|
|
"dates_from_calender": DXFAttr(291),
|
|
"date_input_array_size": DXFAttr(91),
|
|
# represents the number of dates picked
|
|
# 90 Julian day; represents the date. One entry for each date picked.
|
|
# 90 Seconds past midnight; represents the time of day. One entry for each date picked.
|
|
"range_of_dates": DXFAttr(292),
|
|
# 93 Start time. If range of dates flag is true.
|
|
# 94 End time. If range of dates flag is true.
|
|
# 95 Interval in seconds. If range of dates flag is true.
|
|
"hours_count": DXFAttr(73),
|
|
# 290 Hour. One entry for every hour as specified by the number of hours entry above.
|
|
"page_setup_wizard_handle": DXFAttr(
|
|
340
|
|
), # Page setup wizard hard pointer ID
|
|
"view_handle": DXFAttr(341), # View hard pointer ID
|
|
"visual_style_handle": DXFAttr(342), # Visual Style ID
|
|
"shade_plot_type": DXFAttr(74),
|
|
"viewports_per_page": DXFAttr(75),
|
|
"row_count": DXFAttr(76), # Number of rows for viewport distribution
|
|
"column_count": DXFAttr(77), # Number of columns for viewport distribution
|
|
"spacing": DXFAttr(40),
|
|
"lock_viewports": DXFAttr(293),
|
|
"label_viewports": DXFAttr(294),
|
|
"text_style_handle": DXFAttr(343),
|
|
},
|
|
)
|
|
acdb_sunstudy_group_codes = group_code_mapping(acdb_sunstudy)
|