Add calculated arudha, upagraha and extra lagna to varga block

Author: kunjara

src/Base/Data.php

@@ -410,8 +410,13 @@ public function calcExtraLagna(array $lagnaKeys = null)
         $Lagna = new Lagna($this);
         $generateLagna = $Lagna->generateLagna($lagnaKeys);
 
+        if (!isset($this->data[self::BLOCK_VARGA])) {
+            $this->calcVargaData([Varga::KEY_D1]);
+        }
+        
         foreach ($generateLagna as $key => $data) {
             $this->data[self::BLOCK_LAGNA][$key] = $data;
+            $this->data[self::BLOCK_VARGA][Varga::KEY_D1][self::BLOCK_LAGNA][$key] = $data;
         }
         return $this;
     }
@@ -428,8 +433,13 @@ public function calcBhavaArudha(array $arudhaKeys = null, array $options = null)
         $Arudha = new Arudha($this, $options);
         $generateArudha = $Arudha->generateArudha($arudhaKeys);
 
+        if (!isset($this->data[self::BLOCK_VARGA])) {
+            $this->calcVargaData([Varga::KEY_D1]);
+        }
+        
         foreach ($generateArudha as $key => $data) {
             $this->data[self::BLOCK_LAGNA][$key] = $data;
+            $this->data[self::BLOCK_VARGA][Varga::KEY_D1][self::BLOCK_LAGNA][$key] = $data;
         }
         return $this;
     }
@@ -445,8 +455,13 @@ public function calcUpagraha(array $upagrahaKeys = null)
         $Upagraha = new Upagraha($this);
         $generateUpagraha = $Upagraha->generateUpagraha($upagrahaKeys);
 
+        if (!isset($this->data[self::BLOCK_VARGA])) {
+            $this->calcVargaData([Varga::KEY_D1]);
+        }
+        
         foreach ($generateUpagraha as $key => $data) {
             $this->data[self::BLOCK_UPAGRAHA][$key] = $data;
+            $this->data[self::BLOCK_VARGA][Varga::KEY_D1][self::BLOCK_UPAGRAHA][$key] = $data;
         }
         return $this;
     }