-
Notifications
You must be signed in to change notification settings - Fork 2
/
minenergy_constraint.h
93 lines (80 loc) · 3.42 KB
/
minenergy_constraint.h
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
81
82
83
84
85
86
87
88
89
90
91
92
93
/******************************************************************************
* Copyright (C) 2014 Juan Antonio Garcia Martin , Peter Clote, Ivan Dotu *
* *
* This program is free software: you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation, either version 3 of the License, or *
* (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
******************************************************************************/
#include <string>
#include <vector>
#include <map>
#include <limits.h>
#include <ctime>
#include "base/logging.h"
#include "util/string_array.h"
#include "constraint_solver/constraint_solver.h"
#include "constraint_solver/constraint_solveri.h"
#include "rna_plugin.h"
namespace operations_research {
// This constraint maintains:
//
//
class MinEnergyConstraint : public Constraint {
public:
// This constructor does not take any ownership on its arguments.
MinEnergyConstraint(Solver* const s, const std::vector<IntVar*>& vars, int dangles, std::string rnaLib, std::string energyModel, string trgStr, double foldTemp, int cutPoint) : Constraint(s), trgStr_(trgStr),cutPoint_(cutPoint), vars_(vars) {
n=size();
v = newRNAplugin(rnaLib,n,dangles);
v->setEnergyModel(energyModel);
v->setTemperature(foldTemp);
if(cutPoint!= -1){
v->setCutPoint(cutPoint_);
}
if(trgStr_!=""){
v->setTestStructure(trgStr_);
}
vMaxMFE = s->MakeIntVar(-NO_ENERGY_LIMIT*100,NO_ENERGY_LIMIT*100, absl::StrFormat("MaxMFE"));
obj = s->MakeMinimize(vMaxMFE,1);
}
virtual ~MinEnergyConstraint() {
delete v;
}
// Adds observers (named Demon) to variable events. These demons are
// responsible for implementing the propagation algorithm of the
// constraint.
virtual void Post() {
// Create a demon 'global_demon' that will bind events on
// variables to the calling of the 'InitialPropagate()' method. As
// this method is expensive, 'global_demon' has a low priority. As
// such, InitialPropagate will be called after all normal demons
// and constraints have reached a fixed point. Note
// that ownership of the 'global_demon' belongs to the solver.
Demon* const global_demon = solver()->MakeDelayedConstraintInitialPropagateCallback(this);
//Demon* const global_demon = solver()->MakeConstraintInitialPropagateCallback(this);
// Attach to all variables.
for (int i = 0; i < size(); ++i) {
vars_[i]->WhenBound(global_demon);
}
}
virtual void InitialPropagate();
public:
OptimizeVar* obj;
IntVar* vMaxMFE;
protected:
string trgStr_;
int cutPoint_;
std::vector<IntVar*> vars_;
int64 size() const { return vars_.size(); };
RNAPlugin* v;
int n;
};
}