geometry_viewer.py 文件源码

def __init__(self, mol, style=None, display=False, width='100%', height='400px',
                 **kwargs):
        kwargs.update(width=width, height=height)
        super().__init__(**kwargs)
        self.height = in_pixels(height)
        self.width = in_pixels(width)
        self.atom_colors = {}

        # current state
        self.atom_highlights = []
        self._axis_objects = None
        self._colored_as = {}

        self.add_molecule(mol)
        if style is None:
            self.autostyle()
        else:
            self.set_style(style)

        if display:
            dsp.display(self)