refactored entire code base to be in the package core so i can later implement interfaces to interact with the engine

2 files changed, 27 insertions, 0 deletions

diff --git a/core/scoring/field.go b/core/scoring/field.go
new file mode 100644
index 0000000..89f6132
--- /dev/null
+++ b/core/scoring/field.go
@@ -0,0 +1,10 @@
+package scoring
+
+import "searchEngine/core/index"
+
+var TitleWeight = 5.0
+var BodyWeight = 1.0
+
+func FieldScore(p index.Posting) float64 {
+	return (float64(p.TitleCount) * TitleWeight) + (float64(p.BodyCount) * BodyWeight)
+}
diff --git a/core/scoring/idf.go b/core/scoring/idf.go
new file mode 100644
index 0000000..f5aaecd
--- /dev/null
+++ b/core/scoring/idf.go
@@ -0,0 +1,17 @@
+package scoring
+
+import (
+	"math"
+	"searchEngine/core/index"
+)
+
+func CalculateIDF(totalDocs int, docFrequency int) float64 {
+	if docFrequency == 0 {
+		return 0.0
+	}
+	return math.Log10(float64(totalDocs) / float64(docFrequency))
+}
+
+func ScoreTFIDF(p index.Posting, idf float64) float64 {
+	return FieldScore(p) * idf
+}