Source File
	object_mapper.go

Belonging Package
	github.com/pancsta/d2/d2layouts/d2dagrelayout

package d2dagrelayout

import (
	"strconv"

	"github.com/pancsta/d2/d2graph"
)

type objectMapper struct {
	objToID map[*d2graph.Object]string
	idToObj map[string]*d2graph.Object
}

func NewObjectMapper() *objectMapper {
	return &objectMapper{
		objToID: make(map[*d2graph.Object]string),
		idToObj: make(map[string]*d2graph.Object),
	}
}

func (c *objectMapper) Register(obj *d2graph.Object) {
	id := strconv.Itoa(len(c.idToObj))
	c.idToObj[id] = obj
	c.objToID[obj] = id
}

func (c *objectMapper) ToID(obj *d2graph.Object) string {
	return c.objToID[obj]
}

func (c *objectMapper) ToObj(id string) *d2graph.Object {
	return c.idToObj[id]
}

The pages are generated with Golds v0.8.7. (GOOS=linux GOARCH=amd64)
Golds is a Go 101 project developed by Tapir Liu.
PR and bug reports are welcome and can be submitted to the issue list.
Please follow @zigo_101 (reachable from the left QR code) to get the latest news of Golds.