/*

  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

  *

  * Copyright (c) 2017 Oracle and/or its affiliates. All rights reserved.

  *

  * Oracle and Java are registered trademarks of Oracle and/or its affiliates.

  * Other names may be trademarks of their respective owners.

  *

  * The contents of this file are subject to the terms of either the GNU

  * General Public License Version 2 only ("GPL") or the Common

  * Development and Distribution License("CDDL") (collectively, the

  * "License"). You may not use this file except in compliance with the

  * License. You can obtain a copy of the License at

  * http://www.netbeans.org/cddl-gplv2.html

  * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the

  * specific language governing permissions and limitations under the

  * License.  When distributing the software, include this License Header

  * Notice in each file and include the License file at

  * nbbuild/licenses/CDDL-GPL-2-CP.  Oracle designates this

  * particular file as subject to the "Classpath" exception as provided

  * by Oracle in the GPL Version 2 section of the License file that

  * accompanied this code. If applicable, add the following below the

  * License Header, with the fields enclosed by brackets [] replaced by

  * your own identifying information:

  * "Portions Copyrighted [year] [name of copyright owner]"

  *

  * If you wish your version of this file to be governed by only the CDDL

  * or only the GPL Version 2, indicate your decision by adding

  * "[Contributor] elects to include this software in this distribution

  * under the [CDDL or GPL Version 2] license." If you do not indicate a

  * single choice of license, a recipient has the option to distribute

  * your version of this file under either the CDDL, the GPL Version 2 or

  * to extend the choice of license to its licensees as provided above.

  * However, if you add GPL Version 2 code and therefore, elected the GPL

  * Version 2 license, then the option applies only if the new code is

  * made subject to such option by the copyright holder.

  *

  * Contributor(s): Ilia Gromov

  */

 package com.sun.tools.ide.analysis.modernize.impl;

 import java.util.Collection;

 import java.util.HashMap;

 import java.util.HashSet;

 import java.util.Map;

 import java.util.Set;

 import org.netbeans.modules.cnd.api.model.CsmFile;

 /**

  * This cache is designed to help Editor showing correct checks for included

  * files.

  *

  * @param <T>

  */

 public class AnalyzedFiles {

     private final Map<CsmFile, Set<CsmFile>> hierarchyCache = new HashMap<CsmFile, Set<CsmFile>>();

 //    private final Map<FileObject, Set<CsmErrorInfo>> diagnosticsCache = new HashMap<FileObject, Set<CsmErrorInfo>>();

 //    public void cacheDiagnostics(FileObject fo, CsmErrorInfo info) {

 //        Set<CsmErrorInfo> alreadyMapped = diagnosticsCache.get(fo);

 //

 //        if (alreadyMapped == null) {

 //            alreadyMapped = new HashSet<CsmErrorInfo>();

 //            diagnosticsCache.put(fo, alreadyMapped);

 //        }

 //

 //        alreadyMapped.add(info);

 //    }

     public void cacheHierarchy(CsmFile startFile, Collection<CsmFile> includedFiles) {

         Set<CsmFile> alreadyMapped = hierarchyCache.get(startFile);

         if (alreadyMapped == null) {

             alreadyMapped = new HashSet<CsmFile>();

             hierarchyCache.put(startFile, alreadyMapped);

         }

         alreadyMapped.addAll(includedFiles);

     }

 //    public Set<CsmErrorInfo> getDiagnostics(FileObject fo) {

 //        return diagnosticsCache.get(fo);

 //    }

     public Collection<? extends CsmFile> getStartFiles(CsmFile includedFile) {

         Set<CsmFile> result = new HashSet<CsmFile>();

         for (Map.Entry<CsmFile, Set<CsmFile>> entry : hierarchyCache.entrySet()) {

             CsmFile key = entry.getKey();

             Set<CsmFile> value = entry.getValue();

             if (value.contains(includedFile)) {

                 result.add(key);

             }

         }

         return result;

     }

     public boolean isStartFile(CsmFile file) {

         return hierarchyCache.containsKey(file);

     }

     public void clear() {

         hierarchyCache.clear();

     }

     public static AnalyzedFiles getDefault() {

         return HOLDER.INSTANCE;

     }

     private static class HOLDER {

         public static final AnalyzedFiles INSTANCE = new AnalyzedFiles();

     }

 }