#include "TString.h"
#include "TDatabasePDG.h"
#include "TParticlePDG.h"
#include "TObjArray.h"
#include "TObjString.h"

#include <iostream>
#include <iomanip>
#include <fstream>

#include <regex>
#include <string>
#include <vector>
#include <map>
using namespace std;

struct part {

    TString  sid;
    Double_t mass;
    Double_t width;
    //Int_t    parity;
    TString  parity;
    vector<Int_t> vpid;
    vector<TString> vnpid;
    vector<Int_t> vgpid;

    void  reset()
    {
	sid    = "not_set";
	mass   = -1;
	width  = -1;
	parity = "";
	vpid.clear();
	vnpid.clear();
	vgpid.clear();
    }

    void  print()
    {
	cout<<setw(15)<<sid
	    <<" : mass = "<<setw(10)<<mass
	    <<" : width = "<<setw(10)<<width
	    <<" : parity = "<<setw(3)<< parity <<flush;

	for (UInt_t i = 0; i < vpid.size(); i ++){
	    cout<<" "<<setw(10)<<vpid[i]<<"("<<setw(15)<<vnpid[i]<<"|"<<setw(15)<<vgpid[i]<<")"<<flush;
	}
        cout<<endl;
    }
};

void testSMASHmissingPDG()
{
    map< Int_t,Int_t >::iterator it;
    map< Int_t,Int_t > madd;
    madd[ 311] = 16;
    madd[-311] = 16;
    madd[ 113] = 41;
    madd[ 213] = 42;
    madd[-213] = 43;
    madd[ 223] = 52;
    madd[ 331] = 53;
    madd[ 333] = 55;
    madd[2114] = 34;
    madd[2224] = 35;
    madd[2214] = 36;
    madd[1114] = 37;
    madd[ -12] = 4;
    madd[ -14] = 4;
    madd[  14] = 4;
    madd[ -16] = 4;
    madd[  16] = 4;


    regex r {R"(\s+)"};
    const Int_t maxchar = 800;
    Char_t line[maxchar];
    ifstream in;
    in.open("./particles.txt", std::ifstream::in);
    TDatabasePDG pdg;
    while(in.good()){
	in.getline(line,maxchar);
	TString tmp = line;

	string l = line;
        string t = regex_replace(l, r, " ");

	tmp = t.data();
	if(tmp.BeginsWith('#') ){
            cout<<t<<endl;
	    continue;
	}
	if(tmp == " " ) continue;
	if(tmp == "" ) continue;
        //cout<<"start----------------"<<endl;
	part p;
        p.reset();

	TObjArray* a = tmp.Tokenize(" ");
	if(a)
	{
	    for(Int_t i = 0 ; i < a->GetEntries(); i++){
		TObjString* s = (TObjString*) a->At(i);
		TString s1 = s->GetString();
		if(s1 == "#" ) break;
		if ( i == 0 ){
		    p.sid = s1;
		} else if ( i == 1 ) {
                    p.mass = s1.Atof();
		} else if ( i == 2 ) {
                   p.width = s1.Atof();
		} else if ( i == 3 ) {
		    //p.parity = s1.Atoi();
		    p.parity = s1;
		} else {
		    Int_t id = s1.Atoi();
                    p.vpid.push_back(id);

		    TParticlePDG* pa = pdg.GetParticle(id);
		    if(pa) {
                       TString n = pa->GetName();
                       p.vnpid.push_back(n);
		       Int_t gi = pdg.ConvertPdgToGeant3(id);
		       if(gi != 0) p.vgpid.push_back(gi);
		       else {
			   it = madd.find(id);
			   if(it != madd.end()) p.vgpid.push_back(it->second);
                           else  p.vgpid.push_back(0);

		       }
		    } else {
                       TString n = "not_set";
		       p.vnpid.push_back(n);
                       p.vgpid.push_back(0);
		    }


		}
	    }
            if (p.vpid.size()>4) cout<<"test "<<t.data()<<endl;
	    p.print();
            /*
	    cout<<setw(15)<<sid
	    <<" : mass = "<<setw(10)<<mass
	    <<" : width = "<<setw(10)<<width
	    <<" : parity = "<<setw(3)<< parity <<flush;

	    for (UInt_t i = 0; i < vpid.size(); i ++){
                TParticlePDG* pa = TParticlePDGpdg.GetParticle(vpid [i]);


		cout<<" "<<setw(10) << vpid [i] <<flush;
	    }
            */
	    delete a;

	}
        //cout<<"stop----------------"<<endl;



    }


    for (it=madd.begin(); it != madd.end(); ++it)
	cout << it->first << " => " << it->second << '\n';


    return ;
}