build-system/gradle-experimental/src/main/groovy/com/android/build/gradle/ndk/internal/StlNativeToolSpecification.java

/*
 * Copyright (C) 2014 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.build.gradle.ndk.internal;

import com.google.common.collect.Lists;

import org.gradle.nativeplatform.platform.NativePlatform;

import java.io.File;
import java.util.Collections;
import java.util.List;

/**
 * Compiler flags to configure STL.
 */
public class StlNativeToolSpecification extends AbstractNativeToolSpecification {
    private NdkHandler ndkHandler;
    private String stl;
    private String stlName;
    private Boolean isStatic;
    private NativePlatform platform;

    StlNativeToolSpecification(NdkHandler ndkHandler, String stl, NativePlatform platform) {
        this.ndkHandler = ndkHandler;
        this.stl = stl;
        this.stlName = stl.equals("system") ? stl : stl.substring(0, stl.indexOf('_'));
        this.isStatic = stl.endsWith("_static");
        this.platform = platform;
    }


    @Override
    public Iterable<String> getCFlags() {
        return Collections.emptyList();
    }

    @Override
    public Iterable<String> getCppFlags() {

        List<String> cppFlags = Lists.newArrayList();

        List<String> includeDirs = Lists.newArrayList();
        if (stlName.equals("system")) {
            includeDirs.add("system/include");
        } else if (stlName.equals("stlport")) {
            includeDirs.add("stlport/stlport");
            includeDirs.add("gabi++/include");
        } else if (stlName.equals("gnustl")) {
            String gccToolchainVersion = ndkHandler.getGccToolchainVersion(platform.getName());
            includeDirs.add("gnu-libstdc++/" + gccToolchainVersion + "/include");
            includeDirs.add("gnu-libstdc++/" + gccToolchainVersion +
                    "/libs/" + platform.getName() + "/include");
            includeDirs.add("gnu-libstdc++/" + gccToolchainVersion +
                    "/include/backward");
        } else if (stlName.equals("gabi++")) {
            includeDirs.add("gabi++/include");
        } else if (stlName.equals("c++")) {
            includeDirs.add("llvm-libc++/libcxx/include");
            includeDirs.add("gabi++/include");
            includeDirs.add("../android/support/include");
            cppFlags.add("-std=c++11");
        }

        for (String dir : includeDirs) {
            cppFlags.add("-I" +
                    new File(StlConfiguration.getStlBaseDirectory(ndkHandler), dir).toString());
        }
        return cppFlags;
    }

    @Override
    public Iterable<String> getLdFlags() {
        if (stl.equals("system")) {
            return Collections.emptyList();
        }
        List<String> flags = Lists.newArrayList();
        flags.add(getStlLib(platform.getName()).toString());
        return flags;
    }

    public File getStlLib(String abi) {
        String stlLib;
        if (stlName.equals("stlport")) {
            stlLib = "stlport";
        } else if (stlName.equals("gnustl")) {
            stlLib = "gnu-libstdc++/" + ndkHandler.getGccToolchainVersion(abi);
        } else if (stlName.equals("gabi++")) {
            stlLib = "gabi++";
        } else if (stlName.equals("c++")) {
            stlLib = "llvm-libc++";
        } else {
            throw new AssertionError(
                    "Unreachable.  Either stl is invalid or stl is \"system\", " +
                    "in which case there is no library file and getStlLib should not be called.");
        }
        return new File(
                StlConfiguration.getStlBaseDirectory(ndkHandler),
                stlLib + "/libs/" + platform.getName() + "/lib" + stl + (isStatic ? ".a" : ".so"));
    }
}