fastscan: add missing files

src/input/mpegts/fastscan.c

@@ -0,0 +1,176 @@
+/*
+ *  tvheadend, charset list
+ *  Copyright (C) 2014 Jaroslav Kysela
+ *
+ *  This program is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "tvheadend.h"
+#include "settings.h"
+#include "bouquet.h"
+#include "dvb.h"
+#include "fastscan.h"
+
+typedef struct dvb_fastscan_item {
+  LIST_ENTRY(dvb_fastscan_item) ilink;
+
+  const char *name;
+  int         pid;
+
+} dvb_fastscan_item_t;
+
+typedef struct dvb_fastscan {
+  RB_ENTRY(dvb_fastscan) link;
+
+  LIST_HEAD(,dvb_fastscan_item) items;
+
+  int      position;
+  uint32_t frequency;
+} dvb_fastscan_t;
+
+static RB_HEAD(,dvb_fastscan) fastscan_rb;
+static SKEL_DECLARE(fastscan_rb_skel, dvb_fastscan_t);
+
+static int
+_fs_cmp(const void *a, const void *b)
+{
+  int64_t aa = ((dvb_fastscan_t *)a)->position * 1000000000LL +
+               ((dvb_fastscan_t *)a)->frequency;
+  int64_t bb = ((dvb_fastscan_t *)b)->position * 1000000000LL +
+               ((dvb_fastscan_t *)b)->frequency;
+  if (aa < bb)
+    return -1;
+  if (aa > bb)
+    return 1;
+  return 0;
+}
+
+void
+dvb_fastscan_each(void *aux, int position, uint32_t frequency,
+                  void (*job)(void *aux, bouquet_t *bq,
+                              const char *name, int pid))
+{
+  dvb_fastscan_t *fs;
+  dvb_fastscan_item_t *fsi;
+  bouquet_t *bq;
+  char url[64], buf[16];
+
+  SKEL_ALLOC(fastscan_rb_skel);
+  fastscan_rb_skel->position = position;
+  fastscan_rb_skel->frequency = frequency;
+  fs = RB_FIND(&fastscan_rb, fastscan_rb_skel, link, _fs_cmp);
+  if (!fs)
+    return;
+  LIST_FOREACH(fsi, &fs->items, ilink) {
+    dvb_sat_position_to_str(fastscan_rb_skel->position, buf, sizeof(buf));
+    snprintf(url, sizeof(url), "dvb-fastscan://dvbs,%s,%u,%d",
+             buf, frequency, fsi->pid);
+    bq = bouquet_find_by_source(NULL, url, 0);
+    if (bq == NULL || !bq->bq_enabled)
+      continue;
+    bq->bq_shield = 1;
+    job(aux, bq, fsi->name, fsi->pid);
+  }
+}
+
+static void
+dvb_fastscan_create(htsmsg_t *e)
+{
+  dvb_fastscan_t *fs;
+  dvb_fastscan_item_t *fsi;
+  bouquet_t *bq;
+  const char *name;
+  int pid;
+  char url[64], buf[16];
+
+  SKEL_ALLOC(fastscan_rb_skel);
+  if (htsmsg_get_s32(e, "position", &fastscan_rb_skel->position))
+    goto fail;
+  if (htsmsg_get_u32(e, "frequency", &fastscan_rb_skel->frequency))
+    goto fail;
+  if (htsmsg_get_s32(e, "pid", &pid))
+    goto fail;
+  if ((name = htsmsg_get_str(e, "name")) == NULL)
+    goto fail;
+
+  dvb_sat_position_to_str(fastscan_rb_skel->position, buf, sizeof(buf));
+  snprintf(url, sizeof(url), "dvb-fastscan://dvbs,%s,%u,%d",
+           buf, fastscan_rb_skel->frequency, pid);
+  bq = bouquet_find_by_source(name, url, 1);
+  if (bq == NULL)
+    goto fail;
+  bq->bq_shield = 1;
+  if (bq->bq_saveflag)
+    bouquet_save(bq, 1);
+
+  fs = RB_INSERT_SORTED(&fastscan_rb, fastscan_rb_skel, link, _fs_cmp);
+  if (!fs) {
+    fs = fastscan_rb_skel;
+    SKEL_USED(fastscan_rb_skel);
+    LIST_INIT(&fs->items);
+  }
+
+  fsi = calloc(1, sizeof(*fsi));
+  fsi->pid = pid;
+  if ((fsi->name = htsmsg_get_str(e, "name")) == NULL)
+    goto fail;
+  fsi->name = strdup(name);
+
+  LIST_INSERT_HEAD(&fs->items, fsi, ilink);
+
+  return;
+
+fail:
+  tvhwarn("fastscan", "wrong entry format");
+}
+
+/*
+ * Initialize the fastscan list
+ */
+void dvb_fastscan_init ( void )
+{
+  htsmsg_t *c, *e;
+  htsmsg_field_t *f;
+
+  if (!(c = hts_settings_load("fastscan"))) {
+    tvhwarn("fastscan", "configuration file missing");
+    return;
+  }
+  
+  HTSMSG_FOREACH(f, c) {
+    if (!(e = htsmsg_field_get_map(f))) continue;
+    dvb_fastscan_create(e);
+  }
+  htsmsg_destroy(c);
+}
+
+/*
+ *
+ */
+void dvb_fastscan_done ( void )
+{
+  dvb_fastscan_t *fs;
+  dvb_fastscan_item_t *fsi;
+  
+  while ((fs = RB_FIRST(&fastscan_rb)) != NULL) {
+    RB_REMOVE(&fastscan_rb, fs, link);
+    while ((fsi = LIST_FIRST(&fs->items)) != NULL) {
+      LIST_REMOVE(fsi, ilink);
+      free((char *)fsi->name);
+      free(fsi);
+    }
+    free(fs);
+  }
+  SKEL_FREE(fastscan_rb_skel);
+}

src/input/mpegts/fastscan.h

@@ -0,0 +1,32 @@
+/*
+ *  Tvheadend - FastScan support
+ *  Copyright (C) 2014 Jaroslav Kysela
+ *
+ *  This program is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef __TVH_DVB_FASTSCAN_H__
+#define __TVH_DVB_FASTSCAN_H__
+
+struct bouquet;
+
+void
+dvb_fastscan_each(void *aux, int position, uint32_t frequency,
+                  void (*job)(void *aux, struct bouquet *,
+                              const char *name, int pid));
+
+void dvb_fastscan_init ( void );
+void dvb_fastscan_done ( void );
+
+#endif /* TVH_DVB_FASTSCAN_H */