feat: added seralization

huffman/huffman.go

@@ -1,6 +1,8 @@
 package huffman
 
 import (
+	"bytes"
+	"encoding/gob"
 	"fmt"
 )
 
@@ -50,3 +52,36 @@ func Decode(root *Node, encodedData []byte, bitLength int, verbose bool) string
 	}
 	return decodedData
 }
+
+func SerializeTree(root *Node) ([]byte, error) {
+	if root == nil {
+		return nil, nil
+	}
+
+	var buf bytes.Buffer
+	enc := gob.NewEncoder(&buf)
+
+	err := enc.Encode(root)
+	if err != nil {
+		return nil, err
+	}
+
+	return buf.Bytes(), nil
+}
+
+func DeserializeTree(data []byte) (*Node, error) {
+	if data == nil {
+		return nil, nil
+	}
+
+	buf := bytes.NewBuffer(data)
+	dec := gob.NewDecoder(buf)
+
+	var root Node
+	err := dec.Decode(&root)
+	if err != nil {
+		return nil, err
+	}
+
+	return &root, nil
+}

huffman/huffman_util.go

@@ -1,7 +1,9 @@
 package huffman
 
 import (
+	"bytes"
 	"container/heap"
+	"math"
 )
 
 type Node struct {
@@ -114,3 +116,35 @@ func encodeString(s string, codes map[rune]string) string {
 	}
 	return encoded
 }
+
+func bitStringToBytes(s string) []byte {
+	numBytes := int(math.Ceil(float64(len(s)) / 8.0))
+	result := make([]byte, numBytes)
+
+	for i := 0; i < len(s); i++ {
+		if s[i] == '1' {
+			byteIndex := i / 8
+			bitIndex := 7 - (i % 8)
+			result[byteIndex] |= 1 << bitIndex
+		}
+	}
+
+	return result
+}
+
+func bytesToBitString(data []byte, bitLength int) string {
+	var result bytes.Buffer
+
+	for i := 0; i < bitLength; i++ {
+		byteIndex := i / 8
+		bitIndex := 7 - (i % 8)
+
+		if data[byteIndex]&(1<<bitIndex) != 0 {
+			result.WriteByte('1')
+		} else {
+			result.WriteByte('0')
+		}
+	}
+
+	return result.String()
+}

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"flag"
+	"fmt"
 	"github.com/absolutezero000/encoding/huffman"
 	"log"
 	"os"
@@ -11,13 +12,13 @@ func main() {
 	inputFile := flag.String("i", "", "Input file name")
 	outputFile := flag.String("o", "", "Output file name")
 	verbose := flag.Bool("v", false, "Enable verbose mode")
-
 	flag.Parse()
 
 	data, err := os.ReadFile(*inputFile)
 	if err != nil {
 		log.Fatalf("Failed to read file %s: %v", *inputFile, err)
 	}
+
 	var encodedData []byte
 	var bitLength int
 	HuffmanTreeRoot := huffman.MakeTree(string(data))
@@ -31,4 +32,10 @@ func main() {
 		}
 	}
 
+	treeData, _ := huffman.SerializeTree(HuffmanTreeRoot)
+
+	fmt.Println(treeData)
+
+	treeRoot, _ := huffman.DeserializeTree(treeData)
+	_ = huffman.Decode(treeRoot, encodedData, bitLength, *verbose)
 }