Individual 3 of 3

How many businesses close after X time on a Y day in Z state?

items = 'not loaded yet'

$.get('http://bigdatahci2015.github.io/data/yelp/yelp_academic_dataset_business.5000.json.lines.txt')
    .success(function(data){        
        var lines = data.trim().split('\n')

        // convert text lines to json arrays and save them in `items`
        items = _.map(lines, JSON.parse)

        console.log('number of items loaded:', items.length)

        console.log('first item', items[0])
     })
     .error(function(e){
         console.error(e)
     })

function viz(arg1, arg2, arg3){    

    // define a template string
    var tplString = '<g transform="translate(0 ${d.y})"> \
                    <rect x="30"   \
                         width="${d.width}" \
                         height="20"    \
                         style="fill:${d.color};    \
                                stroke-width:3; \
                                stroke:rgb(0,0,0)" />   \
                    <text y="20" x="${d.labelX}">${d.label}</text> \
                    </g>'

    // compile the string to get a template function
    var template = _.template(tplString)

    function computeX(d, i) {
        return 0
    }

    function computeWidth(d, i) {        
        return d.size
    }

    function computeY(d, i) {
        return i * 20
    }

    function computeColor(d, i) {
        return 'red'
    }

    function computeLabel(d, i) {
        return d.name+': '+d.size
    }

    function computeLabelX(d, i){
        return 40
    }

    // TODO: modify the logic here based on your UI
    // take the first 20 items to visualize    
    items = _.take(items, 5000)

    var states=_.groupBy(items, function(b){
            return b['state']
        })
    console.log("states: ", states)
    var businesses=states[arg3]
    console.log("businesses: ", businesses)
    var cities=_.groupBy(businesses, function(c){
            return c['city']
        })

    var pairs=_.pairs(cities)
    console.log('pairs:',pairs)
    var filter_city=_.map(pairs, function(p){
    		var sp1=arg1.split(':')
            return  {
                city: p[0],
                businesses: _.filter(p[1], function(b){
                	if(b['hours'] === undefined || b['hours'][arg2] === undefined || b['hours'][arg2]['close'] === undefined){
                	return false
                	}
                	var sp2=b['hours'][arg2]['close'].split(':')
                    if(parseInt(sp2[0]) > parseInt(sp1[0]))
                    	return true
                    else if(parseInt(sp2[0]) == parseInt(sp1[0])){
                    	return parseInt(sp2[1]) >= parseInt(sp1[1])
                    }
                    return false
                })}
    })
    console.log('city filter:', filter_city)
    var num_cities=_.map(filter_city, function(c){
         return {
            name: c.city,
            size: _.size(c.businesses)}
        })
    console.log('city size:', num_cities)


    var viz = _.map(num_cities, function(d, i){                
                return {
                    x: computeX(d, i),
                    y: computeY(d, i),
                    width: computeWidth(d, i),
                    color: computeColor(d, i),
                    label: computeLabel(d, i),
                    labelX: computeLabelX(d, i)
                }
             })
    console.log('viz', viz)

    var result = _.map(viz, function(d){
             // invoke the compiled template function on each viz data
             return template({d: d})
         })
    console.log('result', result)

    $('.myviz').html('<svg width="100%" height="100%">' + result + '</svg>')
}

$('button#viz').click(function(){    
    var arg1 = $('input#arg1').val()
    var arg2 = $('input#arg2').val()
    var arg3 = $('input#arg3').val()
    viz(arg1, arg2, arg3)
})