int CzXmlNode::Save(const char* filename)
{
// Dont save root node
if (Name == ".")
{
CzXmlNode* node = getFirstNode();
if (node == NULL)
return -1;
return node->Save(filename);
}
CzFile file;
if (file.Open(filename, "wb", true))
{
const char* xml_head = "<?xml version=\"1.0\"?>\n";
if (!file.Write((void*)xml_head, strlen(xml_head)))
return -1;
int err = SaveTree(&file);
if (err < 0)
return err;
}
else
return -1;
return 0;
}
void Pdb::SetTree(const String& exp)
{
SaveTree();
tree.Clear();
NamedVal nv;
try {
CParser p(exp);
nv.val = Exp(p);
}
catch(CParser::Error) {
return;
}
nv.name = exp;
String n = exp;
if(nv.val.type >= 0)
n = GetType(nv.val.type).name;
tree.SetRoot(Null, RawToValue(nv), n + '=' + Visualise(nv.val).GetString());
if(nv.val.type >= 0) {
String w = treetype.Get(n, Null);
LOG("SetTree " << n << ' ' << w);
tree.Open(0);
CParser p(w);
try {
Point sc;
sc.x = p.ReadInt();
sc.y = p.ReadInt();
int cursor = p.ReadInt();
ExpandTreeType(0, p);
tree.ScrollTo(sc);
if(cursor >= 0)
tree.SetCursor(cursor);
}
catch(CParser::Error) {}
}
}
KVSimReader_ELIE::KVSimReader_ELIE(KVString filename) : KVSimReader()
{
// Write your code here
init();
if (!OpenFileToRead(filename)) return;
if (!ReadHeader()) return;
SetROOTFileName(filename+".root");
tree_title.Form("ELIE events %s + %s %.1f MeV/nuc.",
proj.GetSymbol(),targ.GetSymbol(),ebeam);
Run(root_file_name);
SaveTree();
CloseFile();
}
std::vector<unsigned char> CEntityWriter::SaveTree(CSerializationTree& pTree)
{
std::vector<unsigned char> data;
push_long(data, 2); // object definition
std::vector<unsigned char> entity = SaveEntity(pTree.m_Entity);
push_long(data, entity.size());
data.insert(data.end(), entity.begin() , entity.end());
push_long(data, pTree.m_Children.size());
for (unsigned int i = 0 ; i < pTree.m_Children.size() ; i ++ )
{
std::vector<unsigned char> subtree = SaveTree(*pTree.m_Children[i]);
push_long(data, subtree.size());
data.insert(data.end(), subtree.begin() , subtree.end());
}
return data;
}
int main( void )
{
void *info = NULL;
uint32_t free, used, chuncks, freechunks;
SetSystemLog( SYSLOG_FILE, stdout );
printf( WIDE("Lets see... ini files...") );
while( ScanFiles( WIDE("."), WIDE("*.ini"), &info, ParseINI, 0, 0 ) );
GetMemStats( &free, &used, &chuncks, &freechunks );
printf( WIDE("Memory result : free:%ld used:%ld chuncks:%ld freechunks:%ld\n")
, free, used, chuncks, freechunks );
DumpRegisteredNames();
SaveTree();
LoadTree();
DumpRegisteredNames();
return 0;
}
int CIwGameXmlNode::Save(const char* filename)
{
// Dont save root node
if (Name == ".")
{
CIwGameXmlNode* node = GetFirstNode();
if (node == NULL)
return -1;
return node->Save(filename);
}
CIwGameFile file;
if (file.Open(filename, "wb", true))
{
s3eFilePrintf(file.getFileHandle(), "<?xml version=\"1.0\"?>\n");
int err = SaveTree(&file);
if (err < 0)
return err;
}
else
return -1;
return 0;
}
static INT_PTR CALLBACK ButOrderOpts(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
HWND hTree = GetDlgItem(hwndDlg, IDC_BUTTONORDERTREE);
OrderData *dat = (OrderData*)GetWindowLongPtr(hTree, GWLP_USERDATA);
switch (msg) {
case WM_INITDIALOG:
TranslateDialogDefault(hwndDlg);
dat = (OrderData*)malloc( sizeof(OrderData));
SetWindowLongPtr(hTree, GWLP_USERDATA, (LONG_PTR)dat);
dat->dragging = 0;
SetWindowLongPtr(hTree, GWL_STYLE, GetWindowLongPtr(hTree, GWL_STYLE)|TVS_NOHSCROLL);
SetDlgItemInt(hwndDlg, IDC_BUTTHEIGHT, g_ctrl->nButtonHeight, FALSE);
SendDlgItemMessage(hwndDlg, IDC_SPIN_HEIGHT, UDM_SETRANGE, 0, MAKELONG(50,10));
SendDlgItemMessage(hwndDlg, IDC_SPIN_HEIGHT, UDM_SETPOS, 0, MAKELONG(g_ctrl->nButtonHeight,0));
SetDlgItemInt(hwndDlg, IDC_BUTTWIDTH, g_ctrl->nButtonWidth, FALSE);
SendDlgItemMessage(hwndDlg, IDC_SPIN_WIDTH, UDM_SETRANGE, 0, MAKELONG(50,10));
SendDlgItemMessage(hwndDlg, IDC_SPIN_WIDTH, UDM_SETPOS, 0, MAKELONG(g_ctrl->nButtonWidth,0));
SetDlgItemInt(hwndDlg, IDC_BUTTGAP, g_ctrl->nButtonSpace, FALSE);
SendDlgItemMessage(hwndDlg, IDC_SPIN_GAP, UDM_SETRANGE, 0, MAKELONG(20,0));
SendDlgItemMessage(hwndDlg, IDC_SPIN_GAP, UDM_SETPOS, 0, MAKELONG(g_ctrl->nButtonSpace,0));
CheckDlgButton(hwndDlg, IDC_USEFLAT, g_ctrl->bFlatButtons ? BST_CHECKED : BST_UNCHECKED);
CheckDlgButton(hwndDlg, IDC_AUTORESIZE, g_ctrl->bAutoSize ? BST_CHECKED : BST_UNCHECKED);
CheckDlgButton(hwndDlg, IDC_SINGLELINE, g_ctrl->bSingleLine ? BST_CHECKED : BST_UNCHECKED);
BuildTree(hwndDlg);
EnableWindow(GetDlgItem(hwndDlg, IDC_ENAME), FALSE);
EnableWindow(GetDlgItem(hwndDlg, IDC_EPATH), FALSE);
EnableWindow(GetDlgItem(hwndDlg, IDC_DELLBUTTON), FALSE);
OptionshWnd = hwndDlg;
return TRUE;
case WM_COMMAND:
if (HIWORD(wParam) == EN_CHANGE && OptionshWnd) {
switch(LOWORD(wParam)) {
case IDC_ENAME: case IDC_EPATH:
break;
default:
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
}
break;
}
if ((HIWORD(wParam) == BN_CLICKED || HIWORD(wParam) == BN_DBLCLK)) {
int ctrlid = LOWORD(wParam);
//----- Launch buttons -----
if (ctrlid == IDC_BROWSE) {
TCHAR str[MAX_PATH];
OPENFILENAME ofn = {0};
GetDlgItemText(hwndDlg, IDC_EPATH, str, _countof(str));
ofn.lStructSize = OPENFILENAME_SIZE_VERSION_400;
ofn.hwndOwner = hwndDlg;
ofn.hInstance = NULL;
ofn.lpstrFilter = NULL;
ofn.lpstrFile = str;
ofn.Flags = OFN_FILEMUSTEXIST | OFN_PATHMUSTEXIST | OFN_EXPLORER;
ofn.nMaxFile = _countof(str);
ofn.nMaxFileTitle = MAX_PATH;
ofn.lpstrDefExt = _T("exe");
if (!GetOpenFileName(&ofn))
break;
SetDlgItemText(hwndDlg, IDC_EPATH, str);
break;
}
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
if (ctrlid == IDC_LBUTTONSET) {
TVITEM tvi ={0};
tvi.hItem = TreeView_GetSelection(hTree);
if (tvi.hItem == NULL)
break;
tvi.mask = TVIF_PARAM;
TreeView_GetItem(hTree, &tvi);
TopButtonInt* btn = (TopButtonInt*)tvi.lParam;
TCHAR buf [256];
// probably, condition not needs
if (btn->dwFlags & TTBBF_ISLBUTTON) {
if (!(btn->dwFlags & TTBBF_OPTIONAL)) {
// create button
TTBButton ttb = { 0 };
ttb.hIconDn = (HICON)LoadImage(hInst, MAKEINTRESOURCE(IDI_RUN), IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR);
ttb.dwFlags = TTBBF_VISIBLE | TTBBF_ISLBUTTON | TTBBF_INTERNAL | TTBBF_OPTIONAL;
ttb.name = NULL;
ttb.program = NULL;
int id = btn->id;
btn = CreateButton(&ttb);
btn->id = id;
tvi.lParam = (LPARAM)btn;
TreeView_SetItem(hTree, &tvi);
}
GetDlgItemText(hwndDlg, IDC_ENAME, buf, _countof(buf));
replaceStr(btn->pszName, _T2A(buf));
tvi.mask = TVIF_TEXT;
tvi.pszText = buf;
TreeView_SetItem(hTree, &tvi);
GetDlgItemText(hwndDlg, IDC_EPATH, buf, _countof(buf));
replaceStrT(btn->ptszProgram, buf);
}
break;
}
if (ctrlid == IDC_ADDLBUTTON) {
// create button
TTBButton ttb = { 0 };
ttb.hIconDn = (HICON)LoadImage(hInst, MAKEINTRESOURCE(IDI_RUN), IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR);
ttb.dwFlags = TTBBF_VISIBLE | TTBBF_ISLBUTTON | TTBBF_INTERNAL | TTBBF_OPTIONAL;
ttb.name = LPGEN("Default");
ttb.program = _T("Execute Path");
TopButtonInt* b = CreateButton(&ttb);
// get selection for insert
TVITEM tvi = {0};
tvi.hItem = TreeView_GetSelection(hTree);
// insert item
AddLine(hTree, b, tvi.hItem, dat->himlButtonIcons);
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
break;
}
//----- Separators -----
if (ctrlid == IDC_ADDSEP) {
// create button
TTBButton ttb = { 0 };
ttb.dwFlags = TTBBF_VISIBLE | TTBBF_ISSEPARATOR | TTBBF_INTERNAL | TTBBF_OPTIONAL;
TopButtonInt* b = CreateButton(&ttb);
// get selection for insert
TVITEM tvi = {0};
tvi.hItem = TreeView_GetSelection(hTree);
// insert item
AddLine(hTree, b, tvi.hItem, dat->himlButtonIcons);
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
break;
}
if (ctrlid == IDC_REMOVEBUTTON) {
TVITEM tvi = {0};
tvi.hItem = TreeView_GetSelection(hTree);
if (tvi.hItem == NULL)
break;
tvi.mask = TVIF_PARAM;
TreeView_GetItem(hTree, &tvi);
TopButtonInt* btn = (TopButtonInt*)tvi.lParam;
// if button enabled for separator and launch only, no need condition
// except possible service button introducing
if (btn->dwFlags & (TTBBF_ISSEPARATOR | TTBBF_ISLBUTTON)) {
// delete if was added in options
if (btn->dwFlags & TTBBF_OPTIONAL)
delete btn;
TreeView_DeleteItem(hTree,tvi.hItem);
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
}
break;
}
}
break;
case WM_NOTIFY:
switch(((LPNMHDR)lParam)->idFrom) {
case 0:
switch (((LPNMHDR)lParam)->code) {
case PSN_APPLY:
g_ctrl->nButtonHeight = GetDlgItemInt(hwndDlg, IDC_BUTTHEIGHT, NULL, FALSE);
g_ctrl->nButtonWidth = GetDlgItemInt(hwndDlg, IDC_BUTTWIDTH, NULL, FALSE);
g_ctrl->nButtonSpace = GetDlgItemInt(hwndDlg, IDC_BUTTGAP, NULL, FALSE);
g_ctrl->bFlatButtons = (BYTE)IsDlgButtonChecked(hwndDlg, IDC_USEFLAT);
g_ctrl->bAutoSize = (BYTE)IsDlgButtonChecked(hwndDlg, IDC_AUTORESIZE);
g_ctrl->bSingleLine = (BYTE)IsDlgButtonChecked(hwndDlg, IDC_SINGLELINE);
db_set_dw(0, TTB_OPTDIR, "BUTTHEIGHT", g_ctrl->nButtonHeight);
db_set_dw(0, TTB_OPTDIR, "BUTTWIDTH", g_ctrl->nButtonWidth);
db_set_dw(0, TTB_OPTDIR, "BUTTGAP", g_ctrl->nButtonSpace);
db_set_b(0, TTB_OPTDIR, "UseFlatButton", g_ctrl->bFlatButtons);
db_set_b(0, TTB_OPTDIR, "SingleLine", g_ctrl->bSingleLine);
db_set_b(0, TTB_OPTDIR, "AutoSize", g_ctrl->bAutoSize);
SaveTree(hwndDlg);
RecreateWindows();
ArrangeButtons();
}
break;
case IDC_BUTTONORDERTREE:
switch (((LPNMHDR)lParam)->code) {
case TVN_BEGINDRAG:
SetCapture(hwndDlg);
dat->dragging = 1;
dat->hDragItem = ((LPNMTREEVIEW)lParam)->itemNew.hItem;
TreeView_SelectItem(hTree, dat->hDragItem);
break;
case NM_CLICK:
{
TVHITTESTINFO hti;
hti.pt.x = (short)LOWORD(GetMessagePos());
hti.pt.y = (short)HIWORD(GetMessagePos());
ScreenToClient(((LPNMHDR)lParam)->hwndFrom, &hti.pt);
if (TreeView_HitTest(((LPNMHDR)lParam)->hwndFrom, &hti))
if (hti.flags & TVHT_ONITEMSTATEICON) {
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
TreeView_SelectItem(hTree, hti.hItem);
}
}
break;
case TVN_SELCHANGED:
{
HTREEITEM hti = TreeView_GetSelection(hTree);
if (hti == NULL)
break;
TopButtonInt *btn = (TopButtonInt*)((LPNMTREEVIEW)lParam)->itemNew.lParam;
mir_cslock lck(csButtonsHook);
if (btn->dwFlags & TTBBF_ISLBUTTON) {
bool enable = (btn->dwFlags & TTBBF_INTERNAL) !=0;
EnableWindow(GetDlgItem(hwndDlg, IDC_ENAME), enable);
EnableWindow(GetDlgItem(hwndDlg, IDC_EPATH), enable);
EnableWindow(GetDlgItem(hwndDlg, IDC_REMOVEBUTTON), enable);
EnableWindow(GetDlgItem(hwndDlg, IDC_LBUTTONSET), enable);
if (btn->pszName != NULL)
SetDlgItemTextA(hwndDlg, IDC_ENAME, btn->pszName);
else
SetDlgItemTextA(hwndDlg, IDC_ENAME, "");
if (btn->ptszProgram != NULL)
SetDlgItemText(hwndDlg, IDC_EPATH, btn->ptszProgram);
else
SetDlgItemTextA(hwndDlg, IDC_EPATH, "");
}
else {
EnableWindow(GetDlgItem(hwndDlg,IDC_REMOVEBUTTON),
(btn->dwFlags & TTBBF_ISSEPARATOR)?TRUE:FALSE);
EnableWindow(GetDlgItem(hwndDlg, IDC_ENAME), FALSE);
EnableWindow(GetDlgItem(hwndDlg, IDC_EPATH), FALSE);
EnableWindow(GetDlgItem(hwndDlg, IDC_LBUTTONSET), FALSE);
SetDlgItemTextA(hwndDlg, IDC_ENAME, "");
SetDlgItemTextA(hwndDlg, IDC_EPATH, "");
}
}
}
break;
}
break;
case WM_MOUSEMOVE:
if (dat->dragging) {
TVHITTESTINFO hti;
hti.pt.x = (short)LOWORD(lParam);
hti.pt.y = (short)HIWORD(lParam);
ClientToScreen(hwndDlg, &hti.pt);
ScreenToClient(hTree, &hti.pt);
TreeView_HitTest(hTree, &hti);
if (hti.flags & (TVHT_ONITEM | TVHT_ONITEMRIGHT)) {
HTREEITEM it=hti.hItem;
hti.pt.y -= TreeView_GetItemHeight(hTree)/2;
TreeView_HitTest(hTree, &hti);
if (!(hti.flags&TVHT_ABOVE))
TreeView_SetInsertMark(hTree,hti.hItem,1);
else
TreeView_SetInsertMark(hTree,it,0);
}
else {
if (hti.flags & TVHT_ABOVE) SendMessage(hTree, WM_VSCROLL, MAKEWPARAM(SB_LINEUP, 0), 0);
if (hti.flags & TVHT_BELOW) SendMessage(hTree, WM_VSCROLL, MAKEWPARAM(SB_LINEDOWN, 0), 0);
TreeView_SetInsertMark(hTree, NULL, 0);
}
}
break;
case WM_LBUTTONUP:
if (dat->dragging) {
TreeView_SetInsertMark(hTree, NULL, 0);
dat->dragging = 0;
ReleaseCapture();
TVHITTESTINFO hti;
hti.pt.x = (short)LOWORD(lParam);
hti.pt.y = (short)HIWORD(lParam);
ClientToScreen(hwndDlg, &hti.pt);
ScreenToClient(hTree, &hti.pt);
hti.pt.y -= TreeView_GetItemHeight(hTree)/2;
TreeView_HitTest(hTree, &hti);
if (dat->hDragItem == hti.hItem)
break;
if (hti.flags&TVHT_ABOVE)
hti.hItem=TVI_FIRST;
TVITEM tvi;
tvi.mask = TVIF_HANDLE|TVIF_PARAM;
tvi.hItem = (HTREEITEM) dat->hDragItem;
TreeView_GetItem(hTree, &tvi);
if ( (hti.flags & (TVHT_ONITEM | TVHT_ONITEMRIGHT)) || (hti.hItem==TVI_FIRST)) {
TVINSERTSTRUCT tvis;
TCHAR name[128];
tvis.item.mask = TVIF_HANDLE | TVIF_PARAM | TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_STATE;
tvis.item.stateMask = 0xFFFFFFFF;
tvis.item.pszText = name;
tvis.item.cchTextMax = _countof(name);
tvis.item.hItem = dat->hDragItem;
TreeView_GetItem(hTree, &tvis.item);
TreeView_DeleteItem(hTree, dat->hDragItem);
tvis.hParent = NULL;
tvis.hInsertAfter = hti.hItem;
TreeView_SelectItem(hTree, TreeView_InsertItem(hTree, &tvis));
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
}
}
break;
case WM_DESTROY:
if (dat) {
ImageList_Destroy(dat->himlButtonIcons);
free(dat);
}
OptionshWnd = NULL;
break;
}
return FALSE;
}
void Pdb::TreeExpand(int node)
{
if(tree.GetChildCount(node))
return;
Value v = tree.Get(node);
if(!v.Is<NamedVal>())
return;
const NamedVal& nv = ValueTo<NamedVal>(tree.Get(node));
Val val = nv.val;
if(nv.val.ref > 0) {
val = DeRef(val);
if(val.type < 0 || val.ref > 0) {
TreeNode(node, '*' + nv.name, val);
SaveTree();
return;
}
}
if(val.type < 0) {
SaveTree();
return;
}
const Type& t = GetType(val.type);
if(t.vtbl_typeindex == -2) {
int count = GetSymInfo(t.modbase, type.GetKey(val.type), TI_GET_COUNT);
Val prtti;
prtti.ref = 1;
prtti.type = UINT4;
prtti.address = val.address - 4;
Val rtti = GetRVal(prtti);
FnInfo rtf = GetFnInfo(rtti.address);
TreeNode(node, rtf.name, prtti);
for(int i = 0; i < count; i++) {
Val ventry;
ventry.type = PFUNC;
ventry.address = val.address + 4 * i;
TreeNode(node, NFormat("[%d]", i), ventry);
}
return;
}
if(t.vtbl_typeindex >= 0) {
Val vtbl;
vtbl.ref = 1;
vtbl.address = val.address + t.vtbl_offset;
vtbl.type = t.vtbl_typeindex;
TreeNode(node, "virtual", vtbl);
/*
Val vloc = GetRVal(vtbl);
FnInfo vtbl_type = GetFnInfo(vloc.address);
const char *p = vtbl_type.name, *e = p;
while(*e && !(*e == ':' && e[1] == ':'))
e++;
String fullsym(p, e);
FnInfo ffs = GetFnInfo(fullsym);
if(ffs.pdbtype) {
Val typeval;
typeval.address = val.address;
TypeVal(typeval, ffs.pdbtype, t.modbase);
TreeNode(node, String().Cat() << '[' << ffs.name << ']', typeval);
}
*/
}
for(int i = 0; i < t.base.GetCount(); i++) {
Val r = t.base[i];
r.address += val.address;
if(r.type >= 0) {
const Type& bt = GetType(r.type);
TreeNode(node, bt.name, r);
}
}
for(int i = 0; i < t.member.GetCount(); i++) {
Val r = t.member[i];
r.address += val.address;
TreeNode(node, t.member.GetKey(i), r);
}
for(int i = 0; i < t.static_member.GetCount(); i++) {
Val r = t.static_member[i];
TreeNode(node, t.static_member.GetKey(i), r);
}
}
static BOOL CALLBACK GenMenuOpts(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
struct OrderData *dat = (struct OrderData*)GetWindowLong(GetDlgItem(hwndDlg,IDC_MENUITEMS),GWL_USERDATA);
switch (msg) {
case WM_INITDIALOG: {
TranslateDialogDefault(hwndDlg);
dat=(struct OrderData*)mir_alloc(sizeof(struct OrderData));
SetWindowLong(GetDlgItem(hwndDlg,IDC_MENUITEMS),GWL_USERDATA,(LONG)dat);
dat->dragging=0;
MyOldWindowProc=(WNDPROC)GetWindowLong(GetDlgItem(hwndDlg,IDC_MENUITEMS),GWL_WNDPROC);
SetWindowLong(GetDlgItem(hwndDlg,IDC_MENUITEMS),GWL_WNDPROC,(LONG)&LBTNDOWNProc);
//SetWindowLong(GetDlgItem(hwndDlg,IDC_BUTTONORDERTREE),GWL_STYLE,GetWindowLong(GetDlgItem(hwndDlg,IDC_BUTTONORDERTREE),GWL_STYLE)|TVS_NOHSCROLL);
{
HIMAGELIST himlCheckBoxes;
himlCheckBoxes=ImageList_Create(GetSystemMetrics(SM_CXSMICON),GetSystemMetrics(SM_CYSMICON),ILC_COLOR32|ILC_MASK,2,2);
ImageList_AddIcon(himlCheckBoxes,LoadIcon(g_hInst,MAKEINTRESOURCE(IDI_NOTICK)));
ImageList_AddIcon(himlCheckBoxes,LoadIcon(g_hInst,MAKEINTRESOURCE(IDI_TICK)));
TreeView_SetImageList(GetDlgItem(hwndDlg,IDC_MENUOBJECTS),himlCheckBoxes,TVSIL_NORMAL);
TreeView_SetImageList(GetDlgItem(hwndDlg,IDC_MENUITEMS),himlCheckBoxes,TVSIL_NORMAL);
}
BuildMenuObjectsTree(hwndDlg);
// Tree
// BuildTree(hwndDlg);
// OptionsOpened=TRUE;
// OptionshWnd=hwndDlg;
return TRUE;
}
case WM_COMMAND:
{
if ((HIWORD(wParam)==BN_CLICKED|| HIWORD(wParam)==BN_DBLCLK)) {
int ctrlid=LOWORD(wParam);
if (ctrlid==IDC_INSERTSEPARATOR) {
InsertSeparator(hwndDlg);
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
}
if (ctrlid==IDC_GENMENU_DEFAULT) {
TVITEM tvi;
HTREEITEM hti;
char buf[256];
hti=TreeView_GetSelection(GetDlgItem(hwndDlg,IDC_MENUITEMS));
if (hti==NULL)
break;
tvi.mask=TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;
tvi.hItem=hti;
TreeView_GetItem(GetDlgItem(hwndDlg,IDC_MENUITEMS),&tvi);
if ( _tcsstr(((MenuItemOptData *)tvi.lParam)->name, _T("---------------------------------------------")))
break;
ZeroMemory(buf,256);
GetDlgItemTextA(hwndDlg,IDC_GENMENU_CUSTOMNAME,buf,256);
if (((MenuItemOptData *)tvi.lParam)->name) {
mir_free(((MenuItemOptData *)tvi.lParam)->name);
}
((MenuItemOptData *)tvi.lParam)->name = mir_tstrdup( ((MenuItemOptData *)tvi.lParam)->defname );
SaveTree(hwndDlg);
RebuildCurrent(hwndDlg);
}
if (ctrlid==IDC_GENMENU_SET) {
TVITEM tvi;
TCHAR buf[256];
HTREEITEM hti = TreeView_GetSelection( GetDlgItem( hwndDlg,IDC_MENUITEMS ));
if ( hti == NULL )
break;
tvi.mask = TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;
tvi.hItem = hti;
SendDlgItemMessage(hwndDlg, IDC_MENUITEMS, TVM_GETITEM, 0, (LPARAM)&tvi);
if ( _tcsstr(((MenuItemOptData *)tvi.lParam)->name, STR_SEPARATOR ))
break;
ZeroMemory(buf,sizeof( buf ));
GetDlgItemText( hwndDlg, IDC_GENMENU_CUSTOMNAME, buf, SIZEOF( buf ));
if (((MenuItemOptData *)tvi.lParam)->name)
mir_free(((MenuItemOptData *)tvi.lParam)->name);
((MenuItemOptData *)tvi.lParam)->name = mir_tstrdup(buf);
SaveTree(hwndDlg);
RebuildCurrent(hwndDlg);
}
break;
}
if ((HIWORD(wParam)==STN_CLICKED|| HIWORD(wParam)==STN_DBLCLK)) {
int ctrlid=LOWORD(wParam);
}
break;
}
case WM_NOTIFY:
switch(((LPNMHDR)lParam)->idFrom ) {
case 0:
if (((LPNMHDR)lParam)->code == PSN_APPLY ) {
SaveTree(hwndDlg);
RebuildCurrent(hwndDlg);
}
break;
case IDC_MENUOBJECTS:
if (((LPNMHDR)lParam)->code == TVN_SELCHANGEDA )
{
TVITEM tvi;
HTREEITEM hti;
hti=TreeView_GetSelection(GetDlgItem(hwndDlg,IDC_MENUOBJECTS));
if (hti==NULL)
break;
tvi.mask=TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;
tvi.hItem=hti;
TreeView_GetItem(GetDlgItem(hwndDlg,IDC_MENUOBJECTS),&tvi);
BuildTree(hwndDlg,(int)tvi.lParam);
}
break;
case IDC_MENUITEMS:
switch (((LPNMHDR)lParam)->code) {
case NM_CUSTOMDRAW:
{
int i= handleCustomDraw(GetDlgItem(hwndDlg,IDC_MENUITEMS),(LPNMTVCUSTOMDRAW) lParam);
SetWindowLong(hwndDlg, DWL_MSGRESULT, i);
return TRUE;
}
case TVN_BEGINDRAGA:
SetCapture(hwndDlg);
dat->dragging=1;
dat->hDragItem=((LPNMTREEVIEW)lParam)->itemNew.hItem;
TreeView_SelectItem(GetDlgItem(hwndDlg,IDC_MENUITEMS),dat->hDragItem);
//ShowWindow(GetDlgItem(hwndDlg,IDC_BUTTONORDERTREEWARNING),SW_SHOW);
break;
case NM_CLICK:
{
TVHITTESTINFO hti;
hti.pt.x=(short)LOWORD(GetMessagePos());
hti.pt.y=(short)HIWORD(GetMessagePos());
ScreenToClient(((LPNMHDR)lParam)->hwndFrom,&hti.pt);
if (TreeView_HitTest(((LPNMHDR)lParam)->hwndFrom,&hti)) {
if (hti.flags&TVHT_ONITEMICON) {
TVITEM tvi;
tvi.mask=TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;
tvi.hItem=hti.hItem;
TreeView_GetItem(((LPNMHDR)lParam)->hwndFrom,&tvi);
tvi.iImage=tvi.iSelectedImage=!tvi.iImage;
((MenuItemOptData *)tvi.lParam)->show=tvi.iImage;
TreeView_SetItem(((LPNMHDR)lParam)->hwndFrom,&tvi);
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
//all changes take effect in runtime
//ShowWindow(GetDlgItem(hwndDlg,IDC_BUTTONORDERTREEWARNING),SW_SHOW);
}
/*--------MultiSelection----------*/
if (hti.flags&TVHT_ONITEMLABEL) {
/// LabelClicked Set/unset selection
TVITEM tvi;
HWND tvw=((LPNMHDR)lParam)->hwndFrom;
tvi.mask=TVIF_HANDLE|TVIF_PARAM;
tvi.hItem=hti.hItem;
TreeView_GetItem(tvw,&tvi);
if (GetKeyState(VK_CONTROL)&0x8000) {
if (((MenuItemOptData *)tvi.lParam)->isSelected)
((MenuItemOptData *)tvi.lParam)->isSelected=0;
else
((MenuItemOptData *)tvi.lParam)->isSelected=1; //current selection order++.
TreeView_SetItem(tvw,&tvi);
}
else if (GetKeyState(VK_SHIFT)&0x8000) {
; // shifted click
}
else {
// reset all selection except current
HTREEITEM hit;
hit=TreeView_GetRoot(tvw);
if (hit)
do {
TVITEM tvi={0};
tvi.mask=TVIF_HANDLE|TVIF_PARAM;
tvi.hItem=hit;
TreeView_GetItem(tvw,&tvi);
if (hti.hItem!=hit)
((MenuItemOptData *)tvi.lParam)->isSelected=0;
else
((MenuItemOptData *)tvi.lParam)->isSelected=1;
TreeView_SetItem(tvw,&tvi);
}
while (hit=TreeView_GetNextSibling(tvw,hit));
} } }
break;
}
case TVN_SELCHANGING:
{
LPNMTREEVIEW pn;
pn = (LPNMTREEVIEW) lParam;
//((MenuItemOptData *)(pn->itemNew.lParam))->isSelected=1;
/*if (pn->action==NotKeyPressed)
{
remove all selection
}
*/
}
case TVN_SELCHANGEDA:
{
TVITEM tvi;
HTREEITEM hti;
SetDlgItemTextA(hwndDlg,IDC_GENMENU_CUSTOMNAME,"");
SetDlgItemTextA(hwndDlg,IDC_GENMENU_SERVICE,"");
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_CUSTOMNAME),FALSE);
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_DEFAULT),FALSE);
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_SET),FALSE);
hti=TreeView_GetSelection(GetDlgItem(hwndDlg,IDC_MENUITEMS));
if (hti==NULL) {
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_CUSTOMNAME),FALSE);
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_DEFAULT),FALSE);
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_SET),FALSE);
break;
}
tvi.mask=TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;
tvi.hItem=hti;
TreeView_GetItem(GetDlgItem(hwndDlg,IDC_MENUITEMS),&tvi);
if ( tvi.lParam == 0 )
break;
if ( _tcsstr(((MenuItemOptData *)tvi.lParam)->name, STR_SEPARATOR ))
break;
SetDlgItemText(hwndDlg,IDC_GENMENU_CUSTOMNAME,((MenuItemOptData *)tvi.lParam)->name);
SetDlgItemTextA(hwndDlg,IDC_GENMENU_SERVICE,(((MenuItemOptData *)tvi.lParam)->uniqname)?(((MenuItemOptData *)tvi.lParam)->uniqname):"");
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_DEFAULT),TRUE);
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_SET),TRUE);
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_CUSTOMNAME),TRUE);
break;
}
break;
} }
break;
case WM_MOUSEMOVE:
if (!dat||!dat->dragging) break;
{
TVHITTESTINFO hti;
hti.pt.x=(short)LOWORD(lParam);
hti.pt.y=(short)HIWORD(lParam);
ClientToScreen(hwndDlg,&hti.pt);
ScreenToClient(GetDlgItem(hwndDlg,IDC_MENUITEMS),&hti.pt);
TreeView_HitTest(GetDlgItem(hwndDlg,IDC_MENUITEMS),&hti);
if (hti.flags&(TVHT_ONITEM|TVHT_ONITEMRIGHT)) {
HTREEITEM it = hti.hItem;
hti.pt.y -= TreeView_GetItemHeight(GetDlgItem(hwndDlg,IDC_MENUITEMS))/2;
TreeView_HitTest(GetDlgItem(hwndDlg,IDC_MENUITEMS),&hti);
if (!(hti.flags&TVHT_ABOVE))
TreeView_SetInsertMark(GetDlgItem(hwndDlg,IDC_MENUITEMS),hti.hItem,1);
else
TreeView_SetInsertMark(GetDlgItem(hwndDlg,IDC_MENUITEMS),it,0);
}
else {
if (hti.flags&TVHT_ABOVE) SendDlgItemMessage(hwndDlg,IDC_MENUITEMS,WM_VSCROLL,MAKEWPARAM(SB_LINEUP,0),0);
if (hti.flags&TVHT_BELOW) SendDlgItemMessage(hwndDlg,IDC_MENUITEMS,WM_VSCROLL,MAKEWPARAM(SB_LINEDOWN,0),0);
TreeView_SetInsertMark(GetDlgItem(hwndDlg,IDC_MENUITEMS),NULL,0);
} }
break;
case WM_DESTROY:
{
struct OrderData* dat = (struct OrderData*)GetWindowLong( GetDlgItem(hwndDlg,IDC_MENUITEMS), GWL_USERDATA );
if ( dat )
mir_free( dat );
FreeTreeData( hwndDlg );
break;
}
case WM_LBUTTONUP:
if (!dat->dragging)
break;
TreeView_SetInsertMark(GetDlgItem(hwndDlg,IDC_MENUITEMS),NULL,0);
dat->dragging=0;
ReleaseCapture();
{
TVHITTESTINFO hti;
hti.pt.x=(short)LOWORD(lParam);
hti.pt.y=(short)HIWORD(lParam);
ClientToScreen(hwndDlg,&hti.pt);
ScreenToClient(GetDlgItem(hwndDlg,IDC_MENUITEMS),&hti.pt);
hti.pt.y-=TreeView_GetItemHeight(GetDlgItem(hwndDlg,IDC_MENUITEMS))/2;
TreeView_HitTest(GetDlgItem(hwndDlg,IDC_MENUITEMS),&hti);
if (hti.flags&TVHT_ABOVE) hti.hItem=TVI_FIRST;
if (dat->hDragItem==hti.hItem) break;
dat->hDragItem=NULL;
if (hti.flags&(TVHT_ONITEM|TVHT_ONITEMRIGHT)||(hti.hItem==TVI_FIRST)) {
HWND tvw;
HTREEITEM * pSIT;
HTREEITEM FirstItem=NULL;
UINT uITCnt,uSic ;
tvw=GetDlgItem(hwndDlg,IDC_MENUITEMS);
uITCnt=TreeView_GetCount(tvw);
uSic=0;
if (uITCnt) {
pSIT=(HTREEITEM *)malloc(sizeof(HTREEITEM)*uITCnt);
if (pSIT) {
HTREEITEM hit;
hit=TreeView_GetRoot(tvw);
if (hit)
do {
TVITEM tvi={0};
tvi.mask=TVIF_HANDLE|TVIF_PARAM;
tvi.hItem=hit;
TreeView_GetItem(tvw,&tvi);
if (((MenuItemOptData *)tvi.lParam)->isSelected) {
pSIT[uSic]=tvi.hItem;
uSic++;
}
}while (hit=TreeView_GetNextSibling(tvw,hit));
// Proceed moving
{
UINT i;
HTREEITEM insertAfter;
insertAfter=hti.hItem;
for (i=0; i<uSic; i++) {
if (insertAfter) insertAfter=MoveItemAbove(tvw,pSIT[i],insertAfter);
else break;
if (!i) FirstItem=insertAfter;
} }
// free pointers...
free(pSIT);
} }
if (FirstItem) TreeView_SelectItem(tvw,FirstItem);
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
SaveTree(hwndDlg);
} }
break;
}
return FALSE;
}
static INT_PTR CALLBACK GenMenuOpts(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
struct OrderData *dat = (struct OrderData*)GetWindowLongPtr(GetDlgItem(hwndDlg,IDC_MENUITEMS),GWLP_USERDATA);
LPNMHDR hdr;
switch (msg) {
case WM_INITDIALOG:
TranslateDialogDefault(hwndDlg);
dat=(struct OrderData*)mir_alloc(sizeof(struct OrderData));
SetWindowLongPtr(GetDlgItem(hwndDlg,IDC_MENUITEMS),GWLP_USERDATA,(LONG_PTR)dat);
dat->dragging = 0;
dat->iInitMenuValue = DBGetContactSettingByte( NULL, "CList", "MoveProtoMenus", FALSE );
MyOldWindowProc = (WNDPROC)GetWindowLongPtr(GetDlgItem(hwndDlg,IDC_MENUITEMS),GWLP_WNDPROC);
SetWindowLongPtr(GetDlgItem(hwndDlg,IDC_MENUITEMS),GWLP_WNDPROC,(LONG_PTR)&LBTNDOWNProc);
{
HIMAGELIST himlCheckBoxes;
himlCheckBoxes=ImageList_Create(GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON),
(IsWinVerXPPlus() ? ILC_COLOR32 : ILC_COLOR16) | ILC_MASK, 2, 2);
ImageList_AddIcon_IconLibLoaded(himlCheckBoxes, SKINICON_OTHER_NOTICK);
ImageList_AddIcon_IconLibLoaded(himlCheckBoxes, SKINICON_OTHER_TICK);
TreeView_SetImageList(GetDlgItem(hwndDlg,IDC_MENUOBJECTS),himlCheckBoxes,TVSIL_NORMAL);
TreeView_SetImageList(GetDlgItem(hwndDlg,IDC_MENUITEMS),himlCheckBoxes,TVSIL_NORMAL);
}
CheckDlgButton(hwndDlg, dat->iInitMenuValue ? IDC_RADIO2 : IDC_RADIO1, TRUE );
CheckDlgButton(hwndDlg, IDC_DISABLEMENUICONS, bIconsDisabled );
BuildMenuObjectsTree(hwndDlg);
return TRUE;
case WM_COMMAND:
if ( HIWORD(wParam) == BN_CLICKED || HIWORD( wParam ) == BN_DBLCLK ) {
switch ( LOWORD( wParam )) {
case IDC_INSERTSEPARATOR:
InsertSeparator(hwndDlg);
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
break;
case IDC_RESETMENU:
ResetMenuItems( hwndDlg );
SendMessage( GetParent( hwndDlg ), PSM_CHANGED, 0, 0 );
break;
case IDC_DISABLEMENUICONS:
case IDC_RADIO1:
case IDC_RADIO2:
SendMessage( GetParent( hwndDlg ), PSM_CHANGED, 0, 0 );
break;
case IDC_GENMENU_DEFAULT:
{
TVITEM tvi;
HTREEITEM hti;
MenuItemOptData *iod;
hti=TreeView_GetSelection(GetDlgItem(hwndDlg,IDC_MENUITEMS));
if (hti==NULL)
break;
tvi.mask=TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;
tvi.hItem=hti;
TreeView_GetItem(GetDlgItem(hwndDlg,IDC_MENUITEMS),&tvi);
iod = ( MenuItemOptData * )tvi.lParam;
if ( iod->name && _tcsstr( iod->name, STR_SEPARATOR ))
break;
if (iod->name)
mir_free(iod->name);
iod->name = mir_tstrdup( iod->defname );
SaveTree(hwndDlg);
RebuildCurrent(hwndDlg);
SendMessage( GetParent( hwndDlg ), PSM_CHANGED, 0, 0 );
}
break;
case IDC_GENMENU_SET:
{
TVITEM tvi;
TCHAR buf[256];
MenuItemOptData *iod;
HTREEITEM hti = TreeView_GetSelection( GetDlgItem( hwndDlg,IDC_MENUITEMS ));
if ( hti == NULL )
break;
tvi.mask = TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;
tvi.hItem = hti;
SendDlgItemMessage(hwndDlg, IDC_MENUITEMS, TVM_GETITEM, 0, (LPARAM)&tvi);
iod = ( MenuItemOptData * )tvi.lParam;
if ( iod->name && _tcsstr(iod->name, STR_SEPARATOR ))
break;
ZeroMemory(buf,sizeof( buf ));
GetDlgItemText( hwndDlg, IDC_GENMENU_CUSTOMNAME, buf, SIZEOF( buf ));
if (iod->name)
mir_free(iod->name);
iod->name = mir_tstrdup(buf);
SaveTree(hwndDlg);
RebuildCurrent(hwndDlg);
SendMessage( GetParent( hwndDlg ), PSM_CHANGED, 0, 0 );
}
break;
} }
break;
case WM_NOTIFY:
hdr = (LPNMHDR)lParam;
switch( hdr->idFrom ) {
case 0:
if (hdr->code == PSN_APPLY ) {
bIconsDisabled = IsDlgButtonChecked(hwndDlg, IDC_DISABLEMENUICONS) != 0;
DBWriteContactSettingByte(NULL, "CList", "DisableMenuIcons", bIconsDisabled);
SaveTree(hwndDlg);
int iNewMenuValue = IsDlgButtonChecked(hwndDlg, IDC_RADIO1) ? 0 : 1;
if ( iNewMenuValue != dat->iInitMenuValue ) {
RebuildProtoMenus( iNewMenuValue );
dat->iInitMenuValue = iNewMenuValue;
}
RebuildCurrent(hwndDlg);
}
break;
case IDC_MENUOBJECTS:
if (hdr->code == TVN_SELCHANGEDA )
RebuildCurrent( hwndDlg );
break;
case IDC_MENUITEMS:
switch (hdr->code) {
case NM_CUSTOMDRAW:
{
int i= handleCustomDraw(GetDlgItem(hwndDlg,IDC_MENUITEMS),(LPNMTVCUSTOMDRAW) lParam);
SetWindowLongPtr(hwndDlg, DWLP_MSGRESULT, i);
return TRUE;
}
case TVN_BEGINDRAGA:
SetCapture(hwndDlg);
dat->dragging=1;
dat->hDragItem=((LPNMTREEVIEW)lParam)->itemNew.hItem;
TreeView_SelectItem(GetDlgItem(hwndDlg,IDC_MENUITEMS),dat->hDragItem);
break;
case NM_CLICK:
{
TVHITTESTINFO hti;
hti.pt.x=(short)LOWORD(GetMessagePos());
hti.pt.y=(short)HIWORD(GetMessagePos());
ScreenToClient(hdr->hwndFrom,&hti.pt);
if (TreeView_HitTest(hdr->hwndFrom,&hti)) {
if (hti.flags&TVHT_ONITEMICON) {
TVITEM tvi;
tvi.mask=TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;
tvi.hItem=hti.hItem;
TreeView_GetItem(hdr->hwndFrom,&tvi);
tvi.iImage=tvi.iSelectedImage=!tvi.iImage;
((MenuItemOptData *)tvi.lParam)->show=tvi.iImage;
TreeView_SetItem(hdr->hwndFrom,&tvi);
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
//all changes take effect in runtime
//ShowWindow(GetDlgItem(hwndDlg,IDC_BUTTONORDERTREEWARNING),SW_SHOW);
}
/*--------MultiSelection----------*/
if (hti.flags&TVHT_ONITEMLABEL) {
/// LabelClicked Set/unset selection
TVITEM tvi;
HWND tvw=hdr->hwndFrom;
tvi.mask=TVIF_HANDLE|TVIF_PARAM;
tvi.hItem=hti.hItem;
TreeView_GetItem(tvw,&tvi);
if (GetKeyState(VK_CONTROL)&0x8000) {
if (((MenuItemOptData *)tvi.lParam)->isSelected)
((MenuItemOptData *)tvi.lParam)->isSelected=0;
else
((MenuItemOptData *)tvi.lParam)->isSelected=1; //current selection order++.
TreeView_SetItem(tvw,&tvi);
}
else if (GetKeyState(VK_SHIFT)&0x8000) {
; // shifted click
}
else {
// reset all selection except current
HTREEITEM hit;
hit=TreeView_GetRoot(tvw);
if (hit)
do {
TVITEM tvi={0};
tvi.mask=TVIF_HANDLE|TVIF_PARAM;
tvi.hItem=hit;
TreeView_GetItem(tvw,&tvi);
if (hti.hItem!=hit)
((MenuItemOptData *)tvi.lParam)->isSelected=0;
else
((MenuItemOptData *)tvi.lParam)->isSelected=1;
TreeView_SetItem(tvw,&tvi);
}
while (hit=TreeView_GetNextSibling(tvw,hit));
} } }
break;
}
case TVN_SELCHANGING:
{
LPNMTREEVIEW pn;
pn = (LPNMTREEVIEW) lParam;
//((MenuItemOptData *)(pn->itemNew.lParam))->isSelected=1;
/*if (pn->action==NotKeyPressed)
{
remove all selection
}
*/
}
case TVN_SELCHANGEDA:
{
TVITEM tvi;
HTREEITEM hti;
MenuItemOptData *iod;
SetDlgItemTextA(hwndDlg,IDC_GENMENU_CUSTOMNAME,"");
SetDlgItemTextA(hwndDlg,IDC_GENMENU_SERVICE,"");
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_CUSTOMNAME),FALSE);
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_DEFAULT),FALSE);
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_SET),FALSE);
hti=TreeView_GetSelection(GetDlgItem(hwndDlg,IDC_MENUITEMS));
if (hti==NULL)
break;
tvi.mask=TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;
tvi.hItem=hti;
TreeView_GetItem(GetDlgItem(hwndDlg,IDC_MENUITEMS),&tvi);
if ( tvi.lParam == 0 )
break;
iod = ( MenuItemOptData * )tvi.lParam;
if ( iod->name && _tcsstr(iod->name, STR_SEPARATOR))
break;
SetDlgItemText(hwndDlg,IDC_GENMENU_CUSTOMNAME,iod->name);
if (iod->pimi->submenu.first == NULL && iod->uniqname)
SetDlgItemTextA(hwndDlg, IDC_GENMENU_SERVICE, iod->uniqname);
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_DEFAULT), lstrcmp(iod->name, iod->defname) != 0);
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_SET),TRUE);
EnableWindow(GetDlgItem(hwndDlg,IDC_GENMENU_CUSTOMNAME),TRUE);
break;
}
break;
} }
break;
case WM_MOUSEMOVE:
if (!dat||!dat->dragging) break;
{
TVHITTESTINFO hti;
hti.pt.x=(short)LOWORD(lParam);
hti.pt.y=(short)HIWORD(lParam);
ClientToScreen(hwndDlg,&hti.pt);
ScreenToClient(GetDlgItem(hwndDlg,IDC_MENUITEMS),&hti.pt);
TreeView_HitTest(GetDlgItem(hwndDlg,IDC_MENUITEMS),&hti);
if (hti.flags&(TVHT_ONITEM|TVHT_ONITEMRIGHT)) {
HTREEITEM it = hti.hItem;
hti.pt.y -= TreeView_GetItemHeight(GetDlgItem(hwndDlg,IDC_MENUITEMS))/2;
TreeView_HitTest(GetDlgItem(hwndDlg,IDC_MENUITEMS),&hti);
if (!(hti.flags&TVHT_ABOVE))
TreeView_SetInsertMark(GetDlgItem(hwndDlg,IDC_MENUITEMS),hti.hItem,1);
else
TreeView_SetInsertMark(GetDlgItem(hwndDlg,IDC_MENUITEMS),it,0);
}
else {
if (hti.flags&TVHT_ABOVE) SendDlgItemMessage(hwndDlg,IDC_MENUITEMS,WM_VSCROLL,MAKEWPARAM(SB_LINEUP,0),0);
if (hti.flags&TVHT_BELOW) SendDlgItemMessage(hwndDlg,IDC_MENUITEMS,WM_VSCROLL,MAKEWPARAM(SB_LINEDOWN,0),0);
TreeView_SetInsertMark(GetDlgItem(hwndDlg,IDC_MENUITEMS),NULL,0);
} }
break;
case WM_LBUTTONUP:
if (!dat->dragging)
break;
TreeView_SetInsertMark(GetDlgItem(hwndDlg,IDC_MENUITEMS),NULL,0);
dat->dragging=0;
ReleaseCapture();
{
TVHITTESTINFO hti;
hti.pt.x=(short)LOWORD(lParam);
hti.pt.y=(short)HIWORD(lParam);
ClientToScreen(hwndDlg,&hti.pt);
ScreenToClient(GetDlgItem(hwndDlg,IDC_MENUITEMS),&hti.pt);
hti.pt.y-=TreeView_GetItemHeight(GetDlgItem(hwndDlg,IDC_MENUITEMS))/2;
TreeView_HitTest(GetDlgItem(hwndDlg,IDC_MENUITEMS),&hti);
if (hti.flags&TVHT_ABOVE) hti.hItem=TVI_FIRST;
if (dat->hDragItem==hti.hItem) break;
dat->hDragItem=NULL;
if (hti.flags&(TVHT_ONITEM|TVHT_ONITEMRIGHT)||(hti.hItem==TVI_FIRST)) {
HWND tvw;
HTREEITEM * pSIT;
HTREEITEM FirstItem=NULL;
UINT uITCnt,uSic ;
tvw=GetDlgItem(hwndDlg,IDC_MENUITEMS);
uITCnt=TreeView_GetCount(tvw);
uSic=0;
if (uITCnt) {
pSIT=(HTREEITEM *)mir_alloc(sizeof(HTREEITEM)*uITCnt);
if (pSIT) {
HTREEITEM hit;
hit=TreeView_GetRoot(tvw);
if (hit)
do {
TVITEM tvi={0};
tvi.mask=TVIF_HANDLE|TVIF_PARAM;
tvi.hItem=hit;
TreeView_GetItem(tvw,&tvi);
if (((MenuItemOptData *)tvi.lParam)->isSelected) {
pSIT[uSic]=tvi.hItem;
uSic++;
}
}while (hit=TreeView_GetNextSibling(tvw,hit));
// Proceed moving
{
UINT i;
HTREEITEM insertAfter;
insertAfter=hti.hItem;
for (i=0; i<uSic; i++) {
if (insertAfter) insertAfter=MoveItemAbove(tvw,pSIT[i],insertAfter);
else break;
if (!i) FirstItem=insertAfter;
} }
// free pointers...
mir_free(pSIT);
} }
if (FirstItem) TreeView_SelectItem(tvw,FirstItem);
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
SaveTree(hwndDlg);
} }
break;
case WM_DESTROY:
if ( dat )
mir_free( dat );
ImageList_Destroy(TreeView_SetImageList(GetDlgItem(hwndDlg,IDC_MENUOBJECTS),NULL,TVSIL_NORMAL));
FreeTreeData( hwndDlg );
break;
}
return FALSE;
}
int main(int argc, char** argv)
{
//Check if all nedeed arguments to parse are there
if(argc != 2)
{
std::cerr << ">>>>> analysis.cpp::usage: " << argv[0] << " configFileName" << std::endl ;
return 1;
}
// Parse the config file
parseConfigFile (argv[1]) ;
std::string treeName = gConfigParser -> readStringOption("Input::treeName");
std::string inputFile = gConfigParser -> readStringOption("Input::inputFile");
double inputXSection = gConfigParser -> readDoubleOption("Input::inputXSection");
int entryMAX = gConfigParser -> readIntOption("Input::entryMAX");
int entryMIN = gConfigParser -> readIntOption("Input::entryMIN");
int entryMOD = gConfigParser -> readIntOption("Input::entryMOD");
std::cout << ">>>>> input::entryMIN " << entryMIN << std::endl;
std::cout << ">>>>> input::entryMAX " << entryMAX << std::endl;
std::cout << ">>>>> input::entryMOD " << entryMOD << std::endl;
int dataFlag = 0;
if (inputXSection == -1) dataFlag = 1; //==== it's a data sample!!!
std::cerr << ">>>>> input:: --- dataFlag " << dataFlag << std::endl;
// define map with events
std::map<std::pair<int,std::pair<int,int> >,int> eventsMap;
int nStepToDo = 1000;
try {
nStepToDo = gConfigParser -> readIntOption("Input::nStepToDo");
}
catch (char const* exceptionString){
std::cerr << " exception = " << exceptionString << std::endl;
nStepToDo = 1000;
}
std::cout << ">>>>> input::nStepToDo " << nStepToDo << std::endl;
// Open ntple
TChain* chain = new TChain(treeName.c_str());
chain->Add(inputFile.c_str());
treeReader reader((TTree*)(chain));
bool debug = false;
try {
debug = gConfigParser -> readBoolOption("Input::debug");
}
catch (char const* exceptionString){
std::cerr << " exception = " << exceptionString << std::endl;
}
std::cout << ">>>>> input::debug " << debug << std::endl;
///******************
///**** Triggers ****
std::vector<std::string> HLTVector;
try {
HLTVector = gConfigParser -> readStringListOption("Options::HLTVector");
}
catch (char const* exceptionString){
std::cerr << " exception = " << exceptionString << std::endl;
}
std::cout << ">>>>> Options::HLTVector size = " << HLTVector.size() << std::endl;
std::cout << ">>>>> >>>>> ";
for (int iHLT = 0; iHLT < HLTVector.size(); iHLT++){
std::cout << " " << HLTVector.at(iHLT) << ", ";
}
std::cout << std::endl;
///****************************
///**** DATA JSON file ****
std::string inFileNameJSON;
try {
inFileNameJSON = gConfigParser -> readStringOption("Input::inFileNameJSON");
}
catch (char const* exceptionString){
std::cerr << " exception = " << exceptionString << std::endl;
}
std::cout << ">>>>> Input::inFileNameJSON " << inFileNameJSON << std::endl;
std::map<int, std::vector<std::pair<int, int> > > jsonMap;
if( dataFlag == 1 ) {
jsonMap = readJSONFile(inFileNameJSON);
}
///---- Efficiency preselections ----
std::string histoNameEvents = gConfigParser -> readStringOption("Input::histoNameEvents");
std::cout << ">>>>> Input::inputFile " << inputFile << std::endl;
std::cout << ">>>>> Input::inputXSection " << inputXSection << std::endl;
std::cout << ">>>>> Input::histoNameEvents " << histoNameEvents << std::endl;
// Open ntples
TFile File(inputFile.c_str()) ;
TH1F* histoEvents = (TH1F*) File.Get(TString(histoNameEvents.c_str()));
///----------------------
///---- Preselection ----
///~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
///==== only retrocompatibility ====
///=================================
double lepton_efficiency = 1;
double jet_efficiency = 1;
double eff_Channel_Filter = 1;
double preselection_efficiency = 1.;
///~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
///~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
///~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
///-------------------
///---- selection ----
std::string outFileName = gConfigParser -> readStringOption("Output::outFileName");
std::cout << ">>>>> Output::outFileName " << outFileName << std::endl;
int nStep = 10; ///==== number of steps in the analysis
int numEntriesBefore;
// define variable container
Variables vars;
InitializeTree(vars, outFileName);
InitializeTreeTrigger(vars, HLTVector, reader);
vars.XSection = inputXSection;
vars.dataFlag = dataFlag; ///~~~~ 0 = MC 1 = DATA
if (entryMAX == -1) entryMAX = reader.GetEntries();
else if (reader.GetEntries() < entryMAX) entryMAX = reader.GetEntries();
numEntriesBefore = entryMAX - entryMIN;
if (histoEvents) preselection_efficiency = numEntriesBefore / (1. * histoEvents->GetBinContent(1));
else preselection_efficiency = 1;
vars.numEntriesBefore = numEntriesBefore;
vars.preselection_efficiency = preselection_efficiency;
FillEfficiencyTree(vars);
///*************************************
///**** definition of electron ID ****
///**** https://twiki.cern.ch/twiki/bin/viewauth/CMS/SimpleCutBasedEleID
///**** https://twiki.cern.ch/twiki/bin/viewauth/CMS/SimpleCutBasedEleID2011
std::vector<double> BarrelSelections;
std::vector<double> EndcapSelections;
double eleCombinedIsoBarrel = gConfigParser -> readDoubleOption("Selection::eleCombinedIsoBarrel");
double elesigmaIetaIetaBarrel = gConfigParser -> readDoubleOption("Selection::elesigmaIetaIetaBarrel");
double eledPhiBarrel = gConfigParser -> readDoubleOption("Selection::eledPhiBarrel");
double eledEtaBarrel = gConfigParser -> readDoubleOption("Selection::eledEtaBarrel");
double eleCombinedIsoEndcap = gConfigParser -> readDoubleOption("Selection::eleCombinedIsoEndcap");
double elesigmaIetaIetaEndcap = gConfigParser -> readDoubleOption("Selection::elesigmaIetaIetaEndcap");
double eledPhiEndcap = gConfigParser -> readDoubleOption("Selection::eledPhiEndcap");
double eledEtaEndcap = gConfigParser -> readDoubleOption("Selection::eledEtaEndcap");
double elemishits = gConfigParser -> readDoubleOption("Selection::elemishits");
double eledist = gConfigParser -> readDoubleOption("Selection::eledist");
double eledcot = gConfigParser -> readDoubleOption("Selection::eledcot");
double eledzPV = gConfigParser -> readDoubleOption("Selection::eledzPV");
double eledxyPV = gConfigParser -> readDoubleOption("Selection::eledxyPV");
BarrelSelections.push_back(eleCombinedIsoBarrel);
BarrelSelections.push_back(elesigmaIetaIetaBarrel);
BarrelSelections.push_back(eledPhiBarrel);
BarrelSelections.push_back(eledEtaBarrel);
BarrelSelections.push_back(elemishits);
BarrelSelections.push_back(eledist);
BarrelSelections.push_back(eledcot);
BarrelSelections.push_back(eledzPV);
BarrelSelections.push_back(eledxyPV);
EndcapSelections.push_back(eleCombinedIsoEndcap);
EndcapSelections.push_back(elesigmaIetaIetaEndcap);
EndcapSelections.push_back(eledPhiEndcap);
EndcapSelections.push_back(eledEtaEndcap);
EndcapSelections.push_back(elemishits);
EndcapSelections.push_back(eledist);
EndcapSelections.push_back(eledcot);
EndcapSelections.push_back(eledzPV);
EndcapSelections.push_back(eledxyPV);
///**** https://twiki.cern.ch/twiki/bin/viewauth/CMS/SimpleCutBasedEleID --> 2010 Data
///**** 95% ****
/*
BarrelSelections.push_back(0.15); ///==== iso Tk
BarrelSelections.push_back(2.00); ///==== iso em
BarrelSelections.push_back(0.12); ///==== iso had
BarrelSelections.push_back(0.15); ///==== iso combined
BarrelSelections.push_back(0.015); ///==== hOe
BarrelSelections.push_back(0.01); ///==== sigmaIetaIeta
BarrelSelections.push_back(0.8); ///==== dPhi
BarrelSelections.push_back(0.007); ///==== dEta
EndCapSelections.push_back(0.08); ///==== iso Tk
EndCapSelections.push_back(0.06); ///==== iso em
EndCapSelections.push_back(0.05); ///==== iso had
EndCapSelections.push_back(0.10); ///==== iso combined
EndCapSelections.push_back(0.07); ///==== hOe
EndCapSelections.push_back(0.03); ///==== sigmaIetaIeta
EndCapSelections.push_back(0.7); ///==== dPhi
EndCapSelections.push_back(0.01); ///==== dEta
*/
///**** 90% ****
/*
BarrelSelections.push_back(0.12); ///==== iso Tk
BarrelSelections.push_back(0.09); ///==== iso em
BarrelSelections.push_back(0.10); ///==== iso had
BarrelSelections.push_back(0.10); ///==== iso combined
BarrelSelections.push_back(0.12); ///==== hOe
BarrelSelections.push_back(0.01); ///==== sigmaIetaIeta
BarrelSelections.push_back(0.8); ///==== dPhi
BarrelSelections.push_back(0.007); ///==== dEta
EndCapSelections.push_back(0.05); ///==== iso Tk
EndCapSelections.push_back(0.06); ///==== iso em
EndCapSelections.push_back(0.03); ///==== iso had
EndCapSelections.push_back(0.07); ///==== iso combined
EndCapSelections.push_back(0.05); ///==== hOe
EndCapSelections.push_back(0.03); ///==== sigmaIetaIeta
EndCapSelections.push_back(0.7); ///==== dPhi
EndCapSelections.push_back(0.007); ///==== dEta
*/
///**** 80% ****
/*
BarrelSelections.push_back(0.09); ///==== iso Tk
BarrelSelections.push_back(0.07); ///==== iso em
BarrelSelections.push_back(0.10); ///==== iso had
BarrelSelections.push_back(0.07); ///==== iso combined
BarrelSelections.push_back(0.040); ///==== hOe
BarrelSelections.push_back(0.01); ///==== sigmaIetaIeta
BarrelSelections.push_back(0.06); ///==== dPhi
BarrelSelections.push_back(0.004); ///==== dEta
EndCapSelections.push_back(0.04); ///==== iso Tk
EndCapSelections.push_back(0.05); ///==== iso em
EndCapSelections.push_back(0.025); ///==== iso had
EndCapSelections.push_back(0.06); ///==== iso combined
EndCapSelections.push_back(0.025); ///==== hOe
EndCapSelections.push_back(0.03); ///==== sigmaIetaIeta
EndCapSelections.push_back(0.03); ///==== dPhi
EndCapSelections.push_back(0.007); ///==== dEta
*/
///**** 70% ****
/*
BarrelSelections.push_back(0.05); ///==== iso Tk
BarrelSelections.push_back(0.06); ///==== iso em
BarrelSelections.push_back(0.03); ///==== iso had
BarrelSelections.push_back(0.04); ///==== iso combined
BarrelSelections.push_back(0.025); ///==== hOe
BarrelSelections.push_back(0.01); ///==== sigmaIetaIeta
BarrelSelections.push_back(0.004); ///==== dPhi
BarrelSelections.push_back(0.004); ///==== dEta
EndCapSelections.push_back(0.025); ///==== iso Tk
EndCapSelections.push_back(0.025); ///==== iso em
EndCapSelections.push_back(0.02); ///==== iso had
EndCapSelections.push_back(0.03); ///==== iso combined
EndCapSelections.push_back(0.025); ///==== hOe
EndCapSelections.push_back(0.03); ///==== sigmaIetaIeta
EndCapSelections.push_back(0.02); ///==== dPhi
EndCapSelections.push_back(0.005); ///==== dEta
*/
///**** https://twiki.cern.ch/twiki/bin/viewauth/CMS/SimpleCutBasedEleID2011 --> 2011 Data
///**** 95% ****
/*
BarrelSelections.push_back(10000.); ///==== iso Tk
BarrelSelections.push_back(10000.); ///==== iso em
BarrelSelections.push_back(10000.); ///==== iso had
BarrelSelections.push_back(0.150); ///==== iso combined
BarrelSelections.push_back(10000.); ///==== hOe
BarrelSelections.push_back(0.012); ///==== sigmaIetaIeta
BarrelSelections.push_back(0.800); ///==== dPhi
BarrelSelections.push_back(0.007); ///==== dEta
EndCapSelections.push_back(10000.); ///==== iso Tk
EndCapSelections.push_back(10000.); ///==== iso em
EndCapSelections.push_back(10000.); ///==== iso had
EndCapSelections.push_back(0.100); ///==== iso combined
EndCapSelections.push_back(10000.); ///==== hOe
EndCapSelections.push_back(0.031); ///==== sigmaIetaIeta
EndCapSelections.push_back(0.7); ///==== dPhi
EndCapSelections.push_back(0.011); ///==== dEta
*/
///**** 90% ****
/*
BarrelSelections.push_back(10000.); ///==== iso Tk
BarrelSelections.push_back(10000.); ///==== iso em
BarrelSelections.push_back(10000.); ///==== iso had
BarrelSelections.push_back(0.085); ///==== iso combined
BarrelSelections.push_back(10000.); ///==== hOe
BarrelSelections.push_back(0.01); ///==== sigmaIetaIeta
BarrelSelections.push_back(0.071); ///==== dPhi
BarrelSelections.push_back(0.007); ///==== dEta
EndCapSelections.push_back(10000.); ///==== iso Tk
EndCapSelections.push_back(10000.); ///==== iso em
EndCapSelections.push_back(10000.); ///==== iso had
EndCapSelections.push_back(0.051); ///==== iso combined
EndCapSelections.push_back(10000.); ///==== hOe
EndCapSelections.push_back(0.031); ///==== sigmaIetaIeta
EndCapSelections.push_back(0.047); ///==== dPhi
EndCapSelections.push_back(0.011); ///==== dEta
*/
///**** 85% ****
/*
BarrelSelections.push_back(10000.); ///==== iso Tk
BarrelSelections.push_back(10000.); ///==== iso em
BarrelSelections.push_back(10000.); ///==== iso had
BarrelSelections.push_back(0.053); ///==== iso combined
BarrelSelections.push_back(10000.); ///==== hOe
BarrelSelections.push_back(0.01); ///==== sigmaIetaIeta
BarrelSelections.push_back(0.039); ///==== dPhi
BarrelSelections.push_back(0.005); ///==== dEta
EndCapSelections.push_back(10000.); ///==== iso Tk
EndCapSelections.push_back(10000.); ///==== iso em
EndCapSelections.push_back(10000.); ///==== iso had
EndCapSelections.push_back(0.042); ///==== iso combined
EndCapSelections.push_back(10000.); ///==== hOe
EndCapSelections.push_back(0.031); ///==== sigmaIetaIeta
EndCapSelections.push_back(0.028); ///==== dPhi
EndCapSelections.push_back(0.007); ///==== dEta
*/
///**** 80% ****
/*
BarrelSelections.push_back(10000.); ///==== iso Tk
BarrelSelections.push_back(10000.); ///==== iso em
BarrelSelections.push_back(10000.); ///==== iso had
BarrelSelections.push_back(0.040); ///==== iso combined
BarrelSelections.push_back(10000.); ///==== hOe
BarrelSelections.push_back(0.01); ///==== sigmaIetaIeta
BarrelSelections.push_back(0.027); ///==== dPhi
BarrelSelections.push_back(0.005); ///==== dEta
EndCapSelections.push_back(10000.); ///==== iso Tk
EndCapSelections.push_back(10000.); ///==== iso em
EndCapSelections.push_back(10000.); ///==== iso had
EndCapSelections.push_back(0.033); ///==== iso combined
EndCapSelections.push_back(10000.); ///==== hOe
EndCapSelections.push_back(0.031); ///==== sigmaIetaIeta
EndCapSelections.push_back(0.021); ///==== dPhi
EndCapSelections.push_back(0.006); ///==== dEta
*/
///***********************************
///**** definition of muon ID ****
std::vector<double> Selections;
double muCombinedIso = gConfigParser -> readDoubleOption("Selection::muCombinedIso");
double muChi2Ndof = gConfigParser -> readDoubleOption("Selection::muChi2Ndof");
double muValidTrackerHits = gConfigParser -> readDoubleOption("Selection::muValidTrackerHits");
double muValidMuonHits = gConfigParser -> readDoubleOption("Selection::muValidMuonHits");
double mutracker = gConfigParser -> readDoubleOption("Selection::mutracker");
double mustandalone = gConfigParser -> readDoubleOption("Selection::mustandalone");
double muglobal = gConfigParser -> readDoubleOption("Selection::muglobal");
double mudzPV = gConfigParser -> readDoubleOption("Selection::mudzPV");
double mudxyPV = gConfigParser -> readDoubleOption("Selection::mudxyPV");
Selections.push_back(muCombinedIso);
Selections.push_back(muChi2Ndof);
Selections.push_back(muValidTrackerHits);
Selections.push_back(muValidMuonHits);
Selections.push_back(mutracker);
Selections.push_back(mustandalone);
Selections.push_back(muglobal);
Selections.push_back(mudzPV);
Selections.push_back(mudxyPV);
/*
Selections.push_back(0.15); ///==== iso Combined
Selections.push_back(10); ///==== Chi2/ndof
Selections.push_back(10); ///==== n ValidTrackerHits
Selections.push_back(0); ///==== n ValidMuonHits
Selections.push_back(1); ///==== tracker
Selections.push_back(1); ///==== standalone
Selections.push_back(1); ///==== global
//Selections.push_back(1); ///==== goodMuon
*/
double start, end;
std::cout << ">>>>> analysis::entryMIN " << entryMIN << " ==> entryMAX " << entryMAX << ":" << reader.GetEntries() << std::endl;
int step = 0;
start = clock();
for(int iEvent = entryMIN ; iEvent < entryMAX ; ++iEvent) {
reader.GetEntry(iEvent);
if((iEvent%entryMOD) == 0) std::cout << ">>>>> analysis::GetEntry " << iEvent << " : " << entryMAX - entryMIN << std::endl;
///==== define variables ====
std::vector<ROOT::Math::XYZTVector>* jets = reader.Get4V("jets");
// std::vector<ROOT::Math::XYZTVector>* muons = reader.Get4V("muons");
// std::vector<ROOT::Math::XYZTVector>* electrons = reader.Get4V("electrons");
///*********************************************************************************************
///*********************************************************************************************
///=============================
///==== fill MC information ====
SetMCVariables(vars, reader);
///=============================
///==== fill Primary Vertex ====
SetPVVariables(vars, reader);
///================================
///==== fill Event information ====
SetEventVariables(vars, reader);
///***************************************************
///**** STEP -1 - Check no copies in DATA ****
///***************************************************
if (debug) std::cout << " STEP -1 " << std::endl;
if( dataFlag == 1 )
{
std::pair<int,int> eventLSandID(reader.GetInt("lumiId")->at(0), reader.GetInt("eventId")->at(0));
std::pair<int,std::pair<int,int> > eventRUNandLSandID(reader.GetInt("runId")->at(0), eventLSandID);
if( eventsMap[eventRUNandLSandID] == 1 ) continue;
else eventsMap[eventRUNandLSandID] = 1;
}
///*************************************************
///**** Check comparison with JSON file ***
///*************************************************
if( dataFlag == 1 )
{
int runId = reader.GetInt("runId")->at(0);
int lumiId = reader.GetInt("lumiId")->at(0);
if(AcceptEventByRunAndLumiSection(runId, lumiId, jsonMap) == false) continue;
}
///****************************
///**** STEP 0 - Ntuplizer ****
///************* no additional selections applied
step = 0;
if (step > nStepToDo) {
FillTree(vars);
continue;
}
if (debug) std::cout << ">>> STEP 0 <<<" << std::endl;
///*********************************************
///**** STEP 1 - Jet cleaning + min pT ****
///************* it's performed another time here to make sure that the cleaning worked well
///************* Jet - electrons (pT > 5)
///************* Jet - muons (pT > 5)
///************ In addition only jets with pT > 15 are considered from now on!
///************ No selections are applied here
step = 1;
if (step > nStepToDo) {
// FillTree(vars);
continue;
}
if (debug) std::cout << ">>> STEP 1 <<<" << std::endl;
std::vector<ROOT::Math::XYZTVector> leptons_jetCleaning;
// build the collection of electros for jet cleaning
///==== CLEANING WITH ELECTRONS ====
for(unsigned int iEle = 0; iEle < (reader.Get4V("electrons")->size()); ++iEle)
{
if( reader.Get4V("electrons")->at(iEle).pt() < 5. ) continue;
// bool flag = IsEleIsolatedID_VBF(reader,BarrelSelections,EndcapSelections,iEle);
bool flag = IsEleIsolatedIDPUCorrected_VBF(reader,BarrelSelections,EndcapSelections,iEle);
if (!flag) continue;
leptons_jetCleaning.push_back( reader.Get4V("electrons")->at(iEle) );
}
///==== CLEANING WITH MUONS ====
for (int iMu = 0; iMu < reader.Get4V("muons")->size(); iMu++){
if (reader.Get4V("muons")->at(iMu).pt() < 5.0) continue;
if (fabs(reader.Get4V("muons")->at(iMu).Eta()) > 2.5) continue;
// bool flag = IsMuIsolatedID_VBF(reader,Selections,iMu);
bool flag = IsMuIsolatedIDPUCorrected_VBF(reader,Selections,iMu);
if (!flag) continue;
leptons_jetCleaning.push_back( reader.Get4V("muons")->at(iMu) );
}
///==== now clean jet collection ====
int nJets = reader.Get4V("jets")->size();
std::vector<int> whitelistJet; ///~~~~ all jets, 0 if rejected, 1 if accepted
std::vector<int> blacklistJet; ///~~~~ list of numbers of jets that are "rejected"
std::vector<int> blacklistJet_forCJV;
std::vector<int> blacklistJet_forBtag;
for (int iJet = 0; iJet < nJets; iJet++){
bool skipJet = false;
if (reader.Get4V("jets")->at(iJet).Et() < 15.0) skipJet = true;
for(unsigned int iLep = 0; iLep < leptons_jetCleaning.size(); ++iLep) {
ROOT::Math::XYZTVector lep = leptons_jetCleaning.at(iLep);
if (ROOT::Math::VectorUtil::DeltaR(reader.Get4V("jets")->at(iJet),lep) < 0.3 ) skipJet = true;
}
if (skipJet) {
whitelistJet.push_back(0); ///---- reject
blacklistJet.push_back(iJet); ///---- reject ///== black list is in a different format
blacklistJet_forCJV.push_back(iJet); ///---- reject ///== black list is in a different format
blacklistJet_forBtag.push_back(iJet); ///---- reject ///== black list is in a different format
}
else {
whitelistJet.push_back(1); ///---- select
}
}
///**************************************
///**** STEP 2 - Super-Preselections ****
///************* tighter preselections to start the analysis from the same point
///==== construct considered objets
/// Objects considered and selections
/// Muon
/// Pt>10GeV, eta<2.5
/// MuonId & Iso
///
/// Electron
/// Pt>10GeV & |eta|<2.5
/// eleId & Iso
///
/// At least two leptons
/// Jet
/// Antikt5, L2L3 correction jets
/// At least two calo jets or two pf jets with pt>15 GeV
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
step = 2;
if (step > nStepToDo) {
// FillTree(vars);
continue;
}
if (debug) std::cout << ">>> STEP 2 <<<" << std::endl;
/// Electron
/// Pt>10GeV & |eta|<2.5
/// IsoTr / pTele <0.5
/// eleId & Iso
std::vector<int> whitelistEle;
std::vector<int> blacklistEle;
int nEles = reader.Get4V("electrons")->size();
for (int iEle = 0; iEle < nEles; iEle++){
bool skipEle = false;
if (reader.Get4V("electrons")->at(iEle).pt() < 10.0) skipEle = true;
if (fabs(reader.Get4V("electrons")->at(iEle).Eta()) > 2.5) skipEle = true;
// bool flag = IsEleIsolatedID_VBF(reader,BarrelSelections,EndcapSelections,iEle);
bool flag = IsEleIsolatedIDPUCorrected_VBF(reader,BarrelSelections,EndcapSelections,iEle);
if (!flag) skipEle = true;
if (skipEle) {
whitelistEle.push_back(0); ///---- reject
blacklistEle.push_back(iEle); ///---- reject ///== black list is in a different format
}
else {
whitelistEle.push_back(1); ///---- select
}
}
/// Muon
/// Pt>10GeV, eta<2.5
/// MuonId & Iso
std::vector<int> whitelistMu;
std::vector<int> blacklistMu;
int nMus = reader.Get4V("muons")->size();
for (int iMu = 0; iMu < nMus; iMu++){
bool skipMu = false;
if (reader.Get4V("muons")->at(iMu).pt() < 10.0) skipMu = true;
if (fabs(reader.Get4V("muons")->at(iMu).Eta()) > 2.5) skipMu = true;
// bool flag = IsMuIsolatedID_VBF(reader,Selections,iMu);
bool flag = IsMuIsolatedIDPUCorrected_VBF(reader,Selections,iMu);
if (!flag) skipMu = true;
if (skipMu) {
whitelistMu.push_back(0); ///---- reject
blacklistMu.push_back(iMu); ///---- reject ///== black list is in a different format
}
else {
whitelistMu.push_back(1); ///---- select
}
}
/// At least 2 leptons
int numMus_Accepted = GetNumList(whitelistMu);
int numEles_Accepted = GetNumList(whitelistEle);
int numLeptons_Accepted = numMus_Accepted + numEles_Accepted;
if (numLeptons_Accepted < 2) continue;
/// Jet
/// At least two calo jets or two pf jets with pt>20 GeV
int numJets_Accepted = GetNumList(whitelistJet);
if (numJets_Accepted < 2) continue; ///==== at least 2 jets "isolated"
///*************************
///**** STEP 3 - Jet ID ****
///************* Identification of two tag jets
step = 3;
if (step > nStepToDo) {
// FillTree(vars);
continue;
}
if (debug) std::cout << ">>> STEP 3 <<<" << std::endl;
std::vector<int> itSelJet;
double maxPt_jets_selected = SelectJets(itSelJet,*jets,"maxSumPt",-1.,&blacklistJet);
int q1 = itSelJet.at(0);
int q2 = itSelJet.at(1);
///---- check Pt order ----
if (jets->at(q1).Pt() < jets->at(q2).Pt()) {
int tempq = q1;
q1 = q2;
q2 = tempq;
}
if (debug) std::cerr << " q1 = " << q1 << " : q2 = " << q2 << std::endl;
///---- update white/black list jets ----
for (int iJet = 0; iJet < nJets; iJet++){
if (q1 == iJet || q2 == iJet) {
whitelistJet.at(iJet) = 1;
blacklistJet.push_back(iJet); ///===> blacklistJet used for CJV => no 2 tag jets to be considered!
blacklistJet_forCJV.push_back(iJet); ///===> blacklistJet used for CJV => no 2 tag jets to be considered!
}
else {
whitelistJet.at(iJet) = 0;
}
}
SetQJetVariables(vars, reader, q1, q2, blacklistJet_forCJV, blacklistJet_forBtag, blacklistJet_forBtag);
///********************************
///**** STEP 4 - Lepton ID ****
///************* Identification of the two leptons
step = 4;
if (step > nStepToDo) {
FillTree(vars);
continue;
}
if (debug) std::cout << ">>> STEP 4 <<<" << std::endl;
std::vector<ROOT::Math::XYZTVector> leptons;
std::vector<int> leptonFlavours;
std::vector<int> leptonILep;
for(unsigned int iEle = 0; iEle < nEles; iEle++){
if (whitelistEle.at(iEle) == 1){
leptons.push_back( reader.Get4V("electrons")->at(iEle) );
leptonFlavours.push_back(11);
leptonILep.push_back(iEle);
}
}
for(unsigned int iMu = 0; iMu < nMus; iMu++){
if (whitelistMu.at(iMu) == 1){
leptons.push_back( reader.Get4V("muons")->at(iMu) );
leptonFlavours.push_back(13);
leptonILep.push_back(iMu);
}
}
std::vector<int> itSelLep;
double maxPt_lept_selected = SelectJets(itSelLep,leptons,"maxSumPt",-1.,0);
int l1 = itSelLep.at(0);
int l2 = itSelLep.at(1);
///---- check Pt order ----
if (leptons.at(l1).Pt() < leptons.at(l2).Pt()) {
int templ = l1;
l1 = l2;
l2 = templ;
}
if (debug) std::cerr << " l1 = " << l1 << " : l2 = " << l2 << std::endl;
SetLeptonsVariables(vars, reader, leptonILep.at(l1), leptonILep.at(l2),leptonFlavours.at(l1), leptonFlavours.at(l2));
if (debug) std::cerr << ">> Lepton variables set" << std::endl;
SetMetVariables(vars, reader, "PFMet", leptonILep.at(l1), leptonILep.at(l2),leptonFlavours.at(l1), leptonFlavours.at(l2));
if (debug) std::cerr << ">> MET variables set" << std::endl;
//---- lepton veto
std::vector<int> blacklistLepton;
blacklistLepton.push_back(l1);
blacklistLepton.push_back(l2);
vars.Nleptons_pT5 = getNumberPTThreshold(leptons, 5, &blacklistLepton);
vars.Nleptons_pT10 = getNumberPTThreshold(leptons, 10, &blacklistLepton);
vars.Nleptons_pT15 = getNumberPTThreshold(leptons, 15, &blacklistLepton);
vars.Nleptons_pT20 = getNumberPTThreshold(leptons, 20, &blacklistLepton);
vars.Nleptons_pT25 = getNumberPTThreshold(leptons, 25, &blacklistLepton);
vars.Nleptons_pT30 = getNumberPTThreshold(leptons, 30, &blacklistLepton);
if (debug) std::cerr << ">> Lepton multiplicity set" << std::endl;
///*********************************
///**** STEP 5 - Jet Selections ****
///************* Loose selections of tag jets
step = 5;
if (step > nStepToDo) {
FillTree(vars);
continue;
}
if (debug) std::cout << ">>> STEP 5 <<<" << std::endl;
///---- hardcoded fixed preselections ---- VBF (begin) ----
if (vars.q1_pT < 20.) continue;
if (vars.q2_pT < 15.) continue;
if (vars.M_qq < 0.) continue;
if (vars.DEta_qq < 0.) continue;
///---- hardcoded fixed preselections ---- VBF (end) ----
///==== save trigger variables ====
SetTriggerVariables(vars, reader);
///************************************
///**** STEP 6 - Final Production *****
///************************************
///**** No more selections applied ****
step = 6;
if (step > nStepToDo) {
FillTree(vars);
continue;
}
if (debug) std::cout << ">>> STEP 6 <<<" << std::endl;
///==== if not already filled ... ====
FillTree(vars);
///=================================================
}
end = clock();
std::cout <<"Time = " << ((double) (end - start)) << " (a.u.)" << std::endl;
SaveTree(vars);
std::cerr << " === end === " << std::endl;
}
INT_PTR CALLBACK DlgProcToolBar(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
HTREEITEM hti;
switch (uMsg) {
case WM_INITDIALOG:
hToolBarTree = GetDlgItem(hwndDlg, IDC_TOOLBARTREE);
{
LONG_PTR style = GetWindowLongPtr(hToolBarTree, GWL_STYLE);
style ^= TVS_CHECKBOXES;
SetWindowLongPtr(hToolBarTree, GWL_STYLE, style);
style |= TVS_CHECKBOXES;
style |= TVS_NOHSCROLL;
SetWindowLongPtr(hToolBarTree, GWL_STYLE, style);
}
{
mir_cslock lck(ToolBarCS);
BuildMenuObjectsTree(hToolBarTree);
}
Utils::enableDlgControl(hwndDlg, IDC_IMCHECK, FALSE);
Utils::enableDlgControl(hwndDlg, IDC_CHATCHECK, FALSE);
Utils::enableDlgControl(hwndDlg, IDC_CANBEHIDDEN, FALSE);
SendDlgItemMessage(hwndDlg, IDC_SPIN1, UDM_SETRANGE, 0, MAKELONG(10, 0));
SendDlgItemMessage(hwndDlg, IDC_SPIN1, UDM_SETPOS, 0, MAKELONG(PluginConfig.g_iButtonsBarGap, 0));
TranslateDialogDefault(hwndDlg);
bOptionsInit = FALSE;
break;
case WM_LBUTTONUP:
if (drag) {
TreeView_SetInsertMark(hToolBarTree, NULL, 0);
drag = 0;
ReleaseCapture();
TVHITTESTINFO hti;
hti.pt.x = (short)LOWORD(lParam);
hti.pt.y = (short)HIWORD(lParam);
ClientToScreen(hwndDlg, &hti.pt);
ScreenToClient(hToolBarTree, &hti.pt);
hti.pt.y -= TreeView_GetItemHeight(hToolBarTree) / 2;
TreeView_HitTest(hToolBarTree, &hti);
if (hDragItem == hti.hItem) break;
if (hti.flags & TVHT_ABOVE)
hti.hItem = TVI_FIRST;
TVITEM tvi;
tvi.mask = TVIF_HANDLE | TVIF_PARAM;
tvi.hItem = (HTREEITEM)hDragItem;
TreeView_GetItem(hToolBarTree, &tvi);
if (hti.flags & (TVHT_ONITEM | TVHT_ONITEMRIGHT) || (hti.hItem == TVI_FIRST)) {
TVINSERTSTRUCT tvis;
TCHAR strbuf[128];
tvis.item.mask = TVIF_HANDLE | TVIF_PARAM | TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_STATE;
tvis.item.stateMask = 0xFFFFFFFF;
tvis.item.pszText = strbuf;
tvis.item.cchTextMax = sizeof(strbuf);
tvis.item.hItem = (HTREEITEM)hDragItem;
TreeView_GetItem(hToolBarTree, &tvis.item);
TreeView_DeleteItem(hToolBarTree, hDragItem);
tvis.hParent = NULL;
tvis.hInsertAfter = hti.hItem;
TreeView_SelectItem(hToolBarTree, TreeView_InsertItem(hToolBarTree, &tvis));
SendMessage((GetParent(hwndDlg)), PSM_CHANGED, 0, 0);
}
}
break;
case WM_MOUSEMOVE:
if (drag) {
TVHITTESTINFO hti;
hti.pt.x = (short)LOWORD(lParam);
hti.pt.y = (short)HIWORD(lParam);
ClientToScreen(hwndDlg, &hti.pt);
ScreenToClient(hToolBarTree, &hti.pt);
TreeView_HitTest(hToolBarTree, &hti);
if (hti.flags & (TVHT_ONITEM | TVHT_ONITEMRIGHT)) {
HTREEITEM it = hti.hItem;
hti.pt.y -= TreeView_GetItemHeight(hToolBarTree) / 2;
TreeView_HitTest(hToolBarTree, &hti);
if (!(hti.flags & TVHT_ABOVE))
TreeView_SetInsertMark(hToolBarTree, hti.hItem, 1);
else
TreeView_SetInsertMark(hToolBarTree, it, 0);
}
else {
if (hti.flags & TVHT_ABOVE) SendMessage(hToolBarTree, WM_VSCROLL, MAKEWPARAM(SB_LINEUP, 0), 0);
if (hti.flags & TVHT_BELOW) SendMessage(hToolBarTree, WM_VSCROLL, MAKEWPARAM(SB_LINEDOWN, 0), 0);
TreeView_SetInsertMark(hToolBarTree, NULL, 0);
}
}
break;
case WM_COMMAND:
if (HIWORD(wParam) == BN_CLICKED && GetFocus() == (HWND)lParam && (HWND)lParam != hToolBarTree)
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
if ((HIWORD(wParam) == EN_CHANGE) && ((HWND)lParam == GetFocus()))
if (!bOptionsInit)
SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
switch (LOWORD(wParam)) {
case IDC_BBRESET:
CallService(MS_DB_MODULE_DELETE, NULL, LPARAM("TabSRMM_Toolbar"));
CB_HardReInit();
BuildMenuObjectsTree(hToolBarTree);
break;
case IDC_SEPARATOR:
hti = TreeView_GetSelection(hToolBarTree);
if (!hti)
hti = TVI_FIRST;
CustomButtonData *cbd = new CustomButtonData();
cbd->bSeparator = cbd->bHidden = cbd->bIMButton = cbd->bLSided = true;
cbd->dwButtonOrigID = ++dwSepCount;
cbd->pszModuleName = "Tabsrmm_sep";
cbd->iButtonWidth = 22;
cbd->opFlags = BBSF_NTBDESTRUCT;
LButtonsList.insert(cbd);
TVINSERTSTRUCT tvis;
tvis.hParent = NULL;
tvis.hInsertAfter = hti;
tvis.item.mask = TVIF_PARAM | TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE;
tvis.item.pszText = TranslateT("<Separator>");
tvis.item.iImage = tvis.item.iSelectedImage = -1;
tvis.item.lParam = (LPARAM)cbd;
hti = TreeView_InsertItem(hToolBarTree, &tvis);
TreeView_SetCheckState(hToolBarTree, hti, (cbd->bIMButton || cbd->bChatButton));
}
break;
case WM_NOTIFY:
switch (((LPNMHDR)lParam)->idFrom) {
case 0:
switch (((LPNMHDR)lParam)->code) {
case PSN_RESET:
CB_ReInitCustomButtons();
dwSepCount = M.GetDword("TabSRMM_Toolbar", "SeparatorsCount", 0);
return 1;
case PSN_APPLY:
hti = TreeView_GetSelection(hToolBarTree);
if (hti) {
TVITEM tvi;
tvi.mask = TVIF_HANDLE | TVIF_PARAM;
tvi.hItem = hti;
TreeView_GetItem(hToolBarTree, &tvi);
if (tvi.lParam) {
CustomButtonData *cbd = (CustomButtonData*)tvi.lParam;
if (cbd) {
cbd->bIMButton = IsDlgButtonChecked(hwndDlg, IDC_IMCHECK) != 0;
cbd->bChatButton = IsDlgButtonChecked(hwndDlg, IDC_CHATCHECK) != 0;
cbd->bCanBeHidden = IsDlgButtonChecked(hwndDlg, IDC_CANBEHIDDEN) != 0;
}
}
}
SaveTree(hToolBarTree);
CB_ReInitCustomButtons();
PluginConfig.g_iButtonsBarGap = (BYTE)SendDlgItemMessage(hwndDlg, IDC_SPIN1, UDM_GETPOS, 0, 0);
if (PluginConfig.g_iButtonsBarGap != M.GetByte("ButtonsBarGap", 1))
M.BroadcastMessageAsync(WM_SIZE, 0, 0);
db_set_b(0, SRMSGMOD_T, "ButtonsBarGap", PluginConfig.g_iButtonsBarGap);
BuildMenuObjectsTree((HWND)hToolBarTree);
Utils::enableDlgControl(hwndDlg, IDC_IMCHECK, FALSE);
Utils::enableDlgControl(hwndDlg, IDC_CHATCHECK, FALSE);
Utils::enableDlgControl(hwndDlg, IDC_CANBEHIDDEN, FALSE);
return 1;
}
break;
case IDC_TOOLBARTREE:
switch (((LPNMHDR)lParam)->code) {
case TVN_BEGINDRAGA:
case TVN_BEGINDRAGW:
SetCapture(hwndDlg);
drag = 1;
hDragItem = ((LPNMTREEVIEW)lParam)->itemNew.hItem;
TreeView_SelectItem(hToolBarTree, hDragItem);
break;
case TVN_SELCHANGINGA:
case TVN_SELCHANGINGW:
hti = TreeView_GetSelection(hToolBarTree);
if (hti != NULL) {
TCHAR strbuf[128];
TVITEM tvi;
tvi.hItem = hti;
tvi.pszText = strbuf;
tvi.cchTextMax = sizeof(strbuf);
tvi.mask = TVIF_TEXT | TVIF_HANDLE | TVIF_PARAM;
TreeView_GetItem(hToolBarTree, &tvi);
if (tvi.lParam == 0 || !TreeView_GetCheckState(hToolBarTree, tvi.hItem) || !_tcscmp(tvi.pszText, MIDDLE_SEPARATOR))
break;
CustomButtonData *cbd = (CustomButtonData*)tvi.lParam;
if (cbd) {
cbd->opFlags = (IsDlgButtonChecked(hwndDlg, IDC_IMCHECK)) ? BBSF_IMBUTTON : 0;
cbd->opFlags |= (IsDlgButtonChecked(hwndDlg, IDC_CHATCHECK)) ? BBSF_CHATBUTTON : 0;
cbd->opFlags |= (IsDlgButtonChecked(hwndDlg, IDC_CANBEHIDDEN)) ? BBSF_CANBEHIDDEN : 0;
cbd->bIMButton = (IsDlgButtonChecked(hwndDlg, IDC_IMCHECK) ? TRUE : FALSE);
cbd->bChatButton = (IsDlgButtonChecked(hwndDlg, IDC_CHATCHECK) ? TRUE : FALSE);
cbd->bCanBeHidden = (IsDlgButtonChecked(hwndDlg, IDC_CANBEHIDDEN) ? TRUE : FALSE);
}
}
break;
case TVN_SELCHANGEDW:
case TVN_SELCHANGEDA:
hti = TreeView_GetSelection(hToolBarTree);
if (hti != NULL) {
TCHAR strbuf[128];
TVITEM tvi;
tvi.pszText = strbuf;
tvi.cchTextMax = sizeof(strbuf);
tvi.mask = TVIF_TEXT | TVIF_HANDLE | TVIF_PARAM;
tvi.hItem = hti;
TreeView_GetItem(hToolBarTree, &tvi);
if (!TreeView_GetCheckState(hToolBarTree, tvi.hItem) || !_tcscmp(tvi.pszText, MIDDLE_SEPARATOR)) {
Utils::enableDlgControl(hwndDlg, IDC_IMCHECK, FALSE);
Utils::enableDlgControl(hwndDlg, IDC_CHATCHECK, FALSE);
Utils::enableDlgControl(hwndDlg, IDC_CANBEHIDDEN, FALSE);
break;
}
if (tvi.lParam == 0)
break;
CustomButtonData *cbd = (CustomButtonData*)tvi.lParam;
if (cbd) {
Utils::enableDlgControl(hwndDlg, IDC_IMCHECK, TRUE);
Utils::enableDlgControl(hwndDlg, IDC_CHATCHECK, TRUE);
Utils::enableDlgControl(hwndDlg, IDC_CANBEHIDDEN, TRUE);
CheckDlgButton(hwndDlg, IDC_IMCHECK, (cbd->bIMButton) ? 1 : 0);
CheckDlgButton(hwndDlg, IDC_CHATCHECK, (cbd->bChatButton) ? 1 : 0);
CheckDlgButton(hwndDlg, IDC_CANBEHIDDEN, (cbd->bCanBeHidden) ? 1 : 0);
}
}
break;
case NM_CLICK:
TVHITTESTINFO hti = { 0 };
GetCursorPos(&hti.pt);
ScreenToClient(hToolBarTree, &hti.pt);
if (TreeView_HitTest(hToolBarTree, &hti)) {
if (hti.flags&TVHT_ONITEMSTATEICON) {
SendMessage(GetParent(hwndDlg), PSM_CHANGED, (WPARAM)hwndDlg, 0);
if (TreeView_GetCheckState(hToolBarTree, hti.hItem)) {
Utils::enableDlgControl(hwndDlg, IDC_IMCHECK, FALSE);
Utils::enableDlgControl(hwndDlg, IDC_CHATCHECK, FALSE);
Utils::enableDlgControl(hwndDlg, IDC_CANBEHIDDEN, FALSE);
CheckDlgButton(hwndDlg, IDC_IMCHECK, 1);
}
else {
Utils::enableDlgControl(hwndDlg, IDC_IMCHECK, TRUE);
Utils::enableDlgControl(hwndDlg, IDC_CHATCHECK, TRUE);
Utils::enableDlgControl(hwndDlg, IDC_CANBEHIDDEN, TRUE);
}
TreeView_SelectItem(hToolBarTree, hti.hItem);
}
}
}
}
break;
case WM_DESTROY:
HIMAGELIST hIml = TreeView_GetImageList(GetDlgItem(hwndDlg, IDC_TOOLBARTREE), TVSIL_NORMAL);
ImageList_Destroy(hIml);
hIml = TreeView_GetImageList(GetDlgItem(hwndDlg, IDC_TOOLBARTREE), TVSIL_STATE);
ImageList_Destroy(hIml);
break;
}
return FALSE;
}
