Update TV:

- update indexProgrammes() method
webeweb · Jul 30, 2019 · dd325b6 · dd325b6
1 parent bbb8edd
commit dd325b6
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/src/Model/Tv.php b/src/Model/Tv.php
@@ -135,20 +135,25 @@ public function getSourceInfoURL() {
     }
 
     /**
-     * Indexes the programmes.
+     * Indexes the programmes by channel.
      *
-     * @return void
+     * @return Tv Returns this TV.
      */
-    public function indexProgrammes() {
+    public function indexProgrammesByChannel() {
+
+        $this->sortChannels();
+        $this->sortProgrammes();
+
+        $channels = $this->indexChannelsById();
 
         /** @var Programme $current */
         foreach ($this->programmes as $current) {
-
-            $channel = $this->getChannelById($current->getChannel());
-            if (null !== $channel) {
-                $channel->addProgramme($current);
+            if (true === array_key_exists($current->getChannel(), $channels)) {
+                $channels[$current->getChannel()]->addProgramme($current);
             }
         }
+
+        return $this;
     }
 
     /**

diff --git a/src/Parser/Parser.php b/src/Parser/Parser.php
@@ -691,7 +691,7 @@ public static function parseTv(DOMNode $domNode) {
         ParserHelper::parseChildNodes($domNode, "channel", $model);
         ParserHelper::parseChildNodes($domNode, "programme", $model);
 
-        $model->indexProgrammes();
+        $model->indexProgrammesByChannel();
 
         return $model;
     }