1 """

     2     High-level DMX output.

     3 """

     4 

     5 class Head (object) :

     6     CHANNELS = None

     7 

     8     def __init__ (self, channel) :

     9         self.channel = channel

    10         self.channels = self.init()

    11         self.attrs = { }

    12 

    13     def init (self) :

    14         if self.CHANNELS :

    15             return self.CHANNELS

    16         else :

    17             raise NotImplementedError()

    18 

    19     def __setitem__ (self, attr, value) :

    20         if attr in self.channels :

    21             self.attrs[attr] = value

    22         else :

    23             raise KeyError(attr)

    24 

    25     def __getitem__ (self, attr) :

    26         if attr in self.channels :

    27             return self.attrs.get(attr, 0)

    28         else :

    29             raise KeyError(attr)

    30  

    31     def alpha (self, alpha=None) :

    32         if alpha is not None :

    33             self['alpha'] = alpha

    34         

    35         if 'alpha' in self.channels :

    36             return dict(alpha=self['alpha'])

    37         else :

    38             return None

    39 

    40     def color (self, red=None, green=None, blue=None) :

    41         if red is not None :

    42             self['red'] = red

    43         if green is not None :

    44             self['green'] = green

    45         if blue is not None :

    46             self['blue'] = blue

    47         

    48         if 'red' in self.channels and 'green' in self.channels and 'blue' in self.channels :

    49             return dict(red=self['red'], green=self['green'], blue=self['blue'])

    50         else :

    51             return None

    52 

    53     def __iter__ (self) :

    54         for attr in self.channels :

    55             yield self.attrs.get(attr, 0)

    56 

    57     def __call__ (self, dmx) :

    58         dmx[self.channel] = tuple(self)

    59 

    60 class Stairville_LEDPar56 (Head) :

    61     CHANNELS = [

    62         'control1',

    63         'red',

    64         'green',

    65         'blue',

    66         'control2',

    67     ]

    68 

    69 class Dimmer (Head) :

    70     CHANNELS = [

    71         'alpha',

    72     ]

    73 

    74 class AmericanDJ_MegaTri60_Mode2 (Head) :

    75     CHANNELS = [

    76         'red',

    77         'green',

    78         'blue',

    79         'control',

    80         'alpha',

    81     ]

    82