from ladybug.color import *

try:
    # import Dynamo libraries
    import clr
    clr.AddReference("DSCoreNodes")
    from DSCore import Color as DSColor
except ImportError:
    print "Failed to import Dynamo libraries. Make sure path is added to sys.path"


class ColorConvertor(object):
    @staticmethod
    def toDSColor(ladybugColors):
        """Cast a LBColor into Dynamo color"""

        if hasattr(ladybugColors, "__iter__"):
            for col in ladybugColors:
                assert isinstance(col, LBColor), "%s in not an instance of LBColor"%str(col)
                yield DSColor.ByARGB(255, col.r, col.g, col.b)
        else:
            assert isinstance(ladybugColors, LBColor), "%s in not an instance of LBColor"%str(col)
            yield DSColor.ByARGB(255, ladybugColors.r, ladybugColors.g, ladybugColors.b)

    @staticmethod
    def toLBColor(DSColors):
        if hasattr(DSColors, "__iter__"):
            for col in DSColors:
                assert isinstance(col, DSColor), "%s in not an instance of DSColor"%str(col)
                yield LBColor(col.Red, col.Green, col.Blue)
        else:
            assert isinstance(DSColors, DSColor), "%s in not an instance of DSColor"%str(col)
            yield LBColor(DSColors.Red, DSColors.Green, DSColors.Blue)