1: <?php
  2: /**
  3:  * Magento
  4:  *
  5:  * NOTICE OF LICENSE
  6:  *
  7:  * This source file is subject to the Open Software License (OSL 3.0)
  8:  * that is bundled with this package in the file LICENSE.txt.
  9:  * It is also available through the world-wide-web at this URL:
 10:  * http://opensource.org/licenses/osl-3.0.php
 11:  * If you did not receive a copy of the license and are unable to
 12:  * obtain it through the world-wide-web, please send an email
 13:  * to license@magentocommerce.com so we can send you a copy immediately.
 14:  *
 15:  * DISCLAIMER
 16:  *
 17:  * Do not edit or add to this file if you wish to upgrade Magento to newer
 18:  * versions in the future. If you wish to customize Magento for your
 19:  * needs please refer to http://www.magentocommerce.com for more information.
 20:  *
 21:  * @category    Mage
 22:  * @package     Mage_Sales
 23:  * @copyright   Copyright (c) 2012 Magento Inc. (http://www.magentocommerce.com)
 24:  * @license     http://opensource.org/licenses/osl-3.0.php  Open Software License (OSL 3.0)
 25:  */
 26: 
 27: 
 28: class Mage_Sales_Model_Quote_Address_Total_Subtotal extends Mage_Sales_Model_Quote_Address_Total_Abstract
 29: {
 30:     /**
 31:      * Collect address subtotal
 32:      *
 33:      * @param   Mage_Sales_Model_Quote_Address $address
 34:      * @return  Mage_Sales_Model_Quote_Address_Total_Subtotal
 35:      */
 36:     public function collect(Mage_Sales_Model_Quote_Address $address)
 37:     {
 38:         parent::collect($address);
 39:         $address->setTotalQty(0);
 40: 
 41:         $baseVirtualAmount = $virtualAmount = 0;
 42: 
 43:         /**
 44:          * Process address items
 45:          */
 46:         $items = $this->_getAddressItems($address);
 47:         foreach ($items as $item) {
 48:             if ($this->_initItem($address, $item) && $item->getQty() > 0) {
 49:                 /**
 50:                  * Separatly calculate subtotal only for virtual products
 51:                  */
 52:                 if ($item->getProduct()->isVirtual()) {
 53:                     $virtualAmount += $item->getRowTotal();
 54:                     $baseVirtualAmount += $item->getBaseRowTotal();
 55:                 }
 56:             }
 57:             else {
 58:                 $this->_removeItem($address, $item);
 59:             }
 60:         }
 61: 
 62:         $address->setBaseVirtualAmount($baseVirtualAmount);
 63:         $address->setVirtualAmount($virtualAmount);
 64: 
 65:         /**
 66:          * Initialize grand totals
 67:          */
 68:         Mage::helper('sales')->checkQuoteAmount($address->getQuote(), $address->getSubtotal());
 69:         Mage::helper('sales')->checkQuoteAmount($address->getQuote(), $address->getBaseSubtotal());
 70:         return $this;
 71:     }
 72: 
 73:     /**
 74:      * Address item initialization
 75:      *
 76:      * @param  $item
 77:      * @return bool
 78:      */
 79:     protected function _initItem($address, $item)
 80:     {
 81:         if ($item instanceof Mage_Sales_Model_Quote_Address_Item) {
 82:             $quoteItem = $item->getAddress()->getQuote()->getItemById($item->getQuoteItemId());
 83:         }
 84:         else {
 85:             $quoteItem = $item;
 86:         }
 87:         $product = $quoteItem->getProduct();
 88:         $product->setCustomerGroupId($quoteItem->getQuote()->getCustomerGroupId());
 89: 
 90:         /**
 91:          * Quote super mode flag mean what we work with quote without restriction
 92:          */
 93:         if ($item->getQuote()->getIsSuperMode()) {
 94:             if (!$product) {
 95:                 return false;
 96:             }
 97:         }
 98:         else {
 99:             if (!$product || !$product->isVisibleInCatalog()) {
100:                 return false;
101:             }
102:         }
103: 
104:         if ($quoteItem->getParentItem() && $quoteItem->isChildrenCalculated()) {
105:             $finalPrice = $quoteItem->getParentItem()->getProduct()->getPriceModel()->getChildFinalPrice(
106:                $quoteItem->getParentItem()->getProduct(),
107:                $quoteItem->getParentItem()->getQty(),
108:                $quoteItem->getProduct(),
109:                $quoteItem->getQty()
110:             );
111:             $item->setPrice($finalPrice)
112:                 ->setBaseOriginalPrice($finalPrice);
113:             $item->calcRowTotal();
114:         } else if (!$quoteItem->getParentItem()) {
115:             $finalPrice = $product->getFinalPrice($quoteItem->getQty());
116:             $item->setPrice($finalPrice)
117:                 ->setBaseOriginalPrice($finalPrice);
118:             $item->calcRowTotal();
119:             $this->_addAmount($item->getRowTotal());
120:             $this->_addBaseAmount($item->getBaseRowTotal());
121:             $address->setTotalQty($address->getTotalQty() + $item->getQty());
122:         }
123: 
124:         return true;
125:     }
126: 
127:     /**
128:      * Remove item
129:      *
130:      * @param  $address
131:      * @param  $item
132:      * @return Mage_Sales_Model_Quote_Address_Total_Subtotal
133:      */
134:     protected function _removeItem($address, $item)
135:     {
136:         if ($item instanceof Mage_Sales_Model_Quote_Item) {
137:             $address->removeItem($item->getId());
138:             if ($address->getQuote()) {
139:                 $address->getQuote()->removeItem($item->getId());
140:             }
141:         }
142:         elseif ($item instanceof Mage_Sales_Model_Quote_Address_Item) {
143:             $address->removeItem($item->getId());
144:             if ($address->getQuote()) {
145:                 $address->getQuote()->removeItem($item->getQuoteItemId());
146:             }
147:         }
148: 
149:         return $this;
150:     }
151: 
152:     /**
153:      * Assign subtotal amount and label to address object
154:      *
155:      * @param   Mage_Sales_Model_Quote_Address $address
156:      * @return  Mage_Sales_Model_Quote_Address_Total_Subtotal
157:      */
158:     public function fetch(Mage_Sales_Model_Quote_Address $address)
159:     {
160:         $address->addTotal(array(
161:             'code'  => $this->getCode(),
162:             'title' => Mage::helper('sales')->__('Subtotal'),
163:             'value' => $address->getSubtotal()
164:         ));
165:         return $this;
166:     }
167: 
168:     /**
169:      * Get Subtotal label
170:      *
171:      * @return string
172:      */
173:     public function getLabel()
174:     {
175:         return Mage::helper('sales')->__('Subtotal');
176:     }
177: }
178: