#include "IFieldMapDescription.hxx"
#include "IFieldMap.hxx"
#include <iostream>
#include <TSystem.h>
#include <TROOT.h>
#include <TString.h>
#include <set>
#include <cstdlib>
using std::string;

namespace {
    std::ostream& operator<<(std::ostream& stream,const TVector3& vect){
        stream<<"("<<vect.x()<<", "<<vect.y()<<", "<<vect.z() <<")";
        return stream;
    }
}

COMET::IFieldMapDescription::PathList COMET::IFieldMapDescription::fFieldmapPaths;

COMET::IFieldMapDescription::IFieldMapDescription():
    COMET::IElementFieldDescription(),
    fFilename(),
    fCheckSum(NULL),
    fScaling(0),
    fGradient(0),
    fTimeOffset(-999999999),
    fTranslation(),
    fRotation(){
        SetupFilePath();
    }

COMET::IFieldMapDescription::IFieldMapDescription(const std::string& fieldMapFile,
        Double_t scale, Double_t gradient,
        Double_t timeOffset, TVector3 translation, 
        TRotation rotation):
    COMET::IElementFieldDescription(),
    fFilename(fieldMapFile),
    fScaling(scale),
    fGradient(gradient),
    fTimeOffset(timeOffset),
    fTranslation(translation),
    fRotation(rotation)
{
    fCheckSum=TMD5::FileChecksum(fFilename.c_str());
    SetupFilePath();
}

void COMET::IFieldMapDescription::SetupFilePath(){
    if(fFieldmapPaths.empty()){
        fFieldmapPaths.push_back("$PWD");
        fFieldmapPaths.push_back("$ICEDUST_FIELDMAPS");
        fFieldmapPaths.push_back("$OAEMFIELDROOT/fieldmaps");
        fFieldmapPaths.push_back("$ICEDUST_LOCAL_STORAGE/fieldmaps");
    }
}

COMET::IFieldMapDescription::~IFieldMapDescription(){
    if(fCheckSum) delete fCheckSum;
}

void COMET::IFieldMapDescription::ls(Option_t* ) const{
    std::cout<< fFilename<<"("<<fCheckSum->AsString()<<"): ";
    TROOT::IncreaseDirLevel();
    std::string indent(TROOT::GetDirLevel(),' ');
    if(fScaling!=1.0) {
        std::cout<<"\n"<<indent<<"Scaled: "<<fScaling;
    }
    if(fTranslation!=TVector3(0,0,0)){
        std::cout<<"\n"<<indent<<"Translation: ("<<fTranslation.x()<<", "<<fTranslation.y()<<", "<<fTranslation.z()<<")";
    }
    if(!fRotation.IsIdentity()){
        double angle = 0;
        TVector3 axis;
        fRotation.AngleAxis(angle,axis);
        std::cout<<"\n"<<indent<<"Direction: angle="<<angle<<", axis=("<<axis.x()<<", "<<axis.y()<<", "<<axis.z()<<")";
    }
    std::cout<<std::endl;
    TROOT::DecreaseDirLevel();
}

COMET::IElementField* COMET::IFieldMapDescription::Recreate()const{
    std::string filename=FindFullFilename();
    if (filename.empty()) return NULL;
    return new COMET::IFieldMap(filename,fScaling,fGradient,fTimeOffset,fTranslation,fRotation);
}

std::string COMET::IFieldMapDescription::FindFullFilename()const{
    // Remove any prefixed directories
    size_t last_slash=fFilename.find_last_of('/');
    string basename;
    if(last_slash==fFilename.length()){
        COMETError("Seem to have a directory as a fieldmap file path.  Weird....");
        return "";
    }
    if(last_slash!=string::npos) {
        basename=fFilename.substr(last_slash+1);
    }else basename=fFilename;

    std::vector<TString> bad_checksums;
    for(PathList::const_iterator i_path=fFieldmapPaths.begin();
            i_path!=fFieldmapPaths.end();++i_path){
        TString expanded=(*i_path)+"/"+basename;
        gSystem->ExpandPathName(expanded);
        if(!gSystem->AccessPathName(expanded.Data())){
            TMD5* checksum=TMD5::FileChecksum(expanded.Data());
            if(string(checksum->AsString()) == fCheckSum->AsString()){
                return string(expanded.Data());
            }else{
                bad_checksums.push_back(expanded);
            }
        }
    }

    COMETError("Unable to find fieldmap file, '"<<fFilename<<"' in any of the following places:");
    PrintFieldmapSearchPaths(true);

    if(!bad_checksums.empty()){
        COMETError("The following files matched the name but didn't match the MD5 check sum:");
            for(std::vector<TString>::const_iterator i_path=bad_checksums.begin();
                    i_path!=bad_checksums.end();++i_path){
                COMETError("    "<<*i_path);
            }
    }
    return "";
}

void COMET::IFieldMapDescription::PrintFieldmapSearchPaths(bool use_error_stream){
    SetupFilePath();
    for(PathList::const_iterator i_path=fFieldmapPaths.begin();
            i_path!=fFieldmapPaths.end();++i_path){
        if(use_error_stream) COMETError("    "<<*i_path);
        else                 COMETLog  ("    "<<*i_path);
    }
}