Deck color charts

analyzer/models.py

@@ -29,26 +29,64 @@ class Deck(models.Model):
     cards = models.ManyToManyField(Card, through='Included')
     def __unicode__(self):
         return self.name
-    def colors(self):
+    def lands(self):
         symbols = {}
         for symbol in mtg.Mana.types.keys():
             symbols[symbol] = 0
-        total = 0
+        for included in self.included_set.all():
-        for card in self.cards.all():
+            card = included.card
-            # Get symbols from card cost
+            types = card.type.name.split(' ')
-            cost = mtg.ManaCost(card.cost)
+            attributes = [a.name for a in card.attributes.all()]
-            for color, count in cost.mana.mana.iteritems():
+            if 'land' not in types:
-                symbols[color] += count
+                continue
-                total += count
+            if 'forest' in attributes:
+                symbols['G'] += included.count
+            if 'mountain' in attributes:
+                symbols['R'] += included.count
+            if 'island' in attributes:
+                symbols['U'] += included.count
+            if 'plains' in attributes:
+                symbols['W'] += included.count
+            if 'swamp' in attributes:
+                symbols['B'] += included.count
 
             # Get symbols from abilities
             pattern = '{%s}' % mtg.ManaCost.symbolPattern
             costs = [mtg.ManaCost(cost) for cost in re.findall(pattern, str(card.text))]
             for cost in costs:
                 for color, count in cost.mana.mana.iteritems():
-                    symbols[color] += count
+                    symbols[color] += count * included.count
-                    total += count
+        return symbols
-        return (symbols, total)
+    def symbols(self):
+        symbols = {}
+        for symbol in mtg.Mana.types.keys():
+            symbols[symbol] = 0
+        for included in self.included_set.all():
+            card = included.card
+            # Get symbols from card cost
+            cost = mtg.ManaCost(card.cost)
+            for color, count in cost.mana.mana.iteritems():
+                symbols[color] += count * included.count
+
+            # Get symbols from abilities
+            pattern = '{%s}' % mtg.ManaCost.symbolPattern
+            costs = [mtg.ManaCost(cost) for cost in re.findall(pattern, str(card.text))]
+            for cost in costs:
+                for color, count in cost.mana.mana.iteritems():
+                    symbols[color] += count * included.count
+        return symbols
+    def mana_curve(self):
+        curve = {}
+        for included in self.included_set.all():
+            card = included.card
+            if 'land' in card.type.name.split(' '):
+                continue
+            cost = mtg.ManaCost(card.cost).converted()
+            if cost not in curve:
+                curve[cost] = included.count
+            else:
+                curve[cost] += included.count
+        return curve
 class Included(models.Model):
     card = models.ForeignKey(Card)
     deck = models.ForeignKey(Deck)

analyzer/templatetags/math.py

@@ -25,3 +25,11 @@ def getsum(value):
         return sum(value)
     else:
         return 0
+@register.filter
+def getmax(value):
+    if type(value) == dict:
+        return max(value.values())
+    elif type(value) == list:
+        return max(value)
+    else:
+        return 0

decks/views.py

@@ -1,6 +1,8 @@
 from django.http import Http404
 from django.shortcuts import render_to_response
 from mtgweb.analyzer.models import Deck
+from mtgweb.lib.mtg import mtg
+from pygooglechart import PieChart3D
 
 def index(request):
     decks = Deck.objects.all()
@@ -10,4 +12,36 @@ def detail(request, id):
         deck = Deck.objects.get(pk=id)
     except Deck.DoesNotExist:
         raise Http404
-    return render_to_response('decks/detail.html', {'deck': deck})
+
+    # Generate charts
+    colors = {
+        'R': 'FF0000',
+        'G': '00FF00',
+        'U': '0000FF',
+        'W': 'FFFFFF',
+        'B': '000000',
+    }
+    land_chart = PieChart3D(250, 100)
+    lands = dict([(k, v) for k, v in deck.lands().items() if v])
+    land_chart.fill_solid(PieChart3D.BACKGROUND, 'FFFFFF00')
+    land_chart.add_data(lands.values())
+    land_chart.set_pie_labels([mtg.Mana.types[m] for m in lands.keys()])
+    land_chart.set_colours([colors[m] for m in lands.keys()])
+
+    symbol_chart = PieChart3D(250, 100)
+    symbols = dict([(k, v) for k, v in deck.symbols().items() if v])
+    symbol_chart.fill_solid(PieChart3D.BACKGROUND, 'FFFFFF00')
+    symbol_chart.add_data(symbols.values())
+    symbol_chart.set_pie_labels([mtg.Mana.types[m] for m in symbols.keys()])
+    symbol_chart.set_colours([colors[m] for m in symbols.keys()])
+    
+    curve = deck.mana_curve()
+    for i in range(max(curve.keys()) + 1):
+        if i not in curve:
+            curve[i] = 0
+
+    return render_to_response('decks/detail.html', {
+                                                    'deck': deck,
+                                                    'land_chart': land_chart,
+                                                    'symbol_chart': symbol_chart,
+                                                    'curve': curve})

media/mtg.css

@@ -1,6 +1,7 @@
 /* Page styles */
 body {
     font-family: serif;
+    background-color: #eee;
 }
 h1, h2, h3, h4, h5, h6 {
     text-shadow: 2px 2px 2px #999;
@@ -114,3 +115,13 @@ header {
     display: block;
     float: left;
 }
+.mana-curve {
+    width: 50%;
+    border: 1px solid #999;
+}
+.mana-curve .cost {
+    width: 2em;
+}
+.mana-curve .bar {
+    background-color: #999;
+}

templates/decks/detail.html

@@ -1,9 +1,26 @@
 {% extends "main.html" %}
+{% load get_range %}
+{% load math %}
 
 {% block title %}Deck: {{ deck.name }}{% endblock %}
 
 {% block content %}
 <h2>{{deck}}</h2>
+<h3>Mana</h3>
+<h4>Lands</h4>
+<img src="{{ land_chart.get_url }}" />
+<h4>Symbols</h4>
+<img src="{{ symbol_chart.get_url }}" />
+<h4>Curve</h4>
+<table class="mana-curve">
+    {% for cost, count in curve.items %}
+        <tr>
+            <td class="cost">{{ cost }}</td>
+            <td><div class="bar" style="text-align: right; width: {% widthratio count curve.values|getsum 100 %}%">{{ count }}</div></td>
+        </tr>
+    {% endfor %}
+</table>
+<h3>Cards</h3>
 <ul>
     {% for card in deck.included_set.all %}
         <li>{{card.count}}x <a href="{% url mtgweb.cards.views.display name=card.card.name %}">{{ card.card }}</a></li>

templates/decks/index.html

@@ -7,9 +7,9 @@
 {% for deck in decks %}
     <li>
         <div class="manabar">
-        {% for color, count in deck.colors.0.items %}
+        {% for color, count in deck.symbols.items %}
             {% if count > 0 %}
-                <span class="mana {{ color }}" style="width: {{ count|mult:100|div:deck.colors.1 }}%"></span>
+                <span class="mana {{ color }}" style="width: {% widthratio count deck.symbols.values|getsum 100 %}%"></span>
             {% endif %}
         {% endfor %}
         </div>