v30

Documentation

Index

• func Merge(ctx context.Context, cfg MergeConfig) error
• func ValidateOutputIntegrity(outputRoot string, features map[string]meta.FeatureSpec) error
• type MergeConfig
• type StagingConfig
• type StagingWriter
  • func NewStagingWriter(cfg StagingConfig) (*StagingWriter, error)
  • func (w *StagingWriter) AddFrame(ctx context.Context, frame map[string]any) error
  • func (w *StagingWriter) AppendRGBVideoFrame(ctx context.Context, key string, frame video.VideoFrameRGB24) error
  • func (w *StagingWriter) Close() error
  • func (w *StagingWriter) SaveEpisode(ctx context.Context) (manifest.Episode, error)
  • func (w *StagingWriter) SetH264Remux(ctx context.Context, tracks map[string][][]byte) error

Functions

func Merge

func Merge(ctx context.Context, cfg MergeConfig) error

func ValidateOutputIntegrity

func ValidateOutputIntegrity(outputRoot string, features map[string]meta.FeatureSpec) error

ValidateOutputIntegrity checks merged dataset has data parquet and video files.

Types

type MergeConfig

type MergeConfig struct {
	StagingRoot     string
	OutputRoot      string
	RobotType       string
	FPS             int
	Features        map[string]meta.FeatureSpec
	Locator         video.Locator
	Stats           stats.Options
	MaxWorkers      int
	DataFileSizeMB  int
	VideoFileSizeMB int
}

type StagingConfig

type StagingConfig struct {
	Dir       string
	Episode   int
	FPS       int
	Features  map[string]meta.FeatureSpec
	Locator   video.Locator
	VCodec    string
	CRF       int
	UseVideos bool
	Streaming bool
	Stats     stats.Options
	TempRoot  string
	H264Remux bool
}

type StagingWriter

type StagingWriter struct {
	// contains filtered or unexported fields
}

func NewStagingWriter

func NewStagingWriter(cfg StagingConfig) (*StagingWriter, error)

func (*StagingWriter) AddFrame

func (w *StagingWriter) AddFrame(ctx context.Context, frame map[string]any) error

func (*StagingWriter) AppendRGBVideoFrame

func (w *StagingWriter) AppendRGBVideoFrame(ctx context.Context, key string, frame video.VideoFrameRGB24) error

AppendRGBVideoFrame writes one RGB frame into a lazily-created encoder (decode fallback path).

func (*StagingWriter) Close

func (w *StagingWriter) Close() error

func (*StagingWriter) SaveEpisode

func (w *StagingWriter) SaveEpisode(ctx context.Context) (manifest.Episode, error)

func (*StagingWriter) SetH264Remux

func (w *StagingWriter) SetH264Remux(ctx context.Context, tracks map[string][][]byte) error