using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Autodesk.Revit.DB;
using Autodesk.Revit.DB.Analysis;

namespace Revit.SDK.Samples.NetworkPressureLossReport
{
   public class AVFViewer
   {
      private bool? m_isItemzied;
      private View m_view;
      private SpatialFieldManager m_sfm;
      private int m_schemaIdx;
      private XYZ m_maxCorner;
      const string DisplayStyleName = "NetworkFlowDisplayStyle";
      const String SchemaName = "NetworkFlowSchema";
      
      public AVFViewer(View view, bool? isItemized)
      {
         m_view = view;
         m_isItemzied = isItemized;

         m_sfm = SpatialFieldManager.GetSpatialFieldManager(view);
         if (m_sfm == null)
         {
            m_sfm = SpatialFieldManager.CreateSpatialFieldManager(view, 1); 
         }
         m_sfm.Clear();
         m_maxCorner = new XYZ(-Double.MaxValue, -Double.MaxValue, -Double.MaxValue);
      }
      public Document Document
      {
         get { return m_view.Document; }
      }
      public double Scale
      {
         get { return m_view.Scale; }
      }
      public bool IsItemized
      {
         get { return m_isItemzied == true; }
      }
      public void InitAVF()  
      {
         m_view.EnableTemporaryViewPropertiesMode(m_view.Id);
         m_view.TemporaryViewModes.RemoveCustomization();
         m_view.TemporaryViewModes.CustomTitle = "Network Flow Analysis";

         AnalysisResultSchema resultSchema = new AnalysisResultSchema(SchemaName, "");
         m_sfm.SetMeasurementNames(new List<string>() { SchemaName });
         m_schemaIdx = m_sfm.RegisterResult(resultSchema);
      }
      public int AddData(List<XYZ> points, List<VectorAtPoint> valList)
      {
         int idx = m_sfm.AddSpatialFieldPrimitive();
         FieldDomainPointsByXYZ pnts = new FieldDomainPointsByXYZ(points);
         FieldValues vals = new FieldValues(valList);
         m_sfm.UpdateSpatialFieldPrimitive(idx, pnts, vals, m_schemaIdx);
         return idx;
      }
      public void AddCorner(double maxX, double maxY, double maxZ)
      {
         double xx = Math.Max(m_maxCorner.X, maxX);
         double yy = Math.Max(m_maxCorner.Y, maxY);  
         double zz = Math.Max(m_maxCorner.Z, maxZ);
         m_maxCorner = new XYZ(xx, yy, zz);
      }
      private AnalysisDisplayStyle getStyleByName(string name)
      {
         FilteredElementCollector collector = new FilteredElementCollector(Document);
         ICollection<Element> collection = collector.OfClass(typeof(AnalysisDisplayStyle)).ToElements();
         var displayStyle = from element in collection
                              where element.Name == name
                              select element;
         AnalysisDisplayStyle analysisDisplayStyle = null;
         if (displayStyle.Count() != 0)
            analysisDisplayStyle = displayStyle.Cast<AnalysisDisplayStyle>().ElementAt<AnalysisDisplayStyle>(0);
         return analysisDisplayStyle;
      }
      public void FinishDisplayStyle()
      {
         // set the legend to the top right corner so it is close to the AVF display
         m_sfm.LegendPosition = m_maxCorner;

         AnalysisDisplayStyle analysisDisplayStyle = getStyleByName(DisplayStyleName);

         // If display style does not already exist in the document, create it
         AnalysisDisplayColorSettings colorSettings = new AnalysisDisplayColorSettings();
         if (analysisDisplayStyle == null)
         {
            colorSettings.MaxColor = new Color(255, 0, 0);
            colorSettings.MinColor = new Color(0, 255, 0);

            AnalysisDisplayVectorSettings vectorSettings = new AnalysisDisplayVectorSettings();
            vectorSettings.VectorOrientation = AnalysisDisplayStyleVectorOrientation.Linear;
            vectorSettings.ArrowheadScale = AnalysisDisplayStyleVectorArrowheadScale.NoScaling;
            vectorSettings.ArrowLineWeight = 4;
            vectorSettings.VectorTextType = AnalysisDisplayStyleVectorTextType.ShowNone;

            AnalysisDisplayLegendSettings legendSettings = new AnalysisDisplayLegendSettings();
            legendSettings.ShowLegend = false;

            analysisDisplayStyle = AnalysisDisplayStyle.CreateAnalysisDisplayStyle(Document, DisplayStyleName, vectorSettings, colorSettings, legendSettings);
         }

         m_view.AnalysisDisplayStyleId = analysisDisplayStyle.Id;

         m_view.TemporaryViewModes.CustomColor = analysisDisplayStyle.GetColorSettings().MaxColor;

         // Transparent everything so we can see the flow vector.
         ElementId rasterId = new ElementId(BuiltInCategory.OST_RasterImages);
         foreach (Category c in m_view.Document.Settings.Categories)
         {
            if (!m_view.GetCategoryHidden(c.Id))
            {  
               if (c.Id != rasterId && m_view.IsCategoryOverridable(c.Id))
               {
                  OverrideGraphicSettings ogs = m_view.GetCategoryOverrides(c.Id);
                  if (!ogs.Halftone)
                  {
                     m_view.SetCategoryOverrides(c.Id, ogs.SetHalftone(true));
                     m_view.SetCategoryOverrides(c.Id, ogs.SetSurfaceTransparency(50));
                  }
               }
            }
         }
      }

   }
}