Imports System.Collections.ObjectModel

Public Class MapRegion
    Inherits DependencyObject

    Public Shared ReadOnly RegionNameProperty As DependencyProperty _
        = DependencyProperty.Register("RegionName", GetType(String), _
        GetType(MapRegion))
    Public Shared ReadOnly IsSelectedProperty As DependencyProperty _
        = DependencyProperty.Register("IsSelected", GetType(Boolean), _
        GetType(MapRegion))

    '...
    
    Public Property RegionName() As String
        Get
            Return CStr(GetValue(RegionNameProperty))
        End Get
        Set(ByVal value As String)
            SetValue(RegionNameProperty, value)
        End Set
    End Property

    Public Property IsSelected() As Boolean
        Get
            Return CBool(GetValue(IsSelectedProperty))
        End Get
        Set(ByVal value As Boolean)
            SetValue(IsSelectedProperty, value)
        End Set
    End Property

<Style x:Key="RegionPolygonStyle" TargetType="Polygon">
    <Style.Triggers>
        <DataTrigger Binding="{Binding Path=Region.IsSelected}" 
            Value="True">
            <Setter Property="Stroke" Value="Orange" />
            <Setter Property="StrokeThickness" Value="20"/>
        </DataTrigger>
        <DataTrigger Binding="{Binding Path=Region.IsSelected}" 
            Value="False">
            <Setter Property="Stroke" Value="Black"/>
            <Setter Property="StrokeThickness" Value="5" />
        </DataTrigger>
    </Style.Triggers>
</Style>

Public Class MapPolygon
    Inherits DependencyObject

    Private Shared ReadOnly RegionPropertyKey As DependencyPropertyKey _
        = DependencyProperty.RegisterReadOnly("Region",  _
        GetType(MapRegion), GetType(MapPolygon), New PropertyMetadata())
    Public Shared ReadOnly RegionProperty As DependencyProperty

    Private m_Points As PointCollection

    Shared Sub New()
        RegionProperty = RegionPropertyKey.DependencyProperty
    End Sub

    Sub New()
        SetRegion(Nothing)
        m_Points = New PointCollection
    End Sub

    Friend Sub SetRegion(ByVal r As MapRegion)
        SetValue(RegionPropertyKey, r)
    End Sub

    Public ReadOnly Property Region() As MapRegion
        Get
            Return CType(GetValue(RegionProperty), MapRegion)
        End Get
    End Property

    Public ReadOnly Property Points() As PointCollection
        Get
            Return m_Points
        End Get
    End Property
End Class

Private Sub PolygonCollectionChanged(ByVal sender As Object, ByVal e As _
    System.Collections.Specialized.NotifyCollectionChangedEventArgs) _
    Handles m_polygons.CollectionChanged
    Select Case e.Action
        Case Specialized.NotifyCollectionChangedAction.Add
            e.NewItems.Cast(Of MapPolygon).SetRegion(Me)
        Case Specialized.NotifyCollectionChangedAction.Remove
            e.OldItems.Cast(Of MapPolygon).SetRegion(Nothing)
        Case Specialized.NotifyCollectionChangedAction.Replace
            e.NewItems.Cast(Of MapPolygon).SetRegion(Me)
            e.OldItems.Cast(Of MapPolygon).SetRegion(Nothing)
    End Select
End Sub

Private Sub LoadFile(ByVal file As String, ByVal list As ICollection(Of _
    MapRegion))
    Dim doc = XDocument.Load(file)
    Dim q = _
        From v In doc.<File>.<Region>.<Polygon>.<Vertex> _
        Let _
            Polygon = v.Parent, _
            Region = v.Parent.Parent _
        Group By _
            Polygon, _
            Region _
        Into _
            Group, _
            MinX = MinDBL(v.@X), _
            MinY = MinDBL(v.@Y), _
            MaxX = MaxDBL(v.@X), _
            MaxY = MaxDBL(v.@Y) _
        Select _
            TheMapPolygon = (From tmp In Group Select _
            tmp.v).ToMapPolygon(), _
            Region, _
            BoundingBox = New Rect(New Point(MinX, _
            MinY), New Point(MaxX,  MaxY)) _
        Group By _
            Region _
        Into _
            Group, _
            BoundingBox = Enclose(BoundingBox) _
        Select _
            Region = New MapRegion(Region.@Name, (From item In Group _
            Select item.TheMapPolygon), (From f In _
            Region...<FipsCode> Select CInt(f.Value)), BoundingBox)

    list.AddRange(q)
End Sub

<Extension()> _
Function Enclose(Of T)(ByVal collection As IEnumerable(Of T), ByVal _
    selector As Func(Of T, Rect)) As Rect
    Dim ret As Rect
    Dim first As Boolean = True

    For Each item In collection

        If first Then
            ret = selector(item)
            first = False
        Else
            ret.Enclose(selector(item))
        End If
    Next

    Return ret

End Function
<Extension()> _
Sub Enclose(ByRef theBox As Rect, ByVal otherBox As Rect)

    Dim X = Math.Min(theBox.X, otherBox.X)
    Dim Y = Math.Min(theBox.Y, otherBox.Y)
    Dim Width = Math.Max(theBox.BottomRight.X, otherBox.BottomRight.X) _
        - X + 1
    Dim Height = Math.Max(theBox.BottomRight.Y, otherBox.BottomRight.Y) _
        - Y + 1
    theBox.X = X
    theBox.Y = Y
    theBox.Width = Width
    theBox.Height = Height

End Sub

<Canvas Grid.Row="2" Grid.Column="1" Name="MasterCanvas">
    <Canvas.RenderTransform>
        <TransformGroup>
            <TranslateTransform X="{Binding Path=TranslateX}" 
                Y="{Binding Path=TranslateY}"/>
            <ScaleTransform ScaleX="{Binding Path=ScaleX}" 
                ScaleY="{Binding Path=ScaleY}" />
        </TransformGroup>
    </Canvas.RenderTransform>

    <ItemsControl Name="MapViewer" ItemsPanel="{DynamicResource SimpleCanvasTemplate}" ItemsSource="{Binding Path=Regions}" ItemTemplate="{DynamicResource RegionTemplate}" />    
</Canvas>

<Canvas Grid.Row="2" Grid.Column="1" Name="MasterCanvas">
    <Canvas.RenderTransform>
        <TransformGroup>
            <TranslateTransform X="{Binding Path=TranslateX}" 
                Y="{Binding Path=TranslateY}"/>
            <ScaleTransform ScaleX="{Binding Path=ScaleX}" ScaleY="{Binding Path=ScaleY}" />
        </TransformGroup>
    </Canvas.RenderTransform>

    <ItemsControl Name="MapViewer" ItemsPanel="{DynamicResource SimpleCanvasTemplate}" ItemsSource="{Binding Path=Regions}" ItemTemplate=
    "{DynamicResource RegionTemplate}" />
    <ItemsControl Name="DataLayer" ItemsPanel=
        "{DynamicResource SimpleCanvasTemplate}" ItemTemplate=
        "{DynamicResource DataLayerTemplate}"/>
    
</Canvas>

Public Sub LoadData()
    Dim list As New List(Of MapRegion)

    LoadFile("PolygonData\2000\states.xml", list)

    Dim q = From r In list Where Not excludedStates.Contains(r.RegionName)

    m_Map = New Map()

    Dim first As Boolean = True
    Dim bb As Rect

    For Each item In q
        m_Map.Regions.Add(item)

        If (first) Then
            bb = item.BoundingBox
            first = False
        Else
            bb.Enclose(item.BoundingBox)
        End If
    Next

    m_Map.BoundingBox = bb
    MasterCanvas.DataContext = m_Map
    m_Map.ScaleTo(New Size(MasterCanvas.ActualWidth, _
        MasterCanvas.ActualHeight))

    Dim counties = New List(Of MapRegion)

    LoadFile("PolygonData\2000\counties.xml", counties)

    Dim doc = XDocument.Load("PopulationData\CountyPopulation.xml")

    Dim q2 = _
        From _
            state In m_Map.Regions _
        Join _
            county In counties _
        On _
            state.FipsCodes(0) Equals county.FipsCodes(0) _
        Join _
            pd In doc.<CountyPopulationFile>.<CountyPopulation> _
        On _
            CInt(pd.@CountyFipsCode) Equals county.FipsCodes(1) And _
            CInt(pd.@StateFipsCode) Equals state.FipsCodes(0) _
        Where _
            CDate(pd.@Date).Year = 2006 _
        From _
            Polygon In county.Polygons _
        Select _
            Color = New SolidColorBrush(colorMapEntries.Interprolate( _
                CDbl(pd.@Population))), Points = Polygon.Points

    q2 = q2.ToList()

    DataLayer.ItemsSource = q2
    'Force the state boundaries to be on top of the county regions....
    MasterCanvas.Children.Remove(MapViewer)
    MasterCanvas.Children.Add(MapViewer)

    Dim legendInfo(colorMapEntries.Length - 2) As Object

    For i = 0 To colorMapEntries.Length - 2
        legendInfo(colorMapEntries.Length - (i + 2)) = _
        New With{.StartColor = colorMapEntries(i).FillColor, _
        .StopColor = colorMapEntries(i + 1).FillColor, .Value = _
        If(colorMapEntries(i + 1).ShowInIndex, colorMapEntries( _
        i + 1).Value.ToString("n"), "")}
    Next

    Legend.ItemsSource = legendInfo

End Sub

<Extension()> _
Function Interpolate(ByVal entries() As ColorMapEntry, ByVal value As _
    Double) As Color
    Dim upperBound = entries.LeastUpperBound(value)
    Dim c As Color = Nothing

    If upperBound >= entries.Length Then
        c = entries(upperBound - 1).FillColor()
    ElseIf upperBound = 0 Then
        c = entries(0).FillColor
    Else
        c = entries(upperBound - 1).Interpolate(entries( _
            upperBound), value)

    End If

    Return c
End Function
<Extension()> _
Function LeastUpperBound(ByVal entries() As ColorMapEntry, ByVal v As _
    Double) As Integer

    Dim ret = Array.BinarySearch(entries, v, _
        New ColorMapEntryToValueComparer)
    If ret < 0 Then
        ret = ret Xor -1
    End If

    Return ret
End Function