MgrQueryGeneral.cs

using MyCaffe.basecode;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
 
namespace MyCaffe.db.stream
{
    public class MgrQueryGeneral : IXQuery
    {
        CustomQueryCollection m_colCustomQuery = new CustomQueryCollection();
        PropertySet m_schema;
        CancelEvent m_evtCancel = new CancelEvent();
        IXCustomQuery m_iquery;
 
        public MgrQueryGeneral(string strSchema, List<IXCustomQuery> rgCustomQueries)
        {
            m_colCustomQuery.Load();
            m_schema = new PropertySet(strSchema);
 
            m_colCustomQuery.Add(new StandardQueryTextFile());
            m_colCustomQuery.Add(new StandardQueryWAVFile());
 
            foreach (IXCustomQuery icustomquery in rgCustomQueries)
            {
                m_colCustomQuery.Add(icustomquery);
            }
 
            int nConnections = m_schema.GetPropertyAsInt("ConnectionCount");
            if (nConnections != 1)
                throw new Exception("Currently, the general query type only supports 1 connection.");
 
            string strConTag = "Connection0";
            string strCustomQuery = m_schema.GetProperty(strConTag + "_CustomQueryName");
            string strCustomQueryParam = m_schema.GetProperty(strConTag + "_CustomQueryParam");
           
            IXCustomQuery iqry = m_colCustomQuery.Find(strCustomQuery);
            if (iqry == null)
                throw new Exception("Could not find the custom query '" + strCustomQuery + "'!");
 
            if (iqry.QueryType != CUSTOM_QUERY_TYPE.BYTE && iqry.QueryType != CUSTOM_QUERY_TYPE.REAL_FLOAT && iqry.QueryType != CUSTOM_QUERY_TYPE.REAL_DOUBLE)
                throw new Exception("The custom query '" + iqry.Name + "' must support the 'CUSTOM_QUERY_TYPE.BYTE' or 'CUSTOM_QUERY_TYPE.REAL_FLOAT' or 'CUSTOM_QUERY_TYPE.REAL_DOUBLE' query type!");
 
            string strParam = ParamPacker.UnPack(strCustomQueryParam);
            m_iquery = iqry.Clone(strParam);
            m_iquery.Open();
        }
 
        public void AddDirectQuery(IXCustomQuery iqry)
        {
            m_colCustomQuery.Add(iqry);
        }
 
        public void Reset(int nStartOffset)
        {
            m_iquery.Reset();
        }
 
        public void Shutdown()
        {
            m_evtCancel.Set();
            m_iquery.Close();
        }
 
        public List<int> GetQuerySize()
        {
            return m_iquery.GetQuerySize();
        }
 
        public SimpleDatum Query(int nWait)
        {
            if (m_iquery.QueryType == CUSTOM_QUERY_TYPE.REAL_DOUBLE)
            {
                List<double[]> rgData = m_iquery.QueryRealD();
 
                if (rgData == null || rgData.Count == 0 || rgData[0].Length == 0)
                    return null;
 
                double[] rgFullData = new double[rgData[0].Length * rgData.Count];
                int nOffset = 0;
 
                for (int i = 0; i < rgData.Count; i++)
                {
                    Array.Copy(rgData[i], 0, rgFullData, nOffset, rgData[i].Length);
                    nOffset += rgData[i].Length;
                }
 
                return new SimpleDatum(true, rgData.Count, 1, rgData[0].Length, -1, DateTime.MinValue,  rgFullData.ToList(), 0, false, -1);
            }
            else if (m_iquery.QueryType == CUSTOM_QUERY_TYPE.REAL_FLOAT)
            {
                List<float[]> rgData = m_iquery.QueryRealF();
 
                if (rgData == null || rgData.Count == 0 || rgData[0].Length == 0)
                    return null;
 
                float[] rgFullData = new float[rgData[0].Length * rgData.Count];
                int nOffset = 0;
 
                for (int i = 0; i < rgData.Count; i++)
                {
                    Array.Copy(rgData[i], 0, rgFullData, nOffset, rgData[i].Length);
                    nOffset += rgData[i].Length;
                }
 
                return new SimpleDatum(true, rgData.Count, 1, rgData[0].Length, -1, DateTime.MinValue, rgFullData.ToList(), 0, false, -1);
            }
            else
            {
                byte[] rgData = m_iquery.QueryBytes();
 
                if (rgData == null || rgData.Length == 0)
                    return null;
 
                return new SimpleDatum(false, rgData.Length, 1, 1, -1, DateTime.MinValue, rgData.ToList(), 0, false, -1);
            }
        }
 
        public byte[] ConvertOutput(float[] rg, out string type)
        {
            return m_iquery.ConvertOutput(rg, out type);
        }
 
        public Dictionary<string, float> QueryInfo()
        {
            return m_iquery.QueryInfo();
        }
    }
}