ChanSort/source/ChanSort.Api/Model/LookupData.cs

using System.Collections.Concurrent;
using System.Collections.Generic;
using System.IO;
using System.Reflection;

namespace ChanSort.Api
{
  public class LookupData
  {
    private readonly IDictionary<int, NetworkInfo> networks = new Dictionary<int, NetworkInfo>();
    private readonly IDictionary<int, int> transponderNrByFreqInMhz = new Dictionary<int, int>();
    private readonly IDictionary<int, string> serviceTypeDescriptions = new Dictionary<int, string>();
    private readonly IDictionary<int, string> dvbcChannels = new ConcurrentDictionary<int, string>();

    public static readonly LookupData Instance = new LookupData();

    private LookupData()
    {
      this.LoadDataFromCsvFile();
    }

    #region GetNetwork()
    public NetworkInfo GetNetwork(int networkId)
    {
      NetworkInfo network;
      this.networks.TryGetValue((ushort)networkId, out network);
      return network;
    }
    #endregion

    #region GetAstraTransponder(), GetAstraFrequency()
    public int GetAstraTransponder(int frequencyInMhz)
    {
      int number;
      bool found = this.transponderNrByFreqInMhz.TryGetValue(frequencyInMhz, out number) ||
      this.transponderNrByFreqInMhz.TryGetValue(frequencyInMhz - 1, out number) ||
      this.transponderNrByFreqInMhz.TryGetValue(frequencyInMhz + 1, out number) ||
      this.transponderNrByFreqInMhz.TryGetValue(frequencyInMhz - 2, out number) ||
      this.transponderNrByFreqInMhz.TryGetValue(frequencyInMhz + 2, out number);
      return found ? number : 0;
    }

    public int GetAstraFrequency(int transponderNr)
    {
      return this.transponderNrByFreqInMhz.TryGet(transponderNr);
    }
    #endregion

    #region GetServiceTypeDescription()
    public string GetServiceTypeDescription(int serviceType)
    {
      if (this.serviceTypeDescriptions.TryGetValue(serviceType, out var descr))
        return descr;
      return serviceType.ToString();
    }
    #endregion

    #region LoadDataFromCsvFile()
    public void LoadDataFromCsvFile()
    {
      this.networks.Clear();
      this.transponderNrByFreqInMhz.Clear();
      this.serviceTypeDescriptions.Clear();

      string file = Path.Combine(Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location) ?? "", "lookup.csv");
      if (!File.Exists(file))
        return;
      using var reader = new StreamReader(file, System.Text.Encoding.UTF8);
      string line;
      while ((line = reader.ReadLine()) != null)
      {
        var fields = CsvFile.Parse(line, ';');
        if (fields.Count == 0)
          continue;
        switch (fields[0].ToLowerInvariant())
        {
          case "onid": this.ParseNetwork(fields); break;
          case "transp": this.ParseTransponder(fields); break;
          case "dvbc": this.ParseDvbcChannel(fields); break;
          case "servicetype": this.ParseServiceType(fields); break;
        }
      }
    }
    #endregion

    #region AddNetwork()
    private void AddNetwork(NetworkInfo network)
    {
      this.networks[network.OriginalNetworkId] = network;
    }
    #endregion

    #region AddTransponderMapping()
    private void AddTransponderMapping(int transponderNr, int frequencyInMhz)
    {
      this.transponderNrByFreqInMhz[frequencyInMhz] = transponderNr;
    }
    #endregion

    #region AddServiceType()
    public void AddServiceType(int serviceType, string description)
    {
      this.serviceTypeDescriptions[serviceType] = description;
    }
    #endregion

    #region ParseNetwork()
    private void ParseNetwork(IList<string> fields)
    {
      if (fields.Count < 3) 
        return;
      int start = ParseNumber(fields[1]);
      int end = ParseNumber(fields[2]);
      if (start == 0 || end == 0 || start > end)
        return;
      for (int onid = start; onid <= end; onid++)
      {
        var network = new NetworkInfo();
        network.OriginalNetworkId = onid;
        if (fields.Count >= 4)
          network.Name = fields[3];
        if (fields.Count >= 5)
          network.Operator = fields[4];
        this.AddNetwork(network);
      }        
    }
    #endregion

    #region ParseNumber()
    private int ParseNumber(string nr)
    {
      int number;
      if (nr.StartsWith("0x"))
        int.TryParse(nr.Substring(2), System.Globalization.NumberStyles.HexNumber, System.Globalization.NumberFormatInfo.InvariantInfo, out number);
      else
        int.TryParse(nr, System.Globalization.NumberStyles.Integer, System.Globalization.NumberFormatInfo.InvariantInfo, out number);
      return number;
    }
    #endregion

    #region ParseTransponder()
    private void ParseTransponder(IList<string> fields)
    {
      if (fields.Count < 3)
        return;
      int nr, freq;
      int.TryParse(fields[1], out nr);
      int.TryParse(fields[2], out freq);
      if (nr == 0 || freq == 0)
        return;
      this.AddTransponderMapping(nr, freq);
    }
    #endregion

    #region ParseDvbcChannel()
    private void ParseDvbcChannel(IList<string> fields)
    {
      if (fields.Count < 3)
        return;
      int.TryParse(fields[1], out var freq);
      if (freq == 0)
        return;
      freq /= 1000;
      this.dvbcChannels[freq] = fields[2];
    }
    #endregion

    #region ParseServiceType()
    private void ParseServiceType(IList<string> fields)
    {
      if (fields.Count < 3) return;
      int serviceType = this.ParseNumber(fields[1]);
      if (serviceType <= 0) return;
      this.AddServiceType(serviceType, fields[2]);
    }
    #endregion

    #region IsRadioTvOrData()
    public SignalSource IsRadioTvOrData(int dvbServiceType)
    {
      switch (dvbServiceType)
      {
        case 0x01: // SD MPEG1
        case 0x11: // MPEG2-HD
        case 0x16: // H264/AVC-SD
        case 0x19: // H264/AVC-HD
        case 0x1F: // UHD (future use)
        case 0x9F: // UHD (user defined)
        case 0xD3: // Option (Sky channels)
          return SignalSource.Tv;
        case 0x02:
        case 0x0A:
          return SignalSource.Radio;
      }

      return SignalSource.Data;
    }
    #endregion

    #region GetDvbtTransponder()
    public int GetDvbtTransponder(decimal freqInMhz)
    {
      return (int)(freqInMhz - 306)/8;
    }
    #endregion   

    #region GetDvbtFrequency()
    public decimal GetDvbtFrequency(int channelTransponder)
    {
      return channelTransponder * 8 + 306;
    }
    #endregion

    #region GetDvbcTransponder()
    public int GetDvbcTransponder(decimal freqInMhz)
    {
      return (int)(freqInMhz - 106) / 8;
    }
    #endregion

    #region GetDvbcChannelName()
    public string GetDvbcChannelName(decimal freqInMhz)
    {
      // in case the parameter is in Hz or kHz, correct it to MHz to avoid overflow errors. 2 GHz is the largest plausible frequency
      if (freqInMhz > 2000)
        freqInMhz /= 1000;
      if (freqInMhz > 2000)
        freqInMhz /= 1000;

      return dvbcChannels.TryGet((int)freqInMhz) 
             ?? dvbcChannels.TryGet((int)freqInMhz - 1)
             ?? dvbcChannels.TryGet((int)freqInMhz - 2)
             //?? dvbcChannels.TryGet((int)freqInMhz - 3)
             ?? dvbcChannels.TryGet((int)freqInMhz + 1)
             ?? dvbcChannels.TryGet((int)freqInMhz + 2)
             //?? dvbcChannels.TryGet((int)freqInMhz + 3)
             ?? "";      
    }
    #endregion
  }
}