added proximity based scoringHEADmain

1 files changed, 15 insertions, 0 deletions

diff --git a/core/search/search.go b/core/search/search.go
index 721574c..cf1054b 100644
--- a/core/search/search.go
+++ b/core/search/search.go
@@ -24,6 +24,7 @@ func Search(idx *index.InvertedIndex, query string) []string {
 
 	totalDocs := len(idx.DocNames)
 	docMatchCounts := make(map[int]int)
+	docPostings := make(map[int][]index.Posting)
 	scores := make(map[int]float64)
 
 	uniqueMap := make(map[string]bool)
@@ -48,6 +49,8 @@ func Search(idx *index.InvertedIndex, query string) []string {
 			docMatchCounts[p.DocId]++
 			termScore := scoring.ScoreTFIDF(p, idf)
 			scores[p.DocId] += termScore
+
+			docPostings[p.DocId] = append(docPostings[p.DocId], p)
 		}
 	}
 
@@ -57,7 +60,19 @@ func Search(idx *index.InvertedIndex, query string) []string {
 			results = append(results, Result{DocID: docID, Score: scores[docID]})
 		}
 	}
+	for i := range results {
+		proximityBonus := 0.0
+		resultsPostings := docPostings[results[i].DocID]
+		for j := 0; j < len(resultsPostings)-1; j++ {
+			bodyBonus := scoring.CalculateProximity(resultsPostings[j].BodyPositions, resultsPostings[j+1].BodyPositions)
+
+			titleBonus := scoring.CalculateProximity(resultsPostings[j].TitlePositions, resultsPostings[j+1].TitlePositions)
 
+			proximityBonus += bodyBonus + (titleBonus * 5.0)
+		}
+
+		results[i].Score += proximityBonus
+	}
 	slices.SortFunc(results, func(a, b Result) int {
 		return cmp.Compare(b.Score, a.Score)
 	})