DiffVG/pydiffvg/color.py

import pydiffvg
import torch

class LinearGradient:
    def __init__(self,
                 begin = torch.tensor([0.0, 0.0]),
                 end = torch.tensor([0.0, 0.0]),
                 offsets = torch.tensor([0.0]),
                 stop_colors = torch.tensor([0.0, 0.0, 0.0, 0.0])):
        self.begin = begin
        self.end = end
        self.offsets = offsets
        self.stop_colors = stop_colors

class RadialGradient:
    def __init__(self,
                 center = torch.tensor([0.0, 0.0]),
                 radius = torch.tensor([0.0, 0.0]),
                 offsets = torch.tensor([0.0]),
                 stop_colors = torch.tensor([0.0, 0.0, 0.0, 0.0])):
        self.center = center
        self.radius = radius
        self.offsets = offsets
        self.stop_colors = stop_colors