From 371ff473468a72f22bb33392d8ed7f3836a42104 Mon Sep 17 00:00:00 2001
From: Andrew Tomaka <atomaka@gmail.com>
Date: Mon, 26 Nov 2012 03:07:05 -0500
Subject: [PATCH] Process elemental damage on weapon

---
 d3hog_driver.php |  2 +-
 libs/dpclass.php | 19 ++++++++++++-------
 2 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/d3hog_driver.php b/d3hog_driver.php
index ddb76a5..01a30de 100644
--- a/d3hog_driver.php
+++ b/d3hog_driver.php
@@ -19,7 +19,7 @@ if($_POST['submit']) {
   $contents = curl_exec($curl);
   curl_close($curl);
 
-  $character = DPClassFactory::createClassObject($contents);
+  $character = DPClassFactory::createClassObject($contents, $elementalOnWeapon);
 
   if($character === FALSE) {
     die('Bad class.  Either your class could not be detected or we do not support your class at this time.');
diff --git a/libs/dpclass.php b/libs/dpclass.php
index a8208af..8e084f5 100644
--- a/libs/dpclass.php
+++ b/libs/dpclass.php
@@ -1,16 +1,16 @@
 <?php
 
 class DPClassFactory {
-  function createClassObject($characterPage) {
+  function createClassObject($characterPage, $elementalOnWeapon) {
     $class = DPClassFactory::findClass($characterPage);
 
     include_once(__DIR__ . "/$class.php");
 
     switch($class) {
       case 'barbarian':
-        return new Barbarian($characterPage);
+        return new Barbarian($characterPage, $elementalOnWeapon);
       case 'wizard':
-        return new Wizard($characterPage);
+        return new Wizard($characterPage, $elementalOnWeapon);
       default:
         return false;
     }
@@ -30,10 +30,11 @@ class DPClass {
   var $stats = array();
   var $items = array();
 
-  function __construct($characterPage) {
+  function __construct($characterPage, $elementalOnWeapon) {
     $this->dpHTML = $characterPage;
 
     $this->class = get_class($this);
+    $this->elementalOnWeapon = $elementalOnWeapon;
 
     $this->parseStats();
   }
@@ -61,14 +62,18 @@ class DPClass {
   }
 
   function elementalDamage() {
-    $totalElemental = 0;
+    $totalElemental = 1;
     foreach($this->stats as $stat => $value) {
       if(preg_match('/\+DPS \(.*\)/', $stat) > 0) {
         $totalElemental += $value;
       }
     }
 
-    return ($totalElemental > 0) ? $totalElemental + 1 : 0;
+    if($this->elementalOnWeapon && $totalElemental != 1) {
+      $totalElemental *= .5;
+    }
+
+    return ($totalElemental > 0) ? $totalElemental : 0;
   }
 
   function calculateGemLife() {
@@ -93,7 +98,7 @@ class DPClass {
 
   function modifyDPSUnbuffed() {
     $this->stats['DPS Unbuffed'] = $this->getStat('DPS Unbuffed') * 
-      max(1, 1 + ($this->getStat('All Elemental Damage') / 2)) * 
+      $this->getStat('All Elemental Damage') * 
       max(1, 1 + ($this->getStat('+DPS Against Elites') / 2));
   }