﻿//
// BIM IFC library: this library works with Autodesk(R) Revit(R) to export IFC files containing model geometry.
// Copyright (C) 2012  Autodesk, Inc.
// 
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library 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
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
//

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using Autodesk.Revit.DB.IFC;

namespace BIM.IFC.Utility
{
    /// <summary>
    /// Used to keep a cache of the IfcRoot handles mapping to a IfcMaterialLayerSetUsage handle.
    /// </summary>
    public class MaterialLayerRelationsCache : Dictionary<IFCAnyHandle, HashSet<IFCAnyHandle>>
    {
        /// <summary>
        /// Adds the IfcRoot handle to the dictionary.
        /// </summary>
        /// <param name="material">
        /// The material handle.
        /// </param>
        /// <param name="product">
        /// The product handle.
        /// </param>
        public void Add(IFCAnyHandle material, IFCAnyHandle product)
        {
            if (ContainsKey(material))
            {
                this[material].Add(product);
            }
            else
            {
                HashSet<IFCAnyHandle> products = new HashSet<IFCAnyHandle>();
                products.Add(product);
                this[material] = products;
            }
        }
    }
}
