LauCartesianCPCoeffSet.cc

Go to the documentation of this file.

// Copyright University of Warwick 2006 - 2013.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

// Authors:
// Thomas Latham
// John Back
// Paul Harrison

#include <iostream>
#include <fstream>
#include <vector>

#include "TMath.h"
#include "TRandom.h"

#include "LauCartesianCPCoeffSet.hh"
#include "LauComplex.hh"
#include "LauConstants.hh"
#include "LauParameter.hh"
#include "LauPrint.hh"
#include "LauRandom.hh"

ClassImp(LauCartesianCPCoeffSet)


LauCartesianCPCoeffSet::LauCartesianCPCoeffSet(const TString& compName, Double_t x, Double_t y, Double_t deltaX, Double_t deltaY,
                Bool_t xFixed, Bool_t yFixed, Bool_t deltaXFixed, Bool_t deltaYFixed, Bool_t deltaXSecondStage, Bool_t deltaYSecondStage) :
        LauAbsCoeffSet(compName),
        x_(new LauParameter("X", x, minRealImagPart_, maxRealImagPart_, xFixed)),
        y_(new LauParameter("Y", y, minRealImagPart_, maxRealImagPart_, yFixed)),
        deltaX_(new LauParameter("DeltaX", deltaX, minDelta_, maxDelta_, deltaXFixed)),
        deltaY_(new LauParameter("DeltaY", deltaY, minDelta_, maxDelta_, deltaYFixed)),
        particleCoeff_( x+deltaX, y+deltaY ),
        antiparticleCoeff_( x-deltaX, y-deltaY ),
        acp_("ACP", -2.0*(x*deltaX + y*deltaY)/(x*x + deltaX*deltaX + y*y + deltaY*deltaY), -1.0, 1.0, deltaXFixed&&deltaYFixed)
{
        if (deltaXSecondStage && !deltaXFixed) {
                deltaX_->secondStage(kTRUE);
                deltaX_->initValue(0.0);
        }
        if (deltaYSecondStage && !deltaYFixed) {
                deltaY_->secondStage(kTRUE);
                deltaY_->initValue(0.0);
        }
}

LauCartesianCPCoeffSet::LauCartesianCPCoeffSet(const LauCartesianCPCoeffSet& rhs, CloneOption cloneOption, Double_t constFactor) : LauAbsCoeffSet(rhs.name()),
        x_(0),
        y_(0),
        deltaX_(0),
        deltaY_(0),
        particleCoeff_( rhs.particleCoeff_ ),
        antiparticleCoeff_( rhs.antiparticleCoeff_ ),
        acp_( rhs.acp_ )
{
        if ( cloneOption == All || cloneOption == TieRealPart ) {
                x_ = rhs.x_->createClone(constFactor);
        } else {
                x_ = new LauParameter("X", rhs.x_->value(), minRealImagPart_, maxRealImagPart_, rhs.x_->fixed());
                if ( rhs.x_->blind() ) {
                        const LauBlind* blinder = rhs.x_->blinder();
                        x_->blindParameter( blinder->blindingString(), blinder->blindingWidth() );
                }
        }

        if ( cloneOption == All || cloneOption == TieImagPart ) {
                y_ = rhs.y_->createClone(constFactor);
        } else {
                y_ = new LauParameter("Y", rhs.y_->value(), minRealImagPart_, maxRealImagPart_, rhs.y_->fixed());
                if ( rhs.y_->blind() ) {
                        const LauBlind* blinder = rhs.y_->blinder();
                        y_->blindParameter( blinder->blindingString(), blinder->blindingWidth() );
                }
        }

        if ( cloneOption == All || cloneOption == TieCPPars ) {
                deltaX_ = rhs.deltaX_->createClone(constFactor);
                deltaY_ = rhs.deltaY_->createClone(constFactor);
        } else {
                deltaX_ = new LauParameter("DeltaX", rhs.deltaX_->value(), minDelta_, maxDelta_, rhs.deltaX_->fixed());
                deltaY_ = new LauParameter("DeltaY", rhs.deltaY_->value(), minDelta_, maxDelta_, rhs.deltaY_->fixed());
                if ( rhs.deltaX_->secondStage() && !rhs.deltaX_->fixed() ) {
                        deltaX_->secondStage(kTRUE);
                        deltaX_->initValue(0.0);
                }
                if ( rhs.deltaY_->secondStage() && !rhs.deltaY_->fixed() ) {
                        deltaY_->secondStage(kTRUE);
                        deltaY_->initValue(0.0);
                }
                if ( rhs.deltaX_->blind() ) {
                        const LauBlind* blinder = rhs.deltaX_->blinder();
                        deltaX_->blindParameter( blinder->blindingString(), blinder->blindingWidth() );
                }
                if ( rhs.deltaY_->blind() ) {
                        const LauBlind* blinder = rhs.deltaY_->blinder();
                        deltaY_->blindParameter( blinder->blindingString(), blinder->blindingWidth() );
                }
        }
}

std::vector<LauParameter*> LauCartesianCPCoeffSet::getParameters()
{
        std::vector<LauParameter*> pars;
        pars.push_back(x_);
        pars.push_back(y_);
        pars.push_back(deltaX_);
        pars.push_back(deltaY_);
        return pars;
}

void LauCartesianCPCoeffSet::printParValues() const
{
        std::cout<<"INFO in LauCartesianCPCoeffSet::printParValues : Component \""<<this->name()<<"\" has ";
        std::cout<<"x = "<<x_->value()<<",\t";
        std::cout<<"y = "<<y_->value()<<",\t";
        std::cout<<"Delta x = "<<deltaX_->value()<<",\t";
        std::cout<<"Delta y = "<<deltaY_->value()<<"."<<std::endl;
}

void LauCartesianCPCoeffSet::printTableHeading(std::ostream& stream) const
{
        stream<<"\\begin{tabular}{|l|c|c|c|c|}"<<std::endl;
        stream<<"\\hline"<<std::endl;
        stream<<"Component & Real Part & Imaginary Part & $\\Delta$ Real Part & $\\Delta$ Imaginary Part \\\\"<<std::endl;
        stream<<"\\hline"<<std::endl;
}

void LauCartesianCPCoeffSet::printTableRow(std::ostream& stream) const
{
        LauPrint print;
        TString resName = this->name();
        resName = resName.ReplaceAll("_", "\\_");
        stream<<resName<<"  &  $";
        print.printFormat(stream, x_->value());
        stream<<" \\pm ";
        print.printFormat(stream, x_->error());
        stream<<"$  &  $";
        print.printFormat(stream, y_->value());
        stream<<" \\pm ";
        print.printFormat(stream, y_->error());
        stream<<"$  &  $";
        print.printFormat(stream, deltaX_->value());
        stream<<" \\pm ";
        print.printFormat(stream, deltaX_->error());
        stream<<"$  &  $";
        print.printFormat(stream, deltaY_->value());
        stream<<" \\pm ";
        print.printFormat(stream, deltaY_->error());
        stream<<"$ \\\\"<<std::endl;
}

void LauCartesianCPCoeffSet::randomiseInitValues()
{
        if (x_->fixed() == kFALSE) {
                // Choose a value for "X" between -3.0 and 3.0
                Double_t value = LauRandom::zeroSeedRandom()->Rndm()*6.0 - 3.0;
                x_->initValue(value); x_->value(value);
        }
        if (y_->fixed() == kFALSE) {
                // Choose a value for "Y" between -3.0 and 3.0
                Double_t value = LauRandom::zeroSeedRandom()->Rndm()*6.0 - 3.0;
                y_->initValue(value);  y_->value(value);
        }
        if (deltaX_->fixed() == kFALSE && deltaX_->secondStage() == kFALSE) {
                // Choose a value for "Delta X" between -0.5 and 0.5
                Double_t value = LauRandom::zeroSeedRandom()->Rndm()*1.0 - 0.5;
                deltaX_->initValue(value); deltaX_->value(value);
        }
        if (deltaY_->fixed() == kFALSE && deltaY_->secondStage() == kFALSE) {
                // Choose a value for "Delta Y" between -0.5 and 0.5
                Double_t value = LauRandom::zeroSeedRandom()->Rndm()*1.0 - 0.5;
                deltaY_->initValue(value); deltaY_->value(value);
        }
}

void LauCartesianCPCoeffSet::finaliseValues()
{
        // update the pulls
        x_->updatePull();
        y_->updatePull();
        deltaX_->updatePull();
        deltaY_->updatePull();
}

const LauComplex& LauCartesianCPCoeffSet::particleCoeff()
{
        particleCoeff_.setRealImagPart( x_->unblindValue()+deltaX_->unblindValue(), y_->unblindValue()+deltaY_->unblindValue() );
        return particleCoeff_;
}

const LauComplex& LauCartesianCPCoeffSet::antiparticleCoeff()
{
        antiparticleCoeff_.setRealImagPart( x_->unblindValue()-deltaX_->unblindValue(), y_->unblindValue()-deltaY_->unblindValue() );
        return antiparticleCoeff_;
}

void LauCartesianCPCoeffSet::setCoeffValues( const LauComplex& coeff, const LauComplex& coeffBar, Bool_t init )
{
        LauComplex average( coeff );
        average += coeffBar;
        average.rescale( 0.5 );

        Double_t xVal( average.re() );
        Double_t yVal( average.im() );
        Double_t deltaXVal( coeff.re() - average.re() );
        Double_t deltaYVal( coeff.im() - average.im() );

        x_->value( xVal );
        y_->value( yVal );
        deltaX_->value( deltaXVal );
        deltaY_->value( deltaYVal );

        if ( init ) {
                x_->genValue( xVal );
                y_->genValue( yVal );
                deltaX_->genValue( deltaXVal );
                deltaY_->genValue( deltaYVal );

                x_->initValue( xVal );
                y_->initValue( yVal );
                deltaX_->initValue( deltaXVal );
                deltaY_->initValue( deltaYVal );
        }
}

LauParameter LauCartesianCPCoeffSet::acp()
{
        // set the name
        TString parName(this->baseName()); parName += "_ACP";
        acp_.name(parName);

        // work out the ACP value
        Double_t numer = x_->value()*deltaX_->value() + y_->value()*deltaY_->value();
        Double_t denom = x_->value()*x_->value() + deltaX_->value()*deltaX_->value() + y_->value()*y_->value() + deltaY_->value()*deltaY_->value();
        Double_t value = -2.0*numer/denom;

        // is it fixed?
        Bool_t fixed = deltaX_->fixed() && deltaY_->fixed();
        acp_.fixed(fixed);

        // we can't work out the error without the covariance matrix
        Double_t error(0.0);

        // set the value and error
        acp_.valueAndErrors(value,error);

        return acp_;
}

LauAbsCoeffSet* LauCartesianCPCoeffSet::createClone(const TString& newName, CloneOption cloneOption, Double_t constFactor)
{
        LauAbsCoeffSet* clone(0);
        if ( cloneOption == All || cloneOption == TieRealPart || cloneOption == TieImagPart || cloneOption == TieCPPars ) {
                clone = new LauCartesianCPCoeffSet( *this, cloneOption, constFactor );
                clone->name( newName );
        } else {
                std::cerr << "ERROR in LauCartesianCPCoeffSet::createClone : Invalid clone option" << std::endl;
        }
        return clone;
}