materialization

type AggregationConfig struct {
	Strategy       AggregationStrategy    `json:"strategy"`       // Count, Sum, Average, etc.
	Interpretation MetaPathInterpretation `json:"interpretation"` // DirectTraversal, MeetingBased
	Normalization  NormalizationType      `json:"normalization"`  // None, Degree, Max, etc.
	MinWeight      float64                `json:"min_weight"`     // Filter weak edges
	MaxEdges       int                    `json:"max_edges"`      // Keep only top-k edges (0 = no limit)
	Symmetric      bool                   `json:"symmetric"`      // Force symmetric edges
}

type AggregationStats struct {
	EdgeGroupsProcessed  int            `json:"edge_groups_processed"`
	InstancesAggregated  int            `json:"instances_aggregated"`
	EdgesFiltered        int            `json:"edges_filtered"`
	WeightDistribution   map[string]int `json:"weight_distribution"`
	NormalizationApplied bool           `json:"normalization_applied"`
}

type AggregationStrategy int

type EdgeKey struct {
	From string `json:"from"`
	To   string `json:"to"`
}

type GraphStatistics struct {
	NodeCount          int            `json:"node_count"`
	EdgeCount          int            `json:"edge_count"`
	InstanceCount      int            `json:"instance_count"`      // Total meta path instances
	Density            float64        `json:"density"`             // Edge density
	AverageWeight      float64        `json:"average_weight"`      // Average edge weight
	MaxWeight          float64        `json:"max_weight"`          // Maximum edge weight
	MinWeight          float64        `json:"min_weight"`          // Minimum edge weight
	DegreeDistribution map[int]int    `json:"degree_distribution"` // degree -> count
	WeightDistribution map[string]int `json:"weight_distribution"` // weight_bucket -> count
}

type GraphStats struct {
	NodeCount           int            `json:"node_count"`
	EdgeCount           int            `json:"edge_count"`
	NodeTypes           map[string]int `json:"node_types"` // type -> count
	EdgeTypes           map[string]int `json:"edge_types"` // type -> count
	AvgDegree           float64        `json:"avg_degree"`
	MaxDegree           int            `json:"max_degree"`
	ConnectedComponents int            `json:"connected_components"`
}

type GraphVerificationStats struct {
	OriginalGraph     GraphStats    `json:"original_graph"`
	MaterializedGraph GraphStats    `json:"materialized_graph"`
	MetaPath          MetaPathStats `json:"meta_path"`
}

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

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

type HomogeneousGraph struct {
	NodeType   string              `json:"node_type"`  // "Author" for symmetric paths
	Nodes      map[string]Node     `json:"nodes"`      // Nodes in result graph
	Edges      map[EdgeKey]float64 `json:"edges"`      // (from,to) -> weight
	Statistics GraphStatistics     `json:"statistics"` // Graph metrics
	MetaPath   models.MetaPath     `json:"meta_path"`  // Original meta path used
}

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

type MaterializationConfig struct {
	Traversal   TraversalConfig   `json:"traversal"`
	Aggregation AggregationConfig `json:"aggregation"`
	Progress    ProgressConfig    `json:"progress"`
}

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

type MaterializationError struct {
	Component string `json:"component"` // "traversal", "aggregation", "memory", etc.
	Message   string `json:"message"`
	Details   string `json:"details,omitempty"`
}

type MaterializationResult struct {
	HomogeneousGraph *HomogeneousGraph     `json:"homogeneous_graph"`
	Statistics       ProcessingStatistics  `json:"statistics"`
	Config           MaterializationConfig `json:"config"`
	Success          bool                  `json:"success"`
	Error            string                `json:"error,omitempty"`
}

type MetaPathInterpretation int

type MetaPathStats struct {
	Length             int      `json:"length"`
	IsSymmetric        bool     `json:"is_symmetric"`
	StartNodeType      string   `json:"start_node_type"`
	EndNodeType        string   `json:"end_node_type"`
	NodeTypes          []string `json:"node_types"`
	EdgeTypes          []string `json:"edge_types"`
	EstimatedInstances int      `json:"estimated_instances"`
}

type Node struct {
	ID         string                 `json:"id"`
	Type       string                 `json:"type"`
	Properties map[string]interface{} `json:"properties"`
	Degree     int                    `json:"degree"` // Number of connections
}

type NormalizationType int

type PathInstance struct {
	Nodes    []string               `json:"nodes"`    // Sequence of node IDs [a1, p1, a2]
	Edges    []string               `json:"edges"`    // Sequence of edge types used
	Weight   float64                `json:"weight"`   // Accumulated weight along path
	Metadata map[string]interface{} `json:"metadata"` // Additional information
}

type PathState struct {
	Nodes   []string        // Current path of nodes
	Edges   []string        // Current path of edge types
	Weight  float64         // Accumulated weight
	Step    int             // Current step in meta path (0 to len(edges))
	Visited map[string]bool // Visited nodes (for cycle detection)
}

type ProcessingStatistics struct {
	RuntimeMS             int64            `json:"runtime_ms"`
	MemoryPeakMB          int64            `json:"memory_peak_mb"`
	InstancesGenerated    int              `json:"instances_generated"`
	InstancesFiltered     int              `json:"instances_filtered"`
	EdgesCreated          int              `json:"edges_created"`
	NodesInResult         int              `json:"nodes_in_result"`
	TraversalStatistics   TraversalStats   `json:"traversal_stats"`
	AggregationStatistics AggregationStats `json:"aggregation_stats"`
}

type ProgressCallback func(current, total int, message string)

type ProgressConfig struct {
	EnableProgress bool `json:"enable_progress"` // Whether to report progress
	ReportInterval int  `json:"report_interval"` // Report every N instances
}

type TestResult struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	Passed      bool   `json:"passed"`
	Expected    string `json:"expected"`
	Actual      string `json:"actual"`
	ErrorMsg    string `json:"error_msg,omitempty"`
	Severity    string `json:"severity"` // "critical", "warning", "info"
}

type TraversalConfig struct {
	Strategy       TraversalStrategy `json:"strategy"`        // BFS, DFS, etc.
	MaxPathLength  int               `json:"max_path_length"` // Prevent infinite recursion
	AllowCycles    bool              `json:"allow_cycles"`    // Whether to allow node revisits
	MaxInstances   int               `json:"max_instances"`   // Memory safety limit
	TimeoutSeconds int               `json:"timeout_seconds"` // Processing timeout
	Parallelism    int               `json:"parallelism"`     // Number of parallel workers
}

type TraversalStats struct {
	StartingNodes     int         `json:"starting_nodes"`
	NodesVisited      int         `json:"nodes_visited"`
	EdgesTraversed    int         `json:"edges_traversed"`
	PathsExplored     int         `json:"paths_explored"`
	CyclesDetected    int         `json:"cycles_detected"`
	TimeoutOccurred   bool        `json:"timeout_occurred"`
	WorkerUtilization map[int]int `json:"worker_utilization"` // worker_id -> instances_processed
	RuntimeMS         int64       `json:"runtime_ms"`         // Total traversal time
}

type TraversalStrategy int

type VerificationResult struct {
	Passed           bool                   `json:"passed"`
	TotalTests       int                    `json:"total_tests"`
	PassedTests      int                    `json:"passed_tests"`
	FailedTests      int                    `json:"failed_tests"`
	TestResults      []TestResult           `json:"test_results"`
	GraphStats       GraphVerificationStats `json:"graph_stats"`
	RecommendedFixes []string               `json:"recommended_fixes"`
}

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

type WeightStatistics struct {
	Count             int                `json:"count"`
	Sum               float64            `json:"sum"`
	Mean              float64            `json:"mean"`
	Min               float64            `json:"min"`
	Max               float64            `json:"max"`
	Variance          float64            `json:"variance"`
	StandardDeviation float64            `json:"standard_deviation"`
	Percentiles       map[string]float64 `json:"percentiles"`
}