gpu.go RawHistoryFolder

1package system
2
3import "strings"
4
5// GPUVendor represents the GPU vendor.
6type GPUVendor string
7
8const (
9	GPUVendorAMD     GPUVendor = "amd"
10	GPUVendorIntel   GPUVendor = "intel"
11	GPUVendorNVIDIA  GPUVendor = "nvidia"
12	GPUVendorUnknown GPUVendor = "unknown"
13)
14
15// String returns human-readable vendor name.
16func (v GPUVendor) String() string {
17	switch v {
18	case GPUVendorAMD:
19		return "AMD"
20	case GPUVendorIntel:
21		return "Intel"
22	case GPUVendorNVIDIA:
23		return "NVIDIA"
24	default:
25		return "Unknown"
26	}
27}
28
29// GPU represents a detected GPU and its recommended configuration.
30type GPU struct {
31	vendor  GPUVendor
32	model   string
33	drivers []string
34	envVars map[string]string
35}
36
37// NewGPU creates a new GPU value object.
38func NewGPU(vendor GPUVendor, model string, drivers []string, envVars map[string]string) *GPU {
39	if strings.TrimSpace(model) == "" {
40		model = "Unknown"
41	}
42
43	if drivers == nil {
44		drivers = []string{}
45	}
46
47	if envVars == nil {
48		envVars = map[string]string{}
49	}
50
51	return &GPU{
52		vendor:  vendor,
53		model:   model,
54		drivers: drivers,
55		envVars: envVars,
56	}
57}
58
59// Vendor returns the GPU vendor.
60func (g *GPU) Vendor() GPUVendor {
61	return g.vendor
62}
63
64// Model returns the GPU model name.
65func (g *GPU) Model() string {
66	return g.model
67}
68
69// Drivers returns recommended GPU driver packages.
70func (g *GPU) Drivers() []string {
71	return append([]string{}, g.drivers...)
72}
73
74// EnvVars returns recommended environment variables for compositors.
75func (g *GPU) EnvVars() map[string]string {
76	copyVars := make(map[string]string, len(g.envVars))
77	for key, value := range g.envVars {
78		copyVars[key] = value
79	}
80	return copyVars
81}