aboutsummaryrefslogtreecommitdiffstats
path: root/mass_profile/constrained_dbeta.hpp
blob: 24a1ebbf660bdf1b7b3c3125983f4875bbe89ce9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#ifndef CONSTRAINED_DBETA
#define CONSTRAINED_DBETA
#include "projector.hpp"


namespace opt_utilities
{
  template <typename T>
  class constrained_dbeta
    :public model<std::vector<T>,std::vector<T>,std::vector<T> >
  {
  public:
    constrained_dbeta()
    {
      this->push_param_info(param_info<std::vector<T>,std::string>("n01",1));
      this->push_param_info(param_info<std::vector<T>,std::string>("beta1",.66));
      this->push_param_info(param_info<std::vector<T>,std::string>("rc1",100));

      this->push_param_info(param_info<std::vector<T>,std::string>("n02",1));
      this->push_param_info(param_info<std::vector<T>,std::string>("beta2",.67));
      this->push_param_info(param_info<std::vector<T>,std::string>("rc2",110));
      
    }

  public:
    constrained_dbeta<T>* do_clone()const
    {
      return new constrained_dbeta<T>(*this);
    }

    std::vector<T> do_eval(const std::vector<T> & x,
			   const std::vector<T>& p)
    {
      T n01=std::abs(p[0]);
      T beta1=p[1];
      T rc1=p[2];

      T n02=std::abs(p[3]);
      T beta2=p[4];
      T rc2=p[5];

      

      std::vector<T> result(x.size()-1);
      for(int i=1;i<x.size();++i)
	{
	  T xi=(x[i]+x[i-1])/2;
	  T yi=0;
	  yi=n01*pow(1+xi*xi/rc1/rc1,-3./2.*beta1)+n02*pow(1+xi*xi/rc2/rc2,-3./2.*beta2);
	  result[i-1]=yi;
	}
      return result;
    }

    bool do_meets_constraint(const std::vector<T>& p)const
    {
      if(p.size()!=6)
	{
	  cerr<<p.size()<<endl;
	  cerr<<this->get_num_params()<<endl;
	  assert(0);
	}
      
      T rc1=p.at(2);
      T rc2=p.at(5);
      if(rc2>rc1)
	{
	  return true;
	}
      else
	{
	  cerr<<rc2<<"\t"<<rc1<<endl;
	  cerr<<"***"<<endl;
	  return false;
	}
    }
  };
}

#endif