gapid/atom/list.go

// Copyright (C) 2016 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package atom

import (
	"android.googlesource.com/platform/tools/gpu/framework/binary"
	"android.googlesource.com/platform/tools/gpu/framework/log"
)

// List is a list of atoms.
type List struct {
	binary.Generate `java:"AtomList"`
	Atoms           []Atom `variant:"true"`
}

func NewList(atoms ...Atom) *List {
	return &List{Atoms: atoms}
}

// WriteTo writes all atoms in the list to w, terminating with a single EOS
// atom.
func (l *List) WriteTo(ctx log.Context, w Writer) {
	for i, a := range l.Atoms {
		w.Write(ctx, ID(i), a)
	}
}

// Clone makes and returns a shallow copy of the atom list.
func (l *List) Clone() *List {
	c := &List{Atoms: make([]Atom, len(l.Atoms))}
	copy(c.Atoms, l.Atoms)
	return c
}

// Add appends a to the end of the atom list, returning the id of the last added
// atom.
func (l *List) Add(a ...Atom) ID {
	l.Atoms = append(l.Atoms, a...)
	return ID(len(l.Atoms) - 1)
}

// Add adds a to the list before the atom at id.
func (l *List) AddAt(a Atom, id ID) {
	l.Atoms = append(l.Atoms, nil)
	copy(l.Atoms[id+1:], l.Atoms[id:])
	l.Atoms[id] = a
}