Tlsh/BucketProcessor.cs at master · romcab/Tlsh · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
namespace tlsh.buckets
{
    internal class BucketProcessor
    {
        private const int ArrayBucketSize = 256;

        public ProcessedBuckets Process(byte[] data)
        {
            int[] checksum = null;
            int[] bucketArray = new int[ArrayBucketSize];

            SlideWindow slideWindow = new SlideWindow();
            int length = data.Length;

            for (int i = 0; i < length; i++)
            {
                int startWindow = slideWindow.GetPivot();
                slideWindow.Put(data[i]);

                checksum = slideWindow.GetChecksum(startWindow, checksum);

                foreach (var tripletHash in slideWindow.GetTripletHashes(startWindow))
                {
                    bucketArray[tripletHash]++;
                }
            }

            return buildProcessedBuckets(length, bucketArray, checksum);
        }

        //public ProcessedBuckets Process(string data)
        //{
        //    int[] checksum = null;
        //    int[] bucketArray = new int[ArrayBucketSize];

        //    SlideWindow slideWindow = new SlideWindow();
        //    int length = data.Length;

        //    for (int i = 0; i < length; i++)
        //    {
        //        int startWindow = slideWindow.GetPivot();
        //        slideWindow.Put(data[i]);

        //        checksum = slideWindow.GetChecksum(startWindow, checksum);

        //        foreach (var tripletHash in slideWindow.GetTripletHashes(startWindow))
        //        {
        //            bucketArray[tripletHash]++;
        //        }
        //    }

        //    return buildProcessedBuckets(length, bucketArray, checksum);
        //}

        private ProcessedBuckets buildProcessedBuckets(int dataLength, int[] bucketArray, int[] checksum)
        {
            Quartiles quartiles = new Quartiles(bucketArray);
            return new ProcessedBuckets(checksum, bucketArray, dataLength, quartiles);
        }
    }
}