#==========================================================================
#
#   Copyright Insight Software Consortium
#
#   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.txt
#
#   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.
#
#==========================================================================*/

import itkBase
import sys
import os

# Put the ITK_DATA_ROOT setting in the global namespace.  This
# package is only used for testing, so this is okay.

ITK_DATA_ROOT = ""

# Look for the -D command line option.
if not ITK_DATA_ROOT:
    for a in range(len(sys.argv)):
        if sys.argv[a] == "-D" and a < len(sys.argv):
            ITK_DATA_ROOT = sys.argv[a + 1]
            break

# Check for the environment variable ::ITK_DATA_ROOT.
if not ITK_DATA_ROOT and 'ITK_DATA_ROOT' in os.environ:
    ITK_DATA_ROOT = os.environ['ITK_DATA_ROOT']


# Use the default output directory.
if not ITK_DATA_ROOT:
    ITK_DATA_ROOT = itkBase.defaultDataRoot
