RBDataHolder.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Data;
using System.IO;

namespace SeciControls
{
    class RBDataHolder
    {
        int indexTeamInst = -1;
        int indexTeamRole = -1;
        int indexTeamExptr = -1;
        int indexTeamInstitute = -1;
        int indexTeamRBPart = -1;

        int indexDataRBPart = -1;
        int indexDateStart = -1;
        int indexDateDuration = -1;

        private String _instName = "";
        private DataSet _teamData;
        private DataSet _dateData;
        private List<Tuple<DateTime, DateTime>> _dates;

        public String InstName { get { return _instName; } }
        public List<Tuple<DateTime, DateTime>> Dates { get { return _dates; } }

        public RBDataHolder()
        {
            getInstName();
            getData();
        }

        private void getInstName()
        {
            //Get the instrument name
            String inst = Environment.MachineName.ToUpper();

            if (inst.StartsWith("NDX"))
            {
                if (inst == "NDXEVS")
                {
                    inst = "VESUVIO";
                }
                else if (inst == "NDXENGINX")
                {
                    inst = "ENGIN_X";
                }
                else
                {
                    inst = inst.Substring(3);
                }

                _instName = inst;
            }
            else
            {
                //For testing purposes
                _instName = "LET";
            }
        }

        private void getData()
        {
            FileInfo fiTeam = new FileInfo(Seci.Definitions.Status.RBDataFilepath + "\\" + _instName + "Team.xml");
            FileInfo fiDates = new FileInfo(Seci.Definitions.Status.RBDataFilepath + "\\" + _instName + "Dates.xml");

            try
            {
                loadData(fiTeam, fiDates);
            }
            catch (Exception e)
            {
                throw e;
            }
        }

        private void getColumns(DataSet team, DataSet data)
        {
            //From team
            for (int i = 0; i < team.Tables[1].Columns.Count; ++i)
            {
                if (team.Tables[1].Columns[i].ColumnName.ToLower() == "ows_instrument")
                {
                    indexTeamInst = i;
                }
                else if (team.Tables[1].Columns[i].ColumnName.ToLower() == "ows_role")
                {
                    indexTeamRole = i;
                }
                else if (team.Tables[1].Columns[i].ColumnName.ToLower() == "ows_organisation")
                {
                    indexTeamInstitute = i;
                }
                else if (team.Tables[1].Columns[i].ColumnName.ToLower() == "ows_title")
                {
                    indexTeamExptr = i;
                }
                else if (team.Tables[1].Columns[i].ColumnName.ToLower() == "ows_rbpart")
                {
                    indexTeamRBPart = i;
                }
            }

            //From data
            for (int i = 0; i < data.Tables[1].Columns.Count; ++i)
            {
                if (data.Tables[1].Columns[i].ColumnName.ToLower() == "ows_scheduled_date")
                {
                    indexDateStart = i;
                }
                else if (data.Tables[1].Columns[i].ColumnName.ToLower() == "ows_rbpart")
                {
                    indexDataRBPart = i;
                }
                else if (data.Tables[1].Columns[i].ColumnName.ToLower() == "ows_time_allocated")
                {
                    indexDateDuration = i;
                }
            }
        }

        private void loadData(FileInfo team, FileInfo dates)
        {
            if (_teamData == null) _teamData = new DataSet();
            if (_dateData == null) _dateData = new DataSet();

            if (team.Exists && dates.Exists)
            {
                _teamData = RBDataHolder.loadData(team.FullName);
                _dateData = RBDataHolder.loadData(dates.FullName);

                if (_teamData.Tables.Count == 0 || _dateData.Tables.Count == 0)
                {
                    throw new DataException();
                }

                //Get columns
                getColumns(_teamData, _dateData);
            }
        }

        private static DataSet loadData(String filepath)
        {
            DataSet data = new DataSet();
            data.ReadXml(filepath);
            return data;
        }

        public List<string> GetExperimenters()
        {
            List<string> experimenters = new List<string>();

            for (int i = 0; i < _teamData.Tables[1].Rows.Count; ++i)
            {
                //check it has an RB part number ie. 123456_1
                if (!String.IsNullOrEmpty(_teamData.Tables[1].Rows[i].ItemArray[indexTeamRBPart].ToString()))
                {
                    String inst = _teamData.Tables[1].Rows[i].ItemArray[indexTeamInst].ToString();
                    String exptr = _teamData.Tables[1].Rows[i].ItemArray[indexTeamExptr].ToString();
                    String institute = _teamData.Tables[1].Rows[i].ItemArray[indexTeamInstitute].ToString();

                    if (!experimenters.Contains(exptr + " (" + institute + ")") && inst.ToLower() == _instName.ToLower())
                    {
                        experimenters.Add(exptr + " (" + institute + ")");
                    }
                }
            }

            experimenters.Sort();
            return experimenters;
        }

        public List<string> GetDates(String Exptr)
        {
            _dates = new List<Tuple<DateTime, DateTime>>();
            List<string> dateStrings = new List<string>();

            for (int i = 0; i < _teamData.Tables[1].Rows.Count; ++i)
            {
                String inst = _teamData.Tables[1].Rows[i].ItemArray[indexTeamInst].ToString();
                String exptr = _teamData.Tables[1].Rows[i].ItemArray[indexTeamExptr].ToString();
                String rb = _teamData.Tables[1].Rows[i].ItemArray[indexTeamRBPart].ToString();

                if (inst.ToLower() == _instName.ToLower() && Exptr.StartsWith(exptr) && !String.IsNullOrEmpty(rb))
                {
                    Tuple<DateTime, DateTime> allocated = getAllocatedDates(rb);
                    _dates.Add(allocated);

                    if (allocated != null)
                    {
                        dateStrings.Add(allocated.Item1.ToShortDateString() + " - " + allocated.Item2.ToShortDateString() + "   [RB = " + rb.Substring(0, rb.IndexOf('_')) + "]");
                    }
                    else
                    {
                        dateStrings.Add("Unknown [RB = " + rb.Substring(0, rb.IndexOf('_')) + "]");
                    }
                }
            }

            return dateStrings;
        }

        private Tuple<DateTime, DateTime> getAllocatedDates(String rbpart)
        {
            for (int i = 0; i < _dateData.Tables[1].Rows.Count; ++i)
            {
                if (_dateData.Tables[1].Rows[i].ItemArray[indexDataRBPart].ToString() == rbpart)
                {
                    try
                    {
                        DateTime start = Convert.ToDateTime(_dateData.Tables[1].Rows[i].ItemArray[indexDateStart].ToString());
                        String duration = _dateData.Tables[1].Rows[i].ItemArray[indexDateDuration].ToString();
                        DateTime end = start.AddDays(Double.Parse(duration));
                        return new Tuple<DateTime, DateTime>(start, end);
                    }
                    catch
                    {
                        return null;
                    }
                }
            }
            return null;
        }

        private List<Tuple<DateTime, DateTime>> getPossibleDates(String rb)
        {
            List<Tuple<DateTime, DateTime>> dates = new List<Tuple<DateTime, DateTime>>();

            for (int i = 0; i < _dateData.Tables[1].Rows.Count; ++i)
            {
                if (_dateData.Tables[1].Rows[i].ItemArray[indexDataRBPart].ToString().StartsWith(rb))
                {
                    try
                    {
                        DateTime start = Convert.ToDateTime(_dateData.Tables[1].Rows[i].ItemArray[indexDateStart].ToString());
                        String duration = _dateData.Tables[1].Rows[i].ItemArray[indexDateDuration].ToString();
                        DateTime end = start.AddDays(Double.Parse(duration));
                        dates.Add(new Tuple<DateTime, DateTime>(start, end));
                    }
                    catch
                    {
                        //Do nothing
                    }
                }
            }
            return dates;
        }

        public bool DateOK(Tuple<DateTime, DateTime> daterange)
        {
            DateTime today = DateTime.Now;
            if (today >= daterange.Item1.Subtract(new TimeSpan(Seci.Definitions.Status.RBDateLeeway, 0, 0, 0))
                && today <= daterange.Item2.AddDays(Seci.Definitions.Status.RBDateLeeway))
            {
                return true;
            }
            else
            {
                return false;
            }
        }

        public Boolean RetrieveUsersAndCheckDate(String rbnumber, ref List<User> users)
        {
            users = RetrieveUsersByRb(rbnumber);

            Boolean dateokay = false;

            var dates = getPossibleDates(rbnumber);
            foreach(var date in dates)
            {
                if (DateOK(date))
                {
                    dateokay = true;
                    break;
                }
            }

            return dateokay;
        }

        public List<User> RetrieveUsersByRb(String rbnumber)
        {
            if (_teamData.Tables.Count <= 1)
            {
                return new List<User>();
            }

            List<User> PIs = new List<User>();
            List<User> contacts = new List<User>();
            List<User> users = new List<User>();

            for (int i = 0; i < _teamData.Tables[1].Rows.Count; ++i)
            {
                if (_teamData.Tables[1].Rows[i].ItemArray[indexTeamRBPart].ToString().StartsWith(rbnumber))
                {
                    String exptr = _teamData.Tables[1].Rows[i].ItemArray[indexTeamExptr].ToString();
                    String role = _teamData.Tables[1].Rows[i].ItemArray[indexTeamRole].ToString();
                    String institute = _teamData.Tables[1].Rows[i].ItemArray[indexTeamInstitute].ToString();

                    if (role.ToLower() == "pi")
                    {
                        PI temp = new PI(exptr, institute);
                        if (!PIs.Contains(temp))
                        {
                            PIs.Add(temp);
                        }
                    }
                    else if (role.ToLower() == "contact")
                    {
                        Contact temp = new Contact(exptr, institute);
                        if (!contacts.Contains(temp))
                        {
                            contacts.Add(temp);
                        }
                    }
                    else
                    {
                        User temp = new User(exptr, institute);
                        if (!users.Contains(temp))
                        {
                            users.Add(temp);
                        }
                    }
                }
            }

            //Make sure the order is PI(s), Contact(s), User(s)
            users.InsertRange(0, contacts);
            users.InsertRange(0, PIs);

            return users;
        }
    }
}