1 /*

     2  * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.

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

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.  Oracle designates this

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

     9  * by Oracle in the LICENSE file that accompanied this code.

    10  *

    11  * This code is distributed in the hope that it will be useful, but WITHOUT

    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    14  * version 2 for more details (a copy is included in the LICENSE file that

    15  * accompanied this code).

    16  *

    17  * You should have received a copy of the GNU General Public License version

    18  * 2 along with this work; if not, write to the Free Software Foundation,

    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    20  *

    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    22  * or visit www.oracle.com if you need additional information or have any

    23  * questions.

    24  */

    25 

    26 package jdk.jshell;

    27 

    28 import java.util.LinkedHashMap;

    29 import java.util.List;

    30 import javax.tools.Diagnostic;

    31 import javax.tools.JavaFileObject;

    32 import jdk.jshell.Wrap.CompoundWrap;

    33 

    34 /**

    35  * The outer wrap for a set of snippets wrapped in a generated class

    36  * @author Robert Field

    37  */

    38 class OuterSnippetsClassWrap extends OuterWrap {

    39 

    40     private final String className;

    41     private final LinkedHashMap<Wrap, Snippet> wrapToSnippet;

    42 

    43     OuterSnippetsClassWrap(String className, CompoundWrap w, List<Snippet> snippets, List<Wrap> wraps) {

    44         super(w);

    45         assert snippets == null || snippets.size() == wraps.size();

    46         this.className = className;

    47         wrapToSnippet = new LinkedHashMap<>();

    48         for (int i = 0; i < snippets.size(); ++i) {

    49             wrapToSnippet.put(wraps.get(i), snippets.get(i));

    50         }

    51     }

    52 

    53     public Snippet wrapLineToSnippet(int wline) {

    54         Wrap wrap = ((CompoundWrap)w).wrapLineToWrap(wline);

    55         return wrapToSnippet.get(wrap);

    56     }

    57 

    58     @Override

    59     Diag wrapDiag(Diagnostic<? extends JavaFileObject> d) {

    60         return new WrappedDiagnostic(d) {

    61 

    62             @Override

    63             Snippet snippetOrNull() {

    64                 Wrap wrap = ((CompoundWrap) w).wrapIndexToWrap(diag.getPosition());

    65                 Snippet sn = wrapToSnippet.get(wrap);

    66                 if (sn != null) {

    67                     return sn;

    68                 } else {

    69                     return super.snippetOrNull();

    70                 }

    71             }

    72         };

    73     }

    74 

    75     int ordinal(Snippet sn) {

    76         int i = 0;

    77         for (Snippet si : wrapToSnippet.values()) {

    78             if (si == sn) {

    79                 return i;

    80             }

    81             ++i;

    82         }

    83         return -1;

    84     }

    85 

    86     @Override

    87     public String className() {

    88         return className;

    89     }

    90 

    91     @Override

    92     public String toString() {

    93         return "OSCW(" + className + ":" + w + ")";

    94     }

    95 }