#include <TROOT.h>

#include <cometEventLoop.hxx>

class IDumpEvent: public COMET::ICOMETEventLoopFunction {
public:
    IDumpEvent() {
        fLSOption = "";
        fQuiet = false;
        fPrintList = false;
        fTargetDatum=".";
    }

    virtual ~IDumpEvent() {};

    void Usage(void) {
        std::cout << "    -O quiet    Don't call TObject::ls()"
                  << std::endl;
        std::cout << "    -O list     List events in the file"
                  << std::endl;
        std::cout << "    -O datum <target>   Print only the datum named 'target'"
                  << std::endl;
        std::cout << "    -O <option> Set the option for TObject::ls()"
                  << std::endl;
    }

    virtual bool SetOption(std::string option,std::string value="") {
        if (option == "quiet") fQuiet = true;
        else if (option == "datum") fTargetDatum = value;
        else if (option == "list") {
            fPrintList = true;
        }
        else fLSOption = option;
        return true;
    }

    bool operator () (COMET::ICOMETEvent& event) {
        if (COMET::ICOMETLog::GetLogLevel()>COMET::ICOMETLog::QuietLevel && !fQuiet) {
            COMET::IHandle<COMET::IDatum> target 
                = event.Get<COMET::IDatum>(fTargetDatum.c_str());
            if(target) {
                target->ls(fLSOption.c_str());
            }
            else {
                COMETLog("No IDatum called '"<<fTargetDatum<<"' exists for event number "<<event.GetEventId());
            }
        }
        if (fPrintList) {
            COMETLog("Event " << event.GetContext());
        }
        return true;
    }

    // Do nothing... This is here to test compiler warnings.  The warning
    // can be prevented by adding
    //
    // using COMET::ICOMETEventLoopFunction::Finalize;
    void Finalize(COMET::ICOMETOutput*const output) {
    }

private:
    std::string fLSOption;
    std::string fTargetDatum;
    bool fQuiet;
    bool fPrintList;
};

int main(int argc, char **argv) {
    IDumpEvent userCode;
    COMET::cometEventLoop(argc,argv,userCode,1);
}