#-------------------------------------------------------------------------
# Traffic Distribution 
#    Analyze the capture file for traffic chart
#
# You may use this code freely in your commercial and non-commercial work.
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  
#
# Copyright (c) Unleash Networks 2005, All rights reserved
#----------------------------------------------------------------------------

require 'rubygems'
require 'win32ole'
require 'fox16'
include Fox
require 'UnleashCharts'
include UnleashCharts

class ChartWindow < FXMainWindow

	def initialize(theapp)

			# base class
			super(theapp, "Traffic over Time", nil, nil, DECOR_ALL,
						0,0,600,300)

			# single horizontal panel
			@contents = FXHorizontalFrame.new(self,
				LAYOUT_SIDE_TOP|LAYOUT_FILL_X|LAYOUT_FILL_Y,0,0,0,0,0,0,0,0,0)

			# vertical frame
			@canvasFrame = FXVerticalFrame.new(@contents,
				LAYOUT_FILL_X|LAYOUT_FILL_Y|LAYOUT_TOP|LAYOUT_LEFT,
				0,0,0,0,10,10,10,10)

			# actual canvas (whew!)
			@canvas = UnTimeSeriesChart.new(@canvasFrame)

	end


	def create
		super
		show(PLACEMENT_SCREEN)
	end	


	def setModel (mod)
		@canvas.setModel( mod)
	end
	
end


class TimeBuckets

	@captureFile
	@sampleInterval

	attr_reader	:maxval,:minval,:startTime,:endTime
	attr_reader	:valunits

	def initialize(filename, sampleintvl)
		@sampleInterval = sampleintvl
		@captureFile=filename

		# just read first and last in order to set the 
		# scales
		unsniffDB = WIN32OLE.new("Unsniff.Database")
		unsniffDB.OpenForRead(filename)
		packetIndex = unsniffDB.PacketIndex
		firstPkt = packetIndex.Item(0)
		lastPkt  = packetIndex.Item(packetIndex.Count-1)

		# set start timestamp
		@startTime = Time.at(firstPkt.TimestampSecs,firstPkt.TimestampUSecs)
		
		# set end timestamp
		@endTime = Time.at(lastPkt.TimestampSecs,lastPkt.TimestampUSecs)

		print "Start time = #{@startTime}\n"
		print "End time = #{@endTime}\n"

		@maxval = 800000 # 1 mBps max
		@minval = 0
		@valunits = "bps"
		
		unsniffDB.Close()
	end

	def totalTimeUSecs 
		secdiff = @endTime.tv_sec - @startTime.tv_sec 
		usdiff = @endTime.tv_usec - @startTime.tv_usec
		if usdiff < 0
			usdiff =  usdiff + ::USECPERSEC
			secdiff = secdiff-1
		end
		
		totalTimeUSecs = (secdiff * ::USECPERSEC) + usdiff
	end

	def each_val


		unsniffDB = WIN32OLE.new("Unsniff.Database")
		unsniffDB.OpenForRead(@captureFile)
		packetIndex = unsniffDB.PacketIndex
		nPackets = unsniffDB.PacketCount
		currSecs = @startTime.tv_sec
		byteCount = 0
		idx=0
		while (idx < nPackets)

			pkt = packetIndex.Item(idx)

			if pkt.TimestampSecs == currSecs
				byteCount += pkt.Length
			else
				yield Time.at(currSecs), 8*byteCount,  DataPointStyle.new(DataPointStyle::LINEHEIGHT,"o")
				byteCount = 0
				currSecs = pkt.TimestampSecs
			end

			idx+=1

		end

		unsniffDB.Close

	end
	
	def yscalelabel
		"Traffic"
	end
	
	def xscalelabel
		"Time"
	end

end


USAGE = "trafmon <capture-filename> "

if ARGV.length != 1
	puts USAGE
	exit 1
end


# Classify incoming packets into time buckets
lenBuckets = TimeBuckets.new(ARGV[0],1000)
#lenBuckets.each_val do |tm, val|
#	print "Secs = #{tm.tv_sec},   Val = #{val}\n";
#end

# A new Fox Application and MainWindow object
theApp = FXApp.new
theMainWindow = ChartWindow.new(theApp)
theMainWindow.setModel(lenBuckets)


# Run application
theApp.create
theMainWindow.show
theApp.run